Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|