From 2d26e6cf27c06a1dae1b50fa13054f1cd2ce37ca Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Thu, 13 Nov 2025 14:56:32 -0800 Subject: [PATCH] pre-commit --- tests/unit/core/routers/test_vector_io.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/unit/core/routers/test_vector_io.py b/tests/unit/core/routers/test_vector_io.py index beb7eee9a..a6df0694b 100644 --- a/tests/unit/core/routers/test_vector_io.py +++ b/tests/unit/core/routers/test_vector_io.py @@ -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)