Update multi_agent.py

multi_agent.py

@@ -44,6 +44,10 @@ def image_process(image_file: str) -> Dict[str, str]:
         }
 
 # ---- GROQ MODEL WRAPPER ----
+class LLMResponse:
+    def __init__(self, content):
+        self.content = content
+        
 class GroqModel:
     def __init__(self, model_name=""):
         self.model_name = model_name
@@ -60,7 +64,7 @@ class GroqModel:
             stream=False,
             max_tokens=max_tokens,
         )
-        return response.choices[0].message.content
+        return LLMResponse(response.choices[0].message.content)
 
     def generate(self, prompt, max_tokens=8096, **kwargs):
         # For compatibility with agent frameworks