From 32dedbe55139edd5e4a3565335e66b98d163d5c9 Mon Sep 17 00:00:00 2001 From: Tomer Bin Date: Tue, 11 Feb 2025 17:41:57 +0200 Subject: [PATCH] CR --- litellm/proxy/guardrails/guardrail_hooks/aim.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/aim.py b/litellm/proxy/guardrails/guardrail_hooks/aim.py index 033b4ca4fb3..920d9161497 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/aim.py +++ b/litellm/proxy/guardrails/guardrail_hooks/aim.py @@ -30,6 +30,7 @@ from litellm.types.utils import ( ModelResponseStream, ) + class AimGuardrailMissingSecrets(Exception): pass @@ -120,9 +121,9 @@ class AimGuardrail(CustomGuardrail): {"x-aim-user-email": user_email} if user_email else {} ) response = await self.async_handler.post( - f"{self.api_base}/detect", + f"{self.api_base}/detect/output", headers=headers, - json={"user_prompt": output}, + json={"output": output, "messages": request_data.get("messages", [])}, ) response.raise_for_status() res = response.json()