mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-12 23:01:41 +00:00
Fix import error
This commit is contained in:
parent
b3fdb5cc69
commit
a86938a4af
1 changed files with 2 additions and 0 deletions
|
|
@ -4,6 +4,7 @@ Qostodian Nexus (by Qohash) — LiteLLM guardrail integration.
|
|||
import os
|
||||
from typing import TYPE_CHECKING, Literal, Optional, Type
|
||||
|
||||
from litellm.integrations.custom_guardrail import log_guardrail_information
|
||||
from litellm.proxy.guardrails.guardrail_hooks.generic_guardrail_api.generic_guardrail_api import (
|
||||
GenericGuardrailAPI,
|
||||
)
|
||||
|
|
@ -44,6 +45,7 @@ class QostodianNexus(GenericGuardrailAPI):
|
|||
**kwargs,
|
||||
)
|
||||
|
||||
@log_guardrail_information
|
||||
async def apply_guardrail(
|
||||
self,
|
||||
inputs: GenericGuardrailAPIInputs,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue