Update excel_chat.py

excel_chat.py

@@ -10,7 +10,7 @@ from users_management import update_json, users
 
 #users = ['maksG', 'AlmaA', 'YchK']
 
-def ask_llm(query, input, client_index):
+def ask_llm(query, user_input, client_index, user):
     messages = [
         {
             "role": "system",
@@ -18,7 +18,7 @@ def ask_llm(query, input, client_index):
         },
         {
             "role": "user",
-            "content": input,
+            "content": user_input,
         }
     ]
     
@@ -27,12 +27,12 @@ def ask_llm(query, input, client_index):
         "role": "user",
         "content": [{
             "type": "text",
-            "text": input
+            "text": user_input
         }]
     }]
 
     if "Mistral" in client_index:
-        client = MistralClient(api_key=os.environ['MISTRAL_API_KEY'])
+        client = MistralClient(api_key=os.environ[user['api_keys']['mistral']])
         model_map = {
             "Mistral Tiny": "mistral-tiny",
             "Mistral Small": "mistral-small-latest",
@@ -41,7 +41,7 @@ def ask_llm(query, input, client_index):
         chat_completion = client.chat(messages=messages, model=model_map[client_index])
 
     elif "Claude" in client_index:
-        client = anthropic.Anthropic(api_key=os.environ['CLAUDE_API_KEY'])
+        client = anthropic.Anthropic(api_key=os.environ[user['api_keys']['claude']])
         model_map = {
             "Claude Sonnet": "claude-3-sonnet-20240229",
             "Claude Opus": "claude-3-opus-20240229",
@@ -56,7 +56,7 @@ def ask_llm(query, input, client_index):
         return response.content[0].text
 
     elif client_index == "Groq (mixtral)":
-        client = Groq(api_key=os.environ["GROQ_API_KEY"])
+        client = Groq(api_key=os.environ[user['api_keys']['groq']])
         chat_completion = client.chat.completions.create(
             messages=messages,
             model='mixtral-8x7b-32768',