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