pchatzina commited on
Commit
1803c4d
·
verified ·
1 Parent(s): ae7a494

added tool get_days_left_until_next_birthday

Browse files
Files changed (1) hide show
  1. app.py +14 -7
app.py CHANGED
@@ -7,16 +7,23 @@ from tools.final_answer import FinalAnswerTool
7
 
8
  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 my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
13
- #Keep this format for the description / args / args description but feel free to modify the tool
14
- """A tool that does nothing yet
15
  Args:
16
- arg1: the first argument
17
- arg2: the second argument
 
18
  """
19
- return "What magic will you build ?"
 
 
 
 
 
 
 
20
 
21
  @tool
22
  def get_current_time_in_timezone(timezone: str) -> str:
 
7
 
8
  from Gradio_UI import GradioUI
9
 
 
10
  @tool
11
+ def get_days_left_until_next_birthday(year:int, month:int, month_day:int)-> int:
12
+ """A tool that receives as input when the user was born as year, month, month_day
13
+ and returns the days left until their next birthday.
14
  Args:
15
+ year: An integer that represents the birth year.
16
+ month: An integer that represents the birth month.
17
+ month_day: An integer that represents the day of the month the user was born.
18
  """
19
+ today = datetime.date.today()
20
+ birthday = datetime.date(today.year, month, month_day)
21
+
22
+ if birthday < today:
23
+ birthday = datetime.date(today.year + 1, month, month_day)
24
+
25
+ days_left = (birthday - today).days
26
+ return days_left
27
 
28
  @tool
29
  def get_current_time_in_timezone(timezone: str) -> str: