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

pre-commit

Browse source
This commit is contained in:
Ashwin Bharambe 2025-11-13 14:56:32 -08:00
parent dcd4370806
commit 2d26e6cf27
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/unit/core/routers/test_vector_io.py
View file

@ -7,6 +7,8 @@
from unittest.mock import AsyncMock, Mock
import pytest
from llama_stack.core.routers.vector_io import VectorIORouter
from llama_stack_api import (
ModelNotFoundError,
ModelType,
@ -14,9 +16,6 @@ from llama_stack_api import (
OpenAICreateVectorStoreRequestWithExtraBody,
)
from llama_stack.core.routers.vector_io import VectorIORouter
from llama_stack_api import OpenAICreateVectorStoreRequestWithExtraBody
async def test_single_provider_auto_selection():
# provider_id automatically selected during vector store create() when only one provider available
@ -128,7 +127,7 @@ async def test_update_vector_store_same_provider_id_succeeds():
async def test_create_vector_store_with_unknown_embedding_model_raises_error():
"""Test that creating a vector store with an unknown embedding model raises
"""Test that creating a vector store with an unknown embedding model raises
FoundError."""
mock_routing_table = Mock(impls_by_provider_id={"provider": "mock"})
mock_routing_table.get_object_by_identifier = AsyncMock(return_value=None)