ABVM commited on
Commit
28edc7c
·
verified ·
1 Parent(s): f2b38e7

Update multi_agent.py

Browse files
Files changed (1) hide show
  1. 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):