klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Bugfix] fix opensearch db (#1184)

Browse Source 
Co-authored-by: Deven Patel <deven298@yahoo.com>
This commit is contained in:
Deven Patel
2024-01-23 14:22:58 +05:30
committed by GitHub
parent 3cc3a0058d
commit f50f8a444a
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
embedchain/vectordb/opensearch.py
View File

@@ -96,9 +96,9 @@ class OpenSearchDB(BaseVectorDB):
else: else:
query["query"] = {"bool": {"must": []}} query["query"] = {"bool": {"must": []}}
if "app_id" in where: if where:
app_id = where["app_id"] for key, value in where.items():
query["query"]["bool"]["must"].append({"term": {"metadata.app_id.keyword": app_id}}) query["bool"]["must"].append({"term": {f"metadata.{key}.keyword": value}})
# OpenSearch syntax is different from Elasticsearch # OpenSearch syntax is different from Elasticsearch
response = self.client.search(index=self._get_index(), body=query, _source=True, size=limit) response = self.client.search(index=self._get_index(), body=query, _source=True, size=limit)
@@ -176,9 +176,11 @@ class OpenSearchDB(BaseVectorDB):
) )
pre_filter = {"match_all": {}} # default pre_filter = {"match_all": {}} # default
if "app_id" in where: if len(where) > 0:
app_id = where["app_id"] pre_filter = {"bool": {"must": []}}
pre_filter = {"bool": {"must": [{"term": {"metadata.app_id.keyword": app_id}}]}} for key, value in where.items():
pre_filter["bool"]["must"].append({"term": {f"metadata.{key}.keyword": value}})
docs = docsearch.similarity_search_with_score( docs = docsearch.similarity_search_with_score(
input_query, input_query,
search_type="script_scoring", search_type="script_scoring",
@@ -236,10 +238,9 @@ class OpenSearchDB(BaseVectorDB):
def delete(self, where): def delete(self, where):
"""Deletes a document from the OpenSearch index""" """Deletes a document from the OpenSearch index"""
if "doc_id" not in where: query = {"query": {"bool": {"must": []}}}
raise ValueError("doc_id is required to delete a document") for key, value in where.items():
query["query"]["bool"]["must"].append({"term": {f"metadata.{key}.keyword": value}})
query = {"query": {"bool": {"must": [{"term": {"metadata.doc_id": where["doc_id"]}}]}}}
self.client.delete_by_query(index=self._get_index(), body=query) self.client.delete_by_query(index=self._get_index(), body=query)
def _get_index(self) -> str: def _get_index(self) -> str: