Update multi_agent.py
Browse files- 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 |
-
|
46 |
-
|
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,
|