Update excel_chat.py

excel_chat.py

@@ -57,12 +57,17 @@ def ask_llm(query, user_input, client_index, user, keys):
         )
         return response.content[0].text
 
-    elif client_index == "Groq (mixtral)":
+    elif "Groq" in client_index:
         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,
-                model='mixtral-8x7b-32768',
+                model=model_map[client_index],
             )
             response = chat_completion.choices[0].message.content
         except Exception as e: