araeyn commited on
Commit
66c72cf
1 Parent(s): d278532

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -50,7 +50,7 @@ def format_docs(docs):
50
 
51
  #search_kwargs={"score_threshold": 0.3}
52
  #search_type="similarity_score_threshold"
53
- retriever = vectorstore.as_retriever()
54
 
55
  prompt = hub.pull("rlm/rag-prompt")
56
  llm = HuggingFaceEndpoint(repo_id="mistralai/Mistral-7B-Instruct-v0.3", stop_sequences=["Human:"])
@@ -139,7 +139,7 @@ async def echo(websocket):
139
  return
140
  if not "token" in message:
141
  return
142
- m = data["message"] + "\nAssistant: "
143
  token = data["token"]
144
  docs = retriever.get_relevant_documents(m)
145
  rawresponse = conversational_rag_chain.invoke(
 
50
 
51
  #search_kwargs={"score_threshold": 0.3}
52
  #search_type="similarity_score_threshold"
53
+ retriever = vectorstore.as_retriever(k = 4)
54
 
55
  prompt = hub.pull("rlm/rag-prompt")
56
  llm = HuggingFaceEndpoint(repo_id="mistralai/Mistral-7B-Instruct-v0.3", stop_sequences=["Human:"])
 
139
  return
140
  if not "token" in message:
141
  return
142
+ m = data["message"] + "\n\nAssistant: "
143
  token = data["token"]
144
  docs = retriever.get_relevant_documents(m)
145
  rawresponse = conversational_rag_chain.invoke(