scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-08 08:08:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

style(tests): format the Azure AI Search passthrough tests

Browse source
This commit is contained in:
mateo-berri 2026-10-20 19:02:22 -07:00
parent 78e5a0a738
commit 2b53ff4ac1
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/unit/proxy/pass_through_endpoints/test_llm_pass_through_endpoints.py
View file

@ -2142,7 +2142,6 @@ class TestBedrockPassthroughAwsErrors:
assert exc_info.value.aws_error_type == "ValidationException"
assert (exc_info.value.error.message, exc_info.value.error.code) == (message, status_code)
@pytest.mark.asyncio
async def test_bedrock_request_check_is_raised_with_the_aws_error_type_it_carries(
self, monkeypatch: pytest.MonkeyPatch
@ -2183,6 +2182,7 @@ class TestBedrockPassthroughAwsErrors:
await self._answer(error)
assert (exc_info.value.code, exc_info.value.headers.get("x-model-group")) == ("403", "modèle")
@pytest.mark.asyncio
@pytest.mark.parametrize(
("error", "message", "code"),
@ -5558,7 +5558,9 @@ class TestAzureProxyRouteCrossIndexAuthorization:
@pytest.fixture
def azure_ai_search_virtual_index_client(monkeypatch: pytest.MonkeyPatch) -> Iterator[Callable[[UserAPIKeyAuth], TestClient]]:
def azure_ai_search_virtual_index_client(
monkeypatch: pytest.MonkeyPatch,
) -> Iterator[Callable[[UserAPIKeyAuth], TestClient]]:
from litellm.proxy import proxy_server
from litellm.types.vector_stores import IndexCreateLiteLLMParams, LiteLLM_ManagedVectorStoreIndex
from litellm.vector_stores.vector_store_registry import VectorStoreIndexRegistry, VectorStoreRegistry
@ -5677,7 +5679,6 @@ def test_azure_ai_search_index_definition_put_names_the_mapped_index(
assert json.loads(sent.content) == {"name": upstream_name, "fields": fields}
def test_azure_ai_search_index_delete_on_the_virtual_index_deletes_the_mapped_index(
azure_ai_search_virtual_index_client: Callable[[UserAPIKeyAuth], TestClient],
) -> None:
@ -5690,7 +5691,12 @@ def test_azure_ai_search_index_delete_on_the_virtual_index_deletes_the_mapped_in
assert response.status_code == 204
sent: Final = azure.calls.last.request
assert (sent.method, sent.url.path, sent.headers["api-key"]) == ("DELETE", "/indexes/real-idx", "search-admin-key")
assert (sent.method, sent.url.path, sent.headers["api-key"]) == (
"DELETE",
"/indexes/real-idx",
"search-admin-key",
)
class TestAzureProxyRouteServiceLevelIndexCreate:
"""``POST /indexes`` carries no index name, so the managed-index branch cannot