o i@sdZddlZddlmZmZmZmZddlmZ ddl m Z m Z m Z ddlmZddlmZddlmZmZmZmZmZGd d d eZGd d d eeZdS) zOpenAI LLM adapter for Pipecat.N)AnyDictList TypedDict)NotGiven)ChatCompletionMessageParam#ChatCompletionToolChoiceOptionParamChatCompletionToolParam)BaseLLMAdapter) ToolsSchema) LLMContextLLMContextMessageLLMContextToolChoiceLLMSpecificMessagerc@s:eZdZUdZeeed<eeeBed<e eBed<dS)OpenAILLMInvocationParamsz@Context-based parameters for invoking OpenAI ChatCompletion API.messagestools tool_choiceN) __name__ __module__ __qualname____doc__rr__annotations__r OpenAINotGivenrrr]/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/adapters/services/open_ai_adapter.pyrs  rc@seZdZdZedefddZdedefddZ de de e fd d Z dede eeeffd d Zd e ede efddZdeeBdeeBfddZdS)OpenAILLMAdapteraIOpenAI-specific adapter for Pipecat. Handles: - Extracting parameters for OpenAI's ChatCompletion API from a universal LLM context - Converting Pipecat's standardized tools schema to OpenAI's function-calling format. - Extracting and sanitizing messages from the LLM context for logging about OpenAI. returncCsdS)zCGet the identifier used in LLMSpecificMessage instances for OpenAI.openair)selfrrrid_for_llm_specific_messages1sz-OpenAILLMAdapter.id_for_llm_specific_messagescontextcCs"|||||j|jdS)aGet OpenAI-specific LLM invocation parameters from a universal LLM context. Args: context: The LLM context containing messages, tools, etc. Returns: Dictionary of parameters for OpenAI's ChatCompletion API. )rrr) _from_universal_context_messages get_messagesfrom_standard_toolsrr)rr!rrrget_llm_invocation_params6s z*OpenAILLMAdapter.get_llm_invocation_params tools_schemacCs|j}dd|DS)aConvert function schemas to OpenAI's function-calling format. Args: tools_schema: The Pipecat tools schema to convert. Returns: List of OpenAI formatted function call definitions ready for use with ChatCompletion API. cSsg|] }td|dqS)function)typer')r to_default_dict).0funcrrr Qsz=OpenAILLMAdapter.to_provider_tools_format..)standard_tools)rr&functions_schemarrrto_provider_tools_formatFs z)OpenAILLMAdapter.to_provider_tools_formatcCsg}||D]N}t|}d|vrAt|dtrA|dD]#}|ddkr4|dddr4d|dd<|ddkr@d|dd <qd |vrP|d d rPd|d <||q|S) aTGet messages from a universal LLM context in a format ready for logging about OpenAI. Removes or truncates sensitive data like image content for safe logging. Args: context: The LLM context containing messages. Returns: List of messages in a format ready for logging about OpenAI. contentr( image_urlurlz data:image/zdata:image/... input_audioz...data mime_typezimage/)r#copydeepcopy isinstancelist startswithappend)rr!msgsmessagemsgitemrrrget_messages_for_loggingVs         z)OpenAILLMAdapter.get_messages_for_loggingrcCs4g}|D]}t|tr||jq||q|SN)r8rr;r=)rrresultr=rrrr"qs   z1OpenAILLMAdapter._from_universal_context_messagesrcCs|SrAr)rrrrr_from_standard_tool_choice~sz+OpenAILLMAdapter._from_standard_tool_choiceN)rrrrpropertystrr r rr%r rr r/rrr@r rr"rrrrrCrrrrr&s"   r)rr6typingrrrr openai._typesrropenai.types.chatrrr !pipecat.adapters.base_llm_adapterr %pipecat.adapters.schemas.tools_schemar *pipecat.processors.aggregators.llm_contextr r rrrrrrrrs