Skip to content

Commit 1543891

Browse files
Responses API - support tags in metadata
* fix(ui/): fix routing for custom server root path * fix: fix eslint errors * fix(vector_store_pre_call_hook.py): Fix #15724 * fix(responses/main.py): have 'tags' work across metadata + litellm_metadata * fix: add unit testing
1 parent f8d6a6e commit 1543891

File tree

7 files changed

+309
-169
lines changed

7 files changed

+309
-169
lines changed

enterprise/litellm_enterprise/integrations/custom_guardrail.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,11 +29,10 @@ def _should_run_if_mode_by_tag(
2929
if event_hook is None or not isinstance(event_hook, Mode):
3030
return None
3131

32-
metadata: dict = data.get("litellm_metadata") or data.get("metadata", {})
3332
proxy_server_request = data.get("proxy_server_request", {})
3433

3534
request_tags = StandardLoggingPayloadSetup._get_request_tags(
36-
metadata=metadata,
35+
litellm_params=data,
3736
proxy_server_request=proxy_server_request,
3837
)
3938

enterprise/litellm_enterprise/integrations/prometheus.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1226,8 +1226,8 @@ async def async_post_call_failure_hook(
12261226

12271227
try:
12281228
_tags = StandardLoggingPayloadSetup._get_request_tags(
1229-
request_data.get("metadata", {}),
1230-
request_data.get("proxy_server_request", {}),
1229+
litellm_params=request_data,
1230+
proxy_server_request=request_data.get("proxy_server_request", {}),
12311231
)
12321232
enum_values = UserAPIKeyLabelValues(
12331233
end_user=user_api_key_dict.end_user_id,
@@ -1289,7 +1289,8 @@ async def async_post_call_success_hook(
12891289
status_code="200",
12901290
route=user_api_key_dict.request_route,
12911291
tags=StandardLoggingPayloadSetup._get_request_tags(
1292-
data.get("metadata", {}), data.get("proxy_server_request", {})
1292+
litellm_params=data,
1293+
proxy_server_request=data.get("proxy_server_request", {}),
12931294
),
12941295
)
12951296
_labels = prometheus_label_factory(
@@ -2212,8 +2213,9 @@ def _mount_metrics_endpoint(premium_user: bool):
22122213
)
22132214

22142215
# Create metrics ASGI app
2215-
if 'PROMETHEUS_MULTIPROC_DIR' in os.environ:
2216+
if "PROMETHEUS_MULTIPROC_DIR" in os.environ:
22162217
from prometheus_client import CollectorRegistry, multiprocess
2218+
22172219
registry = CollectorRegistry()
22182220
multiprocess.MultiProcessCollector(registry)
22192221
metrics_app = make_asgi_app(registry)

0 commit comments

Comments
 (0)