Update answer_question.py

This commit is contained in:
isafulf 2023-04-02 17:33:25 -07:00
parent 59e10cec11
commit b637c93e74

View File

@ -42,6 +42,7 @@ def get_answer_from_files(question, session_id, pinecone_index):
break break
files_string += file_string files_string += file_string
# Note: this is not the proper way to use the ChatGPT conversational format, but it works for now
messages = [ messages = [
{ {
"role": "system", "role": "system",
@ -69,7 +70,7 @@ def get_answer_from_files(question, session_id, pinecone_index):
choices = response["choices"] # type: ignore choices = response["choices"] # type: ignore
answer = choices[0].message.content.strip() answer = choices[0].message.content.strip()
logging.info(f"[get_answer_from_files] answer: {answer}") logging.info(f"[get_answer_from_files] answer: {answer}")
return jsonify({"answer": answer}) return jsonify({"answer": answer})