heymenn commited on
Commit
2591ca4
1 Parent(s): 7ae4f2a

Update excel_chat.py

Browse files
Files changed (1) hide show
  1. excel_chat.py +2 -2
excel_chat.py CHANGED
@@ -60,7 +60,7 @@ def ask_llm(query, user_input, client_index, user, keys):
60
 
61
  elif "GPT 4o" in client_index:
62
  client = OpenAI(api_key=os.environ["OPENAI_YCHK"])
63
- response = client.messages.create(
64
  model="gpt-4o",
65
  messages=messageC
66
  )
@@ -73,7 +73,7 @@ def ask_llm(query, user_input, client_index, user, keys):
73
  "Perplexity Llama3 70b": "llama-3-70b-instruct",
74
  "Perplexity Llama3 8b": "llama-3-8b-instruct"
75
  }
76
- response = client.messages.create(
77
  model=model_map[client_index],
78
  messages=messageC
79
  )
 
60
 
61
  elif "GPT 4o" in client_index:
62
  client = OpenAI(api_key=os.environ["OPENAI_YCHK"])
63
+ response = client.chat.completions.create(
64
  model="gpt-4o",
65
  messages=messageC
66
  )
 
73
  "Perplexity Llama3 70b": "llama-3-70b-instruct",
74
  "Perplexity Llama3 8b": "llama-3-8b-instruct"
75
  }
76
+ response = client.chat.completions.create(
77
  model=model_map[client_index],
78
  messages=messageC
79
  )