Hugo Bui commited on
Commit
8c13f2b
·
unverified ·
1 Parent(s): 939382b
Files changed (1) hide show
  1. tools/country_info_tool.py +3 -3
tools/country_info_tool.py CHANGED
@@ -1,4 +1,4 @@
1
- from typing import Any, Optional
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) -> Optional[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: