o Úó”iØ$ã@sPdZddlmZddlmZmZmZmZdededBfdd„Z Gd d „d ƒZ dS) z¨ Service loading and validation logic. This module handles loading services, validating configurations, and providing utilities for working with the service registry. é)ÚSERVICE_CONFIGS)ÚBotTypeÚServiceDefinitionÚServiceRegistryÚ ServiceTypeÚpackageÚreturnNcCs,d|vrd|vr| d¡d d¡dSdS)as Extract the extra name from a package string. Args: package: Package string like "pipecat-ai[deepgram]" or "pipecat-ai" Returns: The extra name (e.g., "deepgram") or None if no extra Examples: >>> extract_package_extra("pipecat-ai[deepgram]") 'deepgram' >>> extract_package_extra("pipecat-ai") None ú[ú]réN)Úsplit)r©r úW/home/ubuntu/.local/lib/python3.10/site-packages/pipecat_cli/registry/service_loader.pyÚextract_package_extrasrc @s6eZdZdZedeedededBfdd„ƒZede deefd d „ƒZ ed ededBfd d „ƒZ ed edeefdd„ƒZ ede eeeeBfdeefdd„ƒZed edefdd„ƒZededeefdd„ƒZe dde eeeeBfde eefdedeefdd„ƒZede eeeffdd„ƒZdS)Ú ServiceLoaderz:Handles loading and validating services from the registry.Ú service_listÚvaluerNcCs |D] }|j|kr |SqdS)z÷ Find a service definition by its value. Args: service_list: List of service definitions to search value: Service value to find Returns: Service definition or None if not found N)r)rrÚservicer r rÚget_service_by_value*s  ÿz"ServiceLoader.get_service_by_valueÚ service_typecCs,tjtjtjtjtjtjdœ}| |g¡S)zµ Get all services of a specific type. Args: service_type: Type of service to retrieve Returns: List of service definitions )Ú transportÚsttÚllmÚttsÚrealtime)rÚWEBRTC_TRANSPORTSÚTELEPHONY_TRANSPORTSÚ STT_SERVICESÚ LLM_SERVICESÚ TTS_SERVICESÚREALTIME_SERVICESÚget)rÚtype_mapr r rÚget_all_services_by_type=s û z&ServiceLoader.get_all_services_by_typeÚ service_valuecCs t |¡S)zÞ Get the initialization code for a service. Args: service_value: Service identifier (e.g., "deepgram_stt") Returns: Initialization code string or None if not found )rr!©r$r r rÚget_service_configQs z ServiceLoader.get_service_configcCstj |g¡S)z« Get import statements for a service. Args: service_value: Service identifier Returns: List of import statements )rÚIMPORTSr!r%r r rÚget_service_import^s z ServiceLoader.get_service_importÚservicesc Csìtƒ}| d¡| d¡d|vr@t|dtƒr|dn|dg}tjtj}|D]}t ||¡}|r?t |j ƒ}|r?| |¡q)tj tj tj tjtjdœ}| ¡D]"\}} ||vrs||rst | ||¡} | rst | j ƒ}|rs| |¡qQ|S)aJ Extract all package extras needed for selected services. Args: services: Dict mapping service type to service value(s) e.g., {"transports": ["daily"], "stt": "deepgram_stt", ...} Returns: Set of package extras (e.g., {"daily", "deepgram", "openai"}) ÚrunnerÚsileroÚ transports©rrrrÚvideo)ÚsetÚaddÚ isinstanceÚlistrrrrrrrrrrr ÚVIDEO_SERVICESÚitems) r)ÚextrasÚtransport_listÚall_transportsrÚ transport_defÚextraÚservice_type_maprrÚ service_defr r rÚextract_extras_for_servicesksD    ÿý ÿ   €ûÿ  €z)ServiceLoader.extract_extras_for_servicescCs,|tvrdStjtj}t ||¡rdSdS)zÈ Check if a service exists in the registry. Args: service_value: Service identifier to check Returns: True if service exists, False otherwise TF)rrrrrr)r$r7r r rÚvalidate_service_exists£s    z%ServiceLoader.validate_service_existsÚbot_typecCs |dkrtjS|dkrtjSgS)zÆ Get transport options based on bot type. Args: bot_type: Type of bot ("web" or "telephony") Returns: List of transport service definitions ÚwebÚ telephony)rrr)r>r r rÚget_transport_options¹s  z#ServiceLoader.get_transport_optionsr?ÚfeaturescCs ttjƒ}| tjd¡| tjd¡| tjd¡| tjd¡d|vrKt|dtƒr4|dn|dg}|D]}|tjvrJ| tj|¡q;dD]}||vrd||}|tjvrd| tj|¡qM| d¡rr| tjd¡| d¡r| tjd¡| d ¡rŒ| tjd ¡t|ƒS) aL Get all necessary import statements for selected services and features. Args: services: Dict mapping service type to service value(s) features: Dict of enabled features bot_type: Type of bot ("web" or "telephony") Returns: List of import statements ÚpipelineÚcontextr*Úvadr,r-Ú recordingÚ transcriptionÚ observability) r/rÚ BASE_IMPORTSÚupdateÚFEATURE_IMPORTSr1r2r'r!)r)rBr>Úimportsr6rrr$r r rÚget_imports_for_servicesÊs6   ÿý € €   z&ServiceLoader.get_imports_for_servicescCsŒggdœ}g}| tj¡| tj¡| tj¡| tj¡| tj¡|D]}|j}|tvr7|d  |¡|tj vrC|d  |¡q'|S)z¦ Find services that are defined but missing configs or imports. Returns: Dict with 'missing_configs' and 'missing_imports' lists )Úmissing_configsÚmissing_importsrNrO) Úextendrrrrr r3rrÚappendr')ÚmissingÚ all_servicesrr$r r rÚget_missing_servicesýs       €z"ServiceLoader.get_missing_services)r?)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú staticmethodr2rÚstrrrr#r&r(Údictr/r<Úboolr=rrArMrTr r r rr'sFÿÿ þ  (7ÿÿ ÿÿ þ2 r) rXÚ_configsrÚservice_metadatarrrrrZrrr r r rÚs