Обновить app.py

#353
by Nik347 - opened
Files changed (1) hide show
  1. app.py +27 -4
app.py CHANGED
@@ -33,7 +33,25 @@ def get_current_time_in_timezone(timezone: str) -> str:
33
  except Exception as e:
34
  return f"Error fetching time for timezone '{timezone}': {str(e)}"
35
 
36
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
37
  final_answer = FinalAnswerTool()
38
 
39
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -55,9 +73,15 @@ with open("prompts.yaml", 'r') as stream:
55
 
56
  agent = CodeAgent(
57
  model=model,
58
- tools=[final_answer], ## add your tools here (don't remove final answer)
 
 
 
 
 
 
59
  max_steps=6,
60
- verbosity_level=1,
61
  grammar=None,
62
  planning_interval=None,
63
  name=None,
@@ -65,5 +89,4 @@ agent = CodeAgent(
65
  prompt_templates=prompt_templates
66
  )
67
 
68
-
69
  GradioUI(agent).launch()
 
33
  except Exception as e:
34
  return f"Error fetching time for timezone '{timezone}': {str(e)}"
35
 
36
+ @tool
37
+ def translate_text(text: str, target_language: str) -> str:
38
+ """Инструмент для перевода текста на другой язык
39
+ Аргументы:
40
+ text: текст для перевода
41
+ target_language: целевой язык (например, 'английский', 'испанский')
42
+ """
43
+ # Это упрощенный пример; в реальности вы бы использовали API перевода
44
+ languages = {
45
+ "английский": "Translated to English",
46
+ "испанский": "Traducido al español",
47
+ "французский": "Traduit en français"
48
+ }
49
+
50
+ if target_language.lower() in languages:
51
+ return f"{languages[target_language.lower()]}: {text}"
52
+ else:
53
+ return f"Перевод на '{target_language}' не поддерживается."
54
+
55
  final_answer = FinalAnswerTool()
56
 
57
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 
73
 
74
  agent = CodeAgent(
75
  model=model,
76
+ tools=[
77
+ final_answer, # Необходимый инструмент - не удаляйте
78
+ get_current_time_in_timezone, # Инструмент для получения времени
79
+ DuckDuckGoSearchTool(), # Инструмент для веб-поиска
80
+ image_generation_tool, # Инструмент для генерации изображений
81
+ # Добавьте другие инструменты при необходимости
82
+ ],
83
  max_steps=6,
84
+ verbosity=1,
85
  grammar=None,
86
  planning_interval=None,
87
  name=None,
 
89
  prompt_templates=prompt_templates
90
  )
91
 
 
92
  GradioUI(agent).launch()