h jdZddlmZddlZddlmZmZmZddlm Z ddl m Z ddl m Z Gdd e Zy) z-Wrapper around Elasticsearch vector database.) annotationsN)AnyIterableList)CallbackManagerForRetrieverRun)Document) BaseRetrieverc|eZdZUdZded< ded< e d d dZ d d dZ ddZy )ElasticSearchBM25Retrievera?`Elasticsearch` retriever that uses `BM25`. To connect to an Elasticsearch instance that requires login credentials, including Elastic Cloud, use the Elasticsearch URL format https://username:password@es_host:9243. For example, to connect to Elastic Cloud, create the Elasticsearch URL with the required authentication details and pass it to the ElasticVectorSearch constructor as the named parameter elasticsearch_url. You can obtain your Elastic Cloud URL and login credentials by logging in to the Elastic Cloud console at https://cloud.elastic.co, selecting your deployment, and navigating to the "Deployments" page. To obtain your Elastic Cloud password for the default "elastic" user: 1. Log in to the Elastic Cloud console at https://cloud.elastic.co 2. Go to "Security" > "Users" 3. Locate the "elastic" user and click "Edit" 4. Click "Reset password" 5. Follow the prompts to reset the password The format for Elastic Cloud URLs is https://username:password@cluster_id.region_id.gcp.cloud.es.io:9243. rclientstr index_namecddlm}||}ddddiiidd||d id }d d d ddii}|jj||||||S)aD Create a ElasticSearchBM25Retriever from a list of texts. Args: elasticsearch_url: URL of the Elasticsearch instance to connect to. index_name: Name of the index to use in Elasticsearch. k1: BM25 parameter k1. b: BM25 parameter b. Returns: r) Elasticsearchanalyzerdefaulttypestandard custom_bm25BM25)rk1b)analysis similarity propertiescontenttext)rr)indexmappingssettings)r r) elasticsearchrindicescreate) clselasticsearch_urlrrrresr rs p/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/langchain_community/retrievers/elastic_search_bm25.pyr#z!ElasticSearchBM25Retriever.create,s 0, -$i&*1E%FG"    ""/   XQ"44c ddlm}g}g}t|D]U\}}t t j }d|j||d} |j||j| W||j||r0|jjj|j|S#t$r tdwxYw)a.Run more texts through the embeddings and add to the retriever. Args: texts: Iterable of strings to add to the retriever. refresh_indices: bool to refresh ElasticSearch indices Returns: List of ids from adding the texts into the retriever. r)bulkzbCould not import elasticsearch python package. Please install it with `pip install elasticsearch`.r)_op_type_indexr_id)r) elasticsearch.helpersr* ImportError enumerater uuiduuid4rappendr r"refresh) selftextsrefresh_indicesr*requestsidsirr-requests r' add_textsz$ElasticSearchBM25Retriever.add_textsYs  2  ' %GAtdjjl#C#// G JJsO OOG $ % T[[(#  KK   ' 'doo ' > + F  s B66C cddd|iii}|jj|j|}g}|ddD]#}|jt |dd%|S)Nquerymatchr)rbodyhits_source) page_content)r searchrr3r)r5r> run_manager query_dictresdocsrs r'_get_relevant_documentsz2ElasticSearchBM25Retriever._get_relevant_documentsss)U);<= kk  tZ HVV$ HA KKa l9.EF G H r(N)g@g?) r%r rr rfloatrrKreturnr )T)r6z Iterable[str]r7boolrLz List[str])r>r rErrLzList[Document]) __name__ __module__ __qualname____doc____annotations__ classmethodr#r<rJr(r'r r s2 KO4RV*5 #*514*5:?*5JO*5 #*5*5^!%%%%  %N  *H   r(r )rQ __future__rr1typingrrrlangchain_core.callbacksrlangchain_core.documentsrlangchain_core.retrieversr r rTr(r'rZs+3" &&C-3||r(