o i+@sdZddlZddlmZddlmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZmZddlmZdd lmZeGd d d ZGd d d eZGdddeZGdddeZdS)z;OpenAI LLM service implementation with context aggregators.N) dataclass)AnyOptional) NOT_GIVEN)FunctionCallCancelFrameFunctionCallInProgressFrameFunctionCallResultFrameUserImageRawFrame)LLMAssistantAggregatorParamsLLMAssistantContextAggregatorLLMUserAggregatorParamsLLMUserContextAggregator)OpenAILLMContext)BaseOpenAILLMServicec@s6eZdZUdZded<ded<d ddZd d d Zd S)OpenAIContextAggregatorPairaPair of OpenAI context aggregators for user and assistant messages. .. deprecated:: 0.0.99 `OpenAIContextAggregatorPair` is deprecated and will be removed in a future version. Use the universal `LLMContext` and `LLMContextAggregatorPair` instead. See `OpenAILLMContext` docstring for migration guide. Parameters: _user: User context aggregator for processing user messages. _assistant: Assistant context aggregator for processing assistant messages. OpenAIUserContextAggregator_user OpenAIAssistantContextAggregator _assistantreturncC|jS)zmGet the user context aggregator. Returns: The user context aggregator instance. )rselfrO/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/openai/llm.pyuser1z OpenAIContextAggregatorPair.usercCr)zwGet the assistant context aggregator. Returns: The assistant context aggregator instance. )rrrrr assistant9rz%OpenAIContextAggregatorPair.assistantN)rr)rr)__name__ __module__ __qualname____doc____annotations__rrrrrrrs   rc s|eZdZdZejZddddddeedeedeejdeeffdd Z e e d d e d e d e de fddZZS)OpenAILLMServicezOpenAI LLM service implementation. Provides a complete OpenAI LLM service with context aggregation support. Uses the BaseOpenAILLMService for core functionality and adds OpenAI-specific context aggregator creation. N)model service_tierparamssettingsr$r%r&r'c s|jddtttttdttddid }|dur|dd||_|dur/|s/|jtur/|p.|j}|dur`|d|s`|j|_|j|_|j|_|j|_|j |_ |j |_ |j |_ t |j tr`|j |_ |duri||tjd||d|dS) aInitialize OpenAI LLM service. Args: model: The OpenAI model name to use. Defaults to "gpt-4.1". .. deprecated:: 0.0.105 Use ``settings=OpenAILLMService.Settings(model=...)`` instead. service_tier: Service tier to use (e.g., "auto", "flex", "priority"). params: Input parameters for model configuration. .. deprecated:: 0.0.105 Use ``settings=OpenAILLMService.Settings(...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. **kwargs: Additional arguments passed to the parent BaseOpenAILLMService. zgpt-4.1NF) r$system_instructionfrequency_penaltypresence_penaltyseed temperaturetop_ptop_k max_tokensmax_completion_tokensfilter_incomplete_user_turnsuser_turn_completion_configextrar$r&)r%r'r)Settingsr"_warn_init_param_moved_to_settingsr$r%r)r*r+r,r-r/r0 isinstancer3dict apply_updatesuper__init__)rr$r%r&r'kwargsdefault_settings __class__rrr:LsF     zOpenAILLMService.__init__) user_paramsassistant_paramscontextr?r@rcCs2||t||d}t||d}t||dS)aCreate OpenAI-specific context aggregators. Creates a pair of context aggregators optimized for OpenAI's message format, including support for function calls, tool usage, and image handling. Args: context: The LLM context to create aggregators for. user_params: Parameters for user message aggregation. assistant_params: Parameters for assistant message aggregation. Returns: OpenAIContextAggregatorPair: A pair of context aggregators, one for the user and one for the assistant, encapsulated in an OpenAIContextAggregatorPair. .. deprecated:: 0.0.99 `create_context_aggregator()` is deprecated and will be removed in a future version. Use the universal `LLMContext` and `LLMContextAggregatorPair` instead. See `OpenAILLMContext` docstring for migration guide. )r&)rr)set_llm_adapterget_llm_adapterrrr)rrAr?r@rrrrrcreate_context_aggregators   z*OpenAILLMService.create_context_aggregator)rrr r!rr4rstr InputParamsr:r r rrrD __classcell__rrr=rr#Bs6Mr#c@seZdZdZdS)raOpenAI-specific user context aggregator. Handles aggregation of user messages for OpenAI LLM services. Inherits all functionality from the base LLMUserContextAggregator. .. deprecated:: 0.0.99 `OpenAIUserContextAggregator` is deprecated and will be removed in a future version. Use the universal `LLMContext` and `LLMContextAggregatorPair` instead. See `OpenAILLMContext` docstring for migration guide. N)rrr r!rrrrrs rc@s^eZdZdZdefddZdefddZdefddZ d e d e d e fd d Z de fddZdS)raOpenAI-specific assistant context aggregator. Handles aggregation of assistant messages for OpenAI LLM services, with specialized support for OpenAI's function calling format, tool usage tracking, and image message handling. .. deprecated:: 0.0.99 `OpenAIAssistantContextAggregator` is deprecated and will be removed in a future version. Use the universal `LLMContext` and `LLMContextAggregatorPair` instead. See `OpenAILLMContext` docstring for migration guide. framecsJ|jd|j|jt|jdddgd|jdd|jdd S) zHandle a function call in progress. Adds the function call to the context with an IN_PROGRESS status to track ongoing function execution. Args: frame: Frame containing function call progress information. r)name argumentsfunction)idrKtype)role tool_callstool IN_PROGRESS)rNcontent tool_call_idN)_context add_messagerS function_namejsondumpsrJrrHrrr handle_function_call_in_progresss$  zAOpenAIAssistantContextAggregator.handle_function_call_in_progresscsP|jrtj|jdd}||j|j|IdHdS||j|jdIdHdS)zHandle the result of a function call. Updates the context with the function call result, replacing any previous IN_PROGRESS status. Args: frame: Frame containing the function call result. F) ensure_asciiN COMPLETED)resultrWrX_update_function_call_resultrVrS)rrHr]rrrhandle_function_call_results  zs    "w