juliaturc commited on
Commit
82df3b5
·
1 Parent(s): 2c96b06

Now that we have a reranker, retrieve 25 items from the vector DB.

Browse files
Files changed (1) hide show
  1. 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