youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-29 15:08:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: Revert "feat: initial support for Milvus 2.4.x (#3795)" downgrading to 2.3.x for Linux arm64 installation failure (#5414)

Browse Source
This commit is contained in:
Bowen Liang
2024-06-20 11:18:05 +08:00
committed by GitHub
parent a88aa20824
commit aed56b1a8f
5 changed files with 84 additions and 31 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

@ -6,7 +6,6 @@ from uuid import uuid4
from flask import current_app
from pydantic import BaseModel, model_validator
from pymilvus import MilvusClient, MilvusException, connections
from pymilvus.milvus_client import IndexParams
from core.rag.datasource.entity.embedding import Embeddings
from core.rag.datasource.vdb.field import Field
@ -255,15 +254,11 @@ 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(collection_name=collection_name,
schema=schema, index_params=index_params_obj,
consistency_level=self._consistency_level)
self._client.create_collection_with_schema(collection_name=collection_name,
schema=schema, index_param=index_params,
consistency_level=self._consistency_level)
redis_client.set(collection_exist_cache_key, 1, ex=3600)
def _init_client(self, config) -> MilvusClient: