diff --git a/tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py b/tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py index f1835719b79..834cb235f0c 100644 --- a/tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py +++ b/tests/enterprise/litellm_enterprise/enterprise_callbacks/test_prometheus_logging_callbacks.py @@ -223,6 +223,8 @@ def test_increment_token_metrics(prometheus_logger): api_key_alias="test_alias", team="test_team", team_alias="test_team_alias", + org_id=None, + org_alias=None, requested_model=None, model="gpt-3.5-turbo", model_id="model-123", @@ -237,6 +239,8 @@ def test_increment_token_metrics(prometheus_logger): api_key_alias="test_alias", team="test_team", team_alias="test_team_alias", + org_id=None, + org_alias=None, requested_model=None, model="gpt-3.5-turbo", model_id="model-123", @@ -253,6 +257,8 @@ def test_increment_token_metrics(prometheus_logger): api_key_alias="test_alias", team="test_team", team_alias="test_team_alias", + org_id=None, + org_alias=None, requested_model=None, model="gpt-3.5-turbo", model_id="model-123", @@ -414,6 +420,8 @@ def test_set_latency_metrics(prometheus_logger): api_key_alias="test_alias", team="test_team", team_alias="test_team_alias", + org_id=None, + org_alias=None, requested_model="openai-gpt", model="gpt-3.5-turbo", model_id="model-123", @@ -430,6 +438,8 @@ def test_set_latency_metrics(prometheus_logger): api_key_alias="test_alias", team="test_team", team_alias="test_team_alias", + org_id=None, + org_alias=None, requested_model="openai-gpt", model="gpt-3.5-turbo", model_id="model-123", @@ -446,6 +456,8 @@ def test_set_latency_metrics(prometheus_logger): api_key_alias="test_alias", team="test_team", team_alias="test_team_alias", + org_id=None, + org_alias=None, requested_model="openai-gpt", model="gpt-3.5-turbo", model_id="model-123", @@ -589,6 +601,8 @@ def test_increment_top_level_request_and_spend_metrics(prometheus_logger): api_key_alias="test_alias", team="test_team", team_alias="test_team_alias", + org_id=None, + org_alias=None, model="gpt-3.5-turbo", model_id="model-123", client_ip=None, @@ -605,6 +619,8 @@ def test_increment_top_level_request_and_spend_metrics(prometheus_logger): api_key_alias="test_alias", team="test_team", team_alias="test_team_alias", + org_id=None, + org_alias=None, model="gpt-3.5-turbo", model_id="model-123", client_ip=None, @@ -959,6 +975,8 @@ def test_set_llm_deployment_success_metrics(prometheus_logger): api_key_alias=standard_logging_payload["metadata"]["user_api_key_alias"], team=standard_logging_payload["metadata"]["user_api_key_team_id"], team_alias=standard_logging_payload["metadata"]["user_api_key_team_alias"], + org_id=None, + org_alias=None, ) prometheus_logger.litellm_overhead_latency_metric.labels.assert_called_once_with( api_base="https://api.openai.com",