ArslanFOX commited on
Commit
cf1eff1
·
verified ·
1 Parent(s): d077e2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -20
app.py CHANGED
@@ -7,25 +7,6 @@ from tools.final_answer import FinalAnswerTool
7
 
8
  from Gradio_UI import GradioUI
9
 
10
- @tool
11
- def get_air_quality(city: str) -> str:
12
- """Получает индекс качества воздуха для указанного города.
13
- Args:
14
- city: Название города.
15
- """
16
- url = f"https://api.iqair.com/v2/city?city={city}&key={API_KEY}"
17
- response = requests.get(url)
18
-
19
- if response.status_code == 200:
20
- data = response.json()
21
- pollution_data = data.get("data", {}).get("current", {}).get("pollution", {})
22
- if "aqius" in pollution_data:
23
- return f"Качество воздуха в {city}: AQI {pollution_data['aqius']}"
24
- else:
25
- return "Ошибка: данные о загрязнении отсутствуют."
26
- else:
27
- return f"Ошибка {response.status_code}: {response.text}"
28
-
29
  @tool
30
  def get_current_time_in_timezone(timezone: str) -> str:
31
  """A tool that fetches the current local time in a specified timezone.
@@ -63,7 +44,7 @@ with open("prompts.yaml", 'r') as stream:
63
 
64
  agent = CodeAgent(
65
  model=model,
66
- tools=[final_answer, get_current_time_in_timezone, get_air_quality], ## add your tools here (don't remove final answer) Вот тут не забываем добавить свои инструменты блэт!
67
  max_steps=6,
68
  verbosity_level=1,
69
  grammar=None,
 
7
 
8
  from Gradio_UI import GradioUI
9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  @tool
11
  def get_current_time_in_timezone(timezone: str) -> str:
12
  """A tool that fetches the current local time in a specified timezone.
 
44
 
45
  agent = CodeAgent(
46
  model=model,
47
+ tools=[final_answer, get_current_time_in_timezone], ## add your tools here (don't remove final answer) Вот тут не забываем добавить свои инструменты блэт!
48
  max_steps=6,
49
  verbosity_level=1,
50
  grammar=None,