Standard_Intelligence_Dev

Sleeping

YchKhan commited on May 7

Commit

8817d01

•

1 Parent(s): 8cfa293

Update excel_chat.py

Files changed (1) hide show

excel_chat.py CHANGED Viewed

@@ -61,9 +61,9 @@ def ask_llm(query, user_input, client_index, user, keys):
         try:
             client = Groq(api_key= os.getenv(keys[0]))
             model_map = {
-                "Groq Mixtral" : "llama3-70b-8192"
-                "Groq Llama3 70b" : "llama3-8b-8192"
-                "Groq Llama3 8b" : "mixtral-8x7b-32768"
             }
             chat_completion = client.chat.completions.create(
                 messages=messages,

         try:
             client = Groq(api_key= os.getenv(keys[0]))
             model_map = {
+                "Groq Mixtral": "mixtral-8x7b-32768"
+                "Groq Llama3 70b": "llama3-70b-8192"
+                "Groq Llama3 8b": "llama3-8b-8192"
             }
             chat_completion = client.chat.completions.create(
                 messages=messages,