Skip to content

Commit

Permalink
fix: change retrieval to non-parent retrieval
Browse files Browse the repository at this point in the history
initial eval on parent retrieval doesn't yield good results
  • Loading branch information
parambharat committed Apr 25, 2024
1 parent 8ea83e4 commit d9126dd
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/wandbot/rag/retrieval.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
from langchain.retrievers.document_compressors import CohereRerank
from langchain_core.documents import Document
from langchain_core.runnables import Runnable, RunnablePassthrough

from wandbot.rag.utils import get_web_contexts
from wandbot.retriever.base import VectorStore
from wandbot.retriever.web_search import YouSearch, YouSearchConfig
Expand Down Expand Up @@ -64,7 +63,7 @@ def __init__(
self.top_k = top_k
self.search_type = search_type

self.retriever = self.vectorstore.as_parent_retriever(
self.retriever = self.vectorstore.as_retriever(
search_type=self.search_type, search_kwargs={"k": self.top_k}
)

Expand Down

0 comments on commit d9126dd

Please sign in to comment.