rename __embed__key__ => embed_key

LazyAGI · Dec 16, 2024 · 93866e0 · 93866e0
1 parent b61ee3c
commit 93866e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lazyllm/tools/rag/milvus_store.py b/lazyllm/tools/rag/milvus_store.py
@@ -100,7 +100,7 @@ def __init__(self, group_embed_keys: Dict[str, Set[str]], embed: Dict[str, Calla
                 field_list.append(pymilvus.FieldSchema(name=field_name, dtype=self._type2milvus[datatype],
                                                        **field_kwargs))
                 if isinstance(index_kwargs, list):
-                    embed_key_field_name = "__embed_key__"
+                    embed_key_field_name = "embed_key"
                     for item in index_kwargs:
                         item_key = item.get(embed_key_field_name, None)
                         if not item_key:

diff --git a/tests/basic_tests/test_store.py b/tests/basic_tests/test_store.py
@@ -345,12 +345,12 @@ def setUp(self):
             'uri': self.store_file,
             'index_kwargs': [
                 {
-                    '__embed_key__': 'vec1',
+                    'embed_key': 'vec1',
                     'index_type': 'SPARSE_INVERTED_INDEX',
                     'metric_type': 'IP',
                 },
                 {
-                    '__embed_key__': 'vec2',
+                    'embed_key': 'vec2',
                     'index_type': 'SPARSE_INVERTED_INDEX',
                     'metric_type': 'IP',
                 }