Spaces:
Runtime error
Runtime error
Hugo Bui commited on
fix
Browse files
tools/country_info_tool.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from typing import Any
|
| 2 |
from smolagents.tools import Tool
|
| 3 |
import requests
|
| 4 |
from datetime import datetime, timedelta
|
|
@@ -13,7 +13,7 @@ class CountryInfoTool(Tool):
|
|
| 13 |
description = "Retrieves important contextual information about a country in real-time: security, current events, national holidays, political climate, travel advice."
|
| 14 |
inputs = {
|
| 15 |
'country': {'type': 'string', 'description': 'Country name in French or English (e.g., "France", "United States", "Japan")'},
|
| 16 |
-
'info_type': {'type': 'string', 'description': 'Type of information requested: "all", "security", "events", "holidays", "travel", "politics"'}
|
| 17 |
}
|
| 18 |
output_type = "string"
|
| 19 |
|
|
@@ -690,7 +690,7 @@ ABSOLUTE PRIORITY: Protect travelers - when in doubt, choose the strictest secur
|
|
| 690 |
except Exception:
|
| 691 |
return []
|
| 692 |
|
| 693 |
-
def _get_llm_final_recommendation(self, country: str, full_report: str)
|
| 694 |
"""Uses Claude to generate an intelligent final recommendation"""
|
| 695 |
try:
|
| 696 |
if not self.claude_client:
|
|
|
|
| 1 |
+
from typing import Any
|
| 2 |
from smolagents.tools import Tool
|
| 3 |
import requests
|
| 4 |
from datetime import datetime, timedelta
|
|
|
|
| 13 |
description = "Retrieves important contextual information about a country in real-time: security, current events, national holidays, political climate, travel advice."
|
| 14 |
inputs = {
|
| 15 |
'country': {'type': 'string', 'description': 'Country name in French or English (e.g., "France", "United States", "Japan")'},
|
| 16 |
+
'info_type': {'type': 'string', 'description': 'Type of information requested: "all" (recommended), "security", "events", "holidays", "travel", "politics"', 'nullable': True}
|
| 17 |
}
|
| 18 |
output_type = "string"
|
| 19 |
|
|
|
|
| 690 |
except Exception:
|
| 691 |
return []
|
| 692 |
|
| 693 |
+
def _get_llm_final_recommendation(self, country: str, full_report: str):
|
| 694 |
"""Uses Claude to generate an intelligent final recommendation"""
|
| 695 |
try:
|
| 696 |
if not self.claude_client:
|