first try gwen

.gitignore

@@ -1,3 +1,4 @@
 secret
 test.ipynb
-.env
+.env
+error.txt

agent.py

@@ -1,6 +1,6 @@
 from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
-from dotenv import load_dotenv
-load_dotenv()
+# import asyncio
+import datetime
 
 
 class CustomAgent:
@@ -8,12 +8,14 @@ class CustomAgent:
         self, 
         model_name="Qwen/Qwen2.5-Omni-7B", 
         authorized_imports=["wiki", "json"], 
-        max_steps=5
+        max_steps=5,
+        logging=False
     ):
         """
         Initialize the CustomAgent with a model and tools.
         If no model is provided, a default one is used.
         """
+        self.logging = logging
         model = InferenceClientModel(model_name)
         search_tool = DuckDuckGoSearchTool()
         
@@ -25,8 +27,24 @@ class CustomAgent:
         )
         print("CustomAgent initialized.")
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str) -> tuple[str, str, str | Exception]:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer =self.agent.run(question)
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        try:
+            fixed_answer = self.agent.run(question)
+            
+            if self.logging:
+                timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                with open("log.txt", "a") as log_file:
+                    log_file.write(f"{timestamp}: {question} -> {fixed_answer}\n")
+
+            return question, fixed_answer
+        
+        except Exception as e:
+            print(f"Agent ERROR on task: {e}")
+
+            if self.logging:
+                with open("log.txt", "a") as log_file:
+                    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                    log_file.write(f"{timestamp}: {question} -> ERROR: {e}\n")        
+            
+            return question, e