VirakSM commited on
Commit
3c5fc0b
·
verified ·
1 Parent(s): c1aabcf

Update app.py

Browse files

add tool: translate_text in tools.

Files changed (1) hide show
  1. app.py +3 -10
app.py CHANGED
@@ -1,16 +1,11 @@
1
  from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 
2
  import datetime
3
  import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
 
8
- try:
9
- from google.cloud import translate_v2 as translate
10
- except ImportError:
11
- print("The google-cloud-translate library is not installed.")
12
- print("Please run 'pip install google-cloud-translate' to install it.")
13
- exit(1)
14
 
15
  from Gradio_UI import GradioUI
16
 
@@ -33,12 +28,10 @@ def translate_text(text:str, target_language:str):
33
  Args:
34
  text: The text to be translated.
35
  target_language: The target language code (e.g. 'es' for Spanish).
36
-
37
- Returns:
38
- str: The translated text.
39
  """
40
  translate_client = translate.Client()
41
  result = translate_client.translate(text, target_language=target_language)
 
42
  return result['translatedText']
43
  @tool
44
  def get_current_time_in_timezone(timezone: str) -> str:
@@ -77,7 +70,7 @@ with open("prompts.yaml", 'r') as stream:
77
 
78
  agent = CodeAgent(
79
  model=model,
80
- tools=[final_answer], ## add your tools here (don't remove final answer)
81
  max_steps=6,
82
  verbosity_level=1,
83
  grammar=None,
 
1
  from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
+ from google.cloud import translate_v2 as translate
3
  import datetime
4
  import requests
5
  import pytz
6
  import yaml
7
  from tools.final_answer import FinalAnswerTool
8
 
 
 
 
 
 
 
9
 
10
  from Gradio_UI import GradioUI
11
 
 
28
  Args:
29
  text: The text to be translated.
30
  target_language: The target language code (e.g. 'es' for Spanish).
 
 
 
31
  """
32
  translate_client = translate.Client()
33
  result = translate_client.translate(text, target_language=target_language)
34
+
35
  return result['translatedText']
36
  @tool
37
  def get_current_time_in_timezone(timezone: str) -> str:
 
70
 
71
  agent = CodeAgent(
72
  model=model,
73
+ tools=[final_answer, translate_text], ## add your tools here (don't remove final answer)
74
  max_steps=6,
75
  verbosity_level=1,
76
  grammar=None,