o ig @sdZddlmZmZmZddlmZddlmZddl m Z m Z m Z ddl mZmZGdddeZGd d d eZGd d d e ZGd dde ZdS)zGoogle RTVI processor and observer implementation. This module provides integration with Google's services through the RTVI framework, including models for search responses and an observer for handling Google-specific frame types. )ListLiteralOptional) BaseModel) FramePushed) RTVIObserverRTVIObserverParams RTVIProcessor)LLMSearchOriginLLMSearchResponseFramec@s6eZdZUdZeeed<eeed<eeed<dS)RTVISearchResponseMessageDataaData payload for search response messages in RTVI protocol. Parameters: search_result: The search result text, if available. rendered_content: The rendered content from the search, if available. origins: List of search result origins with metadata. search_resultrendered_contentoriginsN) __name__ __module__ __qualname____doc__rstr__annotations__rr rrP/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/google/rtvi.pyr s   r c@s:eZdZUdZdZeded<dZeded<eed<dS)RTVIBotLLMSearchResponseMessagezRTVI message for bot LLM search responses. Parameters: label: Always "rtvi-ai" for RTVI protocol messages. type: Always "bot-llm-search-response" for this message type. data: The search response data payload. zrtvi-ailabelzbot-llm-search-responsetypedataN) rrrrrrrrr rrrrr%s  rcsFeZdZdZdeffdd Zdeffdd Zdefd d Z Z S) GoogleRTVIObserverzRTVI observer for Google service integration. Extends the base RTVIObserver to handle Google-specific frame types, particularly LLM search response frames from Google services. rtvicst|dS)z{Initialize the Google RTVI observer. Args: rtvi: The RTVI processor to send messages through. N)super__init__)selfr __class__rrr:szGoogleRTVIObserver.__init__rcs<t|IdH|j}t|tr||IdHdSdS)zProcess frames being pushed through the pipeline. Handles Google-specific frames in addition to the base RTVI frame types. Args: data: Frame push event data containing frame and metadata. N)r on_push_frameframe isinstancer !_handle_llm_search_response_frame)r rr$r!rrr#Bs  z GoogleRTVIObserver.on_push_framer$cs0tt|j|j|jdd}||IdHdS)N)r rr)r)rr r rrsend_rtvi_message)r r$messagerrrr&Qsz4GoogleRTVIObserver._handle_llm_search_response_frame) rrrrr rrr#r r& __classcell__rrr!rr3s rc@s(eZdZdZdddeefddZdS)GoogleRTVIProcessorzaRTVI processor for Google service integration. Creates a specific Google RTVI Observer. N)paramsr+cKst|S)zCreates a new RTVI Observer. Args: params: Settings to enable/disable specific messages. **kwargs: Additional arguments passed to the observer. Returns: A new RTVI observer. )r)r r+kwargsrrrcreate_rtvi_observerbs z(GoogleRTVIProcessor.create_rtvi_observer)rrrrrrr-rrrrr*\sr*N)rtypingrrrpydanticrpipecat.observers.base_observerr"pipecat.processors.frameworks.rtvirrr pipecat.services.google.framesr r r rrr*rrrrs  )