Update app.py

app.py

@@ -46,16 +46,25 @@ class GroqLLM:
             print(error_msg)
             return error_msg
             
-    def generate(self, prompt: Union[str, dict, List[Dict]], **kwargs) -> str:
+    def generate(self, prompt: Union[str, dict, List[Dict]], **kwargs) -> object:
         """Add generate method to make compatible with smolagents CodeAgent
         
         Args:
             prompt: The prompt to send to the model
             **kwargs: Additional keyword arguments to support CodeAgent API
                      (stop_sequences, etc.) - these are ignored in the Groq implementation
+        
+        Returns:
+            An object with a 'content' attribute containing the response text
         """
-        # Simply ignore kwargs that aren't needed and pass through to __call__
-        return self.__call__(prompt)
+        response_text = self.__call__(prompt)
+        
+        # Create a simple object with a content attribute
+        class Response:
+            def __init__(self, content):
+                self.content = content
+                
+        return Response(response_text)
 
 class DataAnalysisAgent(CodeAgent):
     """Extended CodeAgent with dataset awareness"""