Update excel_chat.py

excel_chat.py

@@ -57,6 +57,27 @@ def ask_llm(query, user_input, client_index, user, keys):
         )
         return response.content[0].text
 
+    elif "GPT 4o" in client_index:
+        client = OpenAI(api_key=os.environ[user['api_keys']['gpt-4o']])
+        response = client.messages.create(
+            model="gpt-4o",
+            messages=messageC
+        )
+        return response.content[0].text
+
+    elif "Perplexity" in client_index:
+        client = OpenAI(api_key=os.environ[user['api_keys']['perplexity']], base_url="https://api.perplexity.ai")
+        model_map = {
+                "Perplexity Mixtral": "mixtral-8x7b-instruct",
+                "Perplexity Llama3 70b": "llama-3-70b-instruct",
+                "Perplexity Llama3 8b": "llama-3-8b-instruct"
+            }
+        response = client.messages.create(
+            model=model_map[client_index],
+            messages=messageC
+        )
+        return response.content[0].text
+        
     elif "Groq" in client_index:
         try:
             client = Groq(api_key= os.getenv(keys[0]))