phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 09:53:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor(client): replace all AsyncMilvusClient usage of has_collection() with list_collections()

Browse source 
Signed-off-by: Mustafa Elbehery <melbeher@redhat.com>
This commit is contained in:
Mustafa Elbehery 2025-09-09 22:20:02 +02:00
parent c3ce2df439
commit 2347db868d
2 changed files with 15 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/llama_stack/providers/remote/vector_io/milvus/milvus.py
View file

@ -63,7 +63,8 @@ class MilvusIndex(EmbeddingIndex):
async def delete(self):
try:
if await self.client.has_collection(self.collection_name):
collections = await self.client.list_collections()
if self.collection_name in collections:
await self.client.drop_collection(collection_name=self.collection_name)
except Exception as e:
logger.warning(f"Failed to check or delete collection {self.collection_name}: {e}")
@ -74,7 +75,8 @@ class MilvusIndex(EmbeddingIndex):
)
try:
collection_exists = await self.client.has_collection(self.collection_name)
collections = await self.client.list_collections()
collection_exists = self.collection_name in collections
except Exception as e:
logger.error(f"Failed to check collection existence: {self.collection_name} ({e})")
# If it's an event loop issue, try to recreate the client
@ -83,7 +85,8 @@ class MilvusIndex(EmbeddingIndex):
if hasattr(self, "_parent_adapter"):
await self._parent_adapter._recreate_client()
collection_exists = await self.client.has_collection(self.collection_name)
collections = await self.client.list_collections()
collection_exists = self.collection_name in collections
else:
# Assume collection doesn't exist if we can't check
collection_exists = False