o i{ @sdZddlZddlmZmZmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZmZddlmZGd d d eZdS) 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. N)AnyAsyncGeneratorDict)logger) CancelFrameEndFrame ErrorFrameFrame StartFrame) MetricsData)FrameDirectionFrameProcessor)ServiceSettingscseZdZdZd'dedBffdd ZddZdefd d Zde fd d Z de fd dZ dede eeffddZ  d(dededBdefddZddZdedeffdd ZdeedBdffdd Zdefd!d"Zde fd#d$Zde fd%d&ZZS)) 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 s<tjdi||p t|_|i|_d|_d|_dS)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__rO/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/ai_service.pyr's zAIService.__init__cCs |t|j|jjp dddS)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 )rrrrr:s z%AIService._sync_model_name_to_metricsframecs |j|j|_|j|_dS)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#rrrstartHs  zAIService.startcdS)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'rrrstopU zAIService.stopcr))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'rrrcancel`r+zAIService.canceldeltareturncs@|j|}d|vr||rt|jdt||S)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)rr-changedrrr_update_settingsks zAIService._update_settings param_namesettings_field stacklevelcCst|jd}|rd|d|d|d}n d|d|d}ttdtj|t|dWd d S1s=wYd S) aWarn that an ``__init__`` param has moved to ``Settings``. Emits a ``DeprecationWarning`` directing users to the canonical ``settings=ServiceClass.Settings(field=...)`` API. Args: param_name: Name of the deprecated ``__init__`` parameter. settings_field: The corresponding field on the ``Settings`` dataclass, if different from *param_name*. When ``None`` the message omits the field hint. stacklevel: Stack depth for the warning. Default ``3`` targets the caller's caller (i.e. user code that instantiated the service). z .SettingszThe `z)` parameter is deprecated. Use `settings=(zB=...)` instead. If both are provided, `settings` takes precedence.zB(...)` instead. If both are provided, `settings` takes precedence.always)r7N)type__name__warningscatch_warnings simplefilterwarnDeprecationWarning)rr5r6r7labelmsgrrr"_warn_init_param_moved_to_settingss   "z,AIService._warn_init_param_moved_to_settingscCs2|rdt|}t|jd|ddSdS)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 unhandledfieldsrrr _warn_unhandled_updated_settingss z*AIService._warn_unhandled_updated_settings directioncstt||IdHt|tr||IdHdSt|tr)||IdHdSt|tr8||IdHdSdS)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#rJrrrrKs   zAIService.process_frame generatorcsJ|2z3dHW}|r!t|tr||IdHq||IdHq6dS)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)rLrpush_error_frame push_frame)rrPfrrrprocess_generators  zAIService.process_generatorc Zz ||IdHWdSty,}zt|d|d|WYd}~dSd}~wwNz: exception processing z: )r( Exceptionrerrorrr#errrrM(zAIService._startc rUrV)r*rWrrXrYrrrrNr[zAIService._stopc rUrV)r,rWrrXrYrrrrOr[zAIService._cancel)N)Nr4)r; __module__ __qualname____doc__rrrr r(rr*rr,rstrrr3intrCrIr r rKrrTrMrNrO __classcell__rrrrrs.    % r)r^r<typingrrrlogururpipecat.frames.framesrrrr r pipecat.metrics.metricsr "pipecat.processors.frame_processorr r pipecat.services.settingsrrrrrrs