Update multi_agent.py

multi_agent.py

@@ -41,15 +41,26 @@ def get_latest_file(directory, file_extension):
     
 def run_multi_agent(llm, task):
     #llm_config = {"model": llm}
-    llm_config = {
-    "config_list": [
-        {
-            "model": "llama3-8b-8192",
+    if not GROQ_API_KEY_defualt:
+      llm_config = {
+      "config_list": [
+          {
+            "model": llm, #"llama3-8b-8192",
+            "api_key": os.environ.get("GROQ_API_KEY_defualt"),
+            "base_url": "https://api.groq.com/openai/v1"
+          }
+      ]
+      }
+    else GROQ_API_KEY:
+      llm_config = {
+      "config_list": [
+          {
+            "model": llm, #"llama3-8b-8192",
             "api_key": os.environ.get("GROQ_API_KEY"),
             "base_url": "https://api.groq.com/openai/v1"
-        }
-    ]
-    }
+          }
+      ]
+      }
     
     executor = LocalCommandLineCodeExecutor(
         timeout=60,