qi+dZddlZddlmZddlmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZmZddlmZdd lmZmZdd lmZeGd d ZGd deZGddeZGddeZy)z;OpenAI LLM service implementation with context aggregators.N) dataclass)AnyOptional) NOT_GIVEN)FunctionCallCancelFrameFunctionCallInProgressFrameFunctionCallResultFrameUserImageRawFrame)LLMAssistantAggregatorParamsLLMAssistantContextAggregatorLLMUserAggregatorParamsLLMUserContextAggregator)OpenAILLMContext)BaseOpenAILLMServiceOpenAILLMSettings)_warn_deprecated_paramc6eZdZUdZded<ded<d dZd dZy) 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 _assistantc|jS)zmGet the user context aggregator. Returns: The user context aggregator instance. )rselfs M/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/openai/llm.pyuserz OpenAIContextAggregatorPair.user2s zzc|jS)zwGet the assistant context aggregator. Returns: The assistant context aggregator instance. )rrs r assistantz%OpenAIContextAggregatorPair.assistant:s rN)returnr)r!r)__name__ __module__ __qualname____doc____annotations__rr rrrr s  )(22rrc eZdZdZddddddeedeedeejdeeffdZ e e d d e d e d e d e fdZxZS)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 tddtttttdttddi }|tdtd||_|$|s"|jtur|xs |j}|tdt|s|j |_|j |_|j|_|j|_|j|_ |j|_ |j|_ t|jtr|j|_ ||j|t!|Dd||d|y) aInitialize OpenAI LLM service. Args: model: The OpenAI model name to use. Defaults to "gpt-4.1". .. deprecated:: 0.0.105 Use ``settings=OpenAILLMSettings(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=OpenAILLMSettings(...)`` 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')rrrr*r+r0r1r2r3r4r6r7 isinstancer:dict apply_updatesuper__init__)rr*r+r,r-kwargsdefault_settings __class__s rr?zOpenAILLMService.__init__KsA8-#'&! "+).(, "   "7,=w G%*  "  h63F3Fi3W'>6+>+>L   "8-> ?5;5M5M 24:4K4K 1(.  %/5/A/A ,)/ &.4.?.? +9?9U9U 6fllD1-3\\$*    ) )( 3 Xl=MXQWXr) user_paramsassistant_paramscontextrCrDr!c|j|jt||}t||}t ||S)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)rrErCrDrr s rcreate_context_aggregatorz*OpenAILLMService.create_context_aggregatorsC6  4 4 67+7;G4WEUV *)LLr)r"r#r$r%rstrr InputParamsrr?r r rrrI __classcell__)rBs@rr)r)Cs $&*=A04 GY}GYsm GY -99: GY ,- GYZ0G/H9U9W !M!!M- !M 7 !M % !Mrr)ceZdZdZy)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)r"r#r$r%r'rrrrs   rrcTeZdZdZdefdZdefdZdefdZ de de de fd Z de fd Zy ) 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. framec K|jjd|j|jt j |j dddgd|jjdd|jdy w) 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)idrStype)role tool_callstool IN_PROGRESS)rVcontent tool_call_idN)_context add_messager[ function_namejsondumpsrRrrOs r handle_function_call_in_progresszAOpenAIAssistantContextAggregator.handle_function_call_in_progresss !!#$00$)$7$7)-EOO)D%!+     !!( % 2 2  sBBc(K|jrOtj|j}|j|j|j |d{y|j|j|j dd{y757w)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. N COMPLETED)resultr_r`_update_function_call_resultr^r[)rrOres rhandle_function_call_resultzr~sB !   OT<  DrM+rMj  ":  h 'Dh r