mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-12 13:57:57 +00:00
Merge 376eb94def
into d266c59c2a
This commit is contained in:
commit
7774d9e9f3
80 changed files with 32197 additions and 186 deletions
|
@ -29,7 +29,7 @@ from llama_stack.providers.remote.vector_io.qdrant.qdrant import QdrantVectorIOA
|
|||
from llama_stack.providers.remote.vector_io.weaviate.config import WeaviateVectorIOConfig
|
||||
from llama_stack.providers.remote.vector_io.weaviate.weaviate import WeaviateIndex, WeaviateVectorIOAdapter
|
||||
|
||||
EMBEDDING_DIMENSION = 384
|
||||
EMBEDDING_DIMENSION = 768
|
||||
COLLECTION_PREFIX = "test_collection"
|
||||
MILVUS_ALIAS = "test_milvus"
|
||||
|
||||
|
|
|
@ -23,7 +23,7 @@ def loop():
|
|||
@pytest.fixture
|
||||
def embedding_dimension():
|
||||
"""Default embedding dimension for tests."""
|
||||
return 384
|
||||
return 768
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
|
|
@ -39,7 +39,7 @@ def loop():
|
|||
|
||||
@pytest.fixture
|
||||
def embedding_dimension():
|
||||
return 384
|
||||
return 768
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
|
|
@ -52,11 +52,9 @@ def mock_vector_db(vector_db_id) -> MagicMock:
|
|||
mock_vector_db = MagicMock(spec=VectorDB)
|
||||
mock_vector_db.embedding_model = "embedding_model"
|
||||
mock_vector_db.identifier = vector_db_id
|
||||
mock_vector_db.embedding_dimension = 384
|
||||
mock_vector_db.embedding_dimension = 768
|
||||
mock_vector_db.model_dump_json.return_value = (
|
||||
'{"identifier": "'
|
||||
+ vector_db_id
|
||||
+ '", "provider_id": "qdrant", "embedding_model": "embedding_model", "embedding_dimension": 384}'
|
||||
'{"identifier": "' + vector_db_id + '", "embedding_model": "embedding_model", "embedding_dimension": 768}'
|
||||
)
|
||||
return mock_vector_db
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue