heymenn commited on
Commit
fcf5842
1 Parent(s): d6d1995

Update excel_chat.py

Browse files
Files changed (1) hide show
  1. excel_chat.py +2 -2
excel_chat.py CHANGED
@@ -59,7 +59,7 @@ def ask_llm(query, user_input, client_index, user, keys):
59
 
60
  elif client_index == "Groq (mixtral)":
61
  try:
62
- client = Groq(api_key=os.environ[user['api_keys'][keys[0]]])
63
  chat_completion = client.chat.completions.create(
64
  messages=messages,
65
  model='mixtral-8x7b-32768',
@@ -97,7 +97,7 @@ def filter_df(df, column_name, keywords):
97
 
98
  def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col, keywords, client, user):
99
  # API Keys for Groq :
100
- KEYS = ['groqkey', 'groqkey2', 'groqkey3']
101
  GroqKey = KEYS[0]
102
 
103
  new_prompts, new_keywords, new_user, conf_file_path = update_json(user, prompt, keywords)
 
59
 
60
  elif client_index == "Groq (mixtral)":
61
  try:
62
+ client = Groq(api_key= os.getenv(keys[0])
63
  chat_completion = client.chat.completions.create(
64
  messages=messages,
65
  model='mixtral-8x7b-32768',
 
97
 
98
  def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col, keywords, client, user):
99
  # API Keys for Groq :
100
+ KEYS = ['GROQ_API_KEY1', 'GROQ_API_KEY2', 'GROQ_API_KEY3']
101
  GroqKey = KEYS[0]
102
 
103
  new_prompts, new_keywords, new_user, conf_file_path = update_json(user, prompt, keywords)