Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix docstring of RetrievalQA (#3231)

Browse Source 
Structure changed an RetrievalQA now expects BaseRetriever not
VectorStore
This commit is contained in:
Peter Stolz 2023-04-20 19:46:51 +02:00 committed by GitHub
parent b7dea80cba
commit d54b977d4e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/chains/retrieval_qa/base.py
View File

@ -154,8 +154,9 @@ class RetrievalQA(BaseRetrievalQA):
from langchain.llms import OpenAI
from langchain.chains import RetrievalQA
from langchain.faiss import FAISS
vectordb = FAISS(...)
retrievalQA = RetrievalQA.from_llm(llm=OpenAI(), retriever=vectordb)
from langchain.vectorstores.base import VectorStoreRetriever
retriever = VectorStoreRetriever(vectorstore=FAISS(...))
retrievalQA = RetrievalQA.from_llm(llm=OpenAI(), retriever=retriever)
"""