From 7fedcf1ea96fc7852c4c43487c38427b0b6ea836 Mon Sep 17 00:00:00 2001 From: drorbaron Date: Tue, 19 Aug 2025 12:36:29 +0300 Subject: [PATCH] migrate stream ws --- litellm/proxy/guardrails/guardrail_hooks/aim/aim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/aim/aim.py b/litellm/proxy/guardrails/guardrail_hooks/aim/aim.py index e40bafcc463..3c03d70cef8 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/aim/aim.py +++ b/litellm/proxy/guardrails/guardrail_hooks/aim/aim.py @@ -300,7 +300,7 @@ class AimGuardrail(CustomGuardrail): ) call_id = request_data.get("litellm_call_id") async with connect( - f"{self.ws_api_base}/detect/output/ws", + f"{self.ws_api_base}/fw/v1/analyze/stream", additional_headers=self._build_aim_headers( hook="output", key_alias=user_api_key_dict.key_alias,