Update agent.py

agent.py

@@ -14,18 +14,21 @@ class GaiaAgent:
         }
 
     def generate(self, prompt: str, stop: List[str] = []) -> str:
+        print("📤 [generate()] Sending prompt to model.")
+        print(prompt[:200])  # Print first 200 chars of the prompt for sanity
+        
         payload = {
             "inputs": prompt,
             "parameters": {
-                "temperature": 0.0,
-                "max_new_tokens": 1024,
-                "stop": stop,
-            }
+            "temperature": 0.0,
+            "max_new_tokens": 1024,
+            "stop": stop,
         }
+    }
         response = requests.post(self.api_url, headers=self.headers, json=payload)
         response.raise_for_status()
         output = response.json()
-
+        
         if isinstance(output, dict) and "generated_text" in output:
             return output["generated_text"]
         elif isinstance(output, list) and "generated_text" in output[0]: