uPjA UdZddlZddlZddlmZmZddlZddlmZddl m Z m Z m Z m Z ddlmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZmZmZm Z m!Z!m"Z"d d d d d Z#ee$e%efe&d<de%de%e'zdefdZ(de%de'defdZ)de%de%de%defdZ*de'ede'efdZ+de'e$dede'efdZ,de%dzde%dzfdZ-de.de%d e%de%fd!Z/de'ede'efd"Z0d#e$d$e%d%ed&ejbdef d'Z2d(e$de fd)Z3y)*u OCI Generative AI — Generic-format chat transformation helpers. Handles message building, tool definition adaptation, non-streaming response parsing, and streaming chunk parsing for models served with ``apiFormat="GENERIC"`` (e.g. Meta Llama, xAI Grok, Google Gemini). N)AnyFinal)ValidationError)OCIErrorresolve_oci_schema_anyofresolve_oci_schema_refssanitize_oci_schema) OCICompletionResponseOCIContentPartUnionOCIImageContentPart OCIImageUrl OCIMessageOCIRolesOCIStreamChunkOCITextContentPart OCIToolCallOCIToolDefinition OCIVendors)AllMessageValues)ChatCompletionMessageToolCallDelta ModelResponseModelResponseStreamStreamingChoicesUsageSYSTEMUSER ASSISTANTTOOL)systemuser assistanttoolopen_ai_to_generic_oci_role_maprolecontentreturncg}t|tr!tt|t |gddS|D],}t|t s t dd|jd}t|ts t dd|d vrt dd |d |d k(rJ|jd }t|ts t dd |jt ||dk(s|jd}t|t r|jd}t|ts t dd|jtt|/tt||ddS)z@Convert a plain-text or multipart content message to OCI format.textNr%r& toolCalls toolCallIdz&Each content item must be a dictionary status_codemessagetypez1Each content item must have a string `type` field)r* image_urlzContent type `z` is not supported by OCIr*z;Content item of type `text` must have a string `text` fieldr3urlzDProp `image_url` must be a string or an object with a `url` property)r4)imageUrl) isinstancestrrr$rdictrgetappendr r )r%r& new_content content_item item_typer*r3s b/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/oci/chat/generic.py6adapt_messages_to_generic_oci_standard_content_messager?;s46K'306'W56    "Y ,-s4\] ] $$V, )S)K  1 1( 3LM   ##F+DdC( #Y   1t< = + %$((5I)T*%MM%0 i- #b   2KI0adapt_messages_to_generic_oci_standard_tool_callrMrs;"$  )T*s4YZ Z == J .s4[\ \ }}T* ,,s4UV V! j1 -.s4_` `%))&1 --s4`a aj)--k4@ )S)N  ## "#   3  D  ,T 2&  r@rJcDtt|t|gd|S)z,Convert a tool-result message to OCI format.r)Nr+)rr$r)r%rJr&s r>4adapt_messages_to_generic_oci_standard_tool_responserOs(  ,T 2#12  r@messagescNg}|D]}|d}|jd}|jd}|jd}|dk(r;|9t|ts tdd|j t ||||d vrA|?t|t tfs tdd |j t|||d k(st|t s tdd t|t s tdd |j t||||S)z=Convert an OpenAI-format message array to OCI GENERIC format.r%r&rArJr"r.z#Message `tool_calls` must be a listr/)r r!r"z;Message `content` must be a string or list of content partsr#z5Tool result message must have a string `tool_call_id`z.Tool result message `content` must be a string) r9r6listrr:rMr7r?rO)rP new_messagesr1r%r&rArJs r>&adapt_messages_to_generic_oci_standardrTs<Lsv++i([[. {{>2 ; :#9j$/38]^^    PQUWa b c 4 49LgT{3 #Y    VW[]d e f V^lC0 #Sgs+ #L    TUY[gip q r=s@ r@toolsvendorc vg}|D]}|ddk7r tdd|jd}t|ts tdd|jdi}t t t |}|jtd|jd |jd d | |S) zConvert OpenAI-format tool definitions to OCI GENERIC format. Resolves ``$ref``/``$defs`` and ``anyOf`` that the OCI endpoint rejects. r2rCr.z OCI only supports function toolsr/z$Tool `function` must be a dictionary parametersrGrE description)r2rErYrX) rr9r6r8r rrr:r)rUrV new_toolsr#rK raw_paramsresolved_paramss r>%adapt_tool_definition_to_oci_standardr^s I  <: %s4VW W, -.s4Z[ ["&&|R8 -.FG^_iGj.kl "&&v.)--mR@*    ( r@rawc,|y|dk(ry|dk(ry|dvryy)uMap an OCI-specific finish reason to its OpenAI-standard equivalent. OCI emits ``COMPLETE`` / ``MAX_TOKENS`` / ``TOOL_CALL(S)`` plus a long tail of error/cancel reasons (``ERROR``, ``ERROR_TOXIC``, ``ERROR_LIMIT``, ``USER_CANCEL``, ``CONTENT_FILTERED``, ``CANCELLED``, ...). The OpenAI spec only defines ``stop`` / ``length`` / ``tool_calls`` / ... — anything else is collapsed to ``"stop"`` so downstream consumers switching on ``finish_reason`` keep working. A ``None`` input passes through unchanged. NCOMPLETEstop MAX_TOKENSlength) TOOL_CALL TOOL_CALLSrA)r_s r>_normalize_oci_finish_reasonrhs1 { j l )) r@positionrErFctj|d|d|jdjdd}d|S)uKDeterministic synthetic tool-call id derived from chunk content. Used as a fallback when OCI omits ``id`` (always the case for the OCI Cohere protocol, occasionally the case for OCI GENERIC streaming chunks). A random ``uuid4`` per chunk would cause downstream stream-merging consumers — which key off the tool-call ``id`` — to treat re-emissions of the same logical call (e.g. terminal consolidation chunks, retries) as distinct calls. A content-derived digest stays stable across identical re-emissions while differing across truly distinct calls. |F)usedforsecurityNcall_)hashlibsha256encode hexdigest)rirErFdigests r>_synthesize_oci_tool_call_idrtsSNN *AdV1YK(//1ik#2F6( r@c t|Dcgc]W\}}t|jxs!t||j|j d|j|j dYc}}Scc}}w)zAConvert OCI tool-call objects in a response to the OpenAI format.rCrErFrDr2rC) enumeraterrDrtrErF)rUir#s r>adapt_tools_to_openai_standardrzse!'   At &wwT6q$))T^^T"iidnnE   sAA/ json_datamodelmodel_response raw_responsec td i|}|j j }tjj|jdd}t|j|_ |j|_|j jstd|j|j jd}|jdj } |j } | r| j"r@d} | j"D]&} t%| t&s| xsd| j(z} (| | | _| j*rt-| j*| _t1|j2|jd_|j j6} d}| j8r,| j8j:| j8j:}t=| j>| j@xsd| jB| |_|S#ttf$r}td||jd}~wwxYw) zHParse a non-streaming GENERIC OCI response into a LiteLLM ModelResponse.z4Response cannot be casted to OCICompletionResponse: )r1r0NZz+00:00z!OCI response contained no choicesrrZ) prompt_tokenscompletion_tokens total_tokensreasoning_tokensrg)"r TypeErrorrrr0 chatResponse timeCreateddatetime fromisoformatreplaceint timestampcreatedmodelIdr|choicesr1r&r6rr*r,rzrArh finishReason finish_reasonusagecompletionTokensDetailsreasoningTokensr promptTokenscompletionTokens totalTokens)r{r|r}r~completion_responseeiso_strdtresponse_choicer1response_messager*item oci_usagers r>handle_generic_responser-s %:%GY%G)55AAG!!//X0NOB 0N.66N  + + 3 37$00  1==EEaHO#++A.66G-55#  # # $D(00 4d$67 JB$))3D 4"&  % %!?@P@Z@Z![G .J?KgKg.hN1+*77==I#'((Y-N-N-^-^-j$<<LL ,,#449**) N _  ' J1#N$00   s HH=H88H= dict_chunkc b|jdrW|djdrC|ddD]8}|jdd|jdd|jdd: tdi|}|j d |_d }|jr|jjru|jjD]\}t|tr|xsd|jz}'t|tr t dd t dd |j d }|jr|jjryt|jjDcgc]O\}}|j xs!t#||j$|j&d|j$|j&ddQ}}}t)|j*} t-t/|j t1||d d d | gS#ttf$r}t dd| d }~wwxYwcc}}w)zDParse a single GENERIC SSE chunk into a LiteLLM ModelResponseStream.r1r,rFrZrDrEiz*Chunk cannot be parsed as OCIStreamChunk: r/NruDOCI returned image content in a streaming response — not supportedz4Unsupported content type in OCI streaming response: rCrvrw)r&rAprovider_specific_fieldsthinking_blocksreasoning_content)indexdeltar)rrg)r9 setdefaultrrrrrr1r&r6rr*r r2r,rxrDrtrErFrhrrrr) rrI typed_chunkrr*rrArytcrs r>handle_generic_stream_chunkrhs>~~i Z %:%>%>{%K#I.{; -I  b 1  r *   , -  +9j9    D{22::''// D$ 23 dii/D"56 #b  #RSWS\S\R]^ */3J{22<<#;#6#6#@#@A  2eeU;Arww U"GG!#    (DKD\D\']M  !'' )-1$(&* ,    i  ' @D   L  s* H'AH+H(H##H()4__doc__rrotypingrrhttpxpydanticrlitellm.llms.oci.common_utilsrrrr litellm.types.llms.ocir r r r rrrrrrrlitellm.types.llms.openairlitellm.types.utilsrrrrrrr$r8r7__annotations__rRr?rMrOrTr^rhrrtrzResponserrrgr@r>rs $    7   ?tCM':!;44sUYz4^h4n*3*D*U_*ZsRU`chr%#$% *%ZdZTXYjTk:cDjS4Z*3ccc$     '( &88 8"8.. 8  8vMDM5HMr@