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",
"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,

9
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()
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")