Spaces:
Running
Running
Upload tools.py
Browse files- 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
|
| 48 |
-
"
|
|
|
|
| 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(), [])
|