From 59a9b71d216d942a28f3ca1ca1dec7e60652b711 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 20 Nov 2024 05:50:08 +0530 Subject: [PATCH] build: fix test --- litellm/model_prices_and_context_window_backup.json | 6 ++---- tests/test_organizations.py | 9 ++++++--- 2 files changed, 8 insertions(+), 7 deletions(-) diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index f8dd86cbc..815672ff2 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -4674,8 +4674,7 @@ "litellm_provider": "bedrock", "mode": "chat", "supports_assistant_prefill": true, - "supports_function_calling": true, - "supports_vision": true + "supports_function_calling": true }, "us.anthropic.claude-3-opus-20240229-v1:0": { "max_tokens": 4096, @@ -4741,8 +4740,7 @@ "output_cost_per_token": 0.000005, "litellm_provider": "bedrock", "mode": "chat", - "supports_function_calling": true, - "supports_vision": true + "supports_function_calling": true }, "eu.anthropic.claude-3-opus-20240229-v1:0": { "max_tokens": 4096, diff --git a/tests/test_organizations.py b/tests/test_organizations.py index d62380b4a..947f26019 100644 --- a/tests/test_organizations.py +++ b/tests/test_organizations.py @@ -29,6 +29,7 @@ async def new_organization(session, i, organization_alias, max_budget=None): return await response.json() + async def list_organization(session, i): url = "http://0.0.0.0:4000/organization/list" headers = {"Authorization": "Bearer sk-1234", "Content-Type": "application/json"} @@ -46,6 +47,7 @@ async def list_organization(session, i): return await response.json() + @pytest.mark.asyncio async def test_organization_new(): """ @@ -61,6 +63,7 @@ async def test_organization_new(): ] await asyncio.gather(*tasks) + @pytest.mark.asyncio async def test_organization_list(): """ @@ -77,8 +80,8 @@ async def test_organization_list(): ] await asyncio.gather(*tasks) - response_json = await list_organization(session) + response_json = await list_organization(session, i=0) print(len(response_json)) - if len(response_json)==0: - raise Exception(f"Return empty list of organization") + if len(response_json) == 0: + raise Exception("Return empty list of organization")