uPj~dZddlmZmZmZddlmZddlmZddl m Z erddl m Z ddl mZddlmZGd d eZy ) z Cohere Rerank Handler for Unified Guardrails This module provides guardrail translation support for the rerank endpoint. The handler processes only the 'query' parameter for guardrails. ) TYPE_CHECKINGAnyFinal)verbose_proxy_logger)BaseTranslation)GenericGuardrailAPIInputs)CustomGuardrail)Logging)RerankResponsec deZdZdZdZ ddeddddd efd Z dd d ddddd edzdedzd ef dZy)CohereRerankHandlera Handler for processing rerank requests with guardrails. This class provides methods to: 1. Process input query (pre-call hook) 2. Process output response (post-call hook) - not applicable for rerank The handler specifically processes: - The 'query' parameter (string) - The 'instruction' parameter (string), when present Note: Documents are not processed by guardrails as they are the corpus being searched, not user input. )query instructionNdataguardrail_to_applyr litellm_logging_objzLiteLLMLoggingObj | Nonereturnc VK|jDcgc])}t|j|ts#|||f+}}|st j d|St |Dcgc]\}}| c}}}|jd} | r| |d<|j||d|d{} | jdg} t|D]L\} \}} | t| ks| | ||<t j d|t| t||N|Scc}wcc}}w7}w) a Process input text fields ('query' and 'instruction') by applying guardrails and writing the sanitized values back. Args: data: Request data dictionary containing 'query' and optionally 'instruction' guardrail_to_apply: The guardrail instance to apply Returns: Modified data with guardrails applied to query/instruction only z6Rerank: No query/instruction to process or not strings)textsmodelrequest)inputs request_data input_type logging_objNrzDRerank: Applied guardrail to %s. Original length: %d, New length: %d) _SCANNED_FIELDS isinstancegetstrrdebugrapply_guardrail enumeratelen)selfrrrkeyfields_to_scan_valuerrguardrailed_inputsguardrailed_textsidxoriginals }/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/cohere/rerank/guardrail_translation/handler.pyprocess_input_messagesz*CohereRerankHandler.process_input_messages*sC(>B=Q=Q tcU_`d`h`hil`morUs#tCy!1 t t & &'_ `K1~8^81e8^_xx( #F7O*<*L*L + +M+ %  $6#9#9'2#F$-n$=  C#xS*++-c2S $**ZMS N   =!u 9_ % s3D)$D D%D)$ D! 09D))D'*8D)#AD)responser user_api_key_dictrc8Ktjd|Sw)aY Process output response - not applicable for rerank. Rerank responses contain relevance scores and indices, not text, so there's nothing to apply guardrails to. This method returns the response unchanged. Args: response: Rerank response object with rankings guardrail_to_apply: The guardrail instance (unused) litellm_logging_obj: Optional logging object (unused) user_api_key_dict: User API key metadata (unused) Returns: Unmodified response (rankings don't need text guardrails) zURerank: Output processing not applicable (output contains relevance scores, not text))rr )r$r/rrr0rs r-process_output_responsez+CohereRerankHandler.process_output_response^s 0 "" c s)N)NNN) __name__ __module__ __qualname____doc__rdictrr.r2r-r r s &/O ;? 22.28 2 2p;?(,$( ".8  :  Tk  r9r N)r6typingrrrlitellm._loggingrrBs5-,1X9EW3e/er9