qi8dZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZeGd d e ZGd d e Zy )zFFireworks AI service implementation using OpenAI-compatible interface.) dataclass)Optional)logger)OpenAILLMInvocationParams)OpenAILLMSettings)OpenAILLMService)_warn_deprecated_paramceZdZdZy)FireworksLLMSettingsz!Settings for FireworksLLMService.N)__name__ __module__ __qualname____doc__P/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/fireworks/llm.pyr r s+rr c veZdZUdZeZeed<dddddedeeded eeffd Z dfd Z d e d e fdZ xZS)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 td}|tdtd||_||j|t |d|||d|y)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=OpenAILLMSettings(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)r r r apply_updatesuper__init__)selfrrrrkwargsdefault_settings __class__s rrzFireworksLLMService.__init__%s`006ab   "7,@' J%*  "    ) )( 3 a8FVaZ`arc Vtjd|t| ||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)rrrrr!s rr$z!FireworksLLMService.create_clientLs0  :8*EFw$WhA&AArparams_from_contextreturnc|jjd|jj|jj|jj|jj |jj d}|j||j|jj|jjr2|jdg}d|jjdg|z|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_tokensmessagessystem)rolecontent) rrr)r*r+r,r-updateextrasystem_instructionget)rr%paramsr.s rbuild_chat_completion_paramsz0FireworksLLMService.build_chat_completion_paramsZs^^))!%!A!A $ ? ?>>55^^))..33   )* dnn**+ >> , ,zz*b1H!dnn.O.OP""F:  r)NN)r r rrr Settings__annotations__strrrr$rdictr7 __classcell__)r!s@rrrsx $H## $?37 %b%b} %b  %b /0 %bN B$@Y$^b$rrN)r dataclassesrtypingrlogurur)pipecat.adapters.services.open_ai_adapterr pipecat.services.openai.base_llmrpipecat.services.openai.llmrpipecat.services.settingsr r rrrrrDsKM!O>8<  ,   c*cr