Update app.py

app.py

@@ -18,19 +18,22 @@ class GroqLLM:
         self.client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
         self.model_name = model_name
     
-    def __call__(self, prompt: str) -> str:
+    def __call__(self, prompt: Union[str, dict, List[Dict]]) -> str:
         """Make the class callable as required by smolagents"""
-        # Ensure the prompt is a string
-        if not isinstance(prompt, str):
-            return "Error: Prompt must be a string"
-            
         try:
+            # Handle different prompt formats
+            if isinstance(prompt, (dict, list)):
+                # If prompt is a dictionary or list, convert it to a string representation
+                prompt_str = str(prompt)
+            else:
+                prompt_str = str(prompt)
+            
             # Create a properly formatted message
             completion = self.client.chat.completions.create(
                 model=self.model_name,
                 messages=[{
                     "role": "user",
-                    "content": str(prompt)  # Ensure content is string
+                    "content": prompt_str
                 }],
                 temperature=0.7,
                 max_tokens=1024,