Update multiagents.py

multiagents.py

@@ -45,12 +45,19 @@ dotenv.load_dotenv()
 class GrokApiAgent:
     def __init__(self):
         pass  # No setup needed for API-based agent
-    
+
     def __call__(self, prompt):
-        return GrokApi(
+        # You could keep this for backward compatibility if needed
+        return self.generate(prompt)
+
+    def generate(self, prompt):
+        response = GrokApi(
             system_prompt="You are a helpful assistant.",
             user_input=prompt
         )
+        # Assuming GrokApi returns a string or has a property with the answer text
+        # Adjust this depending on the actual return type of GrokApi
+        return str(response)
 
 # Pick your agent here (just change one line to test different ones)
 # My_Agent = BasicAgent