Update multiagents.py

multiagents.py

@@ -3,13 +3,15 @@
 import os
 import dotenv
 from smolagents import CodeAgent
-from smolagents import OpenAIServerModel
+#from smolagents import OpenAIServerModel
 from tools.fetch import fetch_webpage, search_web
 from smolagents import PythonInterpreterTool
 from tools.yttranscript import get_youtube_transcript, get_youtube_title_description
 from tools.stt import get_text_transcript_from_audio_file
 from tools.image import analyze_image
 from common.mylogger import mylog
+from smolagents import LiteLLMModel  # Import LiteLLMModel instead of OpenAIServerModel
+
 import myprompts
 
 from groq_api import GrokApi
@@ -41,70 +43,20 @@ dotenv.load_dotenv()
 #     api_base="https://api.openai.com/v1",
 #     api_key=os.environ["OPENAI_API_KEY"],
 # )
+
+
+
 # --- Agent wrappers ---
-# Fixed GrokApiAgent class
-class GrokApiAgent:
-    def __init__(self):
-        pass  # No setup needed for API-based agent
-
-    def __call__(self, prompt, **kwargs):
-        # You could keep this for backward compatibility if needed
-        return self.generate(prompt, **kwargs)
-
-    def generate(self, messages=None, stop_sequences=None, max_tokens=None, temperature=None, **kwargs):
-        """
-        Generate method compatible with smolagents framework.
-        
-        Args:
-            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
-            response = GrokApi(
-                system_prompt="You are a helpful assistant.",
-                user_input=prompt
-            )
-            
-            # Convert response to string and handle stop_sequences if needed
-            response_text = str(response)
-            
-            # If stop_sequences are provided, truncate the response at the first occurrence
-            if stop_sequences:
-                for stop_seq in stop_sequences:
-                    if stop_seq in response_text:
-                        response_text = response_text.split(stop_seq)[0]
-                        break
-            
-            return response_text
-            
-        except Exception as e:
-            print(f"Error in GrokApiAgent.generate: {e}")
-            return f"Error generating response: {e}"
+groq_model = LiteLLMModel(
+    model_id="qwen/qwen3-32b",  # or any other Groq model like groq/mixtral-8x7b-32768
+    api_key=os.environ["GROQ_API_KEY"],
+    temperature=0.1,
+    max_tokens=4000,
+)
 
 # Pick your agent here (just change one line to test different ones)
 # My_Agent = BasicAgent
-My_Agent = GrokApiAgent
+My_Agent = groq_model
 
 def check_final_answer(final_answer, agent_memory)  -> bool:
     """
@@ -193,7 +145,8 @@ class MultiAgent:
 
 if __name__ == "__main__":
     # Example usage
-   
+
+    asyncio.run(main())
     question = """
 What was the actual enrollment of the Malko competition in 2023?
 """