Merge pull request #22 from hanjuhn/main

fix: intent router 수정

Author: dongyeon1031

core/shared/constants/__init__.py

@@ -1,23 +1,27 @@
 # Shared constants package
 
 from .constants import (
-    TARIFF_SESSION_KEYWORDS,
-    NUMBER_SELECTION_PATTERNS,
     TARIFF_PREDICTION_KEYWORDS,
     CUSTOMS_TRACKING_KEYWORDS,
+    TARIFF_SESSION_KEYWORDS,
+    NUMBER_SELECTION_PATTERNS,
+    QUESTION_PATTERNS,
     INTENT_CLASSIFICATION_PROMPT,
     INTENT_TYPES,
-    DEFAULT_INTENT
+    DEFAULT_INTENT,
+    SESSION_CHECK_MESSAGE_COUNT
 )
 
 from .error_codes import *
 
 __all__ = [
-    'TARIFF_SESSION_KEYWORDS',
-    'NUMBER_SELECTION_PATTERNS', 
     'TARIFF_PREDICTION_KEYWORDS',
     'CUSTOMS_TRACKING_KEYWORDS',
+    'TARIFF_SESSION_KEYWORDS',
+    'NUMBER_SELECTION_PATTERNS',
+    'QUESTION_PATTERNS',
     'INTENT_CLASSIFICATION_PROMPT',
     'INTENT_TYPES',
-    'DEFAULT_INTENT'
+    'DEFAULT_INTENT',
+    'SESSION_CHECK_MESSAGE_COUNT'
 ] 

core/shared/constants/constants.py

@@ -1,5 +1,20 @@
 # Shared constants for the entire application
 
+# 관세 예측 관련 키워드
+TARIFF_PREDICTION_KEYWORDS = [
+    '코드', 'hs', '관세', '세금', '세율', '관세 계산', '관세 예측', '세금 얼마', '관세 얼마',
+    '관세 계산해줘', '관세 예측해줘', '세금 예측', '예상 관세', '뭘 샀어', '뭐 샀어', '샀어', '구매',
+    '저기에', '저기', '그곳에', '그곳', '여기에', '여기', '저기에는', '그곳에는', '여기에는',
+    '적합하지', '맞지', '틀렸', '잘못', '다른 것', '다른 코드', '새로운', '새로', '다시', '재시도'
+]
+
+# 배송 추적 관련 키워드
+CUSTOMS_TRACKING_KEYWORDS = [
+    '운송장', '통관', '조회', '추적', '배송상태', '위치', '도착', '출고', '통관번호', '운송장번호'
+]
+
+# 의도 분류 관련 상수들
+
 # 관세 예측 세션 관련 키워드
 TARIFF_SESSION_KEYWORDS = [
     'hs6 코드 후보', 'hs10 코드 후보', '번호를 선택', '관세 계산', '관세 예측',
@@ -16,18 +31,12 @@
     r'^\d{4}\.\d{2}$',  # 8471.60 등 HS 코드
 ]
 
-# 관세 예측 관련 키워드
-TARIFF_PREDICTION_KEYWORDS = [
-    '코드', 'hs', '관세', '세금', '세율', '관세 계산', '관세 예측', '세금 얼마', '관세 얼마',
-    '관세 계산해줘', '관세 예측해줘', '세금 예측', '예상 관세', '뭘 샀어', '뭐 샀어', '샀어', '구매',
-    '없다', '코드가 없다', '재예측', '다시', '틀렸다', '맞지 않다', '다른', '새로', '다시 예측', '다른 코드', '적합하지 않다', '코드가 없어',
-    '저기에', '저기', '그곳에', '그곳', '여기에', '여기', '저기에는', '그곳에는', '여기에는',
-    '적합하지', '맞지', '틀렸', '잘못', '다른 것', '다른 코드', '새로운', '새로', '다시', '재시도'
-]
-
-# 배송 추적 관련 키워드
-CUSTOMS_TRACKING_KEYWORDS = [
-    '운송장', '통관', '조회', '추적', '배송상태', '위치', '도착', '출고', '통관번호', '운송장번호'
+# 질문 형태 감지 패턴
+QUESTION_PATTERNS = [
+    r'\?$',  # 물음표로 끝나는 경우
+    r'^어떻게', r'^무엇', r'^언제', r'^어디서', r'^왜', r'^누가',
+    r'^무슨', r'^어떤', r'^얼마나', r'^어디', r'^언제', r'^왜',
+    r'알려줘', r'알려주세요', r'궁금해', r'궁금합니다'
 ]
 
 # 의도 분류 프롬프트
@@ -61,4 +70,8 @@
 INTENT_TYPES = ["customs_tracking", "tariff_prediction", "qna"]
 
 # 기본 의도 (오류 시 사용)
-DEFAULT_INTENT = "tariff_prediction"
+DEFAULT_INTENT = "tariff_prediction"
+
+# 세션 연속성 확인을 위한 메시지 수
+SESSION_CHECK_MESSAGE_COUNT = 3
+

core/shared/router/__init__.py

@@ -0,0 +1,7 @@
+# Router package
+
+from .intent_router import intent_router
+
+__all__ = [
+    'intent_router'
+] 

core/shared/router/intent_router.py

@@ -6,9 +6,11 @@
 from core.shared.constants import (
     TARIFF_SESSION_KEYWORDS,
     NUMBER_SELECTION_PATTERNS,
+    QUESTION_PATTERNS,
     INTENT_CLASSIFICATION_PROMPT,
     INTENT_TYPES,
-    DEFAULT_INTENT
+    DEFAULT_INTENT,
+    SESSION_CHECK_MESSAGE_COUNT
 )
 
 def intent_router(state: CustomsAgentState) -> CustomsAgentState:
@@ -24,6 +26,9 @@ def intent_router(state: CustomsAgentState) -> CustomsAgentState:
             is_number_selection = True
             break
 
+    # 질문 형태 감지
+    is_question = any(re.search(pattern, current_query) for pattern in QUESTION_PATTERNS)
+    
     # 이전 대화에서 관세 예측 중인지 확인
     messages = state.get("messages", [])
     is_in_tariff_session = False
@@ -33,7 +38,7 @@ def intent_router(state: CustomsAgentState) -> CustomsAgentState:
         is_in_tariff_session = True
 
     # 2. 최근 메시지들을 확인하여 관세 예측 세션 중인지 판단
-    for msg in messages[-5:]:  # 최근 5개 메시지 확인
+    for msg in messages[-SESSION_CHECK_MESSAGE_COUNT:]:  # 최근 메시지 확인
         if hasattr(msg, 'content') and isinstance(msg.content, str):
             content = msg.content.lower()
             # 관세 예측 관련 키워드가 있거나 HS 코드 선택 메시지가 있으면 관세 예측 세션으로 판단
@@ -45,22 +50,26 @@ def intent_router(state: CustomsAgentState) -> CustomsAgentState:
     if messages and len(messages) > 0:
         last_msg = messages[-1]
         if hasattr(last_msg, 'content') and isinstance(last_msg.content, str):
-            if 'tariff_prediction' in last_msg.content:
+            if 'tariff_prediction' in last_msg.content or '관세 예측 세션' in last_msg.content:
                 is_in_tariff_session = True
 
     # 관세 예측 세션 중이면 무조건 tariff_prediction으로 분류
     if is_in_tariff_session:
         state["intent"] = "tariff_prediction"  # type: ignore
         state["messages"].append(AIMessage(content=f"의도 분류 완료: {state['intent']} (관세 예측 세션 연속성 유지)"))
-        print(state)
+        return state
+    
+    # 질문 형태이면서 관세 예측 세션이 아닌 경우 QnA로 분류
+    if is_question and not is_in_tariff_session:
+        state["intent"] = "qna"  # type: ignore
+        state["messages"].append(AIMessage(content=f"의도 분류 완료: {state['intent']} (질문 형태 감지)"))
         return state
 
     # 숫자 선택이지만 관세 예측 세션이 아닌 경우에도 tariff_prediction으로 분류
     # (HS 코드 직접 입력 등의 경우)
     if is_number_selection:
         state["intent"] = "tariff_prediction"  # type: ignore
         state["messages"].append(AIMessage(content=f"의도 분류 완료: {state['intent']} (숫자 선택 감지)"))
-        print(state)
         return state
 
     # LLM 기반 의도 분류를 수행
@@ -75,10 +84,9 @@ def intent_router(state: CustomsAgentState) -> CustomsAgentState:
         if intent not in INTENT_TYPES:
             intent = DEFAULT_INTENT  # 기본값을 tariff_prediction으로 변경
     except Exception as e:
-        print(f"[DEBUG] LLM 분류 오류: {e}")
+        # LLM 분류 실패 시 기본값 사용
         intent = DEFAULT_INTENT  # 오류 시에도 tariff_prediction으로 분류
 
     state["intent"] = intent  # type: ignore
     state["messages"].append(AIMessage(content=f"의도 분류 완료: {intent} (LLM 분류)"))
-    print(state)
     return state

core/tariff_prediction/constants/constants.py

@@ -168,7 +168,7 @@
 
 💡 **상품 묘사의 정확도가 높을수록 정확한 관세 예측이 가능합니다!**
 
-💡 **배송비는 따로 입력받고 있지 않습니다. 국제배송비(직배송)는 관세기준에 포함되지 않지만 현지 배송비는 관세기준에 포함되니 참고하시어 상품 가격과 같이 입력 부탁드립니다!**
+💡 **배송비는 따로 입력받고 있지 않습니다. 국제 배송비(직배송)는 관세기준에 포함되지 않지만 현지 배송비는 관세기준에 포함되니 참고하시어 상품 가격에 포함 후 입력 부탁드립니다!**
 
 예시:
 • \"아랫창은 고무로 되어있고 하얀색 운동화를 80000원에 독일에서 샀어요\"

core/tariff_prediction/tools/parse_tariff_result.py

@@ -35,7 +35,6 @@ def parse_tariff_result(tariff_result: str) -> Dict[str, Any]:
 
         # 가격 포맷팅
         formatted_price = format_price(parsed['product_price'])
-        formatted_shipping = format_price(parsed['shipping_cost'])
         formatted_tariff = format_price(parsed['tariff_amount'])
         formatted_vat = format_price(parsed['vat_amount'])
         formatted_total = format_price(parsed['total_tax'])