tPjY UdZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZerddlmZd Zeed <Gd d e Zy) u@ Qostodian Nexus (by Qohash) — LiteLLM guardrail integration. N) TYPE_CHECKINGFinalLiteralOptional)log_guardrail_information)GenericGuardrailAPIQostodianNexusConfigModel)GenericGuardrailAPIInputs)Loggingqostodian_nexusGUARDRAIL_NAMEc eZdZ d dedzffd Ze d dedededde d d ef fd Z e d e e dzfd ZxZS)QostodianNexusNapi_basec |xs tjjdd}|jdt|d<gd}|jdxsg}||Dcgc] }||vs| c}z|d<t |dd|i|ycc}w)NQOSTODIAN_NEXUS_API_BASEzhttp://nexus:8800guardrail_name)z#x-qostodian-nexus-identifiers-tracez$x-qostodian-nexus-identifiers-sourcez'x-qostodian-nexus-identifiers-containerz&x-qostodian-nexus-identifiers-identity extra_headersr)osenvirongetrsuper__init__)selfrkwargs nexus_headersexistingh __class__s {/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/guardrails/guardrail_hooks/qohash/qohash.pyrzQostodianNexus.__init__s ^rzz~~.HJ]^#)::.>#O  !**_5;"/h2a!S`J`12a"a    3bs  B'Binputs request_data input_type)requestresponse logging_objLiteLLMLoggingObjreturncFKt|||||d{S7w)ak Apply Qostodian Nexus to the given inputs. NOTE: This override is intentionally a pass-through. It must be present directly in this class's __dict__ so that LiteLLM's unified guardrail routing check (`"apply_guardrail" in type(callback).__dict__` in litellm/proxy/utils.py) routes calls correctly. Do not remove. )r#r$r%r(N)rapply_guardrail)rr#r$r%r(r!s r"r,zQostodianNexus.apply_guardrail1s6 W,%!# -    s !!ctS)z? Returns the config model for Qostodian Nexus. r )clss r"get_config_modelzQostodianNexus.get_config_modelHs )()N)__name__ __module__ __qualname__strrrr dictrrr, classmethodtyper r/ __classcell__)r!s@r"rrs $ * 2 6:  )  12  12  #   ,)&?!@4!G))r0r)__doc__rtypingrrrr%litellm.integrations.custom_guardrailrTlitellm.proxy.guardrails.guardrail_hooks.generic_guardrail_api.generic_guardrail_apir5litellm.types.proxy.guardrails.guardrail_hooks.qohashr litellm.types.utilsr *litellm.litellm_core_utils.litellm_loggingr r)r__annotations__rrr0r"rAsD ::K:W))6)(6)r0