Update app.py

app.py

@@ -14,38 +14,37 @@ import io
 
 class GroqLLM:
     """Compatible LLM interface for smolagents CodeAgent"""
-    def __init__(self, model_name="llama3-8b-8192"):
+    def __init__(self, model_name="llama-3.1-8B-Instant"):
         self.client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
         self.model_name = model_name
-
+    
     def __call__(self, prompt: Union[str, dict, List[Dict]]) -> str:
-        """Allows use as callable (legacy compatibility)"""
-        return self.generate(prompt)
-
-    def generate(self, prompt: Union[str, dict, List[Dict]], **kwargs) -> str:
-        """Generate text using Groq API. Handles optional kwargs from smolagents."""
+        """Make the class callable as required by smolagents"""
         try:
+            # Handle different prompt formats
             if isinstance(prompt, (dict, list)):
                 prompt_str = str(prompt)
             else:
                 prompt_str = str(prompt)
-
-            temperature = kwargs.get("temperature", 0.7)
-            max_tokens = kwargs.get("max_tokens", 1024)
-            # 'stop_sequences' is not supported by Groq API directly; you can ignore or handle manually.
-
+            
+            # Create a properly formatted message
             completion = self.client.chat.completions.create(
                 model=self.model_name,
-                messages=[{"role": "user", "content": prompt_str}],
-                temperature=temperature,
-                max_tokens=max_tokens,
+                messages=[{
+                    "role": "user",
+                    "content": prompt_str
+                }],
+                temperature=0.7,
+                max_tokens=1024,
                 stream=False
             )
-
-            return completion.choices[0].message.content.strip() if completion.choices else "Error: No response generated"
+            
+            return completion.choices[0].message.content if completion.choices else "Error: No response generated"
+            
         except Exception as e:
-            return f"Error generating response: {str(e)}"
-
+            error_msg = f"Error generating response: {str(e)}"
+            print(error_msg)
+            return error_msg
 
 class DataAnalysisAgent(CodeAgent):
     """Extended CodeAgent with dataset awareness"""
@@ -291,4 +290,4 @@ def main():
         st.error(f"An error occurred: {str(e)}")
 
 if __name__ == "__main__":
-    main()
+    main()