aymanemalih commited on
Commit
efd83d7
1 Parent(s): 31370c3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +11 -11
main.py CHANGED
@@ -196,23 +196,23 @@ def chat():
196
  results = []
197
  # Update the model name to "text-davinci-003" (Ada)
198
  prompt = "\n".join([f"{msg['role']}: {msg['content']}" for msg in messages])
199
- response = openai.completions.create(
200
- model="gpt-3.5-turbo-instruct",
201
- prompt=start_message +'\n'+ context + question ,
202
- max_tokens=500,
203
- temperature=0
204
- )
205
  date = time.ctime(time.time())
206
  texte = prompt
207
  data = Question(date, texte)
208
  db.session.add(data)
209
  db.session.commit()
210
  question_id = data.id
211
- resulta = response.choices[0].text
212
- chat_references = filtergpt(resulta)
213
- for law, article in chat_references:
214
- search_results = perform_search_and_get_results_with_filter(collection_names[0], prompt, reference_filter=article)
215
- results.extend(search_results)
216
  for collection_name in collection_names:
217
  search_results = perform_search_and_get_results(collection_name, prompt)
218
  results.extend(search_results)
 
196
  results = []
197
  # Update the model name to "text-davinci-003" (Ada)
198
  prompt = "\n".join([f"{msg['role']}: {msg['content']}" for msg in messages])
199
+ # response = openai.completions.create(
200
+ # model="gpt-3.5-turbo-instruct",
201
+ # prompt=start_message +'\n'+ context + question ,
202
+ # max_tokens=500,
203
+ # temperature=0
204
+ # )
205
  date = time.ctime(time.time())
206
  texte = prompt
207
  data = Question(date, texte)
208
  db.session.add(data)
209
  db.session.commit()
210
  question_id = data.id
211
+ # resulta = response.choices[0].text
212
+ # chat_references = filtergpt(resulta)
213
+ # for law, article in chat_references:
214
+ # search_results = perform_search_and_get_results_with_filter(collection_names[0], prompt, reference_filter=article)
215
+ # results.extend(search_results)
216
  for collection_name in collection_names:
217
  search_results = perform_search_and_get_results(collection_name, prompt)
218
  results.extend(search_results)