qi@dZddlZej5ejdejdeddddddlZddlmZm Z ddl m Z dd l m Z dd lmZmZmZmZmZmZmZmZmZmZmZmZmZdd lmZdd lmZdd l m!Z!ddl"m#Z#m$Z$Gdde Z%eGddZ&eGddZ'GddeZ(eGddeZ)Gdde$Z*Gdde#Z+eGddZ,y#1swYxYw)aContext management for AWS Nova Sonic LLM service. This module provides specialized context aggregators and message handling for AWS Nova Sonic, including conversation history management and role-specific message processing. .. deprecated:: 0.0.91 AWS Nova Sonic no longer uses types from this module under the hood. It now uses ``LLMContext`` and ``LLMContextAggregatorPair``. Using the new patterns should allow you to not need types from this module. BEFORE:: # Setup context = OpenAILLMContext(messages, tools) context_aggregator = llm.create_context_aggregator(context) # Context frame type frame: OpenAILLMContextFrame # Context type context: AWSNovaSonicLLMContext # or context: OpenAILLMContext AFTER:: # Setup context = LLMContext(messages, tools) context_aggregator = LLMContextAggregatorPair(context) # Context frame type frame: LLMContextFrame # Context type context: LLMContext NalwaysaTypes in pipecat.services.aws.nova_sonic.context (or pipecat.services.aws_nova_sonic.context) are deprecated. AWS Nova Sonic no longer uses types from this module under the hood. It now uses `LLMContext` and `LLMContextAggregatorPair`. Using the new patterns should allow you to not need types from this module. BEFORE: ``` # Setup context = OpenAILLMContext(messages, tools) context_aggregator = llm.create_context_aggregator(context) # Context frame type frame: OpenAILLMContextFrame # Context type context: AWSNovaSonicLLMContext # or context: OpenAILLMContext ``` AFTER: ``` # Setup context = LLMContext(messages, tools) context_aggregator = LLMContextAggregatorPair(context) # Context frame type frame: LLMContextFrame # Context type context: LLMContext ```) stacklevel) dataclassfield)Enum)logger) BotStoppedSpeakingFrame DataFrameFrameFunctionCallResultFrameInterruptionFrameLLMFullResponseEndFrameLLMFullResponseStartFrameLLMMessagesAppendFrameLLMMessagesUpdateFrameLLMSetToolChoiceFrameLLMSetToolsFrame TextFrameUserImageRawFrame)OpenAILLMContext)FrameDirection)#AWSNovaSonicFunctionCallResultFrame) OpenAIAssistantContextAggregatorOpenAIUserContextAggregatorc 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__rrr r!Y/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/aws/nova_sonic/context.pyrrnsF DI Dr'rc&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)r"r#r$r%r__annotations__strr&r'r(r*r*~s J Ir'r*cBeZdZUdZdZeed<eeZ ee ed<y)AWSNovaSonicConversationHistoryzComplete conversation history for AWS Nova Sonic initialization. Parameters: system_instruction: System-level instruction for the conversation. messages: List of conversation messages between user and assistant. Nsystem_instruction)default_factorymessages) r"r#r$r%r1r.r-rlistr3r*r&r'r(r0r0s*#"=BSW=XHd9:Xr'r0ceZdZdZdfd ZddefdZedededdfdZ de fdZ fd Z de fd Zd Zdefd Zd ZdZxZS)AWSNovaSonicLLMContextaSpecialized LLM context for AWS Nova Sonic service. Extends OpenAI context with Nova Sonic-specific message handling, conversation history management, and text buffering capabilities. .. deprecated:: 0.0.99 `AWSNovaSonicLLMContext` is deprecated and will be removed in a future version. Use the universal `LLMContext` and `LLMContextAggregatorPair` instead. See `OpenAILLMContext` docstring for migration guide. c Jt|d||d||jy)zInitialize AWS Nova Sonic LLM context. Args: messages: Initial messages for the context. tools: Available tools for the context. **kwargs: Additional arguments passed to parent class. )r3toolsNr&)super__init__$_AWSNovaSonicLLMContext__setup_local)selfr3r8kwargs __class__s r(r:zAWSNovaSonicLLMContext.__init__s( B(%B6B r'r1c.d|_d|_||_y)N)_assistant_text _user_text_system_instruction)r<r1s r( __setup_localz$AWSNovaSonicLLMContext.__setup_locals!#5 r'objreturnc~t|tr,t|tst|_|j ||S)aUpgrade an OpenAI context to AWS Nova Sonic context. Args: obj: The OpenAI context to upgrade. system_instruction: System instruction for the context. Returns: The upgraded AWS Nova Sonic context. ) isinstancerr6r>r;)rEr1s r(upgrade_to_nova_sonicz,AWSNovaSonicLLMContext.upgrade_to_nova_sonics4 c+ ,ZE[5\2CM   0 1 r'c&t|j}|js|Stj|j}|dj ddk(r|j d}|j d}t|tr||_ n)t|tr|dj d|_ |jr|j|_|D]1}|j|}|s|jj|3|S)a;Get conversation history for initializing AWS Nova Sonic session. Processes stored messages and extracts system instruction and conversation history in the format expected by AWS Nova Sonic. Returns: Formatted conversation history with system instruction and messages. )r1rr+systemcontentr,) r0rCr3copydeepcopygetpoprHr.r1r4from_standard_messageappend)r<historyr3rKrLmessagehistory_messages r(%get_messages_for_initializing_historyzs r(rYz:AWSNovaSonicLLMContext.get_messages_for_persistent_storagesP 7>@  # #X(1+//&:QU]:] OOAT=U=UV Wr'c|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 assistantrLr@typer, z+Unhandled content type in context message: z - )r+r,N)rOrHr4r errorr*rupper)r<rTr+rLcs r(rQz,AWSNovaSonicLLMContext.from_standard_messages{{6" ;;v & (GKK,?;,Nkk),G'++i0$7 Y/AuuV}.3v#66 I!%%PV-X[\c[de =4 CU\cdd-Or'c`|xj|jr d|z c_y|z c_y)znBuffer user text for later flushing to context. Args: text: User text to buffer. r_N)rBr<r,s r(buffer_user_textz'AWSNovaSonicLLMContext.buffer_user_texts% Qtf:BdBr'cz|jsy|j}dd|dgd}d|_|j||S)zFlush buffered user text to context as a complete message. Returns: The flushed user text, or empty string if no text was buffered. r@r\r,r^r,rX)rB add_message)r< user_textrTs r(flush_aggregated_user_textz1AWSNovaSonicLLMContext.flush_aggregated_user_textsI OO !';<  !r'c.|xj|z c_y)zxBuffer assistant text for later flushing to context. Args: text: Assistant text to buffer. N)rArds r(buffer_assistant_textz,AWSNovaSonicLLMContext.buffer_assistant_text1s $r'ct|jsydd|jdgd}d|_|j|y)z?Flush buffered assistant text to context as a complete message.Nr]r,rgrXr@)rArh)r<rTs r(flush_aggregated_assistant_textz6AWSNovaSonicLLMContext.flush_aggregated_assistant_text:sA## !'1E1EFG  " !r')NN)r@)r"r#r$r%r:r.r; staticmethodrrIr0rVrYr*rQrerjrlrn __classcell__r>s@r(r6r6s  66  36 !$#7V#J e0Ve:CC$% "r'r6ceZdZUdZeed<y)AWSNovaSonicMessagesUpdateFramezFrame containing updated AWS Nova Sonic context. Parameters: context: The updated AWS Nova Sonic LLM context. contextN)r"r#r$r%r6r-r&r'r(rsrsGs $#r'rscDeZdZdZej fdedeffd ZxZS)!AWSNovaSonicUserContextAggregatoraContext aggregator for user messages in AWS Nova Sonic conversations. Extends the OpenAI user context aggregator to emit Nova Sonic-specific context update frames. .. deprecated:: 0.0.99 `AWSNovaSonicUserContextAggregator` is deprecated and will be removed in a future version. Use the universal `LLMContext` and `LLMContextAggregatorPair` instead. See `OpenAILLMContext` docstring for migration guide. frame directioncKt|||d{t|tr.|j t |j d{yy7C7w)zProcess frames and emit Nova Sonic-specific context updates. Args: frame: The frame to process. direction: The direction the frame is traveling. N)rt)r9 process_framerHr push_framers_contextr<rwrxr>s r(rzz/AWSNovaSonicUserContextAggregator.process_frame`sXg#E9555 e3 4//"A$--"XY Y Y 5 6 Zs!AAs r(rzBAWSNovaSonicAssistantContextAggregator.handle_function_call_resultsO g1%888 oo /U C^E\E\   9  s!AA1AA AA) r"r#r$r%r rrzr rrprqs@r(rrps- ::>:<  7N   r'rc>eZdZUdZeed<eed<defdZdefdZy)!AWSNovaSonicContextAggregatorPairaPair of user and assistant context aggregators for AWS Nova Sonic. .. deprecated:: 0.0.99 `AWSNovaSonicContextAggregatorPair` is deprecated and will be removed in a future version. Use the universal `LLMContext` and `LLMContextAggregatorPair` instead. See `OpenAILLMContext` docstring for migration guide. Parameters: _user: The user context aggregator. _assistant: The assistant context aggregator. _user _assistantrFc|jS)zmGet the user context aggregator. Returns: The user context aggregator instance. )rr<s r(r\z&AWSNovaSonicContextAggregatorPair.users zzr'c|jS)zwGet the assistant context aggregator. Returns: The assistant context aggregator instance. )rrs r(r]z+AWSNovaSonicContextAggregatorPair.assistants r'N) r"r#r$r%rvr-rr\r]r&r'r(rrs/  -,667Ar'r)-r%warningscatch_warnings simplefilterwarnDeprecationWarningrM dataclassesrrenumrlogurur pipecat.frames.framesr r r r rrrrrrrrr1pipecat.processors.aggregators.openai_llm_contextr"pipecat.processors.frame_processorr&pipecat.services.aws.nova_sonic.framesrpipecat.services.openai.llmrrrr*r0r6rsrvrrr&r'r(rs5#JX H(#HMM 6 ; D (O=V 4         Y Y  Yk"-k"^ $i$ $Z(CZ<9 -M9 x    s (DD