lovelymango commited on
Commit
a4cbb07
·
verified ·
1 Parent(s): 3b27f52

Upload tools.py

Browse files
Files changed (1) hide show
  1. src/mcp/tools.py +5 -4
src/mcp/tools.py CHANGED
@@ -44,8 +44,9 @@ TOOLS = [
44
  "chain": {
45
  "type": "string",
46
  "description": "체인/항공사/카드사로 필터링. 여러 체인은 쉼표로 구분 (예: 'MARRIOTT,HILTON'). "
47
- "지원: IHG, MARRIOTT, ACCOR, HILTON, HYATT, KOREAN_AIR, ASIANA, DELTA, UNITED, "
48
- "AMEX, SHINHAN, HYUNDAI, KB, LOTTE"
 
49
  },
50
  "type": {
51
  "type": "string",
@@ -154,7 +155,7 @@ class KBToolHandler:
154
  VALID_CHAINS = {
155
  "IHG", "MARRIOTT", "ACCOR", "HILTON", "HYATT",
156
  "KOREAN_AIR", "ASIANA", "DELTA", "UNITED",
157
- "AMEX", "SHINHAN", "HYUNDAI", "KB", "LOTTE", "SAMSUNG", "WOORI",
158
  "ALLIANCE", "AIRLINE", "CARD", "NEWS"
159
  }
160
 
@@ -205,7 +206,7 @@ class KBToolHandler:
205
  domain_chain_map = {
206
  "hotel": ["MARRIOTT", "HILTON", "IHG", "ACCOR", "HYATT"],
207
  "airline": ["KOREAN_AIR", "ASIANA", "DELTA", "UNITED", "ALLIANCE", "AIRLINE"],
208
- "card": ["AMEX", "SHINHAN", "HYUNDAI", "KB", "SAMSUNG", "WOORI", "LOTTE", "CARD"],
209
  "news": ["NEWS"] # 뉴스 도메인 추가
210
  }
211
  domain_chains = domain_chain_map.get(domain.lower(), [])
 
44
  "chain": {
45
  "type": "string",
46
  "description": "체인/항공사/카드사로 필터링. 여러 체인은 쉼표로 구분 (예: 'MARRIOTT,HILTON'). "
47
+ "지원 호텔: IHG, MARRIOTT, ACCOR, HILTON, HYATT. "
48
+ "지원 항공: KOREAN_AIR, ASIANA, DELTA, UNITED. "
49
+ "지원 카드: AMEX, SHINHAN, HYUNDAI, KB, LOTTE, HANA, SAMSUNG, WOORI"
50
  },
51
  "type": {
52
  "type": "string",
 
155
  VALID_CHAINS = {
156
  "IHG", "MARRIOTT", "ACCOR", "HILTON", "HYATT",
157
  "KOREAN_AIR", "ASIANA", "DELTA", "UNITED",
158
+ "AMEX", "SHINHAN", "HYUNDAI", "KB", "LOTTE", "SAMSUNG", "WOORI", "HANA",
159
  "ALLIANCE", "AIRLINE", "CARD", "NEWS"
160
  }
161
 
 
206
  domain_chain_map = {
207
  "hotel": ["MARRIOTT", "HILTON", "IHG", "ACCOR", "HYATT"],
208
  "airline": ["KOREAN_AIR", "ASIANA", "DELTA", "UNITED", "ALLIANCE", "AIRLINE"],
209
+ "card": ["AMEX", "SHINHAN", "HYUNDAI", "KB", "SAMSUNG", "WOORI", "LOTTE", "HANA", "CARD"],
210
  "news": ["NEWS"] # 뉴스 도메인 추가
211
  }
212
  domain_chains = domain_chain_map.get(domain.lower(), [])