Dinesh310 commited on
Commit
e3f3e21
·
verified ·
1 Parent(s): 9d57136

Update src/rag_engine.py

Browse files
Files changed (1) hide show
  1. src/rag_engine.py +2 -2
src/rag_engine.py CHANGED
@@ -34,7 +34,7 @@ class ProjectRAGEngine:
34
  all_docs.extend(loader.load())
35
 
36
  splitter = RecursiveCharacterTextSplitter(
37
- chunk_size=250,
38
  chunk_overlap=50
39
  )
40
 
@@ -64,7 +64,7 @@ class ProjectRAGEngine:
64
  """
65
 
66
  prompt = ChatPromptTemplate.from_template(template)
67
- retriever = self.vector_store.as_retriever(search_kwargs={"k": 5})
68
 
69
  rag_chain = (
70
  RunnablePassthrough.assign(
 
34
  all_docs.extend(loader.load())
35
 
36
  splitter = RecursiveCharacterTextSplitter(
37
+ chunk_size=500,
38
  chunk_overlap=50
39
  )
40
 
 
64
  """
65
 
66
  prompt = ChatPromptTemplate.from_template(template)
67
+ retriever = self.vector_store.as_retriever(search_type="mmr", search_kwargs={"k": 5, "lambda_mult":0.25})
68
 
69
  rag_chain = (
70
  RunnablePassthrough.assign(