From ed175c5b6548418b423302442a1af2c717ed1b28 Mon Sep 17 00:00:00 2001 From: hbyang Date: Tue, 7 Jul 2026 09:44:21 +0900 Subject: [PATCH] =?UTF-8?q?[feat]=20agent:=20Req=5FChat=20=EC=8A=AC?= =?UTF-8?q?=EB=A6=BC=ED=99=94=20=E2=80=94=20=ED=98=91=EC=83=81=20=EC=BB=A8?= =?UTF-8?q?=ED=85=8D=EC=8A=A4=ED=8A=B8=EB=A5=BC=20DB=20=EC=A1=B0=ED=9A=8C?= =?UTF-8?q?=EB=A1=9C=20=EC=A0=84=ED=99=98=20+=20CRUD=20=EA=B3=84=EC=B8=B5?= =?UTF-8?q?=20=EB=8F=84=EC=9E=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Req_Chat 을 session_id/user_input/client_step 3필드로 축소 — rq_type·목표가·앵커·품목가· 매출액·유통코드·파트너 유형·수용률 필드 전부 제거 - NegotiationContextLoader 신설: 세션 시작 시 공유 DB 1회 조회로 컨텍스트 확정 · rq_type = sessions.qt_type ({1,3}→재협상 / {2,4}→재견적) · anchor = sessions.anchoring_price(박제) — NULL 이면 무할인 폴백 anchor=target (v1.2 정책 승계) · 매출액 = suppliers.total_revenue(KTC 미러), 유통코드 = quotations.supplier_type 매핑 · 파트너 유형 = 상품별 distinct supplier 수 → PartnerType enum(0=NONE/1=SINGLE/2=MULTIPLE) - 가격 수용률은 세션 내 동적 계산: max(0, (첫 제시가−현재가)/첫 제시가) - DB 쿼리를 backend crud 패턴으로 분리: INegoContextCRUD(ABC)+NegoContextCRUD, IChatSessionRepository 인터페이스 추가 (테스트 더블 주입 가능) - 와일드카드 1% 수락 시 합의가=offer_1pct 반영 + Res_Chat.settled_price 신설 — backend 요약/입찰가가 이를 최우선 사용 (19,800원 수락이 20,000원으로 기록되던 버그 수정) - backend: agent 전송 바디 3필드로 축소, 앵커/파트너 조회 메서드 제거, test_anchoring_chat 을 새 구조로 재작업(박제 소비/폴백 검증은 agent 테스트로 이관) - 데모 페이지(/demo·negotiation_demo.html) 제거 — 컨텍스트 주입 경로 폐지로 무의미 - 테스트: agent 83/83, backend 57/57 (컨텍스트 로더 실데이터 왕복 4종 + CRUD 더블 검증 포함) Co-Authored-By: Claude Fable 5 --- agent/negotiation/chat/infra/__init__.py | 0 .../chat/infra/repository/__init__.py | 0 .../infra/repository/nego_context_crud.py | 135 ++++++++++++ agent/negotiation/chat/service/chat_engine.py | 7 + .../chat/service/chat_session_repository.py | 16 +- .../service/negotiation_context_loader.py | 102 +++++++++ .../qtable/domain/model/snapshot.py | 25 ++- agent/router/router.py | 14 +- agent/router/v1/chat/protocol.py | 26 +-- agent/services/chat_service.py | 56 ++++- agent/tests/negotiation_demo.html | 142 ------------ agent/tests/test_agent_integration.py | 7 +- agent/tests/test_context_loader.py | 202 ++++++++++++++++++ agent/tests/test_p7_chat.py | 46 +++- agent/tests/test_p8_session_db.py | 10 +- backend/services/agent_client.py | 34 ++- backend/services/chat_service.py | 59 ++--- backend/tests/test_anchoring_chat.py | 27 ++- 18 files changed, 641 insertions(+), 267 deletions(-) create mode 100644 agent/negotiation/chat/infra/__init__.py create mode 100644 agent/negotiation/chat/infra/repository/__init__.py create mode 100644 agent/negotiation/chat/infra/repository/nego_context_crud.py create mode 100644 agent/negotiation/chat/service/negotiation_context_loader.py delete mode 100644 agent/tests/negotiation_demo.html create mode 100644 agent/tests/test_context_loader.py diff --git a/agent/negotiation/chat/infra/__init__.py b/agent/negotiation/chat/infra/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/agent/negotiation/chat/infra/repository/__init__.py b/agent/negotiation/chat/infra/repository/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/agent/negotiation/chat/infra/repository/nego_context_crud.py b/agent/negotiation/chat/infra/repository/nego_context_crud.py new file mode 100644 index 0000000..39402f8 --- /dev/null +++ b/agent/negotiation/chat/infra/repository/nego_context_crud.py @@ -0,0 +1,135 @@ +"""협상 컨텍스트 CRUD — backend 소유 스키마 read-only 조회 (backend crud 패턴 준용). + +backend/crud/chat_crud.py(IChatCRUD/ChatCRUD)와 같은 형식: ABC 인터페이스 + 구현 클래스, +cdb(AsyncSession) 주입, (ErrorType, data) 반환. 트랜잭션/세션 경계는 호출부 +(NegotiationContextLoader)가 DB_SESSION_MNG.execute_lambda 로 관리한다. + +스키마 소유권: negotiation/partner/quotation 스키마는 backend 소유(01-schema.sql) — +여기서는 read-only 로만 접근한다. ORM 모델 중복 정의를 피하려고 sqlalchemy 경량 +table()/column() 구성을 쓴다(agent DB 매니저는 text() 미지원 — select 구성만 가능). +""" + +from abc import ABC, abstractmethod +from typing import Optional, Tuple + +from sqlalchemy import column, distinct, func, select, table +from sqlalchemy.ext.asyncio import AsyncSession + +from common.database.db_session_manager import DB_SESSION_MNG +from common.enums import ErrorType +from common.logger import LOG + +_SESSIONS = table( + "sessions", + column("session_id"), column("quotation_id"), column("item_id"), column("supplier_id"), + column("qt_type"), column("target_price"), column("anchoring_price"), + column("deleted"), + schema="negotiation", +) +_ITEMS = table("items", column("item_id"), column("price"), column("deleted"), schema="partner") +_SUPPLIERS = table("suppliers", column("supplier_id"), column("total_revenue"), column("deleted"), schema="partner") +_QUOTATIONS = table("quotations", column("qt_id"), column("supplier_type"), column("deleted"), schema="quotation") + + +class INegoContextCRUD(ABC): + @abstractmethod + async def get_session_row(self, cdb: AsyncSession, session_id) -> Tuple[ErrorType, Optional[tuple]]: + """세션 행 (qt_type, target_price, anchoring_price, item_id, quotation_id, supplier_id). 없으면 None.""" + pass + + @abstractmethod + async def get_item_price(self, cdb: AsyncSession, item_id) -> Tuple[ErrorType, int]: + """품목 기준가(items.price). 없으면 0.""" + pass + + @abstractmethod + async def get_supplier_total_revenue(self, cdb: AsyncSession, supplier_id) -> Tuple[ErrorType, float]: + """협력사 총매출액(suppliers.total_revenue — KTC 미러). 없으면 0.0.""" + pass + + @abstractmethod + async def get_quotation_supplier_type(self, cdb: AsyncSession, quotation_id) -> Tuple[ErrorType, Optional[int]]: + """견적의 협력사 유형(quotations.supplier_type: 0=none/1=유통/2=제조/3=총판). 미지정 시 None.""" + pass + + @abstractmethod + async def count_item_suppliers(self, cdb: AsyncSession, item_id) -> Tuple[ErrorType, int]: + """상품에 연결된 협력사 수(협상 세션 이력 기준 distinct supplier).""" + pass + + +class NegoContextCRUD(INegoContextCRUD): + async def get_session_row(self, cdb: AsyncSession, session_id) -> Tuple[ErrorType, Optional[tuple]]: + try: + query = ( + select(_SESSIONS.c.qt_type, _SESSIONS.c.target_price, _SESSIONS.c.anchoring_price, + _SESSIONS.c.item_id, _SESSIONS.c.quotation_id, _SESSIONS.c.supplier_id) + .where(_SESSIONS.c.session_id == session_id, _SESSIONS.c.deleted == False) # noqa: E712 + .limit(1) + ) + err_type, rows = await DB_SESSION_MNG.execute(cdb, query, "get_session_row failed.", raise_error=False) + if err_type != ErrorType.SUCCESS: + return err_type, None + return ErrorType.SUCCESS, (rows[0] if rows else None) + except Exception as ex: + LOG.e_no_callstack(ex) + return ErrorType.DB_RUN_FAILED, None + + async def get_item_price(self, cdb: AsyncSession, item_id) -> Tuple[ErrorType, int]: + try: + query = ( + select(_ITEMS.c.price) + .where(_ITEMS.c.item_id == item_id, _ITEMS.c.deleted == False) # noqa: E712 + .limit(1) + ) + err_type, rows = await DB_SESSION_MNG.execute(cdb, query, "get_item_price failed.", raise_error=False) + if err_type != ErrorType.SUCCESS or not rows or not rows[0]: + return err_type, 0 + return ErrorType.SUCCESS, int(rows[0]) + except Exception as ex: + LOG.e_no_callstack(ex) + return ErrorType.DB_RUN_FAILED, 0 + + async def get_supplier_total_revenue(self, cdb: AsyncSession, supplier_id) -> Tuple[ErrorType, float]: + try: + query = ( + select(_SUPPLIERS.c.total_revenue) + .where(_SUPPLIERS.c.supplier_id == supplier_id, _SUPPLIERS.c.deleted == False) # noqa: E712 + .limit(1) + ) + err_type, rows = await DB_SESSION_MNG.execute(cdb, query, "get_supplier_total_revenue failed.", raise_error=False) + if err_type != ErrorType.SUCCESS or not rows or not rows[0]: + return err_type, 0.0 + return ErrorType.SUCCESS, float(rows[0]) + except Exception as ex: + LOG.e_no_callstack(ex) + return ErrorType.DB_RUN_FAILED, 0.0 + + async def get_quotation_supplier_type(self, cdb: AsyncSession, quotation_id) -> Tuple[ErrorType, Optional[int]]: + try: + query = ( + select(_QUOTATIONS.c.supplier_type) + .where(_QUOTATIONS.c.qt_id == quotation_id, _QUOTATIONS.c.deleted == False) # noqa: E712 + .limit(1) + ) + err_type, rows = await DB_SESSION_MNG.execute(cdb, query, "get_quotation_supplier_type failed.", raise_error=False) + if err_type != ErrorType.SUCCESS or not rows or rows[0] is None: + return err_type, None + return ErrorType.SUCCESS, int(rows[0]) + except Exception as ex: + LOG.e_no_callstack(ex) + return ErrorType.DB_RUN_FAILED, None + + async def count_item_suppliers(self, cdb: AsyncSession, item_id) -> Tuple[ErrorType, int]: + try: + query = ( + select(func.count(distinct(_SESSIONS.c.supplier_id))) + .where(_SESSIONS.c.item_id == item_id, _SESSIONS.c.deleted == False) # noqa: E712 + ) + err_type, rows = await DB_SESSION_MNG.execute(cdb, query, "count_item_suppliers failed.", raise_error=False) + if err_type != ErrorType.SUCCESS or not rows: + return err_type, 0 + return ErrorType.SUCCESS, int(rows[0] or 0) + except Exception as ex: + LOG.e_no_callstack(ex) + return ErrorType.DB_RUN_FAILED, 0 diff --git a/agent/negotiation/chat/service/chat_engine.py b/agent/negotiation/chat/service/chat_engine.py index 8a7e1e8..af362d3 100644 --- a/agent/negotiation/chat/service/chat_engine.py +++ b/agent/negotiation/chat/service/chat_engine.py @@ -89,9 +89,16 @@ class ChatEngine: if price is None: return self._error(session, "가격을 숫자로 입력해 주세요.") session.context["input_price"] = price + # 협력사 첫 제시가 — 가격 수용률(첫 제시가 대비 양보율) 동적 계산의 기준값. + session.context.setdefault("first_offer_price", price) session.context["round"] = session.context.get("round", 0) + 1 nxt = self._default_next(node) elif mode in _CHOICE_MODES: + # 와일드카드 1% 인하 제안을 수락("예")하면 합의가를 제안가(offer_1pct)로 확정한다. + # (멘트에만 쓰이던 offer_1pct 가 input_price 에 반영되지 않아, 요약/입찰가가 + # 직전 제시가로 잡히던 버그 수정 — 수락 시 실제 합의가는 인하가다.) + if session.step == "wild_card_1pct" and user_input == "예" and session.context.get("offer_1pct"): + session.context["input_price"] = float(session.context["offer_1pct"]) nxt = self._choice_next(node, user_input, session) else: nxt = self._default_next(node) diff --git a/agent/negotiation/chat/service/chat_session_repository.py b/agent/negotiation/chat/service/chat_session_repository.py index 5f1fc22..78923ec 100644 --- a/agent/negotiation/chat/service/chat_session_repository.py +++ b/agent/negotiation/chat/service/chat_session_repository.py @@ -2,9 +2,11 @@ 인메모리 대신 learning.chat_sessions 에 진행 상태를 저장 → 서버 재시작/멀티워커 안전. company_id 스코프. ChatSession(dataclass) ↔ row 직렬화. +인터페이스(IChatSessionRepository) + 구현 형식 — backend crud 패턴 준용(테스트 더블 주입 가능). """ import uuid +from abc import ABC, abstractmethod from typing import Optional from sqlalchemy import select @@ -18,7 +20,19 @@ from common.utils.gtime import GTime from negotiation.chat.service.chat_engine import ChatSession -class ChatSessionRepository: +class IChatSessionRepository(ABC): + @abstractmethod + async def get(self, session_id: Optional[str]) -> Optional[ChatSession]: + """세션 조회(자사 company_id 스코프). 없으면 None.""" + pass + + @abstractmethod + async def save(self, session: ChatSession) -> ErrorType: + """진행 상태 upsert(재시작/멀티워커 안전).""" + pass + + +class ChatSessionRepository(IChatSessionRepository): def __init__(self, company_id: str): self.company_id = company_id diff --git a/agent/negotiation/chat/service/negotiation_context_loader.py b/agent/negotiation/chat/service/negotiation_context_loader.py new file mode 100644 index 0000000..ccb6177 --- /dev/null +++ b/agent/negotiation/chat/service/negotiation_context_loader.py @@ -0,0 +1,102 @@ +"""NegotiationContextLoader — 협상 시작 컨텍스트를 DB 에서 1회 조회 (Req_Chat 슬림화). + +backend 가 요청마다 실어 보내던 협상 컨텍스트(rq_type/목표가/앵커링가/품목가/매출액/유통코드/ +파트너 유형)를 세션 시작 시 agent 가 직접 조회한다. session_id 는 backend 와 공유하는 +negotiation.sessions.session_id. 행이 없으면(데모/테스트 직접 호출) None 을 반환하고 +호출부(ChatService)가 기본값으로 폴백한다. + +DB 쿼리는 INegoContextCRUD(negotiation/chat/infra/repository/nego_context_crud.py)에 위임 +— backend crud 패턴 준용(인터페이스 + 함수 호출). 여기는 판정 로직(rq_type·앵커 폴백· +코드 매핑)과 세션 경계(execute_lambda)만 담당한다. + +가격 수용률은 여기서 다루지 않는다 — 세션 내 라운드별 제시가로 매 턴 동적 계산(ChatService). +""" + +import uuid +from dataclasses import dataclass +from typing import Optional + +from common.database.db_session_manager import DB_SESSION_MNG +from common.enums import DBType, DBWRType, ErrorType +from common.logger import LOG +from negotiation.chat.infra.repository.nego_context_crud import INegoContextCRUD, NegoContextCRUD +from negotiation.qtable.domain.model.snapshot import PartnerType + +# 1:1 견적유형 → 재협상 스크립트. QuotationType: 1=renego, 3=new_nego (2=requote, 4=new_quote 는 1:N 재견적). +_ONE_TO_ONE_QT_TYPES = (1, 3) + +# 유통 코드: quotations.supplier_type(1=distribution 유통, 2=manufacture 제조, 3=sole_agency 총판) +# → 테넌트 code_map 키(A/B/C). 제조→A, 총판→B, 유통→C (0=none/NULL 은 미지정 → 호출부 기본값). +_SUPPLIER_TYPE_TO_CODE = {2: "A", 3: "B", 1: "C"} + + +@dataclass(frozen=True) +class NegotiationDbContext: + """세션 시작 시 DB 에서 확정되는 협상 컨텍스트 (라운드 진행 중 불변).""" + + rq_type: str # 재협상(1:1) | 재견적(1:N) — sessions.qt_type 으로 판별 + target_price: int # 목표 매입가(원) — sessions.target_price + anchor_price: int # 앵커링가 — sessions.anchoring_price(생성 시 박제). 없으면 target(무할인 폴백) + item_price: int # 기존 공급가(품목 기준가, items.price) — 인하율 멘트용. 없으면 0 + partner_type: PartnerType # 상품에 연결된 협력사 수(distinct supplier) → NONE/SINGLE/MULTIPLE + revenue_amount: float # 매출액(원) — suppliers.total_revenue(KTC 미러). 없으면 0 + distribution_code: Optional[str] # 유통 코드(A/B/C) — quotations.supplier_type 매핑. 미지정 시 None + + +class NegotiationContextLoader: + def __init__(self, crud: Optional[INegoContextCRUD] = None): + self.crud: INegoContextCRUD = crud or NegoContextCRUD() + + async def load(self, session_id: Optional[str]) -> Optional[NegotiationDbContext]: + """session_id 로 협상 컨텍스트 조회. 행이 없거나 조회 실패 시 None(호출부 기본값 폴백).""" + if not session_id: + return None + try: + sid = uuid.UUID(session_id) + except ValueError: + return None # 데모/테스트의 비-UUID 세션 키 + + async def _load(s) -> Optional[NegotiationDbContext]: + err, row = await self.crud.get_session_row(s, sid) + if err != ErrorType.SUCCESS or row is None: + return None + qt_type, target_price, anchoring_price, item_id, quotation_id, supplier_id = row + target = int(target_price or 0) + + # 앵커링가: 세션 생성 시 박제된 값(anchoring_price)을 그대로 사용 — 협상 중 불변. + # 박제가 없으면(데이터 이상) 무할인 폴백 anchor=target + WARN — 앵커링 v1.2 정책상 + # 앵커를 재계산하지 않으며, 해당 세션은 앵커링 집계에서 자동 제외된다. + anchor = int(anchoring_price or 0) + if anchor <= 0: + LOG.w(f"[NegotiationContextLoader] 앵커가 박제 없음 session_id={session_id} — 무할인 폴백(anchor=target)") + anchor = target + + # 매출액: 협력사 총매출(KTC total_revenue 미러). 미기재 시 0 → 호출부 기본값. + _, revenue_amount = await self.crud.get_supplier_total_revenue(s, supplier_id) + + # 유통 코드: 견적의 협력사 유형(supplier_type) 매핑. 미지정 시 None → 호출부 기본값. + _, supplier_type = await self.crud.get_quotation_supplier_type(s, quotation_id) + + # 기존 공급가(품목 기준가) — 없으면 0(인하율 멘트 미표시). + _, item_price = await self.crud.get_item_price(s, item_id) + + # 파트너사 유형: 상품에 연결된 협력사 수(협상 세션 이력 기준 distinct supplier). 실패 시 SINGLE. + err, supplier_count = await self.crud.count_item_suppliers(s, item_id) + if err != ErrorType.SUCCESS: + supplier_count = 1 + + return NegotiationDbContext( + rq_type="재협상" if int(qt_type) in _ONE_TO_ONE_QT_TYPES else "재견적", + target_price=target, + anchor_price=anchor, + item_price=item_price, + partner_type=PartnerType.from_count(supplier_count), + revenue_amount=revenue_amount, + distribution_code=_SUPPLIER_TYPE_TO_CODE.get(supplier_type) if supplier_type else None, + ) + + try: + return await DB_SESSION_MNG.execute_lambda(DBType.MAIN.value, DBWRType.DB_READ.value, _load) + except Exception as ex: # DB 불가 등 — 컨텍스트 없이 기본값으로 진행(협상 자체는 가능해야 함) + LOG.e_no_callstack(f"[NegotiationContextLoader] 컨텍스트 조회 실패 session_id={session_id}: {ex}") + return None diff --git a/agent/negotiation/qtable/domain/model/snapshot.py b/agent/negotiation/qtable/domain/model/snapshot.py index cda5ea0..6fa5efa 100644 --- a/agent/negotiation/qtable/domain/model/snapshot.py +++ b/agent/negotiation/qtable/domain/model/snapshot.py @@ -7,7 +7,7 @@ """ from dataclasses import dataclass, asdict -from enum import Enum +from enum import Enum, IntEnum from typing import Any, Dict, Optional @@ -19,6 +19,29 @@ class NegotiationOutcome(str, Enum): FAILURE = "failure" +class PartnerType(IntEnum): + """파트너사 유형 — 상품 하나를 취급하는 협력사의 경쟁 구조. + + 상품별 협력사 수 DB 조회(NegotiationContextLoader)로 세션 시작 시 확정한다: + 없음=NONE(0), 하나=SINGLE(1), 여러 곳=MULTIPLE(2). + 값이 협력사 수와 호환되도록 설계됨(0/1/≥2) — snapshot.partner_count 로 그대로 흘러 + state 버킷(_partner_bucket)과 W 가중치 계산에 쓰인다. + """ + + NONE = 0 + SINGLE = 1 + MULTIPLE = 2 + + @classmethod + def from_count(cls, count: int) -> "PartnerType": + """협력사 수 → 유형. 0=NONE, 1=SINGLE, 2 이상=MULTIPLE.""" + if count <= 0: + return cls.NONE + if count == 1: + return cls.SINGLE + return cls.MULTIPLE + + @dataclass class NegotiationSnapshot: # --- 이산 상태 산출 입력 --- diff --git a/agent/router/router.py b/agent/router/router.py index a26c354..ff25ae5 100644 --- a/agent/router/router.py +++ b/agent/router/router.py @@ -1,11 +1,9 @@ -import os import time from contextlib import asynccontextmanager from fastapi import FastAPI, Request from fastapi.middleware.gzip import GZipMiddleware from fastapi.middleware.cors import CORSMiddleware -from fastapi.responses import FileResponse from common.database.db_session_manager import DB_SESSION_MNG from common.logger import LOG @@ -22,7 +20,7 @@ API_SERVER_START_TIME = GTime.UTCStr() @asynccontextmanager async def lifespan(app: FastAPI): - # startup: 공유 베이스(_base) 없으면 자동 시드 (운영 자동화, P5) + # startup: 공유 베이스(_base) 없으면 자동 시드 (운영 자동화) from bootstrap.lifespan import ensure_base_seeded await ensure_base_seeded() yield @@ -58,16 +56,6 @@ async def log_time(request: Request, call_next): async def healthz(): return API_SERVER_START_TIME - -# 간단 테스트 프론트 (tests/negotiation_demo.html). 같은 출처로 서빙 → CORS 불필요. -_DEMO_HTML = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "tests", "negotiation_demo.html") - - -@app.get(path="/demo", include_in_schema=False) -async def demo(): - return FileResponse(_DEMO_HTML) - - # 각 도메인 라우터 등록. 새 기능 추가 시 router.v1.. import 후 include. # (chat / card / learning / qtable 라우터는 P7 에서 Chat_server 14개 API 이식하며 추가) app.include_router(router.v1.health.health.router) diff --git a/agent/router/v1/chat/protocol.py b/agent/router/v1/chat/protocol.py index 380b865..e8bfaa3 100644 --- a/agent/router/v1/chat/protocol.py +++ b/agent/router/v1/chat/protocol.py @@ -8,25 +8,22 @@ from common.models.gmodel import Req_WebPacketProtocol, Res_WebPacketProtocol class Req_Chat(Req_WebPacketProtocol): - """대화 한 턴. session_id 없으면 새 협상 시작(아래 컨텍스트로). tenant 는 헤더로만.""" + """대화 한 턴. session_id 없으면 새 협상 시작. tenant 는 헤더로만. - session_id: Optional[str] = Field(None, description="없으면 새 세션 생성") - rq_type: str = Field("재협상", description="재협상 | 재견적") + 협상 컨텍스트(rq_type/목표가/앵커링가/품목가/매출액/유통코드/파트너 유형)는 요청에 싣지 + 않는다 — 세션 시작 시 agent 가 DB 에서 1회 조회해 확정한다(NegotiationContextLoader): + negotiation.sessions(qt_type·target_price·anchoring_price), partner.items(price), + partner.suppliers(total_revenue), quotation.quotations(supplier_type), 상품별 협력사 수. + 행이 없으면(데모/테스트 직접 호출) 기본값 폴백. + 가격 수용률은 세션 내 라운드별 제시가로 매 턴 동적 계산: max(0, (첫 제시가−현재가)/첫 제시가). + """ + + session_id: Optional[str] = Field(None, description="없으면 새 세션 생성. 운영 경로는 negotiation.sessions.session_id 를 그대로 사용") user_input: Optional[str] = Field(None, description="버튼 선택 텍스트 또는 가격(price 모드)") # ① desync 감지: backend 가 보는 직전 봇 step(내부 step 또는 표시 step). 없으면 검사 생략. # agent 는 자기 세션 step 을 정답으로 보고 진행하되, 불일치 시 경고 로깅하고 응답에 desynced 를 실어 # backend/front 가 agent 응답의 step/client_step 으로 리싱크하게 한다. client_step: Optional[str] = Field(None, description="backend 가 본 직전 봇 step (desync 감지용)") - # 새 세션 시작 시 협상 컨텍스트 (옵션, 기본값 제공) - revenue_amount: float = 20_000_000 - distribution_code: str = "A" - partner_count: int = 1 - acceptance_ratio: float = 0.05 - # 갑(KT/iMK)이 직접 입력. anchor < target. anchor 기본 제안값 = target*(1-0.01). - target_price: int = 10000 # KT 목표 매입가 - anchor_price: int = 9900 # KT 앵커링가(목표가보다 낮음). 제시가 ≤ anchor → 우선협상 - # 기존 공급가(품목 기준가). 가격협상_확인 멘트의 인하율(discount_rate) 산출용. 0 이면 인하율 미표시. - item_price: int = 0 class Res_Chat(Res_WebPacketProtocol): @@ -52,6 +49,9 @@ class Res_Chat(Res_WebPacketProtocol): updated_q: Optional[float] = None visit_count: Optional[int] = None reward_total: Optional[float] = None + # 성공 확정 이후 턴(협상완료 요약·협상종료)에 내려주는 합의가. 와일드카드 1% 인하 수락 등 + # 유저가 직접 입력하지 않은 가격으로 타결될 수 있어, backend 요약/입찰가는 이 값을 최우선 사용한다. + settled_price: Optional[int] = None class Res_ChatSession(Res_WebPacketProtocol): diff --git a/agent/services/chat_service.py b/agent/services/chat_service.py index ef24364..935edad 100644 --- a/agent/services/chat_service.py +++ b/agent/services/chat_service.py @@ -14,16 +14,25 @@ from config.server_configs import agent_config from negotiation.chat.service.chat_engine import ChatEngine, ChatSession, StepView from negotiation.chat.service.indicator import compute_indicator from negotiation.chat.service.chat_session_repository import ChatSessionRepository +from negotiation.chat.service.negotiation_context_loader import NegotiationContextLoader from negotiation.chat.service.script_repository import ScriptRepository from negotiation.policies.base import EpisodeState, PolicyContext, Transition from negotiation.policy.model_store import QTablePolicyStore -from negotiation.qtable.domain.model.snapshot import NegotiationOutcome, NegotiationSnapshot +from negotiation.qtable.domain.model.snapshot import NegotiationOutcome, NegotiationSnapshot, PartnerType from negotiation.qtable.domain.service.reward_calculator import RewardCalculator from negotiation.qtable.domain.service.state_calculator import state_index from negotiation.qtable.infra.repository.learning_repository import LearningRepository from router.v1.chat.protocol import Req_Chat, Res_Chat, Res_ChatSession from tenancy.registry import TenantEngine +# 직접 호출(데모/테스트) 폴백 기본 컨텍스트 — 운영 경로는 NegotiationContextLoader 가 +# DB(negotiation.sessions·partner.items·partner.suppliers·quotation.quotations)에서 조회한다. +_DEFAULT_RQ_TYPE = "재협상" +_DEFAULT_TARGET_PRICE = 10000 # KT 목표 매입가 +_DEFAULT_ANCHOR_PRICE = 9900 # 앵커링가(목표가보다 낮음). 제시가 ≤ anchor → 우선협상 +_DEFAULT_REVENUE_AMOUNT = 20_000_000 # 매출액(원) — suppliers.total_revenue 미기재 시 폴백 +_DEFAULT_DISTRIBUTION_CODE = "A" # 유통 코드 — quotations.supplier_type 미지정 시 폴백 + class ChatService: async def chat(self, engine: TenantEngine, req: Req_Chat) -> Res_Chat: @@ -33,7 +42,11 @@ class ChatService: # 1) 세션 확보 / 시작 (DB 영속 — 재시작/멀티워커 안전, P8-A) session = await sess_repo.get(req.session_id) if req.session_id else None - chat_engine = ChatEngine(repo, rq_type=(session.rq_type if session else req.rq_type)) + # 새 세션 컨텍스트: 요청 페이로드 대신 DB(negotiation.sessions 등)에서 1회 조회. + # 행이 없으면(데모/테스트 직접 호출) 기본값 폴백. + db_ctx = None if session else await NegotiationContextLoader().load(req.session_id) + rq_type = session.rq_type if session else (db_ctx.rq_type if db_ctx else _DEFAULT_RQ_TYPE) + chat_engine = ChatEngine(repo, rq_type=rq_type) # ① step desync 감지: backend 가 본 직전 봇 step(client_step)이 agent 세션 step 과 다르면 경고. # agent 가 자기 step 을 정답으로 보고 진행하고(응답의 step/client_step 으로 backend 가 따라옴), @@ -54,14 +67,22 @@ class ChatService: # 새 uuid 발급 없이 그대로 세션 키로 쓴다. 없으면(직접 호출/데모) 생성. session = ChatSession( session_id=req.session_id or str(uuid.uuid4()), tenant_id=engine.tenant_id, company_id=engine.company_id, - rq_type=req.rq_type, action_space_size=engine.action_space_size, + rq_type=rq_type, action_space_size=engine.action_space_size, context={ - "revenue_amount": req.revenue_amount, "distribution_code": req.distribution_code, - "partner_count": req.partner_count, "acceptance_ratio": req.acceptance_ratio, - # 앵커링값은 갑(KT/iMK)이 직접 입력한 값을 사용 (UI 기본값 = target*(1-rate)). - "anchor_price": req.anchor_price, "target_price": req.target_price, "round": 0, + # 매출액 = suppliers.total_revenue, 유통코드 = quotations.supplier_type 매핑 (loader). + # 미기재/미지정이면 기본값 폴백. + "revenue_amount": db_ctx.revenue_amount if db_ctx and db_ctx.revenue_amount > 0 else _DEFAULT_REVENUE_AMOUNT, + "distribution_code": db_ctx.distribution_code if db_ctx and db_ctx.distribution_code else _DEFAULT_DISTRIBUTION_CODE, + # 파트너 유형(PartnerType 값 0/1/2) — 상품별 협력사 수 DB 조회로 세션 시작 시 1회 확정. + # snapshot.partner_count 로 그대로 사용(값 호환). + # 가격 수용률은 컨텍스트에 두지 않는다 — _snapshot 이 라운드별 제시가로 동적 계산. + "partner_count": int(db_ctx.partner_type) if db_ctx else int(PartnerType.SINGLE), + # 목표가/앵커링가: sessions 행(생성 시 박제된 anchoring_price) → 박제 ‰ → 1% 폴백 (loader). + "anchor_price": db_ctx.anchor_price if db_ctx else _DEFAULT_ANCHOR_PRICE, + "target_price": db_ctx.target_price if db_ctx else _DEFAULT_TARGET_PRICE, + "round": 0, # 기존 공급가(품목 기준가) — 가격협상_확인 인하율 산출용. - "item_price": req.item_price, + "item_price": db_ctx.item_price if db_ctx else 0, }, ) view = chat_engine.start(session) @@ -78,6 +99,10 @@ class ChatService: res.chat_end = view.chat_end res.outcome = view.outcome res.desynced = desynced + # 합의가: 성공 확정 이후 턴(협상완료 요약 → 협상종료)에 내려준다. 와일드카드 수락처럼 + # 유저가 직접 입력하지 않은 가격으로 타결될 수 있어 backend 요약/입찰가는 이 값을 최우선으로 쓴다. + if session.context.get("final_outcome") == "success" and session.context.get("input_price"): + res.settled_price = int(session.context["input_price"]) # 3) 학습 결합 (가격협상 카드선택 → 카드 스크립트·협상지표 / 종료 보상) if view.error is None and engine.action_space_size > 0: @@ -112,11 +137,24 @@ class ChatService: return res # ---- 학습 ---------------------------------------------------------- + @staticmethod + def _acceptance_ratio(context: dict) -> float: + """가격 수용률 동적 계산 — 협력사 첫 제시가 대비 현재 제시가의 양보율 (설계서 공식). + + acceptance = max(0, (첫 제시가 − 현재 제시가) / 첫 제시가). + 첫 제시 라운드(양보 없음)·첫 제시가 미기록(과거 세션 호환)이면 0(low 버킷). + """ + first = context.get("first_offer_price") or 0 + current = context.get("input_price") or 0 + if first <= 0 or current <= 0: + return 0.0 + return max(0.0, (first - current) / first) + def _snapshot(self, session: ChatSession, outcome: NegotiationOutcome) -> NegotiationSnapshot: c = session.context return NegotiationSnapshot( revenue_amount=c["revenue_amount"], distribution_code=c["distribution_code"], - partner_count=c["partner_count"], acceptance_ratio=c["acceptance_ratio"], + partner_count=c["partner_count"], acceptance_ratio=self._acceptance_ratio(c), input_price=c.get("input_price", c["anchor_price"]), anchor_price=c["anchor_price"], target_price=c["target_price"], round_number=c.get("round", 0), outcome=outcome, ) diff --git a/agent/tests/negotiation_demo.html b/agent/tests/negotiation_demo.html deleted file mode 100644 index ff7d852..0000000 --- a/agent/tests/negotiation_demo.html +++ /dev/null @@ -1,142 +0,0 @@ - - - - - -Negosium Agent — 협상 채팅 (테스트) - - - -
-

Negosium Agent — 협상 채팅 /v1/chat

-

KT 구매자 관점: 협력사(판매자)가 제시가를 입력 → 앵커가 이하면 우선협상(타결), 초과면 카드로 인하 협상(카드 소진까지). 낮게 매입할수록 KT 이득.

-
- -
-
- - - - - -
- -
-
-
-
- - - - diff --git a/agent/tests/test_agent_integration.py b/agent/tests/test_agent_integration.py index 21afeb8..3539d69 100644 --- a/agent/tests/test_agent_integration.py +++ b/agent/tests/test_agent_integration.py @@ -33,8 +33,8 @@ async def test_4_1_honors_backend_session_id(db_engine): svc = ChatService() # 첫 턴: backend 의 session_id 를 그대로 키로 써야 함 (새 uuid 발급 X) - r = await svc.chat(eng, Req_Chat(session_id=BACKEND_SESSION_ID, rq_type="재협상", - target_price=10000, anchor_price=9900)) + # 컨텍스트는 DB 조회(행 없음 → 기본값 폴백: target=10000, anchor=9900) + r = await svc.chat(eng, Req_Chat(session_id=BACKEND_SESSION_ID)) assert r.session_id == BACKEND_SESSION_ID assert r.step == "서비스안내" @@ -62,8 +62,7 @@ async def test_4_4_company_id_chat_end_to_end(db_engine): reset_sessions() eng = await _reg().get_engine(COMPANY_ID) # 자동 온보딩 테넌트 svc = ChatService() - r = await svc.chat(eng, Req_Chat(session_id=BACKEND_SESSION_ID, rq_type="재협상", - target_price=10000, anchor_price=9900)) + r = await svc.chat(eng, Req_Chat(session_id=BACKEND_SESSION_ID)) assert r.session_id == BACKEND_SESSION_ID and r.step == "서비스안내" # 카드선택 턴까지 진행 → company_id 스코프로 학습 기록 for ui in ["확인", "예", "확인", "11000", "예"]: diff --git a/agent/tests/test_context_loader.py b/agent/tests/test_context_loader.py new file mode 100644 index 0000000..9d6a3a5 --- /dev/null +++ b/agent/tests/test_context_loader.py @@ -0,0 +1,202 @@ +"""NegotiationContextLoader 검증 — Req_Chat 슬림화 후 세션 시작 컨텍스트 DB 조회. + +backend 소유 스키마(negotiation.sessions / quotation.quotations / partner.items / partner.suppliers)에 +실데이터를 넣고, agent 가 session_id 만으로 rq_type·목표가·앵커링가·품목가·매출액(total_revenue)· +유통코드·파트너 유형을 확정하는지 검증한다. 삽입 행은 테스트 종료 시 삭제. +""" + +import uuid +from datetime import datetime, timedelta, timezone + +import pytest +from sqlalchemy import column, delete, insert, table + +from common.database.db_session_manager import DB_SESSION_MNG +from common.enums import DBType, ErrorType +from negotiation.chat.service.chat_session_repository import ChatSessionRepository +from router.v1.chat.protocol import Req_Chat +from services.chat_service import ChatService, reset_sessions +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") + +# 삽입용 테이블 구성(backend 소유 스키마 — 테스트 데이터 셋업 전용). +_T_SESSIONS = table( + "sessions", + column("session_id"), column("quotation_id"), column("item_id"), column("supplier_id"), + column("qt_number"), column("qt_round"), column("qt_type"), column("target_price"), + column("anchoring_price"), column("status"), column("end_time"), + schema="negotiation", +) +_T_QUOTATIONS = table( + "quotations", + column("qt_id"), column("user_id"), column("qt_setting_id"), column("version_id"), + column("name"), column("number"), column("type"), column("status"), + column("start_time"), column("end_time"), column("supplier_type"), + schema="quotation", +) +_T_ITEMS = table( + "items", + column("item_id"), column("company_id"), column("user_id"), column("name"), column("price"), + schema="partner", +) +_T_SUPPLIERS = table( + "suppliers", + column("supplier_id"), column("company_id"), column("user_id"), column("name"), column("total_revenue"), + schema="partner", +) + + +@pytest.mark.asyncio +async def test_context_loaded_from_db(db_engine): + """세션 시작 시 협상 컨텍스트가 요청이 아니라 DB 에서 확정된다.""" + reset_sessions() + sid, sid2 = uuid.uuid4(), uuid.uuid4() + qid, iid = uuid.uuid4(), uuid.uuid4() + sup1, sup2 = uuid.uuid4(), uuid.uuid4() + now = datetime.now(timezone.utc) + + def _ins_item(s): + return DB_SESSION_MNG.add(s, insert(_T_ITEMS).values( + item_id=iid, company_id=uuid.uuid4(), user_id=uuid.uuid4(), + name="로더 테스트 상품", price=5000, + )) + + def _ins_supplier(s): + return DB_SESSION_MNG.add(s, insert(_T_SUPPLIERS).values( + supplier_id=sup1, company_id=uuid.uuid4(), user_id=uuid.uuid4(), + name="로더 테스트 협력사", total_revenue=55_000_000, # 매출액(KTC total_revenue 미러) + )) + + def _ins_quote(s): + return DB_SESSION_MNG.add(s, insert(_T_QUOTATIONS).values( + qt_id=qid, user_id=uuid.uuid4(), qt_setting_id=uuid.uuid4(), version_id=uuid.uuid4(), + name="로더 테스트", number="QT-LOADER-TEST", type=3, status=2, + start_time=now, end_time=now + timedelta(days=1), + supplier_type=2, # manufacture(제조) → 유통 코드 "A" + )) + + def _ins_sess(s, session_id, supplier_id): + return DB_SESSION_MNG.add(s, insert(_T_SESSIONS).values( + session_id=session_id, quotation_id=qid, item_id=iid, supplier_id=supplier_id, + qt_number="QT-LOADER-TEST", qt_round=1, + qt_type=3, # 신규협상(1:1) → 재협상 스크립트 + target_price=20000, anchoring_price=19000, # 생성 시 박제된 앵커 + status=2, end_time=now + timedelta(days=1), + )) + + err = await DB_SESSION_MNG.execute_lambda_run( + [DBType.MAIN.value], + [_ins_item, _ins_supplier, _ins_quote, + lambda s: _ins_sess(s, sid, sup1), + lambda s: _ins_sess(s, sid2, sup2)], # 같은 상품에 공급사 2곳 → MULTIPLE + ) + assert err == ErrorType.SUCCESS + + try: + reg = TenantEngineRegistry(loader=TenantConfigLoader(tenants_dir=_TENANTS_DIR, cache_ttl_seconds=0)) + eng = await reg.get_engine("ktcommerce") + r = await ChatService().chat(eng, Req_Chat(session_id=str(sid))) + assert r.session_id == str(sid) and r.step == "서비스안내" + + saved = await ChatSessionRepository(eng.company_id).get(str(sid)) + assert saved is not None + assert saved.rq_type == "재협상" # qt_type=3(신규협상 1:1) + c = saved.context + assert c["target_price"] == 20000 + assert c["anchor_price"] == 19000 # sessions.anchoring_price(박제) + assert c["item_price"] == 5000 + assert c["revenue_amount"] == 55_000_000.0 # 매출액 = suppliers.total_revenue + assert c["distribution_code"] == "A" # supplier_type=2(제조) → A + assert c["partner_count"] == 2 # 공급사 2곳 → MULTIPLE + finally: + await DB_SESSION_MNG.execute_lambda_run( + [DBType.MAIN.value], + [lambda s: DB_SESSION_MNG.add(s, delete(_T_SESSIONS).where(_T_SESSIONS.c.quotation_id == qid)), + lambda s: DB_SESSION_MNG.add(s, delete(_T_QUOTATIONS).where(_T_QUOTATIONS.c.qt_id == qid)), + lambda s: DB_SESSION_MNG.add(s, delete(_T_ITEMS).where(_T_ITEMS.c.item_id == iid)), + lambda s: DB_SESSION_MNG.add(s, delete(_T_SUPPLIERS).where(_T_SUPPLIERS.c.supplier_id == sup1))], + ) + + +@pytest.mark.asyncio +async def test_null_anchoring_falls_back_to_target(db_engine): + """박제 앵커(anchoring_price)가 NULL 이면 무할인 폴백 anchor=target (앵커링 v1.2 정책 승계).""" + reset_sessions() + sid, qid, iid = uuid.uuid4(), uuid.uuid4(), uuid.uuid4() + now = datetime.now(timezone.utc) + + def _ins_sess(s): + return DB_SESSION_MNG.add(s, insert(_T_SESSIONS).values( + session_id=sid, quotation_id=qid, item_id=iid, supplier_id=uuid.uuid4(), + qt_number="QT-LOADER-NULL", qt_round=1, qt_type=1, + target_price=30000, anchoring_price=None, # 박제 없음(데이터 이상 경로) + status=2, end_time=now + timedelta(days=1), + )) + + err = await DB_SESSION_MNG.execute_lambda_run([DBType.MAIN.value], [_ins_sess]) + assert err == ErrorType.SUCCESS + try: + reg = TenantEngineRegistry(loader=TenantConfigLoader(tenants_dir=_TENANTS_DIR, cache_ttl_seconds=0)) + eng = await reg.get_engine("ktcommerce") + await ChatService().chat(eng, Req_Chat(session_id=str(sid))) + saved = await ChatSessionRepository(eng.company_id).get(str(sid)) + assert saved is not None + assert saved.context["target_price"] == 30000 + assert saved.context["anchor_price"] == 30000 # 무할인 폴백: anchor = target + finally: + await DB_SESSION_MNG.execute_lambda_run( + [DBType.MAIN.value], + [lambda s: DB_SESSION_MNG.add(s, delete(_T_SESSIONS).where(_T_SESSIONS.c.session_id == sid))], + ) + + +@pytest.mark.asyncio +async def test_loader_with_crud_double(db_engine): + """INegoContextCRUD 인터페이스에 더블을 주입해 쿼리 없이 판정 로직만 검증 (backend crud 패턴).""" + from negotiation.chat.infra.repository.nego_context_crud import INegoContextCRUD + from negotiation.chat.service.negotiation_context_loader import NegotiationContextLoader + from negotiation.qtable.domain.model.snapshot import PartnerType + + class _FakeCRUD(INegoContextCRUD): + async def get_session_row(self, cdb, session_id): + # (qt_type, target, anchoring_price, item_id, quotation_id, supplier_id) — 재견적(2)·앵커 미박제 + return ErrorType.SUCCESS, (2, 50000, None, uuid.uuid4(), uuid.uuid4(), uuid.uuid4()) + + async def get_item_price(self, cdb, item_id): + return ErrorType.SUCCESS, 7000 + + async def get_supplier_total_revenue(self, cdb, supplier_id): + return ErrorType.SUCCESS, 12_000_000.0 + + async def get_quotation_supplier_type(self, cdb, quotation_id): + return ErrorType.SUCCESS, 3 # sole_agency(총판) → "B" + + async def count_item_suppliers(self, cdb, item_id): + return ErrorType.SUCCESS, 0 # 연결 협력사 없음 → NONE + + ctx = await NegotiationContextLoader(crud=_FakeCRUD()).load(str(uuid.uuid4())) + assert ctx is not None + assert ctx.rq_type == "재견적" # qt_type=2(1:N) + assert ctx.target_price == 50000 + assert ctx.anchor_price == 50000 # 미박제 → 무할인 폴백(anchor=target) + assert ctx.item_price == 7000 + assert ctx.revenue_amount == 12_000_000.0 + assert ctx.distribution_code == "B" # supplier_type=3(총판) → B + assert ctx.partner_type is PartnerType.NONE + + +@pytest.mark.asyncio +async def test_context_falls_back_without_db_row(db_engine): + """DB 에 세션 행이 없으면(데모/직접 호출) 기본 컨텍스트로 폴백한다.""" + reset_sessions() + reg = TenantEngineRegistry(loader=TenantConfigLoader(tenants_dir=_TENANTS_DIR, cache_ttl_seconds=0)) + eng = await reg.get_engine("ktcommerce") + r = await ChatService().chat(eng, Req_Chat()) + saved = await ChatSessionRepository(eng.company_id).get(r.session_id) + assert saved is not None + c = saved.context + assert c["target_price"] == 10000 and c["anchor_price"] == 9900 + assert c["partner_count"] == 1 and c["distribution_code"] == "A" diff --git a/agent/tests/test_p7_chat.py b/agent/tests/test_p7_chat.py index 7b9c70d..6d991b6 100644 --- a/agent/tests/test_p7_chat.py +++ b/agent/tests/test_p7_chat.py @@ -31,7 +31,7 @@ async def _run(svc, eng, turns): """turns: user_input 리스트(첫 None=시작). 반환: 응답 리스트.""" sid, out = None, [] for ui in turns: - r = await svc.chat(eng, Req_Chat(session_id=sid, user_input=ui, rq_type="재협상")) + r = await svc.chat(eng, Req_Chat(session_id=sid, user_input=ui)) sid = r.session_id out.append(r) if r.chat_end: @@ -76,6 +76,48 @@ async def test_full_conversation_reaches_completion(db_engine): assert cnt >= 1 +@pytest.mark.asyncio +async def test_wildcard_1pct_accept_settles_at_offer_price(db_engine): + """와일드카드 1% 인하 제안을 '예'로 수락하면 합의가는 인하 제안가(offer_1pct)여야 한다. + + 회귀: 수락해도 input_price 가 직전 제시가로 남아 요약/입찰가가 원래 가격(예: 20000)으로 + 잡히던 버그 — settled_price 가 인하가(예: 19800)로 내려와야 한다. + """ + reset_sessions() + eng = await _reg().get_engine("ktcommerce") + svc = ChatService() + # anchor=9900(기본). 10000 은 anchor*1.02(10098) 이내 → wild_card_1pct 발동, offer_1pct=9900. + out = await _run(svc, eng, [None, "확인", "예", "확인", "10000", "예", "예", + "협상 내용을 확인했으며, 이의가 없음에 동의합니다."]) + steps = [r.step for r in out] + assert "wild_card_1pct" in steps + + # 수락 후 협상완료(요약 턴)·협상종료(종료 턴) 모두 합의가 = 인하 제안가 9900 + done = [r for r in out if r.step in ("협상완료", "협상종료")] + assert done and all(r.settled_price == 9900 for r in done) + + +def test_partner_type_enum_mapping(): + """PartnerType 계약값(0=NONE/1=SINGLE/2=MULTIPLE)과 수→유형 매핑.""" + from negotiation.qtable.domain.model.snapshot import PartnerType + + assert PartnerType.NONE == 0 and PartnerType.SINGLE == 1 and PartnerType.MULTIPLE == 2 + assert PartnerType.from_count(0) is PartnerType.NONE + assert PartnerType.from_count(1) is PartnerType.SINGLE + assert PartnerType.from_count(2) is PartnerType.MULTIPLE + assert PartnerType.from_count(7) is PartnerType.MULTIPLE + + +def test_acceptance_ratio_dynamic_calc(): + """가격 수용률 동적 계산 — 첫 제시가 대비 양보율. 첫 제시/미기록=0, 인상 시 0 클립.""" + calc = ChatService._acceptance_ratio + assert calc({}) == 0.0 # 가격 입력 전 + assert calc({"first_offer_price": 11000, "input_price": 11000}) == 0.0 # 첫 제시(양보 0) + assert calc({"first_offer_price": 11000, "input_price": 10200}) == pytest.approx(800 / 11000) + assert calc({"first_offer_price": 10000, "input_price": 12000}) == 0.0 # 인상(비정상) → 0 + assert calc({"input_price": 9800}) == 0.0 # 과거 세션 호환(첫 제시가 미기록) + + @pytest.mark.asyncio async def test_priority_completes_without_wildcard(db_engine): reset_sessions() @@ -95,7 +137,7 @@ async def test_tenant_brand_isolation(db_engine): reset_sessions() svc = ChatService() reg = _reg() - ik = await svc.chat(await reg.get_engine("imarketkorea"), Req_Chat(session_id=None, rq_type="재협상")) + ik = await svc.chat(await reg.get_engine("imarketkorea"), Req_Chat(session_id=None)) assert "데모상사 B" in ik.script diff --git a/agent/tests/test_p8_session_db.py b/agent/tests/test_p8_session_db.py index 68f7c9b..f7b0f8a 100644 --- a/agent/tests/test_p8_session_db.py +++ b/agent/tests/test_p8_session_db.py @@ -27,19 +27,19 @@ async def test_session_persists_and_resumes_across_instances(db_engine): reg = _eng() eng = await reg.get_engine("ktcommerce") - # 인스턴스 1: 협상 시작 + 몇 턴 진행 + # 인스턴스 1: 협상 시작 + 몇 턴 진행 (컨텍스트는 DB 조회 — 행이 없으므로 기본값 폴백) svc1 = ChatService() - r = await svc1.chat(eng, Req_Chat(rq_type="재협상", target_price=10000, anchor_price=8000)) + 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 에 저장됐는지 확인 + # 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"] == 8000 + assert saved.context["anchor_price"] == 9900 # 인스턴스 2 (재시작/다른 워커 모사): 같은 session_id 로 이어가기 svc2 = ChatService() @@ -53,7 +53,7 @@ async def test_session_company_scoped(db_engine): reg = _eng() eng = await reg.get_engine("ktcommerce") svc = ChatService() - r = await svc.chat(eng, Req_Chat(rq_type="재협상")) + r = await svc.chat(eng, Req_Chat()) sid = r.session_id # 자사(ktcommerce)로는 조회됨 diff --git a/backend/services/agent_client.py b/backend/services/agent_client.py index 92d173c..512c1cf 100644 --- a/backend/services/agent_client.py +++ b/backend/services/agent_client.py @@ -32,26 +32,24 @@ class AgentTurn: indicator_value: Optional[float] = None # agent 가 직접 내려주는 표현 폼(summaryRSP/CM·rejectRSP/CM·indicator). 없으면 backend 가 step+qt_type 으로 폴백. bot_chat_type: Optional[str] = None + # 성공 확정 이후 턴에 agent 가 내려주는 합의가. 와일드카드 1% 인하 수락처럼 유저가 직접 + # 입력하지 않은 가격으로 타결될 수 있어, 요약 표시가·입찰가 확정 시 이 값을 최우선 사용한다. + settled_price: Optional[int] = None ok: bool = True # agent 호출 성공 여부 (False 면 CHAT_AGENT_UNAVAILABLE) timed_out: bool = False # 타임아웃 여부. True 면 agent 가 이미 진행했을 수 있어 desync 위험 → 별도 처리. @dataclass class AgentChatContext: - """새 세션 시작 시 agent 에 주입하는 협상 컨텍스트. 기존 세션이면 user_input 만 의미 있다.""" + """agent 호출 컨텍스트. 협상 컨텍스트(목표가/앵커/품목가/매출액/유통코드/파트너 유형/수용률)는 + 더 이상 전송하지 않는다 — agent 가 session_id 로 DB(negotiation.sessions 등)에서 직접 + 조회·계산한다(Req_Chat 슬림화). rq_type/target_price 는 backend 자체 로직(표현 폴백·테스트 + 더블)용으로만 유지하며 전송되지 않는다. + """ tenant_id: str # X-Tenant-ID = 견적(갑) 회사 company_id - rq_type: str = "재협상" # 재협상 | 재견적 - target_price: int = 0 # 갑 목표 매입가(원) - anchor_price: int = 0 # 앵커링가(목표가보다 낮음). 세션 생성 시 박제된 sessions.anchoring_price. - item_price: int = 0 # 기존 공급가(품목 기준가). agent 가격협상_확인 인하율 산출용. - # 핸드오프 #4: agent 의 RL 상태(state) 계산 입력. - # partner_count 는 견적당 세션 수로 산출(실데이터). 나머지 3개는 우리 스키마에 데이터 소스가 없어 - # 기본값으로 보낸다 → agent 가 실제값을 받으려면 backend 스키마에 컬럼 추가 필요(HANDOFF.md ④). - revenue_amount: float = 20_000_000 # 매출액(원) — DB 소스 없음(기본값) - distribution_code: str = "A" # 유통 코드(agent config code_map 키) — DB 소스 없음(기본값) - partner_count: int = 1 # 공급사 수 — 견적당 세션 수로 산출 - acceptance_ratio: float = 0.05 # 가격 수용률 0~1 — DB 소스 없음(기본값) + rq_type: str = "재협상" # 재협상(1:1) | 재견적(1:N) — backend 로컬 용도 + target_price: int = 0 # 갑 목표 매입가(원) — backend 로컬 용도 # 핸드오프 #1: backend 가 보는 현재 step(직전 봇 step). agent 가 자기 세션 step 과 대조해 desync 감지에 쓸 수 있다. client_step: Optional[str] = None extra: dict = field(default_factory=dict) @@ -72,16 +70,9 @@ class HttpAgentClient(IAgentClient): body = { "session_id": session_id, # 핸드오프 #1: agent 가 이 값을 세션 키로 그대로 사용해야 함 - "rq_type": ctx.rq_type, + # 협상 컨텍스트(rq_type/목표가/앵커/품목가/매출액/유통코드/파트너 유형/수용률)는 보내지 + # 않는다 — agent 가 session_id 로 DB 에서 직접 조회·계산한다(NegotiationContextLoader). "user_input": user_input, - "target_price": ctx.target_price, - "anchor_price": ctx.anchor_price, - "item_price": ctx.item_price, - # 핸드오프 #4: RL state 입력 (agent Req_Chat 이 받는 필드). 현재 기본값. - "revenue_amount": ctx.revenue_amount, - "distribution_code": ctx.distribution_code, - "partner_count": ctx.partner_count, - "acceptance_ratio": ctx.acceptance_ratio, # 핸드오프 #1: backend 가 보는 직전 step. agent 가 desync 감지에 사용(미구현 시 무시됨). "client_step": ctx.client_step, } @@ -111,6 +102,7 @@ class HttpAgentClient(IAgentClient): card_id=data.get("card_id"), indicator_value=data.get("indicator_value"), bot_chat_type=data.get("bot_chat_type"), + settled_price=data.get("settled_price"), ok=True, ) diff --git a/backend/services/chat_service.py b/backend/services/chat_service.py index 9ebedc3..ddcf4f3 100644 --- a/backend/services/chat_service.py +++ b/backend/services/chat_service.py @@ -361,7 +361,9 @@ class ChatService: summary = None if bot_chat_type in ("summaryRSP", "summaryCM"): final_price = price if price is not None else (sess.bid_price or sess.target_price) - summary = await self._build_summary(sess, quote, item, final_price, last_price) + # 합의가는 agent 가 내려준 settled_price 최우선 (와일드카드 1% 인하 수락 등 + # 유저 미입력 가격 타결 케이스 — 마지막 유저 제시가와 다를 수 있다). + summary = await self._build_summary(sess, quote, item, final_price, turn.settled_price or last_price) # 봇 메시지 + 종료 시 확정(성공=DONE+입찰가 / 실패=REJECTED+거부사유·제시가). 한 트랜잭션. bot_msg = self._build_bot_chat(sess, seq=max_seq + 2, turn=turn, bot_chat_type=bot_chat_type, summary=summary) @@ -374,10 +376,11 @@ class ChatService: if turn.chat_end: if turn.outcome == "success": new_status = SessionStatus.DONE.value - # 입찰가 = 이번 턴 가격(보통 None) → 마지막 제시가 → 목표가 순으로 확정. - bid = price if price is not None else (last_price if last_price else sess.target_price) + # 입찰가 = agent 합의가(settled_price, 와일드카드 수락 등) → 이번 턴 가격(보통 None) + # → 마지막 제시가 → 목표가 순으로 확정. + bid = turn.settled_price or (price if price is not None else (last_price if last_price else sess.target_price)) # ⑤ 협상된 제시가가 하나도 없어 목표가로 폴백하면, 합의가가 실제 협상과 다를 수 있어 경고. - if price is None and not last_price: + if not turn.settled_price and price is None and not last_price: LOG.w(f"[chat] 합의가 폴백→목표가 session_id={sess.session_id} bid={bid} " f"— 협상 중 가격 제시가 기록되지 않음(프론트 user_input_type='price' 누락 의심)") funcs.append(lambda s: self.chat_crud.finalize_session(s, sess.session_id, new_status, bid_price=bid)) @@ -417,52 +420,18 @@ class ChatService: tenant_id = str(item.company_id) else: LOG.w(f"[chat] tenant_id 해석 실패(item.company_id 없음) session_id={sess.session_id} — agent 400 위험") - rq_type = "재협상" if sess.qt_type == 1 else "재견적" + # 1:1 견적유형(1=renego, 3=new_nego) → 재협상, 1:N(2=requote, 4=new_quote) → 재견적. + # agent 도 sessions.qt_type 으로 동일 판별(NegotiationContextLoader) — backend 로컬 용도. + rq_type = "재협상" if sess.qt_type in (1, 3) else "재견적" target_price = int(sess.target_price or 0) - # 앵커가: 세션 생성 시 박제된 값(anchoring_price)을 그대로 사용 — 협상 중 불변. - anchor = await self._resolve_anchor_price(sess, target_price) - # 공급사 수: 같은 견적에 속한 세션 수(재협상=1, 재견적=N). agent partner 차원(single/multiple/none) 입력. - partner_count = await self._count_partners(sess) - # revenue_amount / distribution_code / acceptance_ratio 는 현재 스키마에 데이터 소스가 없어 - # AgentChatContext 기본값으로 보낸다(HANDOFF #4 — 컬럼 추가/소스 합의 필요). - # 기존 공급가(품목 기준가) — agent 가격협상_확인 인하율(discount_rate) 산출 입력. - item_price = int(item.price) if item is not None and item.price else 0 + # 협상 컨텍스트(앵커가/품목가/매출액/유통코드/파트너 유형/수용률)는 더 이상 계산·전송하지 않는다 — + # agent 가 session_id 로 DB(negotiation.sessions·partner.items/suppliers·quotations)에서 직접 조회한다. + # (앵커가 박제·무할인 폴백 정책 — schedules/anchoring/docs/개발용.md §9.2 — 은 agent loader 가 승계.) return AgentChatContext( tenant_id=tenant_id, rq_type=rq_type, - target_price=target_price, anchor_price=anchor, partner_count=partner_count, - item_price=item_price, client_step=client_step, + target_price=target_price, client_step=client_step, ) - async def _resolve_anchor_price(self, sess, target_price: int) -> int: - """세션에 박제된 앵커가(anchoring_price — negodata 가 생성 시 기록)를 그대로 사용. - - 박제값 사용이 정상 경로다: 협상 진행 중 앵커링 배치 조정·재기동이 껴도 앵커가 흔들리지 않는다 - ("제안 당시 값" 판정의 전제 — schedules/anchoring/docs/개발용.md §9.2). backend 는 앵커를 계산하지 않는다. - 박제가 없으면(데이터 이상 — 사실상 발생하지 않음) 무할인 폴백 anchor=target + WARN. - 이때 박제하지 않으므로 해당 세션은 앵커링 집계에서 자동 제외(EXCLUDED)된다 — 학습 무오염. - """ - if not target_price: - return 0 - if sess.anchoring_price is not None: - return int(sess.anchoring_price) - LOG.w(f"[chat] 앵커가 박제 없음 session_id={sess.session_id} — 무할인 폴백(anchor=target), 집계 제외") - return target_price - - async def _count_partners(self, sess) -> int: - """같은 견적(quotation_id)에 속한 협상 세션 수 = 참여 공급사 수. 실패 시 1 폴백.""" - def _q(s): - stmt = ( - select(func.count()) - .select_from(sessions) - .where(sessions.quotation_id == sess.quotation_id, sessions.deleted == False) # noqa: E712 - ) - return DB_SESSION_MNG.execute(s, stmt) - - err_type, rows = await DB_SESSION_MNG.execute_lambda(sessions.DBType(), DBWRType.DB_READ.value, _q) - if err_type != ErrorType.SUCCESS or not rows or not rows[0]: - return 1 - return int(rows[0]) - async def _last_user_price(self, sess) -> Optional[int]: """세션에서 가장 최근 유저 제시가(negotiation.chats.target_price>0). 없으면 None.""" def _q(s): diff --git a/backend/tests/test_anchoring_chat.py b/backend/tests/test_anchoring_chat.py index 7470b58..3aaede0 100644 --- a/backend/tests/test_anchoring_chat.py +++ b/backend/tests/test_anchoring_chat.py @@ -32,15 +32,17 @@ class _AnchorAgent(IAgentClient): """결정론적 더블: 서비스안내(오프닝) → 가격 입력 요청 → 합의 종료. 앵커보다 높은 가격이면 같은 step 을 반복(마지막 제시가 덮어쓰기 검증용). - 매 턴 수신한 ctx.anchor_price 를 기록해 backend 의 앵커 해석을 관찰한다. - (script 에 앵커가 보이는 건 테스트 관찰 편의일 뿐 — 실제 agent 는 비노출.) + 앵커 해석(박제값 소비/무할인 폴백)은 agent 가 DB 에서 직접 수행하도록 이관됐다 + (NegotiationContextLoader — agent tests/test_context_loader.py 가 검증). 여기 더블은 + 자체 앵커 상수로 수락 여부만 판정하고, backend 가 보내는 ctx.target_price 를 관찰한다. """ - def __init__(self): - self.seen_anchors: list[int] = [] + def __init__(self, anchor: int = ANCHOR): + self.anchor = anchor + self.seen_targets: list[int] = [] async def chat(self, session_id, user_input, ctx) -> AgentTurn: - self.seen_anchors.append(ctx.anchor_price) + self.seen_targets.append(ctx.target_price) sid = session_id or "fake-session" if user_input is None: # 오프닝(턴0) return AgentTurn(session_id=sid, step="서비스안내", client_step="서비스안내", @@ -48,10 +50,10 @@ class _AnchorAgent(IAgentClient): input_options=["네, 시작할게요"]) if ctx.client_step == "서비스안내": return AgentTurn(session_id=sid, step="기존가격제시", client_step="기존가격제시", - script=f"저희가 제안드리는 첫 목표 가격은 {ctx.anchor_price}원입니다. " + script=f"저희가 제안드리는 첫 목표 가격은 {self.anchor}원입니다. " f"제안하실 가격을 입력해 주세요.", input_mode="price") price = _parse_price(user_input) - if price is not None and price <= ctx.anchor_price: + if price is not None and price <= self.anchor: return AgentTurn(session_id=sid, step="협상종료", client_step="협상종료", script=f"{price:,}원으로 합의되었습니다.", chat_end=True, outcome="success") return AgentTurn(session_id=sid, step="기존가격제시", client_step="기존가격제시", @@ -159,7 +161,7 @@ async def test_snapshot_consumed_and_last_offer_recorded(client, db_engine, anch assert r.status_code == 200 and r.json()["message"]["step"] == "기존가격제시" row = await _anchor_columns(db_engine, sid) assert row.last_offer_price is None - assert _fake_agent.seen_anchors[-1] == ANCHOR # backend 가 박제값을 그대로 전달 + assert _fake_agent.seen_targets[-1] == TARGET # backend 로컬 컨텍스트(target) 전달 확인 r = await _send(client, token, sid, "99,500", "price") # 앵커 초과 → 같은 step 반복 assert r.json()["message"]["step"] == "기존가격제시" @@ -175,7 +177,8 @@ async def test_snapshot_consumed_and_last_offer_recorded(client, db_engine, anch assert (row.anchoring_price, row.anchoring_value) == (ANCHOR, 10) # 박제 불변 -# ── 폴백 경로: 박제 NULL → 무할인(anchor=target) + 미박제 유지 ── +# ── 폴백 경로: 박제 NULL 세션도 backend 채팅 경로가 정상 동작 + 미박제 유지 ── +# (무할인 폴백 anchor=target 자체는 agent NegotiationContextLoader 가 수행/검증 — agent 테스트 소관) async def test_null_snapshot_falls_back_to_target(client, db_engine, anchor_seed, _fake_agent): sid = str(anchor_seed["sids"]["N"]) token = await _login_token(client) @@ -183,11 +186,13 @@ async def test_null_snapshot_falls_back_to_target(client, db_engine, anchor_seed await _messages(client, token, sid) r = await _send(client, token, sid, "네, 시작할게요") assert r.json()["message"]["step"] == "기존가격제시" - assert _fake_agent.seen_anchors[-1] == TARGET # 무할인 폴백: anchor = target - r = await _send(client, token, sid, "97,000", "price") # 가격 입력(폴백 앵커 이하 → 종료) + r = await _send(client, token, sid, "97,000", "price") # 가격 입력(더블 앵커 이하 → 종료) assert r.json()["session_status"] == 3 row = await _anchor_columns(db_engine, sid) + assert (row.anchoring_price, row.anchoring_value) == (None, None) # 미박제 유지(집계 제외 조건) + assert row.bid_price == 97_000 + row = await _anchor_columns(db_engine, sid) assert row.anchoring_price is None # backend 는 박제하지 않음(앵커 없음 → 집계 제외) assert row.anchoring_value is None assert row.last_offer_price == 97_000 # 가격 흔적 기록은 정상 동작