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