aymanemalih commited on
Commit
0299dbf
1 Parent(s): 1c62393

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -7
main.py CHANGED
@@ -208,19 +208,14 @@ def chat():
208
  results = []
209
  # Update the model name to "text-davinci-003" (Ada)
210
  prompt = "\n".join([f"{msg['role']}: {msg['content']}" for msg in messages])
211
- response = openai.completions.create(
212
- model="gpt-3.5-turbo-instruct",
213
- prompt=start_message +'\n'+ context + question ,
214
- max_tokens=500,
215
- temperature=0
216
- )
217
  date = time.ctime(time.time())
218
  texte = prompt
219
  data = Question(date, texte)
220
  db.session.add(data)
221
  db.session.commit()
222
  question_id = data.id
223
- resulta = response.choices[0].text
224
  chat_references = filtergpt(resulta)
225
  for law, article in chat_references:
226
  search_results = perform_search_and_get_results_with_filter(collection_names[0], prompt, reference_filter=article)
 
208
  results = []
209
  # Update the model name to "text-davinci-003" (Ada)
210
  prompt = "\n".join([f"{msg['role']}: {msg['content']}" for msg in messages])
211
+ response = response = openai.chat.completions.create(model="gpt-4",messages=[{"role": "user","content": start_message +'\n'+ context + question,},],)
 
 
 
 
 
212
  date = time.ctime(time.time())
213
  texte = prompt
214
  data = Question(date, texte)
215
  db.session.add(data)
216
  db.session.commit()
217
  question_id = data.id
218
+ resulta = response.choices[0].message.content
219
  chat_references = filtergpt(resulta)
220
  for law, article in chat_references:
221
  search_results = perform_search_and_get_results_with_filter(collection_names[0], prompt, reference_filter=article)