Spaces:
Running
Running
Now that we have a reranker, retrieve 25 items from the vector DB.
Browse files- sage/chat.py +1 -1
sage/chat.py
CHANGED
|
@@ -28,7 +28,7 @@ def build_rag_chain(args):
|
|
| 28 |
"""Builds a RAG chain via LangChain."""
|
| 29 |
llm = build_llm_via_langchain(args.llm_provider, args.llm_model)
|
| 30 |
|
| 31 |
-
retriever = vector_store.build_from_args(args).to_langchain().as_retriever()
|
| 32 |
|
| 33 |
if args.reranker_provider == "none":
|
| 34 |
compressor = None
|
|
|
|
| 28 |
"""Builds a RAG chain via LangChain."""
|
| 29 |
llm = build_llm_via_langchain(args.llm_provider, args.llm_model)
|
| 30 |
|
| 31 |
+
retriever = vector_store.build_from_args(args).to_langchain().as_retriever(search_kwargs={"k": 25})
|
| 32 |
|
| 33 |
if args.reranker_provider == "none":
|
| 34 |
compressor = None
|