o iz@svdZddlmZddlmZddlmZddlmZddl m Z ddl m Z eGdd d e j ZGd d d e Zd S) zFCerebras LLM service implementation using OpenAI-compatible interface.) dataclass)Optional)logger)OpenAILLMInvocationParams)BaseOpenAILLMService)OpenAILLMServicec@seZdZdZdS)CerebrasLLMSettingsz Settings for CerebrasLLMService.N)__name__ __module__ __qualname____doc__r r Q/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/cerebras/llm.pyrsrc sreZdZUdZeZeed<dddddededeed eeffd d Z dfd d Z de de fddZ ZS)CerebrasLLMServicezA service for interacting with Cerebras's API using the OpenAI-compatible interface. This service extends OpenAILLMService to connect to Cerebras's API endpoint while maintaining full compatibility with OpenAI's interface and functionality. _settingszhttps://api.cerebras.ai/v1N)base_urlmodelsettingsapi_keyrrrc sV|jdd}|dur|dd||_|dur||tjd|||d|dS)aInitialize the Cerebras LLM service. Args: api_key: The API key for accessing Cerebras's API. base_url: The base URL for Cerebras API. Defaults to "https://api.cerebras.ai/v1". model: The model identifier to use. Defaults to "gpt-oss-120b". .. deprecated:: 0.0.105 Use ``settings=CerebrasLLMService.Settings(model=...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. **kwargs: Additional keyword arguments passed to OpenAILLMService. z gpt-oss-120b)rNr)rrrr )Settings"_warn_init_param_moved_to_settingsr apply_updatesuper__init__)selfrrrrkwargsdefault_settings __class__r rr$s   zCerebrasLLMService.__init__c s&td|tj||fi|S)aCreate OpenAI-compatible client for Cerebras API endpoint. Args: api_key: The API key for authentication. If None, uses instance key. base_url: The base URL for the API. If None, uses instance URL. **kwargs: Additional arguments passed to the client constructor. Returns: An OpenAI-compatible client configured for Cerebras API. z"Creating Cerebras client with api )rdebugr create_client)rrrrrr rr Ks z CerebrasLLMService.create_clientparams_from_contextreturncCs|jjd|jj|jj|jj|jjd}||||jj|jjrH| dg}|r<|d ddkrs