o i1"@sdZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZddlmZdd lmZmZdd lmZmZGd d d eZeGd ddZGddde ZGdddeeZdS)z'AWS Nova Sonic LLM adapter for Pipecat.N) dataclass)Enum)AnyDictListOptional TypedDict)logger)BaseLLMAdapter)FunctionSchema) AdapterType ToolsSchema) LLMContextLLMContextMessagec@s eZdZdZdZdZdZdZdS)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__rrrrrrd/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/adapters/services/aws_nova_sonic_adapter.pyrs  rc@s"eZdZUdZeed<eed<dS)&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  rc@s>eZdZUdZeeed<eeed<ee ee fed<dS)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) rrrrrrrrrrrrrrrr 4s   r c@seZdZdZedefddZdedefddZ de e ee ffdd Z eGd d d Zd e edefd dZdefddZedede ee ffddZdede e ee ffddZdS)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. returncCsdS)zKGet the identifier used in LLMSpecificMessage instances for AWS Nova Sonic.zaws-nova-sonicrselfrrrid_for_llm_specific_messagesFsz3AWSNovaSonicLLMAdapter.id_for_llm_specific_messagescontextcCs,|||}|j|j||jpgdS)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"rrrget_llm_invocation_paramsKs  z0AWSNovaSonicLLMAdapter.get_llm_invocation_paramscCs|||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. )r*r+r")r'r)rrrget_messages_for_logging^s z/AWSNovaSonicLLMAdapter.get_messages_for_loggingc@s.eZdZUdZeeed<dZee ed<dS)z(AWSNovaSonicLLMAdapter.ConvertedMessageszIContainer for Google-formatted messages converted from universal context.r"Nr!) rrrrrrrr!rrrrrrConvertedMessagesms  r/universal_context_messagescCsd}g}|s |St|}|dddkr;|d}|d}t|tr*|}n t|tr6|dd}|r;||_|D]}| |}|rK| |q=|j||dS)Nrrsystemcontentr)r"r!) r/copydeepcopygetpop isinstancerlist_system_instruction_from_universal_context_messageappend)r'r0r!r"r1r2universal_context_messagemessagerrrr*ts(       z7AWSNovaSonicLLMAdapter._from_universal_context_messagescCs|d}|ddks|ddkrU|d}t|dtrId}|dD]!}|ddkr:|d|d7}q'td |dd |q'|rWtt||d Sd Sd S) 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. ruser assistantr2typer z+Unhandled content type in context message: z - )rrN)r5r7r8r errorrrupper)r'r=rr2crrrr:s  z6AWSNovaSonicLLMAdapter._from_universal_context_messagefunctionc Cs*d|j|jdtd|j|jdidiS)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)rA propertiesrequired)name description inputSchema)rLrMrHdumpsrJrK)rFrrr"_to_aws_nova_sonic_function_formats z9AWSNovaSonicLLMAdapter._to_aws_nova_sonic_function_format tools_schemacs:|j}fdd|D}g}|jr|jtjg}||S)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. csg|]}|qSr)rP).0funcr&rr s zCAWSNovaSonicLLMAdapter.to_provider_tools_format..)standard_tools custom_toolsr5r SHIM)r'rQfunctions_schemarU shimmed_toolsrr&rto_provider_tools_formats  z/AWSNovaSonicLLMAdapter.to_provider_tools_formatN)rrrrpropertyrr(rr r-rrrr.rr/rr*rr: staticmethodr rPr rZrrrrr$?s"  "r$)rr3rH dataclassesrenumrtypingrrrrrlogurur !pipecat.adapters.base_llm_adapterr (pipecat.adapters.schemas.function_schemar %pipecat.adapters.schemas.tools_schemar r *pipecat.processors.aggregators.llm_contextrrrrr r$rrrrs