Cachoups commited on
Commit
a9ddb96
·
verified ·
1 Parent(s): 0a1c5f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -120,7 +120,7 @@ def ra(user_question):
120
  stream=False,
121
  temperature=0.1,
122
  stop = ['('])
123
- return res["choices"][0]["message"]["content"]
124
 
125
  # chain = RunnablePassthrough() | RunnableLambda(ra) | prompt_template | client.chat_completion() | StrOutputParser() for notebook
126
 
@@ -165,7 +165,7 @@ Instructions:
165
  if val[1]:
166
  messages.append({"role": "assistant", "content": val[1]})
167
  messages.append({"role": "user", "content": "Context: "+str(retriever(message))+"\n\nQuestion: "+message +"\n\nAnswer: "})
168
- print("Context: "+str(retriever(message))+"\n\nQuestion: "+message +"\n\nAnswer: ")
169
  response = ""
170
 
171
  for message in client.chat_completion(
 
120
  stream=False,
121
  temperature=0.1,
122
  stop = ['('])
123
+ return copy.deepcopy(res["choices"][0]["message"]["content"])
124
 
125
  # chain = RunnablePassthrough() | RunnableLambda(ra) | prompt_template | client.chat_completion() | StrOutputParser() for notebook
126
 
 
165
  if val[1]:
166
  messages.append({"role": "assistant", "content": val[1]})
167
  messages.append({"role": "user", "content": "Context: "+str(retriever(message))+"\n\nQuestion: "+message +"\n\nAnswer: "})
168
+ print("Context: "+str(retriever(new_query))+"\n\nQuestion: "+new_query +"\n\nAnswer: ")
169
  response = ""
170
 
171
  for message in client.chat_completion(