phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-15 14:08:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
Ashwin Bharambe 2025-08-12 14:36:59 -07:00
parent 35b9da04e5
commit 61bffe8b56
2 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
llama_stack/providers/registry/vector_io.py
View file

@ -466,6 +466,7 @@ See [Weaviate's documentation](https://weaviate.io/developers/weaviate) for more
""", """,
), ),
api_dependencies=[Api.inference], api_dependencies=[Api.inference],
optional_api_dependencies=[Api.files],
), ),
InlineProviderSpec( InlineProviderSpec(
api=Api.vector_io, api=Api.vector_io,

6
llama_stack/providers/remote/vector_io/milvus/milvus.py
View file

@ -290,10 +290,12 @@ class MilvusIndex(EmbeddingIndex):
async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeletion]) -> None: async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeletion]) -> None:
"""Remove a chunk from the Milvus collection.""" """Remove a chunk from the Milvus collection."""
chunk_ids_str = ",".join(f"'{c.chunk_id}'" for c in chunks_for_deletion) chunk_ids = [c.chunk_id for c in chunks_for_deletion]
try: try:
# Use IN clause with square brackets and single quotes for VARCHAR field
chunk_ids_str = ", ".join(f"'{chunk_id}'" for chunk_id in chunk_ids)
await asyncio.to_thread( await asyncio.to_thread(
self.client.delete, collection_name=self.collection_name, filter=f"chunk_id in ({chunk_ids_str})" self.client.delete, collection_name=self.collection_name, filter=f"chunk_id in [{chunk_ids_str}]"
) )
except Exception as e: except Exception as e:
logger.error(f"Error deleting chunks from Milvus collection {self.collection_name}: {e}") logger.error(f"Error deleting chunks from Milvus collection {self.collection_name}: {e}")