YchKhan commited on
Commit
803a58c
1 Parent(s): 62df666

api keys update

Browse files
Files changed (1) hide show
  1. excel_chat.py +2 -2
excel_chat.py CHANGED
@@ -19,13 +19,13 @@ def ask_llm(query, input, client_index):
19
  ]
20
 
21
  if client_index == 0:
22
- client = Groq(api_key=userdata.get('GROQ_API_KEY'))
23
  chat_completion = client.chat.completions.create(
24
  messages=messages,
25
  model='mixtral-8x7b-32768',
26
  )
27
  else:
28
- client = MistralClient(api_key=userdata.get('MISTRAL_API_KEY'))
29
  chat_completion = client.chat(
30
  messages=messages,
31
  model='mistral-small-latest',
 
19
  ]
20
 
21
  if client_index == 0:
22
+ client = Groq(api_key=os.environ["GROQ_API_KEY"])
23
  chat_completion = client.chat.completions.create(
24
  messages=messages,
25
  model='mixtral-8x7b-32768',
26
  )
27
  else:
28
+ client = MistralClient(api_key=os.environ['MISTRAL_API_KEY']))
29
  chat_completion = client.chat(
30
  messages=messages,
31
  model='mistral-small-latest',