qi2dZddlZddlZddlZddlmZddlmZmZm Z m Z m Z ddl m Z ddlmZddlmZddlmZdd lmZmZmZmZmZGd d e ZGd d eeZy)z$AWS Bedrock LLM adapter for Pipecat.N) dataclass)AnyDictListOptional TypedDict)logger)BaseLLMAdapter)FunctionSchema) ToolsSchema) LLMContextLLMContextMessageLLMContextToolChoiceLLMSpecificMessageLLMStandardMessagecpeZdZUdZeeeeefe d<eeeefe d<eeeefe d<e e d<y)AWSBedrockLLMInvocationParamsz"" S#X %%r rceZdZdZedefdZdedefdZ de e ee ffdZ eGddZd e edefd Zd edeee ffd Zd edeee ffd Zedede ee ffdZdede e ee ffdZy)AWSBedrockLLMAdapterzAdapter for AWS Bedrock LLM integration with Pipecat. Provides conversion utilities for transforming Pipecat function schemas into AWS Bedrock's expected tool format for function calling capabilities. returncy)zHGet the identifier used in LLMSpecificMessage instances for AWS Bedrock.awsr)selfs r!id_for_llm_specific_messagesz1AWSBedrockLLMAdapter.id_for_llm_specific_messages-sr contextc|j|j|}|j|j|j |j xsg|j dS)a Get AWS Bedrock-specific LLM invocation parameters from a universal LLM context. Args: context: The LLM context containing messages, tools, etc. Returns: Dictionary of parameters for invoking AWS Bedrock's LLM API. )rrrr) _from_universal_context_messages get_messagesrrfrom_standard_toolsrr)r'r)rs r!get_llm_invocation_paramsz.AWSBedrockLLMAdapter.get_llm_invocation_params2s[889J9J79SToo ))--gmm<B#..  r c4|j|j|j}g}|D]f}tj|}d|vr:t |dt r'|dD]}|jdsd|ddd<!|j|h|S)a^Get messages from a universal LLM context in a format ready for logging about AWS Bedrock. Removes or truncates sensitive data like image content for safe logging. Args: context: The LLM context containing messages. Returns: List of messages in a format ready for logging about AWS Bedrock. contentimagez...sourcebytes) r+r,rcopydeepcopy isinstancelistgetappend)r'r)rmessages_for_loggingmessagemsgitems r!get_messages_for_loggingz-AWSBedrockLLMAdapter.get_messages_for_loggingGs889J9J79ST]] " -G--(CCc)nd3 #IE88G,?DDM(3G<E ! ' ' , -$#r c<eZdZUdZeeeefed<e eed<y)&AWSBedrockLLMAdapter.ConvertedMessageszLContainer for Anthropic-formatted messages converted from universal context.rrN) rrrrrrrrrrrr r!ConvertedMessagesr@as"ZtCH~&& r rAuniversal_context_messagescRd}g} |Dcgc]}|j|}}|r$|dddk(r|dd}|j d|D]}|ddk(s d|d<d}|t |dz kr||}||dz} |d| dk(rjt |dtr d|dd g|d<t | dtr d| dd g| d<|dj| d|j |dzn|dz }|t |dz kr|D]R}t |dtr|dd k(rd |d<$t |dts8t |ddk(sJdd d g|d<T|j|| Scc}w#t$r#}tjd|Yd}~}d}~wwxYw) NzError mapping messages: rrolerr0usertext)typerG(empty))rr) _from_universal_context_message Exceptionr errorpoplenr6rextendr7rA) r'rBrrmer;icurrent_message next_messages r!r+z5AWSBedrockLLMAdapter._from_universal_context_messageshs  9IcdA<>oi8#>!'1KL2OI.l95s;8> U^H_/`.aL+ *11,y2IJ QU#Q!#h-!##&  KG'),c2wy7IR7O%. "GI.5#gi>P:QUV:V/5y&I%J "  K %%x%GGUe 9 LL3A37 8 8 9s'E:E5E:5E:: F&F!!F&r;ct|trtj|jS|j |S)N)r6rr4r5r;_from_standard_message)r'r;s r!rKz4AWSBedrockLLMAdapter._from_universal_context_messages1 g1 2==1 1**733r c $tj|}|ddk(ry |djjdr@|djj drt j |d}d|ig}nd|dig}dd |d |d igd S|jd rR|d }dgd }|D]A}|d}t j |d}d|d|d|di} |dj| C|S|jd} t| tr| dk(r |dddigd S|dd| igd St| trg} | D]} | jdddk(r"| ddk7r| dnd} | jd| i| ddk(sC| ddjdr| dd}|jddjdd}|jdd}d |d!t!j"|jd"did#i}| j|| dd}t%j&d$|t)| D cgc] \}} d | vs |}}} t)| D cgc] \}} d| vs |}}} t+|dk(r4|r2|d}|d}||kDr| j-|}| j/||d| d S|S#t jttf$r d|dig}YywxYwcc} }wcc} }w)%axConvert standard format message to AWS Bedrock format. Handles conversion of text content, tool calls, and tool results. Empty text content is converted to "(empty)". Args: message: Message in standard format. Returns: Message in AWS Bedrock format. Examples: Standard format input:: { "role": "assistant", "tool_calls": [ { "id": "123", "function": {"name": "search", "arguments": '{"q": "test"}'} } ] } AWS Bedrock format output:: { "role": "assistant", "content": [ { "toolUse": { "toolUseId": "123", "name": "search", "input": {"q": "test"} } } ] } rDtoolr0{}jsonrGrE toolResult tool_call_id) toolUseIdr0)rDr0 tool_calls assistantfunction argumentstoolUseidname)r_rfinputrIrJrH image_urlurlzdata::rF;r/r1r3,)formatr2zUnsupported 'image_url': )r4r5strip startswithendswithr\loadsJSONDecodeError ValueErrorAttributeErrorr8r9r6rr7splitbase64 b64decoder warning enumeraterOrNinsert)r'r; content_jsontool_result_contenttcret tool_callrbrc new_tool_user0 new_contentr= text_contentri mime_type image_formatnew_itemrS image_indices text_indicesimg_idx first_txt_idx image_items r!rWz+AWSBedrockLLMAdapter._from_standard_messagesP--( 6?f $ E9%++-88='C%'((3-C($(::gi.@#AL,2L+A*B',2GI4F+G*H' %)0)@':'   ;;| $&B&26C 4 $Z0 JJx '<= %.t_ ( 0!*   I%%l3 4J++i( gs #"} 'fi=P<;.K(/::7C";/6$'IIcN1$5$;$;C$@$C '0s';A'> #*6+2F4D4DSYYs^TUEV4W*X&$ $**84";/6)B3%'HI- J6/8 .DX71dSWQXMX-6{-CV'!TvQU~AVLV=!Q&<'* ,Q ]*!,!9J""=*=#FO D D]((*nE E(. 0B'C&D# EFYVs*A*K L$L9 L L &LLrbcnd|j|jdd|j|jdidiS)zConvert a function schema to Bedrock's tool format. Args: function: The function schema to convert. Returns: Dictionary formatted for Bedrock's tool specification. toolSpecr\object)rH propertiesrequired)rf description inputSchema)rfrrr)rbs r!_to_bedrock_function_formatz0AWSBedrockLLMAdapter._to_bedrock_function_format$sH  '33 (&.&9&9$,$5$5    r tools_schemacb|j}|Dcgc]}|j|c}Scc}w)zConvert function schemas to Bedrock's function-calling format. Args: tools_schema: The tools schema containing functions to convert. Returns: List of Bedrock formatted function call definitions. )standard_toolsr)r'rfunctions_schemafuncs r!to_provider_tools_formatz-AWSBedrockLLMAdapter.to_provider_tools_format<s/(66CST4006TTTs,N)rrrrpropertyrr(r rr.rrrr>rrArr+rrKrrW staticmethodr rr rrr r!r#r#&s c  @] *$4S#X3G$4 2H*./@*A2H 2Hh47H4TRUWZRZ^4 A.@AT#s(^AF n c3h  . U[ UT$sTWx.EY Ur r#)rrwr4r\ dataclassesrtypingrrrrrlogurur !pipecat.adapters.base_llm_adapterr (pipecat.adapters.schemas.function_schemar %pipecat.adapters.schemas.tools_schemar *pipecat.processors.aggregators.llm_contextr rrrrrr#rr r!rsT+ !77<C=&I&`U>*GH`Ur