ddeng1990 commited on
Commit
417dfe8
·
verified ·
1 Parent(s): 19c19fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -6
app.py CHANGED
@@ -19,16 +19,23 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
19
  return "What magic will you build ?"
20
 
21
  @tool
22
- def calculate_square_root(number: float) -> float:
23
- """Calculates the square root of a given number.
24
 
25
  Args:
26
- number: The number to calculate the square root of.
 
27
 
28
  Returns:
29
- The square root of the number.
30
  """
31
- return math.sqrt(number)
 
 
 
 
 
 
32
 
33
  @tool
34
  def get_current_time_in_timezone(timezone: str) -> str:
@@ -71,7 +78,7 @@ agent = CodeAgent(
71
  tools=[final_answer,
72
  get_current_time_in_timezone,
73
  image_generation_tool,
74
- calculate_square_root], ## add your tools here (don't remove final answer)
75
  max_steps=6,
76
  verbosity_level=1,
77
  grammar=None,
 
19
  return "What magic will you build ?"
20
 
21
  @tool
22
+ def reverse_and_add_magic_number(text: str, magic_number: int) -> int:
23
+ """Reverses the given text, converts it to an integer, and adds a magic number.
24
 
25
  Args:
26
+ text: The string to reverse and convert.
27
+ magic_number: The number to add.
28
 
29
  Returns:
30
+ The resulting integer.
31
  """
32
+ try:
33
+ reversed_text = text[::-1]
34
+ reversed_int = int(reversed_text)
35
+ return reversed_int + magic_number
36
+ except ValueError:
37
+ return None # Handle cases where the reversed string is not a valid integer
38
+
39
 
40
  @tool
41
  def get_current_time_in_timezone(timezone: str) -> str:
 
78
  tools=[final_answer,
79
  get_current_time_in_timezone,
80
  image_generation_tool,
81
+ reverse_and_add_magic_number], ## add your tools here (don't remove final answer)
82
  max_steps=6,
83
  verbosity_level=1,
84
  grammar=None,