Upload agent.py

agent.py

@@ -115,6 +115,7 @@ def arvix_search(query: str) -> str:
 
 @tool
 def wolfram_alpha_query(query: str) -> str:
+    """Query Wolfram Alpha with the given question and return the result."""
     import wolframalpha
     client = wolframalpha.Client(os.environ['WOLFRAM_APP_ID'])
     res = client.query(query)
@@ -124,6 +125,7 @@ def wolfram_alpha_query(query: str) -> str:
         return "No result found."
 
 
+
 # load the system prompt from the file
 with open("system_prompt.txt", "r", encoding="utf-8") as f:
     system_prompt = f.read()