fix: 모델 state랑 response에서 세션 제거

Author: hanjuhn

app/dto/response.py

@@ -10,7 +10,6 @@ class Response(BaseModel):
     success: Optional[bool] = None
     progress_details: Optional[List[ProgressDetail]] = None
     error_reason: Optional[str] = None
-    session_id: Optional[str] = None
 
     @staticmethod
     def cargo_progres_result_to_response(result: CargoProgressResult) -> "Response":

app/routes.py

@@ -40,10 +40,6 @@
                     'answer': {
                         'type': 'string',
                         'example': '이 물건은 8%의 부가세가 부과됩니다.'
-                    },
-                    'session_id': {
-                        'type': 'string',
-                        'example': 'uuid-1234'
                     }
                 }
             }
@@ -54,13 +50,6 @@ def predict():
     request_data = Request(**request.get_json())
     session_id = request_data.session_id or request.cookies.get("session_id")
     answer = run_model(question=request_data.message, session_id=session_id)
-    # 응답 dict에서 session_id 제거
     answer_dict = answer.model_dump()
-    answer_dict.pop("session_id", None)
     response = make_response(jsonify(answer_dict))
-    # 쿠키는 그대로 관리
-    if answer.session_id:
-        response.set_cookie("session_id", answer.session_id)
-    else:
-        response.delete_cookie("session_id")
     return response

app/service.py

@@ -29,6 +29,5 @@ def run_model(question: str, session_id: str = None) -> "Response":
         reply=ai_reply,
         progress_details=state.get("progress_details"),
         error_reason=state.get("error_reason"),
-        success=True,
-        session_id=session_id  # session_id를 응답에 포함
+        success=True
     )

core/graphs/runner.py

@@ -22,8 +22,7 @@ def run_customs_agent(query: str, session_id: str = None, messages=None) -> Cust
         final_response="",
         intermediate_results={},
         error_reason=None,
-        progress_details=None,
-        session_id=session_id
+        progress_details=None
     )
 
     # 그래프 실행

core/shared/router/intent_router.py

@@ -118,7 +118,7 @@ def intent_router(state: CustomsAgentState) -> CustomsAgentState:
         if prev_intent == "tariff_prediction":
             try:
                 from core.tariff_prediction.agent.tariff_prediction_agent import workflow_manager
-                workflow_manager.cleanup_session(state["session_id"])
+                workflow_manager.cleanup_session()
             except Exception:
                 pass
         return state
@@ -130,7 +130,7 @@ def intent_router(state: CustomsAgentState) -> CustomsAgentState:
         if prev_intent == "tariff_prediction" and keyword_intent != "tariff_prediction":
             try:
                 from core.tariff_prediction.agent.tariff_prediction_agent import workflow_manager
-                workflow_manager.cleanup_session(state["session_id"])
+                workflow_manager.cleanup_session()
             except Exception:
                 pass
         return state
@@ -144,7 +144,7 @@ def intent_router(state: CustomsAgentState) -> CustomsAgentState:
     if prev_intent == "tariff_prediction" and intent != "tariff_prediction":
         try:
             from core.tariff_prediction.agent.tariff_prediction_agent import workflow_manager
-            workflow_manager.cleanup_session(state["session_id"])
+            workflow_manager.cleanup_session()
         except Exception:
             pass
     return state

core/shared/states/states.py

@@ -15,5 +15,4 @@ class CustomsAgentState(TypedDict):
     final_response: str
     intermediate_results: Dict[str, Any]
     progress_details: Optional[List[ProgressDetail]]
-    error_reason: Optional[str]
-    session_id: Optional[str]  # 세션 ID 추가
+    error_reason: Optional[str]

core/tariff_prediction/agent/utils.py

@@ -3,11 +3,8 @@
 from core.tariff_prediction.constants import RESPONSE_MESSAGES
 
 def get_or_create_workflow(state: CustomsAgentState) -> TariffPredictionWorkflow:
-    session_id = state.get("session_id")
-    if session_id:
-        return workflow_manager.get_session(session_id)
-    else:
-        return TariffPredictionWorkflow()  # 임시 워크플로우로 단발성 처리
+    # session_id는 더 이상 사용하지 않음
+    return TariffPredictionWorkflow()  # 임시 워크플로우로 단발성 처리
 
 def extract_context_from_messages(messages: list) -> str:
     context = ""
@@ -40,5 +37,4 @@ def run_workflow_with_context(workflow: TariffPredictionWorkflow, state: Customs
 
 def build_response_state(state: CustomsAgentState, response: str) -> CustomsAgentState:
     state["final_response"] = response
-    # session_id는 외부에서 관리하므로 그대로 둠
     return state