o2o-negosium-original/agent/tests/test_p8_session_db.py
hbyang 1682481b01 [feat] agent: 협상 고도화 — LLM 표현/이해층 + 카드 전술 실행계층 + ktcommerce 정리
카드 재설계("멘트 카드 → 전술 카드"):
- tactics.py 신설: 카드번호→전술(카운터 산식) 레지스트리, min(counter,target) 클램프
- 카운터 수락=즉시 타결(pending_counter_price 일반화, 구 offer_1pct 흡수)
- 목표가 초과 타결 금지(성공스텝 진입 가드) + "카드 소진=실패" 폐지→종결 국면
- 선택형 와일드카드(WC-*) 발동 + card.wild_cards 멘트 DB 어댑터

LLM 계층:
- Phase 2 표현층 ScriptNaturalizer(카드 멘트 자연화, 마커·치환자·숫자 보존 검증)
- Phase 3 이해층 InputInterpreter(자유발화 NLU→기대입력, 한국어 가격 파서)
- OPENAI_API_KEY env override(server_configs) + 전역 자격증명 게이트

결정 스택(Phase 1):
- 협상 규칙 데이터화(negotiation.wildcard_*_ratio/max_counter_rounds)
- 선택카드 우선순위 prior(UCB 방문수 감쇠, Q-table 오염 없음)

버그픽스:
- 인하율 음수 표기 제거 + 인상/동일/인하 구분(discount_phrase)
- 자연화 강조마커 보존(볼드/색 소실 시 원본 폴백)
- 카드 시드 가격변수(prev_partner_price·target_mid_price·middle_price 등) 치환

정리:
- ktcommerce 테넌트 삭제 + 테스트 21파일 imarketkorea/_base 로 마이그레이션
- 실 LLM 호출 차단 conftest 가드

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
2026-07-10 13:37:36 +09:00

70 lines
2.8 KiB
Python

"""P8-A 검증 — /chat 세션 상태 DB 영속화 (재시작/멀티워커 안전).
1. 세션이 DB(learning.chat_sessions)에 저장된다.
2. 새 ChatService 인스턴스(=다른 워커/재시작 모사)로 같은 session_id 를 이어가도 진행 상태 복원.
3. company_id 격리(타테넌트 session_id 로는 조회 안 됨).
"""
import pytest
from negotiation.chat.service.chat_session_repository import ChatSessionRepository
from router.v1.chat.protocol import Req_Chat
from services.chat_service import ChatService
from tenancy.config_loader import TenantConfigLoader
from tenancy.registry import TenantEngineRegistry
import os
_TENANTS_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "tenants")
def _eng():
reg = TenantEngineRegistry(loader=TenantConfigLoader(tenants_dir=_TENANTS_DIR, cache_ttl_seconds=0))
return reg
@pytest.mark.asyncio
async def test_session_persists_and_resumes_across_instances(db_engine):
reg = _eng()
eng = await reg.get_engine("imarketkorea")
# 인스턴스 1: 협상 시작 + 몇 턴 진행 (컨텍스트는 DB 조회 — 행이 없으므로 기본값 폴백)
svc1 = ChatService()
r = await svc1.chat(eng, Req_Chat())
sid = r.session_id
for ui in ["확인", "예", "확인", "11000"]:
r = await svc1.chat(eng, Req_Chat(session_id=sid, user_input=ui))
step_before = r.step
assert step_before == "가격협상_확인" # 11000 제시 후 확인 단계
# DB 에 저장됐는지 확인 (기본 컨텍스트: anchor=9900)
saved = await ChatSessionRepository(eng.company_id).get(sid)
assert saved is not None and saved.step == step_before
assert saved.context["anchor_price"] == 9900
# 인스턴스 2 (재시작/다른 워커 모사): 같은 session_id 로 이어가기
svc2 = ChatService()
r2 = await svc2.chat(eng, Req_Chat(session_id=sid, user_input="예"))
assert r2.step == "가격협상" # 상태가 복원되어 다음 step 으로 진행
assert r2.card_id is not None # 가격협상 턴 → 카드 선택 이어짐
@pytest.mark.asyncio
async def test_session_company_scoped(db_engine):
reg = _eng()
eng = await reg.get_engine("imarketkorea")
svc = ChatService()
r = await svc.chat(eng, Req_Chat())
sid = r.session_id
# 자사(imarketkorea)로는 조회됨
assert await ChatSessionRepository(eng.company_id).get(sid) is not None
# 타테넌트 company_id 로는 조회 안 됨 (격리)
assert await ChatSessionRepository("00000000-0000-0000-0000-0000000000aa").get(sid) is None
@pytest.mark.asyncio
async def test_get_none_for_missing(db_engine):
repo = ChatSessionRepository("imarketkorea")
assert await repo.get(None) is None
assert await repo.get("00000000-0000-0000-0000-000000000000") is None