o i?@sdZddlZddlZddlmZddlmZmZmZm Z ddl m Z m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZmZGd dde ZGdddeeZdS)z"Anthropic LLM adapter for Pipecat.N) dataclass)AnyDictList TypedDict) NOT_GIVENNotGiven) MessageParam)ToolUnionParam)logger)BaseLLMAdapter)FunctionSchema) ToolsSchema) LLMContextLLMContextMessageLLMSpecificMessageLLMStandardMessagec@s6eZdZUdZeeBed<eeed<ee ed<dS)AnthropicLLMInvocationParamsz:Context-based parameters for invoking Anthropic's LLM API.systemmessagestoolsN) __name__ __module__ __qualname____doc__strr__annotations__rr r rr_/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/adapters/services/anthropic_adapter.pyrs   rc@s eZdZdZedefddZdedede fddZ dede e ee ffd d ZeGd d d Zd e edefddZdedefddZdedefddZdedefddZde ede efddZedede ee ffddZdede e ee ffddZd S)!AnthropicLLMAdapterzAdapter for converting tool schemas to Anthropic's function-calling format. This adapter handles the conversion of Pipecat's standard function schemas to the specific format required by Anthropic's Claude models for function calling. returncCsdS)zFGet the identifier used in LLMSpecificMessage instances for Anthropic. anthropicrselfrrrid_for_llm_specific_messages-sz0AnthropicLLMAdapter.id_for_llm_specific_messagescontextenable_prompt_cachingcCs<|||}|j|r||jn|j||jpgdS)aRGet Anthropic-specific LLM invocation parameters from a universal LLM context. Args: context: The LLM context containing messages, tools, etc. enable_prompt_caching: Whether prompt caching should be enabled. Returns: Dictionary of parameters for invoking Anthropic's LLM API. )rrr) _from_universal_context_messages get_messagesr_with_cache_control_markersrfrom_standard_toolsr)r#r%r&rrrrget_llm_invocation_params2s z-AnthropicLLMAdapter.get_llm_invocation_paramscCs|||j}g}|D]9}t|}d|vrAt|dtrA|dD]}|ddkr1d|dd<|ddkr@|dr@d|d<q#||q |S) aZGet messages from a universal LLM context in a format ready for logging about Anthropic. 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 Anthropic. contenttypeimagez...sourcedatathinking signature) r'r(rcopydeepcopy isinstancelistgetappend)r#r%rmessages_for_loggingmessagemsgitemrrrget_messages_for_loggingJs      z,AnthropicLLMAdapter.get_messages_for_loggingc@s*eZdZUdZeeed<eeBed<dS)z%AnthropicLLMAdapter.ConvertedMessageszLContainer for Anthropic-formatted messages converted from universal context.rrN) rrrrrr rrrrrrrConvertedMessagesfs  r>universal_context_messagesc st}g}z fdd|D}Wnty)}ztd|WYd}~nd}~ww|rL|dddkrLt|dkrAd|dd<n |dd }|d|D] }|ddkrZd|d<qNd}|t|dkr||}||d}|d|dkrt|d trd |d d g|d <t|d trd |d d g|d <|d |d ||dn|d7}|t|dkse|D]+}t|d tr|d d krd |d <qt|d t rt|d dkrd d d g|d <qj ||dS)Ncg|]}|qSr)_from_universal_context_message).0mr"rr uzHAnthropicLLMAdapter._from_universal_context_messages..zError mapping messages: rroleruserr,textr-rI(empty))rr) r Exceptionr errorlenpopr5rextendr6r>) r#r?rrer:icurrent_message next_messagerr"rr'msL       z4AnthropicLLMAdapter._from_universal_context_messagesr:cCst|tr ||S||S)N)r5r _from_anthropic_specific_message_from_standard_message)r#r:rrrrAs   z3AnthropicLLMAdapter._from_universal_context_messagecCs\t|jtr(|jddkr(|jd}r(|jd}r(dd||dgdSt|jS) a'Convert LLMSpecificMessage to Anthropic format. Anthropic-specific messages may either be special thought messages that need to be handled in a special way, or messages already in Anthropic format. Args: message: Anthropic-specific message. r-thoughtrIr2 assistantr1)r-r1r2rFr,)r5r:dictr7r3r4)r#r:rIr2rrrrVs  z4AnthropicLLMAdapter._from_anthropic_specific_messagecCst|}|ddkrdd|d|ddgdS|d rK|d }d gd}|D]}|d }t|d }d |d|d|d}|d|q)|S|d}t|tr]|dkr[d}|St|tr |D]r} | ddkrw| ddkrwd| d<| ddkr| dd dr| dd} | dd dd} d| d<d| | ddd| d <| d=qe| dd d!rd| d<d| ddd"| d <| d=qe| dd} t d#| qed$d%t |D} d&d%t |D} t| dkr | r | d}| d}||kr ||}||||S)'aeConvert standard universal context message to Anthropic format. Handles conversion of text content, tool calls, and tool results. Empty text content is converted to "(empty)". Args: message: Message in standard universal context format. Returns: Message in Anthropic format. Examples: Input standard format:: { "role": "assistant", "tool_calls": [ { "id": "123", "function": {"name": "search", "arguments": '{"q": "test"}'} } ] } Output Anthropic format:: { "role": "assistant", "content": [ { "type": "tool_use", "id": "123", "name": "search", "input": {"q": "test"} } ] } rFtoolrH tool_result tool_call_idr,)r- tool_use_idr,rZ tool_callsrYfunction argumentstool_useidname)r-rdreinputrKrLr-rI image_urlurlzdata::rG;rr.base64,)r- media_typer0r/http)r-rhzUnsupported 'image_url': cS g|] \}}|ddkr|qS)r-r.rrBrSr<rrrrD7 z>AnthropicLLMAdapter._from_standard_message..cSrorJrrprrrrD8rq)r3r4r7jsonloadsr8r5rr6 startswithsplitr warning enumeraterOrPinsert)r#r:tcret tool_callrarb new_tool_user,r<rh mime_type image_indices text_indicesimg_idx first_txt_idx image_itemrrrrWsv '     ,           z*AnthropicLLMAdapter._from_standard_messagerc Csdtfdd}z7t|}g}tt|dddD]}||ddkr0||t|dkr0nq|D]}|||q3|WStyZ}ztd ||WYd }~Sd }~ww) zAdd cache control markers to messages for prompt caching. Args: messages: List of messages in Anthropic format. Returns: List of messages with cache control markers added. r:cSs:t|dtrd|ddg|d<ddi|ddd<dS)Nr,rIrJr- ephemeral cache_control)r5r)r:rrradd_cache_control_markerMszQAnthropicLLMAdapter._with_cache_control_markers..add_cache_control_markerrGrrFrHz#Error adding cache control marker: N) r r3r4rangerOr8rMr rN)r#rrmessages_with_markersuser_message_indicesrSindexrRrrrr)Cs$    z/AnthropicLLMAdapter._with_cache_control_markersracCs|j|jd|j|jddS)zConvert a single function schema to Anthropic's format. Args: function: The function schema to convert. Returns: Dictionary containing the function definition in Anthropic's format. object)r- propertiesrequired)re description input_schema)rerrr)rarrr_to_anthropic_function_formatrs z1AnthropicLLMAdapter._to_anthropic_function_format tools_schemacs|j}fdd|DS)zConvert function schemas to Anthropic's function-calling format. Args: tools_schema: The tools schema containing functions to convert. Returns: List of function definitions formatted for Anthropic's API. cr@r)r)rBfuncr"rrrDrEz@AnthropicLLMAdapter.to_provider_tools_format..)standard_tools)r#rfunctions_schemarr"rto_provider_tools_formats z,AnthropicLLMAdapter.to_provider_tools_formatN)rrrrpropertyrr$rboolrr+rrrr=rr>rr'r rArrVrrWr) staticmethodr rrrrrrrr&s4  ;#s/"r) rr3rr dataclassesrtypingrrrrr!rranthropic.types.message_paramr anthropic.types.tool_union_paramr logurur !pipecat.adapters.base_llm_adapterr (pipecat.adapters.schemas.function_schemar %pipecat.adapters.schemas.tools_schemar*pipecat.processors.aggregators.llm_contextrrrrrrrrrrs