qidZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZeGd d e ZGd d e Zy )zFCerebras LLM service implementation using OpenAI-compatible interface.) dataclass)Optional)logger)OpenAILLMInvocationParams)OpenAILLMSettings)OpenAILLMService)_warn_deprecated_paramceZdZdZy)CerebrasLLMSettingsz Settings for CerebrasLLMService.N)__name__ __module__ __qualname____doc__O/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/cerebras/llm.pyr r s*rr c veZdZUdZeZeed<dddddededeed eeffd Z dfd Z d e d e fdZ xZS)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 td}|tdtd||_||j|t |d|||d|y)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=OpenAILLMSettings(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)r r r apply_updatesuper__init__)selfrrrrkwargsdefault_settings __class__s rrzCerebrasLLMService.__init__%s_0/^D   "7,? I%*  "    ) )( 3 a8FVaZ`arc Vtjd|t| ||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)rrrrr!s rr$z CerebrasLLMService.create_clientLs0  9(DEw$WhA&AArparams_from_contextreturnc|jjd|jj|jj|jj|jj d}|j ||j |jj|jjr2|jdg}d|jjdg|z|d<|S)aBuild parameters for Cerebras chat completion request. Cerebras supports a subset of OpenAI parameters, focusing on core completion settings without advanced features like frequency/presence penalties. Args: params_from_context: Parameters, derived from the LLM context, to use for the chat completion. Contains messages, tools, and tool choice. Returns: Dictionary of parameters for the chat completion request. T)rstreamseed temperaturetop_pmax_completion_tokensmessagessystem)rolecontent) rrr)r*r+r,updateextrasystem_instructionget)rr%paramsr-s rbuild_chat_completion_paramsz/CerebrasLLMService.build_chat_completion_paramsZs^^))NN''>>55^^))%)^^%I%I    )* dnn**+ >> , ,zz*b1H!dnn.O.OP""F:  r)NN)r r rrr Settings__annotations__strrrr$rdictr6 __classcell__)r!s@rrrsx #H"" 5#26 %b%b %b } %b ./ %bN B#@Y#^b#rrN)r dataclassesrtypingrlogurur)pipecat.adapters.services.open_ai_adapterr pipecat.services.openai.base_llmrpipecat.services.openai.llmrpipecat.services.settingsr r rrrrrCsKM!O>8<  +   b)br