estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

dep: initial support for Milvus 2.4.x (#6084)

Browse Source
This commit is contained in:
Bowen Liang
2024-07-29 19:56:45 +08:00
committed by GitHub
parent f78d0082ae
commit b414ea41d6
4 changed files with 45 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
api/core/rag/datasource/vdb/milvus/milvus_vector.py
View File

@@ -5,6 +5,7 @@ from uuid import uuid4
from pydantic import BaseModel, model_validator
from pymilvus import MilvusClient, MilvusException, connections
from pymilvus.milvus_client import IndexParams
from configs import dify_config
from core.rag.datasource.entity.embedding import Embeddings
@@ -250,11 +251,15 @@ class MilvusVector(BaseVector):
# Since primary field is auto-id, no need to track it
self._fields.remove(Field.PRIMARY_KEY.value)
# Create Index params for the collection
index_params_obj = IndexParams()
index_params_obj.add_index(field_name=Field.VECTOR.value, **index_params)
# Create the collection
collection_name = self._collection_name
self._client.create_collection_with_schema(collection_name=collection_name,
schema=schema, index_param=index_params,
consistency_level=self._consistency_level)
self._client.create_collection(collection_name=collection_name,
schema=schema, index_params=index_params_obj,
consistency_level=self._consistency_level)
redis_client.set(collection_exist_cache_key, 1, ex=3600)
def _init_client(self, config) -> MilvusClient: