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

fix(unittest): add pymilvus and milvus-lite to unit dep group

Browse source 
This commit resolves unittest-3.12 issues.

- It adds pymilvus and milvus-lite to unit dep group
- It rename VectorDB to VectorStore

Signed-off-by: Mustafa Elbehery <melbeher@redhat.com>
This commit is contained in:
Mustafa Elbehery 2025-11-03 23:38:44 +01:00
parent 3e8291c1b9
commit c6bf292f07
4 changed files with 17 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/unit/providers/vector_io/conftest.py
View file

@ -11,7 +11,6 @@ import numpy as np
import pytest
from pymilvus import AsyncMilvusClient, connections
from llama_stack.apis.vector_dbs import VectorDB
from llama_stack.apis.vector_io import Chunk, ChunkMetadata, QueryChunksResponse
from llama_stack.apis.vector_stores import VectorStore
from llama_stack.core.storage.datatypes import KVStoreReference, SqliteKVStoreConfig
@ -205,8 +204,8 @@ async def milvus_vec_adapter(milvus_vec_db_path, mock_inference_api):
files_api=None,
)
await adapter.initialize()
await adapter.register_vector_db(
VectorDB(
await adapter.register_vector_store(
VectorStore(
identifier=adapter.metadata_collection_name,
provider_id="test_provider",
embedding_model="test_model",