Spaces:
Sleeping
Sleeping
Update multi_agent.py
Browse files- multi_agent.py +5 -0
multi_agent.py
CHANGED
|
@@ -62,6 +62,11 @@ class GroqModel:
|
|
| 62 |
)
|
| 63 |
return response.choices[0].message.content
|
| 64 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
# ---- MULTI-AGENT SYSTEM ----
|
| 66 |
class MultyAgentSystem:
|
| 67 |
def __init__(self):
|
|
|
|
| 62 |
)
|
| 63 |
return response.choices[0].message.content
|
| 64 |
|
| 65 |
+
def generate(self, prompt, max_tokens=8096, **kwargs):
|
| 66 |
+
# For compatibility with agent frameworks
|
| 67 |
+
return self.__call__(prompt, max_tokens=max_tokens)
|
| 68 |
+
|
| 69 |
+
|
| 70 |
# ---- MULTI-AGENT SYSTEM ----
|
| 71 |
class MultyAgentSystem:
|
| 72 |
def __init__(self):
|