qig dZddlmZmZmZddlmZddlmZddl m Z m Z m Z ddl mZmZGddeZGd d eZGd d e ZGd de Zy)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)LLMSearchOriginLLMSearchResponseFramecBeZdZUdZeeed<eeed<eeed<y)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 N/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/google/rtvi.pyrrs(C= sm# / ""rrcDeZdZUdZdZeded<dZeded<eed<y)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) rrrrrrrrrrrrrr%s.!*E79 )/HD'+ ,H ''rrcDeZdZdZdeffd Zdeffd ZdefdZ xZ 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. rtvic$t||y)z{Initialize the Google RTVI observer. Args: rtvi: The RTVI processor to send messages through. N)super__init__)selfr" __class__s rr%zGoogleRTVIObserver.__init__:s rrcKt||d{|j}t|tr|j |d{yy7;7w)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's rr)z GoogleRTVIObserver.on_push_frameBsUg#D)))  e3 488? ? ? 5 * @s!AA4A A AAr*cKtt|j|j|j}|j |d{y7w)N)rrr)r)rrrrrsend_rtvi_message)r&r*messages rr,z4GoogleRTVIObserver._handle_llm_search_response_frameQsG1.#11 !&!7!7 $$W---sA A AA) rrrrr r%rr)r r, __classcell__)r's@rr!r!3s/ ] @ @.=S.rr!c(eZdZdZdddeefdZy)GoogleRTVIProcessorzaRTVI processor for Google service integration. Creates a specific Google RTVI Observer. N)paramsr3c t|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&r3kwargss rcreate_rtvi_observerz(GoogleRTVIProcessor.create_rtvi_observerbs"$''r)rrrrrr r6rrrr2r2\s  NR (h7I.J (rr2N)rtypingrrrpydanticrpipecat.observers.base_observerr"pipecat.processors.frameworks.rtvirr r pipecat.services.google.framesr r rrr!r2rrrr<sR+*7^^R #I # (i (&.&.R(-(r