refactor: 통관진행조회 가능 여부 질문 시 필요한 파라미터를 안내하도록 수정

Author: dongyeon1031

core/customs_tracking/agent/customs_tracking_agent.py

@@ -12,15 +12,15 @@ def customs_tracking_agent(state: CustomsAgentState) -> CustomsAgentState:
     llm = get_llm()
 
     prompt = ChatPromptTemplate.from_messages([
-        ("system", """
-        당신은 통관 및 배송 추적 전문가입니다.
-        사용할 수 있는 도구 목록이 아래 제공됩니다.
-
-        다음 사용자 쿼리를 분석하여 정확히 하나의 카테고리로 분류해주세요:
+        ("system", """ 당신은 통관 진행 조회 서비스를 사용자에게 직접 제공하는 관세청 도우미입니다.
+        아래 도구를 통해 실제 통관 진행 정보를 조회할 수 있으므로, 가능한 경우에는 반드시 도구를 사용하여 답변하세요.
+        
+        사용자의 질문이 화물번호나 BL 정보를 포함하지 않은 경우에는 다음 문장으로 안내하세요:
+        "통관 진행을 조회하기 위해서는 화물번호 또는 BL 번호가 필요합니다. 화물번호(MT 번호) 또는 연도, MBL, HBL 번호를 제공해 주시면 통관 진행 정보를 조회해 드리겠습니다."
+        
+        도구 목록:
         1. get_cargo_progress_details_by_mt: 화물번호를 통한 통관 진행 조회
         2. get_cargo_progress_details_by_bl: 연도, MBL, HBL 번호를 통한 통관 진행 조회
-
-        도구를 이용해 통관 상태를 확인하고, 호출 결과는 가공하지 말고 그대로 반환하세요.
         """),
         MessagesPlaceholder(variable_name="messages"),
         ("system", "{agent_scratchpad}"),
@@ -30,6 +30,7 @@ def customs_tracking_agent(state: CustomsAgentState) -> CustomsAgentState:
         tools=tools,
         prompt=prompt
     )
+    print(prompt)
 
     agent_executor = AgentExecutor(
         agent=agent,
@@ -56,6 +57,7 @@ def customs_tracking_agent(state: CustomsAgentState) -> CustomsAgentState:
 
         state["final_response"] = ""
     else:
+        print(result["output"])
         state["final_response"] = result["output"]
 
     return state

core/graphs/workflow.py

@@ -37,7 +37,7 @@ def create_customs_graph():
     )
 
     # 각 에이전트에서 final_agent로
-    workflow.add_edge("customs_tracking", "final_agent")
+    workflow.add_edge("customs_tracking", END)
     workflow.add_edge("tariff_prediction", "final_agent")
     workflow.add_edge("qna", "final_agent")
     # final_agent에서 종료점으로