phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

CR

Browse source
This commit is contained in:
Tomer Bin 2025-02-11 17:41:57 +02:00
parent 4a31b32a88
commit 32dedbe551
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/guardrails/guardrail_hooks/aim.py
View file

@ -30,6 +30,7 @@ from litellm.types.utils import (
ModelResponseStream, ModelResponseStream,
) )
class AimGuardrailMissingSecrets(Exception): class AimGuardrailMissingSecrets(Exception):
pass pass
@ -120,9 +121,9 @@ class AimGuardrail(CustomGuardrail):
{"x-aim-user-email": user_email} if user_email else {} {"x-aim-user-email": user_email} if user_email else {}
) )
response = await self.async_handler.post( response = await self.async_handler.post(
f"{self.api_base}/detect", f"{self.api_base}/detect/output",
headers=headers, headers=headers,
json={"user_prompt": output}, json={"output": output, "messages": request_data.get("messages", [])},
) )
response.raise_for_status() response.raise_for_status()
res = response.json() res = response.json()