fix: fix linting errors

This commit is contained in:
Krrish Dholakia 2025-03-13 19:40:18 -07:00
parent 997f2f0b3e
commit 58a7351a73

View file

@ -121,7 +121,7 @@ class AimGuardrail(CustomGuardrail):
async def call_aim_guardrail_on_output( async def call_aim_guardrail_on_output(
self, request_data: dict, output: str, hook: str self, request_data: dict, output: str, hook: str
) -> None: ) -> Optional[str]:
user_email = ( user_email = (
request_data.get("metadata", {}).get("headers", {}).get("x-aim-user-email") request_data.get("metadata", {}).get("headers", {}).get("x-aim-user-email")
) )