uPju< UdZddlZddlZddlmZmZmZddlmZm Z ddl Z ddl m Z m Z mZddlmZmZddlmZmZmZmZmZmZddlmZmZmZmZmZmZm Z m!Z!dd l"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+d e d e,e-e ffd Z.d e d e/e fdZ0d e d e-fdZ1dee-e2fd e-fdZ3de-eee-e2fzdzd e-fdZ4ded e-fdZ5e e,e-e2fZ6e e7d<de-e,e-e2fzd e,e-e2ffdZ8de$d efdZ9de/e#d e/efdZ:de,e-e fd e fdZ;de,e-e fded#ee-e fd efd$Z?d%eee-e fd e/efd&Z@d'ee-e fd(e-d)e(d*e jd e(f d+ZB d0d,ee-e fd-esC eT "eii&76&A))FB'DdC(40b0r-contentc|yt|tr|St|trdjd|DSt|S)Nr2c32K|]}t|ywr&)r:).0r5s r* z _content_text..KsGe*51Gs)r'r3r/joinr;s r* _content_textrBEsB'3'4 wwGwGGG w<r-ct|S)z@Return the plain-text representation of a message content value.)rBrAs r*_extract_text_contentrDOs  !!r-_TOOL_ARGUMENTS_ADAPTER raw_argumentsctt|ts|S tj|S#t$ricYSwxYwr&)r'r3rE validate_jsonr )rFs r*_parsed_tool_argumentsrIWs< mS )&44]CC  s ) 77 tool_callc |jdi}tt|jddt|jddS)Nfunctionnamer2 argumentsz{}rM parameters)r9rr3rI)rJfunction_fieldss r*_to_cohere_tool_callrR`sI&]]:r:O  $$VR0 1)/*=*=k4*PQ r-messagesc Dcic]D}|jddk(r.d|vr*|dxsgD]}|jddt| F}}}tfdtt dz ddDd }|n#t Dcgc] \}}||k7s |c}}}g}|D]B}|jd} t |jd } | dk(r%d|vr!|dr|dDcgc] }t|c}nd } | d k(r|jtd | }| dk(r|jtd| | | dk(st|jddxsd} |j| tdi} t| d| ig}|r3t|dtr |djj|'|jt|gE|Scc}}wcc}}wcc}w)uABuild a Cohere ``chatHistory`` list from an OpenAI-format message array. - All messages except the *last user message* are included. The caller pulls the last user message into the request's top-level ``message`` field, so trailing tool results (the standard agentic continuation pattern) still appear in ``chatHistory`` and reach the model. - If no user message exists, every message is included (no slice). - System messages must be filtered out by the caller (they are routed into ``preambleOverride`` separately) — they are not represented in ``chatHistory``. - Tool results are expressed as OCI ``CohereToolMessage.toolResults`` entries, with the originating call's name and parameters resolved from the preceding assistant message via a ``tool_call_id`` lookup. role assistant tool_callsidr2c3RK|]}|jddk(s| yw)rUuserN)r9)r>irSs r*r?z4adapt_messages_to_cohere_standard..s&Zq 8OSY8YZs''Nr;rZUSER)rUmessageCHATBOT)rUr_ toolCallstool tool_call_idrOoutput)calloutputs) toolResults)r9rRnextrangelen enumeraterDappendrr3rrr'rrg)rSmsgrJtool_call_lookuplast_user_indexr[mhistory_source chat_historyrUr;rWrc cohere_call tool_results` r*!adapt_messages_to_cohere_standardruhs)*  776?k )lc.A\*0b    dB!5i!@@"ZE#h-!+R4Z O $+ (@S1l1WX\kWk!1l02LRwwv' (:;{"|s':sO P !) , P  6>    67 K L [    9gYc d e V^sww~r:@bAL*..|^QS`b=cdK* "G,-K <+;=N OR ,,33K@##$5;-$PQ;R> _2m QsA G3 G9G9/G?raw_parametersc<ttt|Sr&)rrr)rvs r*_resolved_oci_parameter_schemarxs 78OP^8_` aar- param_schema is_requiredc t|jdxsd}ttt|jd|t j|||S)Nr7string description)r}r7 isRequired)r4r9rrr )ryrz json_types r*_cohere_parameter_definitionrsW !1!1&!9:FhI $3Il>N>N}>]4^`lm % ) ))Y ? r-resolved_schemac t|}t|jd}t|jdjDcic]\}}|t t|||vc}}Scc}}w)Nrequired properties)r+r0r9itemsr)r schema_fieldsr param_namerys r*_cohere_parameter_definitionsrsx%o6M !2!2:!>?H)3=3D3D\3R(S(Y(Y([  $J  0L1I:YaKabb  s"A7rbc t|jd}tt|jdt|jdt t t|jdS)NrLrMr}rP)rMr}parameterDefinitions)r+r9rr4rrx)rb function_defs r*_to_cohere_toolrsh$TXXj%9:L  |''/ 0l..}=>: *:l6F6F|6T+U V  r-toolsc>|Dcgc] }t|c}Scc}w)unAdapt OpenAI-format tool definitions to the OCI Cohere format. - Resolves ``$ref``/``$defs`` and ``anyOf`` patterns that OCI rejects. - Maps JSON Schema type names to Python type names (``"string"`` → ``"str"``). - Embeds unsupported constraints (enum, format, range, pattern) into the parameter description so the model can still see them. )r)rrbs r*)adapt_tool_definitions_to_cohere_standardrs/4 4dOD ! 44 4s json_responsemodelmodel_response raw_responsec  tj|}||_ttjjj|_ |jj}t|jj}d}|jj rt#|jj D cgc]i\} } t%| | j&t)j*| j,dd| j&t)j*| j,ddk}} } |r|nd} d | d } ||| d <t/d | | g|_|jj2} | 3t5| j6| j8| j:|_|St5d d d |_|S#ttf$r}t d||j d}~wwxYwcc} } w)zGParse a non-streaming Cohere OCI response into a LiteLLM ModelResponse.z/Response cannot be casted to CohereChatResult: )r_ status_codeNT sort_keysrLrMrNrXr7rLrV)rUr;rWr)indexr_ finish_reason) prompt_tokenscompletion_tokens total_tokens)rmodel_validate TypeErrorr rrrintdatetimenow timestampcreated chatResponser8r finishReasonrarkr rMjsondumpsrPrchoicesusager" promptTokenscompletionTokens totalTokens)rrrrcohere_responsee response_textrrWr[tcr;r_ usage_infos r*handle_cohere_responsers !1!@!@!O!N !2!2!6!6!8!B!B!DEN*77<A.G/>G, G''G, dict_chunkprior_tool_calls_emittedprior_text_emittedc  tj|}|j d|_|j duxr|jdu}|r|rdn |j}|r|rdn |j}d}|rt|D cgc]i\} } t| | jtj| jdd| jtj| jd d k}} } t!|j} t#t%|j t'||ddd | g S#ttf$r}t dd|d}~wwxYwcc} } w)aParse a single Cohere SSE chunk into a LiteLLM ModelResponseStream. ``prior_tool_calls_emitted`` lets the caller signal whether tool calls were already emitted in earlier chunks of the same stream. When set, the terminal consolidation chunk's tool calls are suppressed (they would duplicate prior deltas); otherwise they are passed through so a stream that delivers tool calls only on the terminal chunk doesn't silently drop them. ``prior_text_emitted`` plays the analogous role for the ``text`` field: when set, the terminal consolidation chunk's ``text`` is suppressed (it would re-emit the full assembled response on top of prior deltas); when unset (e.g. a degenerate stream that delivers the entire response in a single SSE event carrying both ``chatHistory`` and ``finishReason``), the text is passed through so the response content isn't silently lost. iz-Chunk cannot be parsed as CohereStreamChunk: )rr_NrTrrLrr)r;rWprovider_specific_fieldsthinking_blocksreasoning_content)rdeltar)r)rrrr rr chatHistoryrr8rarkr rMrrrPr r r!r) rrr typed_chunkris_terminal_consolidationr8cohere_tool_callsrWr[rrs r*handle_cohere_stream_chunkrsy* .==jI   (3'>'>d'J's{OgOgosOs(AEWd^i^n^nD";?W^i^s^s15J##45 231bggtzz"--cg?hi"GG!%BMM!:   8 8P8PQM  !'' )-1$(&* ,    s  ' CA3G   N sD5>A.E5EEE)FF)D__doc__rrcollections.abcrrrtypingrrhttpxpydanticrr r litellm.llms.oci.chat.genericr r litellm.llms.oci.common_utilsr rrrrrlitellm.types.llms.ocirrrrrrrrlitellm.types.llms.openairrlitellm.types.utilsrrrr r!r"r(r3r+r/r0r4objectr:rBrDrE__annotations__rIrRrurxboolrrrrResponserrr-r*rs 77 <<   X4i4Di$844i4DO43Y3331wsF{3113'#v+*>!??$F3"3"3" "-T#v+->!??#S&[0A*Ad3PV;FW$CB#$B -BJb4Y3GbIbtCN/CRV[t9cKdFdAe'#y.1j 5 GCN+ , 5 * 5@3 >*@ @"@.. @  @J&+$^Y'^"^^ ^r-