`gjStdZddlmZmZmZmZddlmZddlm Z m Z m Z GddeZ ddl mZede y ) aAWS Bedrock Converse API transport. Delegates to the existing adapter functions in agent/bedrock_adapter.py. Bedrock uses its own boto3 client (not the OpenAI SDK), so the transport owns format conversion and normalization, while client construction and boto3 calls stay on AIAgent. )AnyDictListOptional)ProviderTransport)NormalizedResponseToolCallUsagec eZdZdZedefdZdeeee fde fdZ deeee fde fdZ dd edeeee fde eeee fdeee ffd Z d e defd Zd e defd ZdedefdZy)BedrockTransportz*Transport for api_mode='bedrock_converse'.returncy)Nbedrock_converse)selfs K/root/.hermes/venv/lib/python3.12/site-packages/agent/transports/bedrock.pyapi_modezBedrockTransport.api_modes!messagesc ddlm}||S)z3Convert OpenAI messages to Bedrock Converse format.r)convert_messages_to_converse)agent.bedrock_adapterr)rrkwargsrs rconvert_messagesz!BedrockTransport.convert_messagessF+H55rtoolscddlm}||S)z;Convert OpenAI tool schemas to Bedrock Converse toolConfig.r)convert_tools_to_converse)rr)rrrs r convert_toolszBedrockTransport.convert_toolssC(//rNmodelc ddlm}|jdd}|jd}|||||jdd|jd| }d |d <||d <|S) u^Build Bedrock converse() kwargs. Calls convert_messages and convert_tools internally. params: max_tokens: int — output token limit (default 4096) temperature: float | None guardrail_config: dict | None — Bedrock guardrails region: str — AWS region (default 'us-east-1') r)build_converse_kwargsregionz us-east-1guardrail_config max_tokensi temperature)rrrr$r%r#T__bedrock_converse____bedrock_region__)rr!get) rrrrparamsr!r" guardrailrs r build_kwargszBedrockTransport.build_kwargs st" @Hk2JJ12 &zz,5 =1&  *.%&'-#$ rresponsec ddlm}t|dr|jr|}n||}|jd}|j}|j xsd}d}|j rW|j D cgc]B} t| j| jj| jjD}} d} t|drS|jrG|j} tt| ddxsdt| d dxsdt| d dxsd } t|d dxs t|d d} t|j ||| | Scc} w)zNormalize Bedrock response to NormalizedResponse. Handles two shapes: 1. Raw boto3 dict (from direct converse() calls) 2. Already-normalized SimpleNamespace with .choices (from dispatch site) r)normalize_converse_responsechoicesstopN)idname argumentsusage prompt_tokenscompletion_tokens total_tokens)r5r6r7 reasoningreasoning_content)content tool_calls finish_reasonr8r4)rr.hasattrr/messager<r;r r1functionr2r3r4r getattrrr:) rr,rr.nschoicemsgr<r;tcr4ur8s rnormalize_responsez#BedrockTransport.normalize_responseCsM F 8Y 'H,<,<B-X6BAnn,,6  >>..   uu)) kk33J 2w BHHA%a!<A")!-@!"D"I$Q:?aE Cd3^wsDWY]7^ !KK!'   )s1AEct|yt|trd|vSt|drt|jSy)uCheck Bedrock response structure. After normalize_converse_response, the response has OpenAI-compatible .choices — same check as chat_completions. Foutputr/) isinstancedictr=boolr/)rr,s rvalidate_responsez"BedrockTransport.validate_responsevs@   h %x' ' 8Y '(() )r raw_reasonc8ddddddd}|j|dS)zMap Bedrock stop reason to OpenAI finish_reason. The adapter already does this mapping inside normalize_converse_response, so this is only used for direct access to raw responses. r0r;lengthcontent_filter)end_turntool_user$ stop_sequenceguardrail_intervenedcontent_filtered)r()rrM_MAPs rmap_finish_reasonz"BedrockTransport.map_finish_reasons0$"#$4 0  xx F++r)N)__name__ __module__ __qualname____doc__propertystrrrrrrrrr+rrFrKrLrWrrrr r s4 "#""6d38n)=6C6 04S#X#70C015 !!tCH~&!T#s(^,- ! c3h !F1 31 =O1 f#$ ,C,C,rr )register_transportrN)r[typingrrrragent.transports.baseragent.transports.typesrr r r agent.transportsr^rrrrcs<-,3FFE,(E,R0%'78r