feat: 통관진행조회 반환타입 수정 (str->progress_details)

Author: dongyeon1031

app/service.py

@@ -8,8 +8,10 @@ def run_model(question: str) -> "Response":
         ex : "안녕, 너는 뭘 할 수 있어?"
         예시 코드는 아래와 같습니다.
     """
-    reply = run_customs_agent(question)
+    state = run_customs_agent(question)
     return Response(
-        reply=reply,
+        reply=state.get("final_response"),
+        progress_details=state.get("progress_details"),
+        error_reason=state.get("error_reason"),
         success=True
     )

core/customs_tracking/agent/customs_tracking_agent.py

@@ -1,6 +1,7 @@
 from langchain.agents import create_openai_functions_agent, AgentExecutor
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 
+from core.customs_tracking.dto.cargo_progress_result import CargoProgressResult
 from core.customs_tracking.tools.get_cargo_progress_details import get_cargo_progress_details_by_bl, get_cargo_progress_details_by_mt
 from core.shared.states.states import CustomsAgentState
 from core.shared.utils.llm import get_llm
@@ -14,7 +15,7 @@ def customs_tracking_agent(state: CustomsAgentState) -> CustomsAgentState:
         ("system", """
         당신은 통관 및 배송 추적 전문가입니다.
         사용할 수 있는 도구 목록이 아래 제공됩니다.
-        
+
         다음 사용자 쿼리를 분석하여 정확히 하나의 카테고리로 분류해주세요:
         1. get_cargo_progress_details_by_mt: 화물번호를 통한 통관 진행 조회
         2. get_cargo_progress_details_by_bl: 연도, MBL, HBL 번호를 통한 통관 진행 조회
@@ -29,12 +30,32 @@ def customs_tracking_agent(state: CustomsAgentState) -> CustomsAgentState:
         tools=tools,
         prompt=prompt
     )
-    agent_executor = AgentExecutor(agent=agent, tools=tools)
+
+    agent_executor = AgentExecutor(
+        agent=agent,
+        tools=tools,
+        handle_parsing_errors=True,
+        return_intermediate_steps=True
+    )
 
     result = agent_executor.invoke({
         "input": state["query"],
         "messages": state["messages"]
     })
 
-    state["final_response"] = result["output"]
+    steps = result.get("intermediate_steps", [])
+    if steps:
+        _, tool_result = steps[-1]
+        # tool_result가 dict라면 아래처럼 처리
+        if isinstance(tool_result, CargoProgressResult):
+            state["progress_details"] = tool_result.progress_details
+            state["error_reason"] = tool_result.error_reason
+        else:
+            state["progress_details"] = None
+            state["error_reason"] = None
+
+        state["final_response"] = ""
+    else:
+        state["final_response"] = result["output"]
+
     return state

core/graphs/runner.py

@@ -4,7 +4,7 @@
 from core.shared.states.states import CustomsAgentState
 
 
-def run_customs_agent(query: str) -> str:
+def run_customs_agent(query: str) -> CustomsAgentState:
     """관세청 에이전트를 실행합니다."""
 
     # 그래프 생성
@@ -16,10 +16,12 @@ def run_customs_agent(query: str) -> str:
         query=query,
         intent=None,
         final_response="",
-        intermediate_results={}
+        intermediate_results={},
+        error_reason=None,
+        progress_details=None
     )
 
     # 그래프 실행
     result = app.invoke(initial_state)
 
-    return result["final_response"]
+    return result

core/shared/states/states.py

@@ -1,14 +1,18 @@
 # 상태 정의
 from typing import List, Optional, Literal, Dict, Any
+
 from typing_extensions import TypedDict
 
 from langchain_core.messages import BaseMessage
 
+from core.customs_tracking.dto.progress_detail import ProgressDetail
 
 class CustomsAgentState(TypedDict):
     """관세청 에이전트 상태"""
     messages: List[BaseMessage]
     query: str
     intent: Optional[Literal["customs_tracking", "tariff_prediction", "qna"]]
     final_response: str
-    intermediate_results: Dict[str, Any]
+    intermediate_results: Dict[str, Any]
+    progress_details: Optional[List[ProgressDetail]]
+    error_reason: Optional[str]