uPj~dZddlmZmZmZddlmZddlmZddl m Z erddl m Z ddl mZddl mZGd d eZy ) z OpenAI Embeddings Handler for Unified Guardrails This module provides guardrail translation support for OpenAI's embeddings endpoint. The handler processes the 'input' parameter for guardrails. ) TYPE_CHECKINGAnyFinal)verbose_proxy_logger)BaseTranslation)GenericGuardrailAPIInputs)CustomGuardrail)Logging)EmbeddingResponsec eZdZdZ ddedddddefd Zded edddddef d Zded e ee ze e zdddddef d Z dd ddddddedzdedzdef dZ y)OpenAIEmbeddingsHandlera% Handler for processing OpenAI embeddings requests with guardrails. This class provides methods to: 1. Process input text (pre-call hook) 2. Process output response (post-call hook) - embeddings don't typically need output guardrails The handler specifically processes the 'input' parameter which can be: - A single string - A list of strings (for batch embeddings) - A list of integers (token IDs - not processed by guardrails) - A list of lists of integers (batch token IDs - not processed by guardrails) Ndataguardrail_to_applyr litellm_logging_objzLiteLLMLoggingObj | Nonereturnc`K|jd}|tjd|St|tr|j ||||d{}|St|t r|j||||d{}|Stjdt||S7U7)w)ah Process input text by applying guardrails to text content. Args: data: Request data dictionary containing 'input' parameter guardrail_to_apply: The guardrail instance to apply litellm_logging_obj: Optional logging object Returns: Modified data with guardrails applied to input inputNz1OpenAI Embeddings: No input found in request datazFOpenAI Embeddings: Unexpected input type: %s. Expected string or list.) getrdebug isinstancestr_process_string_inputlist_process_list_inputwarningtype)selfrrr input_datas /home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/openai/embeddings/guardrail_translation/handler.pyprocess_input_messagesz.OpenAIEmbeddingsHandler.process_input_messages#s"!HHW-   & &'Z [K j# &33D*FXZmnnD  D )11$ DVXkllD ! ( (XZ    ols$AB.B*-B.B,(B.,B.rc$Kt|g}|jdx}r||d<|j||d|d{}|jdx}r4|d|d<tjd t |t |d|S7Mw) z4Process a single string input through the guardrail.textsmodelrequestinputs request_data input_type logging_objNr#rrzYOpenAI Embeddings: Applied guardrail to string input. Original length: %d, New length: %d)rrapply_guardrailrrlen) rrrrrr'r$guardrailed_inputsguardrailed_textss rrz-OpenAIEmbeddingsHandler._process_string_inputEs2 EHHW% %5 %#F7O*<*L*L + +M+ % !3 6 6w ? ?  ?-a0DM & &kJDM"   % s>BBABcKt|dk(r|S|d}t|ttfrt j d|St|t s!t jdt||St|}|jdx}r||d<|j||d|d{}|jd x} r$| |d <t j d t| |S7=w) zDProcess a list input through the guardrail (if it contains strings).rzDOpenAI Embeddings: Input is token IDs, skipping guardrail processingz6OpenAI Embeddings: Unexpected input list item type: %sr"r$r%r&Nr#rz1OpenAI Embeddings: Applied guardrail to %d inputs) r,rintrrrrrrrrr+) rrrrr first_itemr'r$r-r.s rrz+OpenAIEmbeddingsHandler._process_list_inputbs  z?a K&qM  j3+ . & &'m nK*c* ( (HZ  K2 CHHW% %5 %#F7O*<*L*L + +M+ % !3 6 6w ? ?  ?-DM & &C%&   % sB0C22C03>C2responser user_api_key_dictr(c8Ktjd|Sw)ap Process output response - embeddings responses contain vectors, not text. For embeddings, the output is numerical vectors, so there's typically no text content to apply guardrails to. This method is a no-op but is included for interface consistency. Args: response: Embedding response object guardrail_to_apply: The guardrail instance to apply litellm_logging_obj: Optional logging object user_api_key_dict: User API key metadata Returns: Unmodified response (embeddings don't have text output to guard) z\OpenAI Embeddings: Output response processing skipped - embeddings contain vectors, not text)rr)rr2rrr3r(s rprocess_output_responsez/OpenAIEmbeddingsHandler.process_output_responses 0 "" j s)N)NNN) __name__ __module__ __qualname____doc__dictrr rrrr0rr5rr r s $;?  . 8  D.  8   :,,sT#Y./,. , 8 ,  ,d;?(,$( %.8  :  Tk  r<r N)r9typingrrrlitellm._loggingrrDs5-,1X9EW5WoWr<