sosa123454321 commited on
Commit
218e48a
·
verified ·
1 Parent(s): 790eb88

Update multi_agent.py

Browse files
Files changed (1) hide show
  1. multi_agent.py +10 -1
multi_agent.py CHANGED
@@ -40,7 +40,16 @@ def get_latest_file(directory, file_extension):
40
  return latest_file
41
 
42
  def run_multi_agent(llm, task):
43
- llm_config = {"model": llm}
 
 
 
 
 
 
 
 
 
44
 
45
  executor = LocalCommandLineCodeExecutor(
46
  timeout=60,
 
40
  return latest_file
41
 
42
  def run_multi_agent(llm, task):
43
+ #llm_config = {"model": llm}
44
+ llm_config = {
45
+ "config_list": [
46
+ {
47
+ "model": "llama3-8b-8192",
48
+ "api_key": os.environ.get("GROQ_API_KEY"),
49
+ "base_url": "https://api.groq.com/openai/v1"
50
+ }
51
+ ]
52
+ }
53
 
54
  executor = LocalCommandLineCodeExecutor(
55
  timeout=60,