phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +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,
)
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()