Update multiagents.py

multiagents.py

@@ -51,28 +51,39 @@ class GrokApiAgent:
         # You could keep this for backward compatibility if needed
         return self.generate(prompt, **kwargs)
 
-    def generate(self, prompt, stop_sequences=None, max_tokens=None, temperature=None, **kwargs):
+    def generate(self, messages=None, stop_sequences=None, max_tokens=None, temperature=None, **kwargs):
         """
         Generate method compatible with smolagents framework.
         
         Args:
-            prompt: The input prompt
+            messages: Can be a string prompt or list of messages
             stop_sequences: List of strings where generation should stop (optional)
             max_tokens: Maximum number of tokens to generate (optional)
             temperature: Sampling temperature (optional)
             **kwargs: Other parameters that might be passed by the framework
         """
         try:
+            # Handle different input formats that smolagents might send
+            if messages is None:
+                # If no messages provided, check if prompt is in kwargs
+                prompt = kwargs.get('prompt', '')
+            elif isinstance(messages, str):
+                # If messages is a string, use it as prompt
+                prompt = messages
+            elif isinstance(messages, list) and len(messages) > 0:
+                # If messages is a list, extract the content from the last message
+                last_message = messages[-1]
+                if isinstance(last_message, dict):
+                    prompt = last_message.get('content', str(last_message))
+                else:
+                    prompt = str(last_message)
+            else:
+                prompt = str(messages) if messages else ''
+            
             # Call your GrokApi with the prompt
-            # Note: You might need to adjust this based on how GrokApi actually works
-            # and whether it supports the additional parameters
             response = GrokApi(
                 system_prompt="You are a helpful assistant.",
                 user_input=prompt
-                # Add other parameters here if GrokApi supports them:
-                # max_tokens=max_tokens,
-                # temperature=temperature,
-                # etc.
             )
             
             # Convert response to string and handle stop_sequences if needed