diff --git a/enterprise/enterprise_hooks/aporia_ai.py b/enterprise/enterprise_hooks/aporia_ai.py index 2b427bea5c..373c75925a 100644 --- a/enterprise/enterprise_hooks/aporia_ai.py +++ b/enterprise/enterprise_hooks/aporia_ai.py @@ -42,8 +42,8 @@ class AporiaGuardrail(CustomGuardrail): self.async_handler = get_async_httpx_client( llm_provider=httpxSpecialProvider.GuardrailCallback ) - self.aporia_api_key = api_key or os.environ["APORIO_API_KEY"] - self.aporia_api_base = api_base or os.environ["APORIO_API_BASE"] + self.aporia_api_key = api_key or os.environ["APORIA_API_KEY"] + self.aporia_api_base = api_base or os.environ["APORIA_API_BASE"] super().__init__(**kwargs) #### CALL HOOKS - proxy only #### @@ -94,10 +94,10 @@ class AporiaGuardrail(CustomGuardrail): _json_data = json.dumps(data) """ - export APORIO_API_KEY= + export APORIA_API_KEY= curl https://gr-prd-trial.aporia.com/some-id \ -X POST \ - -H "X-APORIA-API-KEY: $APORIO_API_KEY" \ + -H "X-APORIA-API-KEY: $APORIA_API_KEY" \ -H "Content-Type: application/json" \ -d '{ "messages": [ diff --git a/litellm/proxy/guardrails/guardrail_hooks/aporia_ai.py b/litellm/proxy/guardrails/guardrail_hooks/aporia_ai.py index 3c39b90b0a..9108fcf8ef 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/aporia_ai.py +++ b/litellm/proxy/guardrails/guardrail_hooks/aporia_ai.py @@ -46,8 +46,8 @@ class AporiaGuardrail(CustomGuardrail): self.async_handler = get_async_httpx_client( llm_provider=httpxSpecialProvider.GuardrailCallback ) - self.aporia_api_key = api_key or os.environ["APORIO_API_KEY"] - self.aporia_api_base = api_base or os.environ["APORIO_API_BASE"] + self.aporia_api_key = api_key or os.environ["APORIA_API_KEY"] + self.aporia_api_base = api_base or os.environ["APORIA_API_BASE"] super().__init__(**kwargs) #### CALL HOOKS - proxy only #### @@ -105,10 +105,10 @@ class AporiaGuardrail(CustomGuardrail): _json_data = json.dumps(data) """ - export APORIO_API_KEY= + export APORIA_API_KEY= curl https://gr-prd-trial.aporia.com/some-id \ -X POST \ - -H "X-APORIA-API-KEY: $APORIO_API_KEY" \ + -H "X-APORIA-API-KEY: $APORIA_API_KEY" \ -H "Content-Type: application/json" \ -d '{ "messages": [