fix add segment when dataset and document is empty (#3021)

Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong
2024-03-29 13:06:00 +08:00
committed by GitHub
parent 2c43393bf1
commit a6cd0f0e73
4 changed files with 24 additions and 2 deletions

View File

@@ -144,6 +144,16 @@ class MilvusVector(BaseVector):
utility.drop_collection(self._collection_name, None, using=alias)
def text_exists(self, id: str) -> bool:
alias = uuid4().hex
if self._client_config.secure:
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)
from pymilvus import utility
if not utility.has_collection(self._collection_name, using=alias):
return False
result = self._client.query(collection_name=self._collection_name,
filter=f'metadata["doc_id"] == "{id}"',