6i!dZddlmZddlmZddlmZ d dZd dZ ddZ dd Z ejee d ejee d iZ y )zCohere-specific utilities. This module contains utilities specific to the Cohere provider, including reask functions, response handlers, and message formatting. ) annotations)Any)Modec|jd}t|drd}|j}nt|dr~t|jdrhd}|jj}d}|r>|D]9}t|ds|j dk(s t|ds-|j}n|s*t |}nt |}|d |vrd}n d |vsd|vrd}d |d |}|dk(r|d jd |d|S|dk(r>|jdd}d |vr|d jd |dn d |dg|d <||d<|Std|d)a^ Handle reask for Cohere tools and JSON schema modes. Supports both V1 and V2 formats. V1 kwargs modifications: - Adds/Modifies: "chat_history" (appends prior message) - Modifies: "message" (user prompt describing validation errors) V2 kwargs modifications: - Modifies: "messages" (appends error correction message) _cohere_client_versiontextv1messagecontentv2typemessages chat_historyzMCorrect the following JSON response, based on the errors given below: JSON: z Exceptions: userroler rr #Unsupported Cohere client version: . Expected 'v1' or 'v2'.) gethasattrr r r rstrappend ValueError) kwargsresponse exceptionclient_version response_text content_itemsitemcorrection_msgr s l/root/.openclaw/workspace/visionaryfx/venv/lib/python3.12/site-packages/instructor/providers/cohere/utils.pyreask_cohere_toolsr&s"ZZ 89Nx  9 %'(2B2BI*N ((00  % D&) V+f-$(IIM MMH  !V#!%6)Y&-@!%  1) > z!!6n"MN& M% 4 **Y+ V # > " ) )6g*N O/5'&J%KF> "*y M 1.1AB% &  c|j}|jd}|dk(r2d|vrd|vr|jd|d<|jddd|fS|dk(rz|jdg}g}|dd D]}|j|d |d d |d d |d <||d<d|vrd|vr|jd|d<|jddd|fStd|d)a Convert OpenAI-style messages to Cohere format. Handles both V1 and V2 client formats. V1 format: - Removes: "messages" - Adds: "message" (last user message) - Adds: "chat_history" (prior messages) V2 format: - Keeps: "messages" (compatible with OpenAI format) Both versions: - Renames: "model_name" -> "model" - Removes: "strict" - Removes: "_cohere_client_version" (internal marker) rr model_namemodelstrictNr rrr rr rrr)copypoprr) new_kwargsr rrr s r%handle_cohere_modesr0\s:$"J^^$<=N : %'*C",..">Jw x&0  / 4 >>*b1 } G   #FO&y1  !) Y 7 9%1 >" : %'*C",..">Jw x&   1.1AB% &  r'cl| t|Sd|jd|d<t|\}}||fS)a Handle Cohere JSON schema mode. When response_model is None: - Converts messages from OpenAI format to Cohere format (message + chat_history) - No schema is added to the request When response_model is provided: - Converts messages from OpenAI format to Cohere format - Adds the model's JSON schema to response_format Kwargs modifications: - Removes: "messages" (converted to message + chat_history) - Adds: "message" (last message content) - Adds: "chat_history" (all messages except last) - Modifies: "model" (if "model_name" exists, renames to "model") - Removes: "strict" - Adds: "response_format" (with JSON schema) - only when response_model provided json_object)rschemaresponse_format)r0model_json_schema)response_modelr/_s r%handle_cohere_json_schemar8sN,":.. 224%J !( 3MAz : %%r'c | t|St|\}}d|jd|jd|jd|jd }d|vr|djdd|d ||fSd|d g|d z|d <||fS) aQ Handle Cohere tools mode. When response_model is None: - Converts messages from OpenAI format to Cohere format (message + chat_history for V1, messages for V2) - No tools or schema instructions are added - Allows for unstructured responses from Cohere When response_model is provided: - Converts messages from OpenAI format to Cohere format - Prepends extraction instructions to the chat history (V1) or messages (V2) - Includes the model's JSON schema in the instructions - The model is instructed to extract a valid object matching the schema Kwargs modifications: - All modifications from handle_cohere_modes (message format conversion) - Modifies: "chat_history" (V1) or "messages" (V2) to prepend extraction instruction - only when response_model provided zExtract a valid z= object based on the chat history and the json schema below. z= The JSON schema was obtained by running: ```python schema = zG.model_json_schema() ``` The output must be a valid JSON object that `z}.model_validate_json()` can successfully parse. Respond with JSON only. Do not include code fences, markdown, or extra text. rrrrrr)r0__name__r5insert)r6r/r7 instructions r%handle_cohere_toolsr=s*":..' 3MAz(()*!!#$%   ! ! "#./=.E.E-FG KZ:%%a&[)QR : %% 4& ~ &&' >" : %%r')reaskrN)rdict[str, Any]rrr Exception)r/r?returnztuple[None, dict[str, Any]])r6ztype[Any] | Noner/r?rAz'tuple[type[Any] | None, dict[str, Any]])__doc__ __future__rtypingrmoderr&r0r8r= COHERE_TOOLSCOHERE_JSON_SCHEMACOHERE_HANDLERSr'r%rJs #K KKK\2j &$ &2@ &, &F1&$1&2@1&,1&l #' #- r'