kk20krishna commited on
Commit
17bfa6c
·
verified ·
1 Parent(s): f7c887c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -9,12 +9,12 @@ from Gradio_UI import GradioUI
9
 
10
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
  @tool
12
- def state_capital(state: str) -> str:
13
- """A tool that takes a state name and returns a the state capital name
14
  Args:
15
- state: A state name
16
  """
17
- return "Palakkad"
18
 
19
  @tool
20
  def get_current_time_in_timezone(timezone: str) -> str:
@@ -50,7 +50,7 @@ with open("prompts.yaml", 'r') as stream:
50
 
51
  agent = CodeAgent(
52
  model=model,
53
- tools=[state_capital, final_answer], ## add your tools here (don't remove final answer)
54
  max_steps=6,
55
  verbosity_level=1,
56
  grammar=None,
 
9
 
10
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
  @tool
12
+ def current_temprature_in_planet(planet: str) -> str:
13
+ """A tool that takes a planet name and returns a the temprature in that planet
14
  Args:
15
+ state: A planet name
16
  """
17
+ return "1008 degrees"
18
 
19
  @tool
20
  def get_current_time_in_timezone(timezone: str) -> str:
 
50
 
51
  agent = CodeAgent(
52
  model=model,
53
+ tools=[current_temprature_in_planet, final_answer], ## add your tools here (don't remove final answer)
54
  max_steps=6,
55
  verbosity_level=1,
56
  grammar=None,