qidZddlZddlmZmZmZmZddlmZ ddl m Z m Z m Z ddlmZddlmZddlmZmZmZmZmZGd d eZGd d eeZy) zOpenAI LLM adapter for Pipecat.N)AnyDictList TypedDict)NotGiven)ChatCompletionMessageParam#ChatCompletionToolChoiceOptionParamChatCompletionToolParam)BaseLLMAdapter) ToolsSchema) LLMContextLLMContextMessageLLMContextToolChoiceLLMSpecificMessagercHeZdZUdZeeed<eeezed<e ezed<y)OpenAILLMInvocationParamsz@Context-based parameters for invoking OpenAI ChatCompletion API.messagestools tool_choiceN) __name__ __module__ __qualname____doc__rr__annotations__r OpenAINotGivenr [/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/adapters/services/open_ai_adapter.pyrrs-J-.. ' (> 994~EErrceZdZdZedefdZdedefdZ de de e fdZ dede eeeffdZd e ede efd Zd eezdeezfd Zy )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. returncy)zCGet the identifier used in LLMSpecificMessage instances for OpenAI.openair)selfs rid_for_llm_specific_messagesz-OpenAILLMAdapter.id_for_llm_specific_messages1srcontextc|j|j||j|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)r$r&s rget_llm_invocation_paramsz*OpenAILLMAdapter.get_llm_invocation_params6sC==d>O>OPW>XY--gmm<"..   r tools_schemacv|j}|Dcgc]}td|jc}Scc}w)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. function)typer.)standard_toolsr to_default_dict)r$r,functions_schemafuncs rto_provider_tools_formatz)OpenAILLMAdapter.to_provider_tools_formatFsB(66)  $d>R>R>T U   s"6cng}|j|D]}tj|}d|vrUt|dtrB|dD]:}|ddk(r|ddj drd|dd<|ddk(s3d|dd <<d |vr|d j d rd|d <|j ||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)r$r&msgsmessagemsgitems rget_messages_for_loggingz)OpenAILLMAdapter.get_messages_for_loggingVs((1 G--(CCc)nd3 #I@<;6#K07BB=Q;K[ 1% 8<=8:?D/7 @ c!c+&6&A&A(&K#F KK   rrcg}|D]?}t|tr|j|j/|j|A|SN)r>rrArC)r$rresultrCs rr(z1OpenAILLMAdapter._from_universal_context_messagesqsH 'G'#56 goo. g&  ' rrc|SrHr)r$rs r_from_standard_tool_choicez+OpenAILLMAdapter._from_standard_tool_choice~s rN)rrrrpropertystrr%r rr+r rr r4rrrFrrr(rrr rrKrrrr r &sc  @Y  [ TJaEb  tDcN?S6 ./  ( ) /(: ,~ =rr )rr<typingrrrr openai._typesrropenai.types.chatrr r !pipecat.adapters.base_llm_adapterr %pipecat.adapters.schemas.tools_schemar *pipecat.processors.aggregators.llm_contextr rrrrr rrrrTsT& --4 ==F F\~&?@\r