NgddlmZddlmZmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZGdd eZd S) ) annotations)AnyDictListOptionalcast)uuid4)CallbackManagerForRetrieverRun)Document) BaseRetriever) ConfigDictmodel_validatorceZdZUdZdZded< ded< ded< dZd ed < d Zd ed < ded< dZded< e de d(dZ e dZ d)dZddddd*d'ZdS)+WeaviateHybridSearchRetrieverz{`Weaviate hybrid search` retriever. See the documentation: https://weaviate.io/blog/hybrid-search-explained Nrclientstr index_nametext_keyg?floatalphaintk List[str] attributesTboolcreate_schema_if_missingbefore)modevaluesDict[str, Any]returncR ddl}n#t$rtdwxYwt|d|js'|d}t dt ||dg|d<tt|d |d|ddr]|d |dd gd gd d }|dj |d s |dj ||S)Nrz_Could not import weaviate python package. Please install it with `pip install weaviate-client`.rz5client should be an instance of weaviate.Client, got rrrTrtext)namedataTypeztext2vec-openai)class properties vectorizer) weaviate ImportError isinstanceClient ValueErrortypegetrrappendschemaexists create_class)clsr r*r class_objs q/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/retrievers/weaviate_hybrid_search.pyvalidate_clientz-WeaviateHybridSearchRetriever.validate_client"s^   OOOO   H   &*HO<< H%FVV VV  ::l # # +#%F<  T6,'((//z0BCCC ::0$ 7 7 @ -(.z(:QQR/I (#*11&2FGG @x '44Y??? s!)arbitrary_types_alloweddocsList[Document]kwargsc rddlm}|jj5}g}t |D]y\}}|jpi}|j|ji|} d|vr|d|} n|t} | | |j | | | z dddn #1swxYwY|S)zUpload documents to Weaviate.r)get_valid_uuiduuidsN) weaviate.utilr>rbatch enumeratemetadatar page_contentr add_data_objectrr1) selfr:r<r>rAidsidocrCdata_properties_ids r7 add_documentsz+WeaviateHybridSearchRetriever.add_documentsJs000000 [  %C#D// 3<-2#'=#2B"Oh"Of$$ /!,CC(.11C%%otLLL 3                 sB B,,B03B0F) where_filterscorehybrid_search_kwargsquery run_managerr rMOptional[Dict[str, object]]rNrOc0|jj|j|j}|r||}|r|ddg}|i}|j|fd|ji| |j  }d|vrtd|dg}|dd|jD]@} | |j} |t!| | A|S) aDLook up similar documents in Weaviate. query: The query to search for relevant documents of using weviate hybrid search. where_filter: A filter to apply to the query. https://weaviate.io/developers/weaviate/guides/querying/#filtering score: Whether to include the score, and score explanation in the returned Documents meta_data. hybrid_search_kwargs: Used to pass additional arguments to the .with_hybrid() method. The primary uses cases for this are: 1) Search specific properties only - specify which properties to be used during hybrid search portion. Note: this is not the same as the (self.attributes) to be returned. Example - hybrid_search_kwargs={"properties": ["question", "answer"]} https://weaviate.io/developers/weaviate/search/hybrid#selected-properties-only 2) Weight boosted searched properties - Boost the weight of certain properties during the hybrid search portion. Example - hybrid_search_kwargs={"properties": ["question^2", "answer"]} https://weaviate.io/developers/weaviate/search/hybrid#weight-boost-searched-properties 3) Search with a custom vector - Define a different vector to be used during the hybrid search portion. Example - hybrid_search_kwargs={"vector": [0.1, 0.2, 0.3, ...]} https://weaviate.io/developers/weaviate/search/hybrid#with-a-custom-vector 4) Use Fusion ranking method Example - from weaviate.gql.get import HybridFusion hybrid_search_kwargs={"fusion": fusion_type=HybridFusion.RELATIVE_SCORE} https://weaviate.io/developers/weaviate/search/hybrid#fusion-ranking-method rN explainScoreNrerrorszError during query: dataGet)rDrC)rrPr0rr with_wherewith_additional with_hybridr with_limitrdor.poprr1r ) rFrPrQrMrNrO query_objresultr:resr$s r7_get_relevant_documentsz5WeaviateHybridSearchRetriever._get_relevant_documents_s3XK%))$/4?KK  ;!,,\::I  M!117N2KLLI  '#%  "I !% R Rtz R=Q R R Z   RTT  v  FF84DFFGG G&>%(9 C CC774=))D KKdSAAA B B B B )r r!r"r)r:r;r<rr"r) rPrrQr rMrRrNrrOrRr"r;)__name__ __module__ __qualname____doc__r__annotations__rrrr classmethodr8r model_configrLrarbr7rr s- F;OOO'MMM*E: AJJJJ*2%)))));_(###[$#B: $L 459<@CCCCCCCCrbrN) __future__rtypingrrrrruuidr langchain_core.callbacksr langchain_core.documentsr langchain_core.retrieversr pydanticr rrrjrbr7rrs""""""22222222222222CCCCCC------33333300000000VVVVVMVVVVVrb