Update agent.py

agent.py

@@ -1,5 +1,5 @@
-from smolagents import LiteLLMModel, ToolCallingAgent, CodeAgent
-from tools import DuckDuckGoSearchTool
+from smolagents import HfApiModel, ToolCallingAgent, CodeAgent
+from tools import DuckDuckGoSearchTool, get_current_time_in_timezone
 import os
 
 class BasicAgent:
@@ -10,15 +10,15 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]} ...")
 
         # Using Qwen
-        qwen_model = os.getenv(key="QWEN_MODEL")  # e.g. 'Qwen/Qwen1.5-7B-Chat'
-        model = LiteLLMModel(
-            model_id=qwen_model,
-            api_key=os.getenv(key="QWEN_API_KEY")
+        model = HfApiModel(
+            max_tokens=2096,
+            temperature=0.5,
+            model_id = os.getenv("QWEN_MODEL"),# it is possible that this model may be overloaded
+            custom_role_conversions=None,
         )
-
         agent = CodeAgent(
             model=model,
-            tools=[DuckDuckGoSearchTool()]
+            tools=[DuckDuckGoSearchTool(),get_current_time_in_timezone]
         )
         prompt = f"""You are a general AI assistant. I will ask you a question. Report your thoughts, and 
 finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should 
@@ -29,7 +29,11 @@ write the digits in plain text unless specified otherwise. If you are asked for
 the above rules depending of whether the element to be put in the list is a number or a string.
 {question}
 """
-        answer = agent.run(prompt)
+        try:
+            answer = agent.run(prompt)
+        except Exception as e:
+            print(f"Agent failed: {e}")
+            answer = "Error: Agent could not complete the request."
         
         print(f"Agent returning answer: {answer}")
         return answer