Update tools.py

tools.py

@@ -1,4 +1,4 @@
-from langchain.tools import Tool
+from langchain.tools import tool
 import datetime
 import requests
 import openai
@@ -7,13 +7,19 @@ import tempfile
 from urllib.parse import urlparse
 from openai import OpenAI
 
-def current_date(_):
+@tool
+def current_date(_) -> str :
+    """ Returns the current date in YYYY-MM-DD format """
     return datetime.datetime.now().strftime("%Y-%m-%d")
 
-def day_of_week(_):
+@tool
+def day_of_week(_) -> str :
+    """ Returns the current day of the week (e.g., Monday, Tuesday) """
     return datetime.datetime.now().strftime("%A")
 
-def days_until(date_str):
+@tool
+def days_until(date_str: str) -> str :
+    """ Returns the number of days from today until a given date (input format: YYYY-MM-DD) """
     try:
         future_date = datetime.datetime.strptime(date_str, "%Y-%m-%d").date()
         today = datetime.date.today()
@@ -23,24 +29,9 @@ def days_until(date_str):
     except Exception as e:
         return f"Error parsing date: {str(e)}"
 
-datetime_tools = [
-    Tool(
-        name="current_date",
-        func=current_date,
-        description="Returns the current date in YYYY-MM-DD format."
-    ),
-    Tool(
-        name="current_day_of_week",
-        func=day_of_week,
-        description="Returns the current day of the week (e.g., Monday, Tuesday)."
-    ),
-    Tool(
-        name="days_until",
-        func=days_until,
-        description="Returns the number of days from today until a given date (input format: YYYY-MM-DD)."
-    )
-]
+datetime_tools = [current_date, day_of_week, days_until]
 
+@tool
 def transcribe_audio(audio_input: str) -> str:
     """
     Transcribes an audio file from a local file or a URL
@@ -78,10 +69,4 @@ def transcribe_audio(audio_input: str) -> str:
         return transcription.text
 
     except Exception as e:
-        return f"Transcription failed: {e}"
-
-transcribe_audio_tool = Tool(
-        name="transcribe_audio", 
-        func=transcribe_audio,
-        description="Transcribes an audio file from a local file or a URL"
-)
+        return f"Transcription failed: {e}"