6iSJdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddZ dd Z dd Zdd Zdd Zdd Z ddZ ddZ ddZe j,eede j.eediZy)zAWS Bedrock-specific utilities. This module contains utilities specific to the AWS Bedrock provider, including reask functions, response handlers, and message formatting. ) annotationsN)dedent)Any)Modec|j}d|j|jxsd|jdd|idiS)aD Generate Bedrock tool schema from a Pydantic model. Bedrock Converse API expects tools in this format: { "toolSpec": { "name": "tool_name", "description": "tool description", "inputSchema": { "json": { JSON Schema } } } } toolSpeczCorrectly extracted `z5` with all the required parameters with correct typesjson)name description inputSchema)model_json_schema__name____doc__)response_modelschemas m/root/.openclaw/workspace/visionaryfx/venv/lib/python3.12/site-packages/instructor/providers/bedrock/utils.pygenerate_bedrock_schemarsZ - - /F "++)11v&~'>'>&??tu"F+   c|j}|ddg}|jddd|igd|dj||S)z Handle reask for Bedrock JSON mode when validation fails. Kwargs modifications: - Adds: "messages" (user message requesting JSON correction) outputmessageusertextz@Correct your JSON ONLY RESPONSE, based on the following errors: rolecontentmessagescopyappendextend)kwargsresponse exception reask_msgss rreask_bedrock_jsonr'.sl[[]F8$Y/0J_`i_jk   :j) Mrc  |j}|dd}|g}d}d|vr|dD]}d|vs|dd}n|r#|jdd|d d |d igd d igdn|jdd d|d igd|dj||S)z Handle reask for Bedrock tools mode when validation fails. Kwargs modifications: - Adds: "messages" (assistant message with tool use, then user message with tool result error) rrNrtoolUse toolUseIdr toolResultrzValidation Error found: z. Recall the function correctly, fix the errorserror)r*rstatusrz,Validation Error due to no tool invocation: rr)r#r$r%assistant_messager& tool_use_id content_blocks rreask_bedrock_toolsr1Is[[]F!*95#$JK%%.y9 MM)+I6{C   %)4%+.G {SB-C!"( '.'    ( "OPY{[J!K   :j) Mrc|sy|jj}d|vr|jddd}|dvry|dvry|dvry|d vry y) zy Map common/variant image types to Bedrock's required image.format enum: one of {'gif','jpeg','png','webp'}. jpeg/)jpgpjpegzx-jpegzx-jpg)pngzx-pngr8)gifzx-gifr9)webpz image/webpr:)striplowersplit) mime_or_extvals r_normalize_bedrock_image_formatr@sp      # # %C cziiQ" 11   $$ rcf|jdxsijd}|s tdtj|dxsd}t |}|j drA |j dd\}}d |vr td d |dtj|idiS|j drg tj|d}|j|jjd}|rd|vr t |}d |d|jidiStd#t$r}td |d }~wwxYw#tjj$r}td||d }~wtjj$r}td|d||d }~wtjj $r}td|d||d }~wtjj"$r}td|d||d }~wt$$r}td|d||d }~wwxYw)z Convert OpenAI-style image part: {"type":"image_url","image_url":{"url": ""}} into Bedrock Converse image content: {"image":{"format": "","source":{"bytes": }}} image_urlurlz6image_url.url is required for OpenAI-style image partsrz image/jpegzdata:,r5z!Invalid data URL in image_url.urlNz;base64z/Only base64 data URLs are supported for Bedrockimagebytes)formatsource)zhttp://zhttps://)timeoutz Content-Typer4z$Timed out while fetching image from z+Connection error while fetching image from z: z%HTTP error while fetching image from z(Request error while fetching image from z+Unexpected error while fetching image from zFUnsupported image_url scheme. Use http(s) or data:image/...;base64,...)get ValueError mimetypes guess_typer@ startswithr=base64 b64decoderequestsraise_for_statusheadersr exceptionsTimeoutConnectionError HTTPErrorRequestException Exception) partrB guessed_mimefmtheaderb64erespctypes r_openai_image_part_to_bedrockrcs^+&,"11%8I QRR'' 215EL ), 7CG$ I#//#q1KFC F "NO OCGV=M=Mc=R3STUU   5 6 << 26D  ! ! #LL$$^4E5e<7NOP P( T  E I@Aq H I""** XCI;OPVW W""22 =i[1#N "",, 7 {"QCH ""33 :9+RsK  =i[1#N  sa+D.=A%E . E7 EE H0(E77 H0F)) H0 G H0;H  H0H++H0ct|trd|igSt|trg}|D]}t|trd|vr|j d}|dvr<|j dxs|j dxsd}|j d|ii|dk(r|j t |td|t|trd|vrBt|dtr/t|jdhk(r|j |d|vr&t|dtr|j |d |vr&t|d tr|j |Atd |t|tr|j d|ittd t||Std t|) a Normalize content into Bedrock Converse content list. Allowed inputs: - string -> [{"text": "..."}] - list of parts: OpenAI-style: {"type":"text","text":"..."} {"type":"input_text","text":"..."} {"type":"image_url","image_url":{"url":""}} Bedrock-native (passed through as-is): {"text":"..."} {"image":{"format":"jpeg|png|gif|webp","source":{"bytes": }}} {"document":{"format":"pdf|csv|doc|docx|xls|xlsx|html|txt|md","name":"...","source":{"bytes": }}} Note: - We do not validate or normalize Bedrock-native image/document blocks here. Caller is responsible for providing valid 'format' and raw 'bytes'. rtype)r input_textrfrBz0Unsupported OpenAI-style part type for Bedrock: rEdocumentz&Unsupported dict content for Bedrock: z&Unsupported content part for Bedrock: z.Unsupported message content type for Bedrock: ) isinstancestrlistdictrKr!rcrLsetkeysre)ritemspttxts r_to_bedrock_content_itemsrss*'3!""'4 &(' QA!T"v{EE&M..%%-D155+>D"CLL&#/ #LL!>q!AB #STUSV!WXX!T"aK"1V9c2AFFH &1LLOa|jd }t|t r|g|d<nGt|tr7||d<n1d|vr|jd|d<nd|vr|jd|d<|r1d|vr(|d}|j D]\}}||vs |||<n||d<d|vrt|dtr|jd}g} g} |D]} t| ts| j| %| j} | jd} | jd}| dk(r "modelId" - Collects: temperature, max_tokens, top_p, stop into inferenceConfig - Converts: messages content to Bedrock format systemrrrrmodelmodelId temperature max_tokens maxTokenstop_ptopPstop stopSequencesstop_sequencesinferenceConfigrrz\System message content must be a string for Bedrock processing by this handler. Found type: .) rirkpoprlinsertrjror!r rKrLrers) call_kwargssystem_content system_textinference_config_paramsstop_valexisting_inference_configkeyvalueoriginal_input_messagesbedrock_system_list$bedrock_user_assistant_messages_listmsg_dictcurrent_message_for_apirrs r)_prepare_bedrock_converse_kwargs_internalrsH;:k(.CT#J$2 >!,d3.++)+F3K,*, J'  # * *H=  +);">!,!9 I! #1<1O .{"/:|/L ,  #/:{/K ,+*5//'*B' ; *5//&*A'??6* h $8@z #O 4 $ '7? #O 4 [ (3>??CS3T0;&3>???3S0   +)44E(F %5;;= ; U775:-c2 ;.EK) *[ Z J0G%N"-//*"=46EG,/ UHh- 5;;HE'/mmo #*..v6D-11Gxgs+'../@A$''+G}oQ8  779R:+I65;;rjrrj)r[rrr)rrrzlist[dict[str, Any]])rrrr)rrrrrz tuple[type[Any], dict[str, Any]])rztype[Any] | Nonerrrz'tuple[type[Any] | None, dict[str, Any]])r __future__rrPr rMrRtextwraprtypingrmoderrr'r1r@rcrsrrr BEDROCK_JSON BEDROCK_TOOLSBEDROCK_HANDLERSrrrs # 6 6; ;;;|*6 rEWPAAAH$&$&+9$&%$&N&$&2@&,&> #' $( r