Просмотр исходного кода

fix milvus database name parameter missed (#3229)

tags/0.6.1
Jyong 1 год назад
Родитель
Сommit
8fcf459285
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 4 добавлений и 2 удалений
  1. 4
    2
      api/core/rag/datasource/vdb/milvus/milvus_vector.py

+ 4
- 2
api/core/rag/datasource/vdb/milvus/milvus_vector.py Просмотреть файл

@@ -130,7 +130,8 @@ class MilvusVector(BaseVector):
uri = "https://" + str(self._client_config.host) + ":" + str(self._client_config.port)
else:
uri = "http://" + str(self._client_config.host) + ":" + str(self._client_config.port)
connections.connect(alias=alias, uri=uri, user=self._client_config.user, password=self._client_config.password)
connections.connect(alias=alias, uri=uri, user=self._client_config.user, password=self._client_config.password,
db_name=self._client_config.database)

from pymilvus import utility
if utility.has_collection(self._collection_name, using=alias):
@@ -142,7 +143,8 @@ class MilvusVector(BaseVector):
uri = "https://" + str(self._client_config.host) + ":" + str(self._client_config.port)
else:
uri = "http://" + str(self._client_config.host) + ":" + str(self._client_config.port)
connections.connect(alias=alias, uri=uri, user=self._client_config.user, password=self._client_config.password)
connections.connect(alias=alias, uri=uri, user=self._client_config.user, password=self._client_config.password,
db_name=self._client_config.database)

from pymilvus import utility
if not utility.has_collection(self._collection_name, using=alias):

Загрузка…
Отмена
Сохранить