araeyn commited on
Commit
2607d44
·
verified ·
1 Parent(s): 1a09aef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -87,7 +87,6 @@ qa_prompt = ChatPromptTemplate.from_messages(
87
  ("system", qa_system_prompt),
88
  MessagesPlaceholder("chat_history"),
89
  ("user", "{input}"),
90
- ("assistant", "")
91
  ]
92
  )
93
  question_answer_chain = create_stuff_documents_chain(llm, qa_prompt)
@@ -123,7 +122,7 @@ async def echo(websocket):
123
  return
124
  m = data["message"]
125
  token = data["token"]
126
- docs = retriever.get_relevant_documents(m)
127
  response = conversational_rag_chain.invoke(
128
  {"input": m},
129
  config={
 
87
  ("system", qa_system_prompt),
88
  MessagesPlaceholder("chat_history"),
89
  ("user", "{input}"),
 
90
  ]
91
  )
92
  question_answer_chain = create_stuff_documents_chain(llm, qa_prompt)
 
122
  return
123
  m = data["message"]
124
  token = data["token"]
125
+ docs = retriever.get_relevant_documents(m + "\n\nAssistant: ")
126
  response = conversational_rag_chain.invoke(
127
  {"input": m},
128
  config={