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,
|