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

Merge a7920706cd into b82af5b826

Browse source
This commit is contained in:
ranadkar 2025-04-24 01:00:38 -07:00 • committed by GitHub
commit f10ce65f0f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

8
enterprise/enterprise_hooks/aporia_ai.py
View file

@ -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=<your key>
export APORIA_API_KEY=<your 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": [

8
litellm/proxy/guardrails/guardrail_hooks/aporia_ai.py
View file

@ -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=<your key>
export APORIA_API_KEY=<your 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": [