NgOdZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZGd d eZd ed edefdZdS)zMilvus RetrieverN)AnyDictListOptional)CallbackManagerForRetrieverRun)Document) Embeddings) BaseRetriever)model_validator)Milvusc `eZdZUdZeed<dZeed<dZe e ee fed<dZ e e ee fed<dZ eed <dZe eed <eed <eed <ed ede de fdZ ddeede eeddfdZdedede deefdZdS)MilvusRetrieveraoMilvus API retriever. See detailed instructions here: https://python.langchain.com/docs/integrations/retrievers/milvus_hybrid_search/ Setup: Install ``langchain-milvus`` and other dependencies: .. code-block:: bash pip install -U pymilvus[model] langchain-milvus Key init args: collection: Milvus Collection Instantiate: .. code-block:: python retriever = MilvusCollectionHybridSearchRetriever(collection=collection) Usage: .. code-block:: python query = "What are the story about ventures?" retriever.invoke(query) .. code-block:: none [Document(page_content="In 'The Lost Expedition' by Caspian Grey...", metadata={'doc_id': '449281835035545843'}), Document(page_content="In 'The Phantom Pilgrim' by Rowan Welles...", metadata={'doc_id': '449281835035545845'}), Document(page_content="In 'The Dreamwalker's Journey' by Lyra Snow..", metadata={'doc_id': '449281835035545846'})] Use within a chain: .. code-block:: python from langchain_core.output_parsers import StrOutputParser from langchain_core.prompts import ChatPromptTemplate from langchain_core.runnables import RunnablePassthrough from langchain_openai import ChatOpenAI prompt = ChatPromptTemplate.from_template( """Answer the question based only on the context provided. Context: {context} Question: {question}""" ) llm = ChatOpenAI(model="gpt-3.5-turbo-0125") def format_docs(docs): return "\n\n".join(doc.page_content for doc in docs) chain = ( {"context": retriever | format_docs, "question": RunnablePassthrough()} | prompt | llm | StrOutputParser() ) chain.invoke("What novels has Lila written and what are their contents?") .. code-block:: none "Lila Rose has written 'The Memory Thief,' which follows a charismatic thief..." embedding_functionLangChainCollectioncollection_nameNcollection_propertiesconnection_argsSessionconsistency_level search_paramsstore retrieverbefore)modevaluesreturnct|d|d|d|d|d|d<|dd|di |d <|S) z&Create the Milvus store and retriever.rrrrrrparamr) search_kwargsr)r as_retriever)clsrs a/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/retrievers/milvus.pycreate_retrieverz MilvusRetriever.create_retriever`s}! ' ( $ % * + $ % & '   w%Wo::"F?$;<;  { texts metadatasc<|j||dS)zAdd text to the Milvus store Args: texts (List[str]): The text metadatas (List[dict]): Metadata dicts, must line up with existing store N)r add_texts)selfr%r&s r"r(zMilvusRetriever.add_textsps" UI.....r$query run_managerkwargsc P|jj|fd|i|S)Nr+)rinvoke get_child)r)r*r+r,s r"_get_relevant_documentsz'MilvusRetriever._get_relevant_documents{sA%t~$   *4466 :@   r$)N)__name__ __module__ __qualname____doc__r __annotations__rstrrrrrrrrdictr r r classmethodr#rr(rrr0r$r"rrs}BBH#"""0OS0006:8DcN3:::04OXd38n-444&s&&&$(M8D>((( MMM_(### d s   [$# CG / /#Y /+3DJ+? /  / / / /    4    h       r$rargsr,rcNtjdtt|i|S)zDeprecated MilvusRetreiver. Please use MilvusRetriever ('i' before 'e') instead. Args: *args: **kwargs: Returns: MilvusRetriever zfMilvusRetreiver will be deprecated in the future. Please use MilvusRetriever ('i' before 'e') instead.)warningswarnDeprecationWarningr)r:r,s r"MilvusRetreiverr?s5 M ? D +F + ++r$)r4r<typingrrrrlangchain_core.callbacksrlangchain_core.documentsrlangchain_core.embeddingsr langchain_core.retrieversr pydanticr 'langchain_community.vectorstores.milvusr rr?r9r$r"rGs ,,,,,,,,,,,,CCCCCC------000000333333$$$$$$:::::: s s s s s ms s s l,3,#,/,,,,,,r$