Update src/rag_engine.py
Browse files- 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=
|
| 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(
|