Update app.py

app.py

@@ -129,6 +129,22 @@ class PollinationsModel:
         except Exception as e:
             logger.exception(f"PollinationsModel generate failed: {e}")
             return f"Error generating response: {str(e)}"
+            
+    def __call__(self, prompt=None, messages=None, **kwargs):
+        """Make the model callable as required by smolagents CodeAgent."""
+        # Determine content to send based on what's provided
+        if messages:
+            # If we have chat messages, extract content from the last one
+            content = messages[-1].get("content", "") if messages else ""
+        else:
+            content = prompt or ""
+        
+        # Generate the response
+        response_text = self.generate(content, **kwargs)
+        
+        # Return in a format similar to OpenAI API responses
+        return {"choices": [{"message": {"content": response_text}}]}
+
 
 logger.info(f"Initializing Pollinations LLM connection: {MODEL_ID}")
 try: