tjddlmZddlmZmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZe d d d Gd deZdS)) annotations)AnyDictListOptionalcast)uuid4) deprecated)CallbackManagerForRetrieverRun)Document) BaseRetriever) ConfigDictmodel_validatorz0.3.18z1.0z&langchain_weaviate.WeaviateVectorStore)sinceremovalalternative_importceZdZUdZdZded< 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 rrr!Trtext)namedataTypeztext2vec-openai)class properties vectorizer) weaviate ImportError isinstanceClient ValueErrortypegetrrappendschemaexists create_class)clsr$r.r class_objs s/var/www/html/CA-Chatbot/venv/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.utilrBrbatch enumeratemetadatar page_contentr add_data_objectrr5) selfr>r@rBrEidsidocrGdata_properties_ids r; add_documentsz+WeaviateHybridSearchRetriever.add_documentsPs000000 [  %C#D// 3<-2#'=#2B"Oh"Of$$ /!,CC(.11C%%otLLL 3                 sB B,,B03B0F) where_filterscorehybrid_search_kwargsquery run_managerr rQOptional[Dict[str, object]]rRrSc0|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 rR explainScoreNrerrorszError during query: dataGet)rHrG)rrTr4rr with_wherewith_additional with_hybridr with_limitrdor2poprr5r ) rJrTrUrQrRrS query_objresultr>resr(s r;_get_relevant_documentsz5WeaviateHybridSearchRetriever._get_relevant_documentses3XK%))$/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) rTrrUr rQrVrRr rSrVr&r?)__name__ __module__ __qualname____doc__r__annotations__rrr!r classmethodr<r model_configrPrerfr;rr s-  F;OOO'MMM*E: AJJJJ*2%)))));_(###[$#B: $L 459<@CCCCCCCCrfrN) __future__rtypingrrrrruuidr langchain_core._apir langchain_core.callbacksr langchain_core.documentsr langchain_core.retrieversr pydanticrrrrnrfr;rws""""""22222222222222******CCCCCC------33333300000000   ? VVVVVMVV  VVVrf