o i@svdZddlmZddlmZddlmZddlmZddl m Z ddl m Z eGdd d e j ZGd d d e Zd S) zFFireworks AI service implementation using OpenAI-compatible interface.) dataclass)Optional)logger)OpenAILLMInvocationParams)BaseOpenAILLMService)OpenAILLMServicec@seZdZdZdS)FireworksLLMSettingsz!Settings for FireworksLLMService.N)__name__ __module__ __qualname____doc__r r R/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/fireworks/llm.pyrsrc sreZdZUdZeZeed<dddddedeeded eeffd d Z dfd d Z de de fddZ ZS)FireworksLLMServicezA service for interacting with Fireworks AI using the OpenAI-compatible interface. This service extends OpenAILLMService to connect to Fireworks' API endpoint while maintaining full compatibility with OpenAI's interface and functionality. _settingsNz%https://api.fireworks.ai/inference/v1)modelbase_urlsettingsapi_keyrrrc sV|jdd}|dur|dd||_|dur||tjd|||d|dS)aInitialize the Fireworks LLM service. Args: api_key: The API key for accessing Fireworks AI. model: The model identifier to use. Defaults to "accounts/fireworks/models/firefunction-v2". .. deprecated:: 0.0.105 Use ``settings=FireworksLLMService.Settings(model=...)`` instead. base_url: The base URL for Fireworks API. Defaults to "https://api.fireworks.ai/inference/v1". settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. **kwargs: Additional keyword arguments passed to OpenAILLMService. z)accounts/fireworks/models/firefunction-v2)rNr)rrrr )Settings"_warn_init_param_moved_to_settingsr apply_updatesuper__init__)selfrrrrkwargsdefault_settings __class__r rr$s   zFireworksLLMService.__init__c s&td|tj||fi|S)aCreate OpenAI-compatible client for Fireworks API endpoint. Args: api_key: API key for authentication. If None, uses instance default. base_url: Base URL for the API. If None, uses instance default. **kwargs: Additional arguments passed to the client constructor. Returns: Configured OpenAI client instance for Fireworks API. z#Creating Fireworks client with api )rdebugr create_client)rrrrrr rr Ks z!FireworksLLMService.create_clientparams_from_contextreturncCs|jjd|jj|jj|jj|jj|jjd}||||jj|jj rK| dg}|r?|d ddkr?t |dd|jj dg||d<|S) aBuild parameters for Fireworks chat completion request. Fireworks doesn't support some OpenAI parameters like seed, max_completion_tokens, and stream_options. 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)rstreamfrequency_penaltypresence_penalty temperaturetop_p max_tokensmessagesrrolesystemzc: Both system_instruction and an initial system message in context are set. This may be unintended.)r*content) rrr$r%r&r'r(updateextrasystem_instructiongetrwarning)rr!paramsr)r r rbuild_chat_completion_paramsYs*   z0FireworksLLMService.build_chat_completion_params)NN)r r r r rr__annotations__strrrr rdictr3 __classcell__r r rrrs$ 'rN)r dataclassesrtypingrlogurur)pipecat.adapters.services.open_ai_adapterr pipecat.services.openai.base_llmrpipecat.services.openai.llmrrrrr r r rs