uPj,UdZddlmZmZmZddlZddlZddlmZddl m Z ddl m Z m Z mZmZmZmZddlmZmZmZddlmZmZdd lmZmZer dd lmZeZneZd Z ee!d <d ddddZ"ee!d<Gdde Z#e#Z$ee!d<y)uI OCI Generative AI — Embedding transformation. Endpoint: POST /20231130/actions/embedText Supported models: cohere.embed-english-v3.0, cohere.embed-multilingual-v3.0, cohere.embed-v4.0, and all other Cohere embed variants available on OCI (including dedicated endpoints). Authentication follows the same RSA-SHA256 / OCI SDK signer pattern as chat. The base handler (base_llm_http_handler.embedding) calls sign_request after building the body, so signing happens automatically. Supported models: - cohere.embed-english-v3.0 - cohere.embed-english-light-v3.0 - cohere.embed-multilingual-v3.0 - cohere.embed-multilingual-light-v3.0 - cohere.embed-english-image-v3.0 - cohere.embed-multilingual-image-v3.0 - cohere.embed-v4.0 Reference: https://docs.oracle.com/en-us/iaas/api/#/en/generative-ai-inference/latest/EmbedTextResult/EmbedText ) TYPE_CHECKINGAnyFinalN)BaseLLMException)BaseEmbeddingConfig)OCI_API_VERSIONOCIErrorget_oci_base_urlresolve_oci_credentialssign_oci_requestvalidate_oci_environment)OCIEmbedRequestOCIEmbedResponseOCIServingMode)AllEmbeddingInputValuesAllMessageValues)EmbeddingResponseUsage)Logging`OCI_EMBED_BATCH_LIMITSEARCH_DOCUMENT SEARCH_QUERYCLASSIFICATION CLUSTERING)search_document search_queryclassification clustering_INPUT_TYPE_MAPceZdZdZdedeefdZ ddededededef dZ dd eded ee ded ed ed zded zdefdZ d ded zd ed zdeded eded zdefdZ d!d edededed ed zded zded zded zde eeffdZdededed edef dZdedej&deded ed zdeded edefdZdeded eej0zdefdZy )"OCIEmbedConfigu Transformation config for OCI Generative AI embeddings. Supports both text and (on cohere.embed-v4.0) multimodal inputs. Authentication — same two modes as chat: - **OCI SDK signer**: pass ``oci_signer`` in optional_params. - **Manual RSA-SHA256**: pass ``oci_user``, ``oci_fingerprint``, ``oci_tenancy``, and ``oci_key`` or ``oci_key_file``, or set the corresponding ``OCI_*`` env vars. Required call-time params (via optional_params or env vars): - ``oci_compartment_id`` / ``OCI_COMPARTMENT_ID`` - ``oci_region`` / ``OCI_REGION`` (default: ``us-ashburn-1``) Optional call-time params: - ``oci_serving_mode``: ``"ON_DEMAND"`` (default) or ``"DEDICATED"`` - ``oci_endpoint_id``: endpoint OCID for dedicated serving mode - ``input_type``: ``SEARCH_DOCUMENT``, ``SEARCH_QUERY``, ``CLASSIFICATION``, ``CLUSTERING`` - ``truncate``: ``NONE``, ``START``, or ``END`` (default ``END``) - ``dimensions``: output embedding dimensions (cohere.embed-v4.0+) modelreturncdgS)N dimensions)selfr#s j/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/oci/embed/transformation.pyget_supported_openai_paramsz*OCIEmbedConfig.get_supported_openai_paramsZs ~non_default_paramsoptional_params drop_paramscL|jD]\}}|dk(s ||d<|S)Nr&outputDimensions)items)r(r,r-r#r.keyvalues r)map_openai_paramsz OCIEmbedConfig.map_openai_params]s;-224  DEDICATEDrWz4oci_serving_mode must be 'ON_DEMAND' or 'DEDICATED'.rXoci_endpoint_id) servingType endpointId)rZmodelId input_typetruncateENDr0) compartmentId servingModeinputs inputTyper^r0T) exclude_none)r r isinstancestrlistappendlenrrEupperrr lowerr model_dump)r(r#rTr-r5rFcompartment_idtextsitemserving_mode_type endpoint_id serving_moder]requests r)transform_embedding_requestz*OCIEmbedConfig.transform_embedding_requests/? %&: ;X  eS !GE t $E KdD)"$'Y Zc%:TD J KZLE u:- -45J4KLJ<'>@ $3#6#67I;#W#]#]#_ $> >N   +!0!4!45F!NK)kkZL)k5QL%((6 (,,Z-=-=-?AQAQASTJ(($ $((U;,001CD  !!t!44r+ raw_responsemodel_response logging_objc |jdk7r!t|j|j |j} t d i| } | j |_t| jD cgc] \} } d| | d c} } |_ | j)t| j}t|||_|S| jrs0-, FN P8X* "u!)"& wH(wHv+E*r+