qi1"dZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZddlmZdd lmZmZdd lmZmZGd d eZeGd dZGdde ZGddeeZy)z'AWS Nova Sonic LLM adapter for Pipecat.N) dataclass)Enum)AnyDictListOptional TypedDict)logger)BaseLLMAdapter)FunctionSchema) AdapterType ToolsSchema) LLMContextLLMContextMessagec eZdZdZdZdZdZdZy)Rolea5Roles supported in AWS Nova Sonic conversations. Parameters: SYSTEM: System-level messages (not used in conversation history). USER: Messages sent by the user. ASSISTANT: Messages sent by the assistant. TOOL: Messages sent by tools (not used in conversation history). SYSTEMUSER ASSISTANTTOOLN)__name__ __module__ __qualname____doc__rrrrb/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/adapters/services/aws_nova_sonic_adapter.pyrrsF DI Drrc&eZdZUdZeed<eed<y)&AWSNovaSonicConversationHistoryMessagezA single message in AWS Nova Sonic conversation history. Parameters: role: The role of the message sender (USER or ASSISTANT only). text: The text content of the message. roletextN)rrrrr__annotations__strrrrrr's J IrrcLeZdZUdZeeed<eeed<ee ee fed<y)AWSNovaSonicLLMInvocationParamszContext-based parameters for invoking AWS Nova Sonic LLM API. This is a placeholder until support for universal LLMContext machinery is added for AWS Nova Sonic. system_instructionmessagestoolsN) rrrrrr#r"rrrrrrrr%r%4s2 ! %9:: S#X rr%ceZdZdZedefdZdedefdZ de e ee ffdZ eGddZd e edefd Zdefd Zed ede ee ffd Zdede e ee ffdZy)AWSNovaSonicLLMAdapterzAdapter for AWS Nova Sonic language models. Converts Pipecat's standard function schemas into AWS Nova Sonic's specific function-calling format, enabling tool use with Nova Sonic models. returncy)zKGet the identifier used in LLMSpecificMessage instances for AWS Nova Sonic.zaws-nova-sonicr)selfs rid_for_llm_specific_messagesz3AWSNovaSonicLLMAdapter.id_for_llm_specific_messagesFs rcontextc|j|j|}|j|j|j |j xsgdS)a|Get AWS Nova Sonic-specific LLM invocation parameters from a universal LLM context. This is a placeholder until support for universal LLMContext machinery is added for AWS Nova Sonic. Args: context: The LLM context containing messages, tools, etc. Returns: Dictionary of parameters for invoking AWS Nova Sonic's LLM API. )r&r'r() _from_universal_context_messages get_messagesr&r'from_standard_toolsr()r-r/r's rget_llm_invocation_paramsz0AWSNovaSonicLLMAdapter.get_llm_invocation_paramsKsR889J9J79ST"*"="= ))--gmm<B   rcV|j|j|jS)aGet messages from a universal LLM context in a format ready for logging about AWS Nova Sonic. Removes or truncates sensitive data like image content for safe logging. This is a placeholder until support for universal LLMContext machinery is added for AWS Nova Sonic. Args: context: The LLM context containing messages. Returns: List of messages in a format ready for logging about AWS Nova Sonic. )r1r2r')r-r/s rget_messages_for_loggingz/AWSNovaSonicLLMAdapter.get_messages_for_logging^s&44T5F5Fw5OPYYYrc6eZdZUdZeeed<dZee ed<y)(AWSNovaSonicLLMAdapter.ConvertedMessageszIContainer for Google-formatted messages converted from universal context.r'Nr&) rrrrrrr"r&rr#rrrConvertedMessagesr8ms W=>>,0HSM0rr9universal_context_messagescd}g}|s|jStj|}|djddk(rb|j d}|jd}t |t r|}n$t |tr|djd}|r||_|D]'}|j|}|s|j|)|j||S)Nrr systemcontentr!)r'r&) r9copydeepcopygetpop isinstancer#list_system_instruction_from_universal_context_messageappend)r-r:r&r'r<r=universal_context_messagemessages rr1z7AWSNovaSonicLLMAdapter._from_universal_context_messagests"*))+ +%)]]3M%N" &a ( , ,V 4 @/33A6Fjj+G'3'%,"GT*%,QZ^^F%;"!+=(*D ) %::;TUG( ) %%xL^%__rc|jd}|jddk(s|jddk(r|jd}t|jdtrnd}|jdD]X}|jddk(r|d|jdzz }/tjd |jdd |Z|r"t t |j| Sy y ) zConvert standard message format to Nova Sonic format. Args: message: Standard message dictionary to convert. Returns: Nova Sonic conversation history message, or None if not convertible. r user assistantr=typer! z+Unhandled content type in context message: z - )r r!N)r@rBrCr errorrrupper)r-rHr r=cs rrEz6AWSNovaSonicLLMAdapter._from_universal_context_messages{{6" ;;v & (GKK,?;,Nkk),G'++i0$7 Y/AuuV}.3v#66 I!%%PV-X[\c[de =4 CU\cdd-Orfunctionc d|j|jdtjd|j|j didiS)zConvert a function schema to AWS Nova Sonic format. Args: function: The function schema to convert. Returns: Dictionary in AWS Nova Sonic function format with toolSpec structure. toolSpecjsonobject)rM propertiesrequired)name description inputSchema)rYrZrUdumpsrWrX)rRs r"_to_aws_nova_sonic_function_formatz9AWSNovaSonicLLMAdapter._to_aws_nova_sonic_function_formatsS  '33DJJ$,*2*=*=(0(9(9    r tools_schemac|j}|Dcgc]}|j|}}g}|jr*|jjtj g}||zScc}w)aConvert tools schema to AWS Nova Sonic function-calling format. Args: tools_schema: The tools schema containing function definitions to convert. Returns: List of dictionaries in AWS Nova Sonic function format. )standard_toolsr] custom_toolsr@r SHIM)r-r^functions_schemafuncr` shimmed_toolss rto_provider_tools_formatz/AWSNovaSonicLLMAdapter.to_provider_tools_formatss(66FV >BD 3 3D 9    $ $(5599+:J:JBOM -- sA(N)rrrrpropertyr#r.rr%r4rrrr6rr9rr1rrE staticmethodr r]rrfrrrr*r*?s  c    @_ & Z4S#X3G Z111 `*./@*A` `@e:`e: ^ SRUX  2.[.T$sTWx.EY.rr*)rr>rU dataclassesrenumrtypingrrrrr logurur !pipecat.adapters.base_llm_adapterr (pipecat.adapters.schemas.function_schemar %pipecat.adapters.schemas.tools_schemar r*pipecat.processors.aggregators.llm_contextrrrrr%r*rrrrqsp. !77<CJT 4        i d.^,KLd.r