scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

Merge pull request #35845 from BerriAI/litellm_/cci-failing-tests-f73740

Browse source 
test: repair three failing suites on litellm_internal_staging
This commit is contained in:
yuneng-jiang 2026-08-04 16:42:58 -07:00 • committed by GitHub
commit 0e593645d1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 35 additions and 50 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py
View file

@ -1741,26 +1741,16 @@ async def test_initialize_remaining_budget_metrics_exception_handling(
# Verify all five errors were logged (teams, keys, users, orgs, and user/team count)
assert mock_logger.call_count == 5
assert (
"Error initializing teams budget metrics"
in mock_logger.call_args_list[0][0][0]
)
assert (
"Error initializing keys budget metrics"
in mock_logger.call_args_list[1][0][0]
)
assert (
"Error initializing users budget metrics"
in mock_logger.call_args_list[2][0][0]
)
assert (
"Error initializing orgs budget metrics"
in mock_logger.call_args_list[3][0][0]
)
assert (
"Error initializing user/team count metrics"
in mock_logger.call_args_list[4][0][0]
)
logged = [
call.args[0] % call.args[1:] for call in mock_logger.call_args_list
]
assert logged == [
"Error initializing teams budget metrics: Database error",
"Error initializing keys budget metrics: Key listing error",
"Error initializing users budget metrics: User database error",
"Error initializing orgs budget metrics: Org database error",
"Error initializing user/team count metrics: User count error",
]
# Verify the metrics were never called
prometheus_logger.litellm_remaining_team_budget_metric.assert_not_called()

30
tests/local_testing/test_completion.py
View file

@ -213,36 +213,6 @@ def test_completion_empower():
pytest.fail(f"Error occurred: {e}")
def test_completion_github_api():
litellm.set_verbose = True
messages = [
{
"role": "user",
"content": "\nWhat is the query for `console.log` => `console.error`\n",
},
{
"role": "assistant",
"content": "\nThis is the GritQL query for the given before/after examples:\n<gritql>\n`console.log` => `console.error`\n</gritql>\n",
},
{
"role": "user",
"content": "\nWhat is the query for `console.info` => `consdole.heaven`\n",
},
]
try:
# test without max tokens
response = completion(
model="github/gpt-4o",
messages=messages,
)
# Add any assertions, here to check response args
print(response)
except litellm.AuthenticationError:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_completion_claude_3_empty_response():
litellm.set_verbose = True

25
tests/proxy_behavior/management/test_team_metadata_schema.py Normal file
View file

@ -0,0 +1,25 @@
"""GET /team/metadata_schema — the behavior world declares no
``general_settings.team_metadata_schema``, so the route is an info route that
returns an empty field list to every authenticated actor and 401s without a key.
"""
import pytest
from .actors import Actor
pytestmark = pytest.mark.asyncio(loop_scope="session")
@pytest.mark.parametrize("actor", list(Actor), ids=[a.value for a in Actor])
async def test_team_metadata_schema_default_is_empty(actor: Actor, proxy_client, world):
resp = await proxy_client.get(
"/team/metadata_schema",
headers={"Authorization": f"Bearer {world.keys[actor].cleartext}"},
)
assert resp.status_code == 200, f"{actor.value}: {resp.status_code} {resp.text}"
assert resp.json() == {"fields": []}
async def test_team_metadata_schema_requires_auth(proxy_client, world):
resp = await proxy_client.get("/team/metadata_schema")
assert resp.status_code == 401, resp.text