sosa123454321 commited on
Commit
57a9dca
·
verified ·
1 Parent(s): 27b7edf

Update multi_agent.py

Browse files
Files changed (1) hide show
  1. multi_agent.py +6 -6
multi_agent.py CHANGED
@@ -42,15 +42,15 @@ def get_latest_file(directory, file_extension):
42
  def run_multi_agent(llm, task):
43
  #llm_config = {"model": llm}
44
  if not GROQ_API_KEY_defualt:
45
- llm_config = {
46
- "config_list": [
47
  {
48
  "model": llm, #"llama3-8b-8192",
49
  "api_key": os.environ.get("GROQ_API_KEY_defualt"),
50
  "base_url": "https://api.groq.com/openai/v1"
51
  }
52
- ]
53
- }
54
  else GROQ_API_KEY:
55
  llm_config = {
56
  "config_list": [
@@ -59,8 +59,8 @@ def run_multi_agent(llm, task):
59
  "api_key": os.environ.get("GROQ_API_KEY"),
60
  "base_url": "https://api.groq.com/openai/v1"
61
  }
62
- ]
63
- }
64
 
65
  executor = LocalCommandLineCodeExecutor(
66
  timeout=60,
 
42
  def run_multi_agent(llm, task):
43
  #llm_config = {"model": llm}
44
  if not GROQ_API_KEY_defualt:
45
+ llm_config = {
46
+ "config_list": [
47
  {
48
  "model": llm, #"llama3-8b-8192",
49
  "api_key": os.environ.get("GROQ_API_KEY_defualt"),
50
  "base_url": "https://api.groq.com/openai/v1"
51
  }
52
+ ]
53
+ }
54
  else GROQ_API_KEY:
55
  llm_config = {
56
  "config_list": [
 
59
  "api_key": os.environ.get("GROQ_API_KEY"),
60
  "base_url": "https://api.groq.com/openai/v1"
61
  }
62
+ ]
63
+ }
64
 
65
  executor = LocalCommandLineCodeExecutor(
66
  timeout=60,