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

fix: Update embedding field name for Elastiscearch mapping (#425)

Browse Source
This commit is contained in:
Taranjeet Singh
2023-08-10 21:13:52 -07:00
committed by GitHub
parent e56f91a239
commit 1dbe7daac1
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
embedchain/vectordb/elasticsearch_db.py
View File

@@ -44,7 +44,7 @@ class ElasticsearchDB(BaseVectorDB):
"mappings": { "mappings": {
"properties": { "properties": {
"text": {"type": "text"}, "text": {"type": "text"},
"text_vector": {"type": "dense_vector", "index": False, "dims": self.vector_dim}, "embeddings": {"type": "dense_vector", "index": False, "dims": self.vector_dim},
} }
} }
} }
@@ -84,12 +84,12 @@ class ElasticsearchDB(BaseVectorDB):
""" """
docs = [] docs = []
embeddings = self.embedding_fn(documents) embeddings = self.embedding_fn(documents)
for id, text, metadata, text_vector in zip(ids, documents, metadatas, embeddings): for id, text, metadata, embeddings in zip(ids, documents, metadatas, embeddings):
docs.append( docs.append(
{ {
"_index": self.es_index, "_index": self.es_index,
"_id": id, "_id": id,
"_source": {"text": text, "metadata": metadata, "text_vector": text_vector}, "_source": {"text": text, "metadata": metadata, "embeddings": embeddings},
} }
) )
bulk(self.client, docs) bulk(self.client, docs)
@@ -109,7 +109,7 @@ class ElasticsearchDB(BaseVectorDB):
"script_score": { "script_score": {
"query": {"bool": {"must": [{"exists": {"field": "text"}}]}}, "query": {"bool": {"must": [{"exists": {"field": "text"}}]}},
"script": { "script": {
"source": "cosineSimilarity(params.input_query_vector, 'text_vector') + 1.0", "source": "cosineSimilarity(params.input_query_vector, 'embeddings') + 1.0",
"params": {"input_query_vector": query_vector}, "params": {"input_query_vector": query_vector},
}, },
} }