qi\dZddlmZmZmZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZmZddlmZGdd eZy ) zBase AI service implementation. Provides the foundation for all AI services in the Pipecat framework, including model management, settings handling, and frame processing lifecycle methods. )AnyAsyncGeneratorDict)logger) CancelFrameEndFrame ErrorFrameFrame StartFrame) MetricsData)FrameDirectionFrameProcessor)ServiceSettingsceZdZdZddedzffd ZdZdefdZde fdZ de fd Z d ed e eeffd Zd Zdedeffd ZdeedzdffdZdefdZde fdZde fdZxZS) AIServicea'Base class for all AI services. Provides common functionality for AI services including model management, settings handling, session properties, and frame processing lifecycle. Subclasses should implement specific AI functionality while leveraging this base infrastructure. Nsettingsc t|di||xs t|_|j i|_d|_d|_y)zInitialize the AI service. Args: settings: The runtime-updatable settings for the AI service. **kwargs: Additional arguments passed to the parent FrameProcessor. FN)super__init__r _settings_sync_model_name_to_metrics_session_properties_tracing_enabled_tracing_context)selfrkwargs __class__s M/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/ai_service.pyrzAIService.__init__&sR "6"  !  ((*35 &+ $c|jt|j|jjxsdy)akSync the current AI model name (in `self._settings.model`) for usage in metrics. We don't store model name here because there's already a single source of truth for it in `self._settings.model`. This method is just for syncing the model name to the metrics data. Args: model: The name of the AI model to use. ) processormodelN)set_core_metrics_datar namerr$)rs rrz%AIService._sync_model_name_to_metrics9s/ "" $))4>>3G3G3M2 N r framecK|jj|j|_|j|_yw)a Start the AI service. Called when the service should begin processing. Subclasses should override this method to perform service-specific initialization. Args: frame: The start frame containing initialization parameters. N)rvalidate_completeenable_tracingrtracing_contextrrr's rstartzAIService.startGs4 ((* % 4 4 % 5 5s>Ac Kyw)zStop the AI service. Called when the service should stop processing. Subclasses should override this method to perform cleanup operations. Args: frame: The end frame. Nrr,s rstopzAIService.stopT  c Kyw)zCancel the AI service. Called when the service should cancel all operations. Subclasses should override this method to handle cancellation logic. Args: frame: The cancel frame. Nrr,s rcancelzAIService.cancel_r0r1deltareturncK|jj|}d|vr|j|r-tj|j dt ||Sw)agApply a settings delta and return the changed fields. The delta is applied to ``_settings`` and a dict mapping each changed field name to its **pre-update** value is returned. The ``model`` field is handled specially: when it changes, ``set_model_name`` is called. Concrete services should override this method (calling ``super()``) to react to specific changed fields (e.g. reconnect on voice change). Args: delta: A delta-mode settings object. Returns: Dict mapping changed field names to their previous values. r$z: updated settings fields: )r apply_updaterrinfor&set)rr4changeds r_update_settingszAIService._update_settingsjsX"..--e4 g   , , .  KK499+%@WO PsA!A#c|r@djt|}tj|jd|dyy)a;Log a warning for settings changes that won't take effect at runtime. Convenience helper for ``_update_settings`` overrides. Accepts any iterable of field names (a ``dict``, ``set``, ``dict_keys``, etc.). Args: unhandled: Field names that changed but are not applied. z, z: runtime update of [z] is not currently supportedN)joinsortedrwarningr&)r unhandledfieldss r _warn_unhandled_updated_settingsz*AIService._warn_unhandled_updated_settingss> YYvi01F NNdii[(=fXEab c r directioncJKt|||d{t|tr|j |d{yt|t r|j |d{yt|tr|j|d{yy77\747 w)a"Process frames and handle service lifecycle. Automatically handles StartFrame, EndFrame, and CancelFrame by calling the appropriate lifecycle methods. Args: frame: The frame to process. direction: The direction of frame processing. N) r process_frame isinstancer _startr_stopr_cancel)rr'rCrs rrEzAIService.process_framesg#E9555 eZ (++e$ $ $ x (**U# # # { +,,u% % %, 6 % # %sEB#B(B#B)B#*B+)B#B!B#B#B#!B# generatorcK|23d{}|s t|tr|j|d{6|j|d{P7K7!7 6yw)aProcess frames from an async generator. Takes an async generator that yields frames and processes each one, handling error frames specially by pushing them as errors. Args: generator: An async generator that yields Frame objects or None. N)rFr push_error_frame push_frame)rrJfs rprocess_generatorzAIService.process_generators_! - -!a,//222//!,,,  -3, !sLAAAAA$AAAAAAAAAcK |j|d{y7#t$r'}tj|d|d|Yd}~yd}~wwxYwwNz: exception processing z: )r- Exceptionrerrorrr'es rrGzAIService._startsP G**U# # # G LLD6!8r!E F F G6A  A AA A AAcK |j|d{y7#t$r'}tj|d|d|Yd}~yd}~wwxYwwrQ)r/rRrrSrTs rrHzAIService._stopsP G))E" " " G LLD6!8r!E F F GrVcK |j|d{y7#t$r'}tj|d|d|Yd}~yd}~wwxYwwrQ)r3rRrrSrTs rrIzAIService._cancelsP G++e$ $ $ G LLD6!8r!E F F GrV)N)__name__ __module__ __qualname____doc__rrrr r-rr/rr3rstrrr;rBr r rErrOrGrHrI __classcell__)rs@rrrs%4!7%&   6 6      +  OS#X6 d&&>&&- d@R1S- G*G GG G;Gr rN)r\typingrrrlogururpipecat.frames.framesrrr r r pipecat.metrics.metricsr "pipecat.processors.frame_processorr rpipecat.services.settingsrrrr rres: -,0M5hGhGr