Fix import error

This commit is contained in:
Sameer Kankute 2026-04-16 19:20:41 +05:30
parent b3fdb5cc69
commit a86938a4af
No known key found for this signature in database

View file

@ -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,