o `+ i-&@sdZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZdd lmZmZmZdd lmZdd lmZd efd dZGdddeZd ee fddZGdddeZdS)z'Vectorstore stubs for the indexing api.)AnyOptional) BaseLoader)Document) Embeddings)BaseLanguageModel) VectorStore)RecursiveCharacterTextSplitter TextSplitter) BaseModel ConfigDictField)RetrievalQAWithSourcesChain) RetrievalQAreturncCs tdddS)z=Return the default text splitter used for chunking documents.ir) chunk_sizeZ chunk_overlap)r rri/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/langchain/indexes/vectorstore.py_get_default_text_splitters rc @seZdZUdZeed<edddZ  ddede e d e e ee fd e d ef d d Z  ddede e d e e ee fd e d ef ddZ  ddede e d e e ee fd e d e f ddZ  ddede e d e e ee fd e d e f ddZdS)VectorStoreIndexWrapperz-Wrapper around a vectorstore for easy access. vectorstoreTforbidZarbitrary_types_allowedextraNquestionllmretriever_kwargskwargsrcKsV|dur d}t||p i}tj|fd|jjdi|i|}||j|i|jS)aQuery the vectorstore using the provided LLM. Args: question: The question or prompt to query. llm: The language model to use. Must not be None. retriever_kwargs: Optional keyword arguments for the retriever. **kwargs: Additional keyword arguments forwarded to the chain. Returns: The result string from the RetrievalQA chain. NThis API has been changed to require an LLM. Please provide an llm to use for querying the vectorstore. For example, from langchain_openai import OpenAI llm = OpenAI(temperature=0) retrieverr)NotImplementedErrorrfrom_chain_typer as_retrieverinvoke input_key output_keyselfrrrrmsgchainrrrquery szVectorStoreIndexWrapper.querycs^|dur d}t||pi}tj|fd|jjdi|i|}||j|iIdH|jS)aAsynchronously query the vectorstore using the provided LLM. Args: question: The question or prompt to query. llm: The language model to use. Must not be None. retriever_kwargs: Optional keyword arguments for the retriever. **kwargs: Additional keyword arguments forwarded to the chain. Returns: The asynchronous result string from the RetrievalQA chain. Nrrr)r rr!rr"ainvoker$r%r&rrraqueryCszVectorStoreIndexWrapper.aquerycKsP|dur d}t||p i}tj|fd|jjdi|i|}||j|iS)aQuery the vectorstore and retrieve the answer along with sources. Args: question: The question or prompt to query. llm: The language model to use. Must not be None. retriever_kwargs: Optional keyword arguments for the retriever. **kwargs: Additional keyword arguments forwarded to the chain. Returns: A dictionary containing the answer and source documents. Nrrr)r rr!rr"r# question_keyr&rrrquery_with_sourcesfsz*VectorStoreIndexWrapper.query_with_sourcescsX|dur d}t||pi}tj|fd|jjdi|i|}||j|iIdHS)aAsynchronously query the vectorstore and retrieve the answer and sources. Args: question: The question or prompt to query. llm: The language model to use. Must not be None. retriever_kwargs: Optional keyword arguments for the retriever. **kwargs: Additional keyword arguments forwarded to the chain. Returns: A dictionary containing the answer and source documents. Nrrr)r rr!rr"r+r-r&rrraquery_with_sourcessz+VectorStoreIndexWrapper.aquery_with_sources)NN)__name__ __module__ __qualname____doc__r__annotations__r model_configstrrrdictrr*r,r.r/rrrrrsv  & & &rc CsRddl}zddlm}Wnty}zd}t||d}~ww|jddd|S)zGet the InMemoryVectorStore.rN)InMemoryVectorStorezBPlease install langchain-community to use the InMemoryVectorStore.zUsing InMemoryVectorStore as the default vectorstore.This memory store won't persist data. You should explicitlyspecify a vectorstore when using VectorstoreIndexCreator) stacklevel)warningsZ)langchain_community.vectorstores.inmemoryr8 ImportErrorwarn)r;r8er(rrr_get_in_memory_vectorstores r?c@seZdZUdZeedZeee d<e e d<ee dZ e e d<eedZee d<eddd Zd eed efd d Zd eed efddZdeed efddZdeed efddZdS)VectorstoreIndexCreatorzLogic for creating indexes.)default_factoryvectorstore_cls embedding text_splittervectorstore_kwargsTrrloadersrcCs&g}|D] }||q||S)zCreate a vectorstore index from a list of loaders. Args: loaders: A list of `BaseLoader` instances to load documents. Returns: A `VectorStoreIndexWrapper` containing the constructed vectorstore. )extendloadfrom_documentsr'rFdocsloaderrrr from_loaderss  z$VectorstoreIndexCreator.from_loaderscs>g}|D]}|dd|2IdHq||IdHS)aAsynchronously create a vectorstore index from a list of loaders. Args: loaders: A list of `BaseLoader` instances to load documents. Returns: A `VectorStoreIndexWrapper` containing the constructed vectorstore. csg|z3dHW}|q6S)Nr).0docrrr sz9VectorstoreIndexCreator.afrom_loaders..N)rGZ alazy_loadafrom_documentsrJrrr afrom_loaderss   z%VectorstoreIndexCreator.afrom_loaders documentscCs0|j|}|jj||jfi|j}t|dS)zCreate a vectorstore index from a list of documents. Args: documents: A list of `Document` objects. Returns: A `VectorStoreIndexWrapper` containing the constructed vectorstore. r)rDsplit_documentsrBrIrCrErr'rSZsub_docsrrrrrIs  z&VectorstoreIndexCreator.from_documentscs8|j|}|jj||jfi|jIdH}t|dS)zAsynchronously create a vectorstore index from a list of documents. Args: documents: A list of `Document` objects. Returns: A `VectorStoreIndexWrapper` containing the constructed vectorstore. NrT)rDrUrBrQrCrErrVrrrrQs  z'VectorstoreIndexCreator.afrom_documentsN)r0r1r2r3r r?rBtyperr4rrrDr r7rEr r5listrrrMrRrrIrQrrrrr@s( r@N)r3typingrrZlangchain_core.document_loadersrZlangchain_core.documentsrZlangchain_core.embeddingsrZlangchain_core.language_modelsrZlangchain_core.vectorstoresrZlangchain_text_splittersr r Zpydanticr r r Z*langchain.chains.qa_with_sources.retrievalrZ"langchain.chains.retrieval_qa.baserrrrWr?r@rrrrs