phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

build: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-20 05:50:08 +05:30
parent cf579fe644
commit 59a9b71d21
2 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/model_prices_and_context_window_backup.json
View file

@ -4674,8 +4674,7 @@
"litellm_provider": "bedrock", "litellm_provider": "bedrock",
"mode": "chat", "mode": "chat",
"supports_assistant_prefill": true, "supports_assistant_prefill": true,
"supports_function_calling": true, "supports_function_calling": true
"supports_vision": true
}, },
"us.anthropic.claude-3-opus-20240229-v1:0": { "us.anthropic.claude-3-opus-20240229-v1:0": {
"max_tokens": 4096, "max_tokens": 4096,
@ -4741,8 +4740,7 @@
"output_cost_per_token": 0.000005, "output_cost_per_token": 0.000005,
"litellm_provider": "bedrock", "litellm_provider": "bedrock",
"mode": "chat", "mode": "chat",
"supports_function_calling": true, "supports_function_calling": true
"supports_vision": true
}, },
"eu.anthropic.claude-3-opus-20240229-v1:0": { "eu.anthropic.claude-3-opus-20240229-v1:0": {
"max_tokens": 4096, "max_tokens": 4096,

7
tests/test_organizations.py
View file

@ -29,6 +29,7 @@ async def new_organization(session, i, organization_alias, max_budget=None):
return await response.json() return await response.json()
async def list_organization(session, i): async def list_organization(session, i):
url = "http://0.0.0.0:4000/organization/list" url = "http://0.0.0.0:4000/organization/list"
headers = {"Authorization": "Bearer sk-1234", "Content-Type": "application/json"} headers = {"Authorization": "Bearer sk-1234", "Content-Type": "application/json"}
@ -46,6 +47,7 @@ async def list_organization(session, i):
return await response.json() return await response.json()
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_organization_new(): async def test_organization_new():
""" """
@ -61,6 +63,7 @@ async def test_organization_new():
] ]
await asyncio.gather(*tasks) await asyncio.gather(*tasks)
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_organization_list(): async def test_organization_list():
""" """
@ -77,8 +80,8 @@ async def test_organization_list():
] ]
await asyncio.gather(*tasks) await asyncio.gather(*tasks)
response_json = await list_organization(session) response_json = await list_organization(session, i=0)
print(len(response_json)) print(len(response_json))
if len(response_json) == 0: if len(response_json) == 0:
raise Exception(f"Return empty list of organization") raise Exception("Return empty list of organization")