Fix tests

Author: Sameerlite

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