6i"dZddlmZddlmZmZddlmZerddlm Z n ddlm Z ddZ dd Z dd Z dd Z dd Zej"e ed ej$eed iZy#e $rdZ YVwxYw)zxAI-specific utilities. This module contains utilities specific to the xAI provider, including reask functions, response handlers, and message formatting. ) annotations)Any TYPE_CHECKING)Mode)chatNc4t tdg}|D]}|d}|jdd}t|trtj |}n t d|dk(r%|jtj|u|dk(r%|jtj||dk(r%|jtj||d k(r%|jtj|t d ||S) z,Convert OpenAI-style messages to xAI format.z$xai_sdk is required for xAI providerrolecontentz.Only string content supported for xAI provideruser assistantsystemtoolzUnsupported role: ) xchat ImportErrorget isinstancestrtext ValueErrorappendr rr tool_result)messages convertedmr r cs i/root/.openclaw/workspace/visionaryfx/venv/lib/python3.12/site-packages/instructor/providers/xai/utils.py_convert_messagesrs }@AAI :y%% 2& gs # 7#AMN N 6>   UZZ] + [   U__Q/ 0 X    U\\!_ - V^   U..w7 81$89 9!:" cd|j}dd|d|d}|dj||S)z Handle reask for xAI JSON mode when validation fails. Kwargs modifications: - Modifies: "messages" (appends user message requesting correction) r zValidation Errors found: zO Recall the function correctly, fix the errors found in the following attempt: r r r)copyr)kwargsresponse exception reask_msgs rreask_xai_jsonr(0sT[[]F/ {;LMULVWI :i( Mr c|j}dt|d}|dj|dd|dd}|dj||S)z Handle reask for xAI tools mode when validation fails. Kwargs modifications: - Modifies: "messages" (appends assistant and user messages for tool correction) rr"rr zValidation Error found: z. Recall the function correctly, fix the errors)r#rr)r$r%r& assistant_msgr's rreask_xai_toolsr+Dsj[[]Fx=M :m,.yk9hiI :i( Mr c|jdg}t||d<|jdd|jdd|jdd|jdd||fS)a Handle xAI JSON mode. When response_model is None: - Converts messages from OpenAI format to xAI format - No schema is added to the request When response_model is provided: - Converts messages from OpenAI format to xAI format - Sets up the model for JSON parsing mode Kwargs modifications: - Modifies: "messages" (converts from OpenAI to xAI format) - Removes: instructor-specific kwargs (max_retries, validation_context, context, hooks) r x_messages max_retriesNvalidation_contextcontexthooks)rrpopresponse_model new_kwargsrs rhandle_xai_jsonr6asi&~~j"-H0:J|NN=$'NN'.NN9d#NN7D! : %%r cl|jdg}t||d<|jdd|jdd|jdd|jdd|GtAtj|j |j xsd|j |d <||fS) ax Handle xAI tools mode. When response_model is None: - Converts messages from OpenAI format to xAI format - No tools are configured When response_model is provided: - Converts messages from OpenAI format to xAI format - Sets up tool schema from the response model - Configures tool choice for automatic tool selection Kwargs modifications: - Modifies: "messages" (converts from OpenAI to xAI format) - Adds: "tool" (xAI tool schema) - only when response_model provided - Removes: instructor-specific kwargs (max_retries, validation_context, context, hooks) rr-r.Nr/r0r1r )name description parametersr)rrr2rr__name____doc__model_json_schemar3s rhandle_xai_toolsr>s*~~j"-H0:J|NN=$'NN'.NN9d#NN7D!!e&7"ZZ((&..4"%779 6 : %%r )reaskr%)rzlist[dict[str, Any]])r$dict[str, Any]r%rr& Exception)r4ztype[Any] | Noner5r@returnz'tuple[type[Any] | None, dict[str, Any]])r< __future__rtypingrrmoderxai_sdkrrrrr(r+r6r>XAI_JSON XAI_TOOLS XAI_HANDLERSr rrKs #%%) 4 ( :&$&2@&,&>&&$&&2@&&,&&V MM# NN $  q sA44A>=A>