qi?dZddlZddlZddlmZddlmZmZmZm Z ddl m Z m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZmZGd de ZGddeeZy)z"Anthropic LLM adapter for Pipecat.N) dataclass)AnyDictList TypedDict) NOT_GIVENNotGiven) MessageParam)ToolUnionParam)logger)BaseLLMAdapter)FunctionSchema) ToolsSchema) LLMContextLLMContextMessageLLMSpecificMessageLLMStandardMessagecBeZdZUdZeezed<eeed<ee ed<y)AnthropicLLMInvocationParamsz:Context-based parameters for invoking Anthropic's LLM API.systemmessagestoolsN) __name__ __module__ __qualname____doc__strr __annotations__rr r ]/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/adapters/services/anthropic_adapter.pyrrs%D (N<   r rceZdZdZedefdZdedede fdZ dede e ee ffdZeGdd Zd e edefd Zd edefd Zd edefdZd edefdZde ede efdZedede ee ffdZdede e ee ffdZy)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. returncy)zFGet the identifier used in LLMSpecificMessage instances for Anthropic. anthropicr)selfs r!id_for_llm_specific_messagesz0AnthropicLLMAdapter.id_for_llm_specific_messages-sr contextenable_prompt_cachingc|j|j|}|j|r|j|jn |j|j |j xsgdS)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*rs r!get_llm_invocation_paramsz-AnthropicLLMAdapter.get_llm_invocation_params2sm889J9J79SToo)001B1BC&&--gmm<B  r cZ|j|j|j}g}|D]y}tj|}d|vrMt |dt r:|dD]2}|ddk(rd|dd<|ddk(s|jds.d|d<4|j|{|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_loggingmessagemsgitems r!get_messages_for_loggingz,AnthropicLLMAdapter.get_messages_for_loggingJs889J9J79ST]] " -G--(CCc)nd3 #I6<725:DN62<:5$((;:O05D- 6 ! ' ' , -$#r c2eZdZUdZeeed<eezed<y)%AnthropicLLMAdapter.ConvertedMessageszLContainer for Anthropic-formatted messages converted from universal context.rrN) rrrrrr rrr rr r!ConvertedMessagesrEfsZ|$$hr rFuniversal_context_messagesct}g} |Dcgc]}|j|}}|r;|dddk(r0t |dk(r d|dd<n|dd}|j d|D]}|ddk(s d|d<d}|t |dz kr||}||dz} |d| dk(rjt|dtr d|dd g|d<t| dtr d| dd g| d<|dj| d|j |dzn|dz }|t |dz kr|D]R}t|dtr|dd k(rd |d<$t|dts8t |ddk(sJdd d g|d<T|j|| Scc}w#t$r#}tjd|Yd}~d}~wwxYw) NzError mapping messages: rroleruserr2textr3rL(empty))rr) r_from_universal_context_message Exceptionr errorlenpopr;rextendr<rF) r'rGrrmer@icurrent_message next_messages r!r,z4AnthropicLLMAdapter._from_universal_context_messagesms) 9IcdA<>oi8#>!'1KL2OI.l95s;8> U^H_/`.aL+ *11,y2IJ QU#Q!#h-!##&  KG'),c2wy7IR7O%. "GI.5#gi>P:QUV:V/5y&I%J "  K %%x%GGce 9 LL3A37 8 8 9s'FFFF GF<<Gr@cft|tr|j|S|j|S)N)r;r _from_anthropic_specific_message_from_standard_message)r'r@s r!rPz3AnthropicLLMAdapter._from_universal_context_messages/ g1 288A A**733r c8t|jtrb|jjddk(rD|jjdx}r'|jjdx}r dd||dgdSt j |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. r3thoughtrLr8 assistantr7)r3r7r8rIr2)r;r@dictr=r9r:)r'r@rLr8s r!r\z4AnthropicLLMAdapter._from_anthropic_specific_messages w -##F+y8 ,,V444%oo11+>>>$!+$(%.  }}W__--r ctj|}|ddk(rdd|d|ddgdS|jd rQ|d }d gd}|D]@}|d }tj|d }d |d|d|d}|dj |B|S|jd}t |tr |dk(rd}|St |trj|D]} | ddk(r | ddk(rd| d<| ddk(s!| ddjdrR| dd} | jddjdd} d| d<d| | jddd| d <| d=| ddjd!rd| d<d| ddd"| d <| d=| dd} tjd#| t|D cgc]\} } | ddk(s| } } } t|D cgc]\} } | ddk(s| }} } t| dk(r4|r2| d}|d}||kDr#|j|}|j!|||Scc} } wcc} } w)$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"} } ] } rItoolrK tool_result tool_call_idr2)r3 tool_use_idr2ra tool_callsr`function argumentstool_useidname)r3rlrminputrNrOr3rL image_urlurlzdata::rJ;rr4base64,)r3 media_typer6r5http)r3rpzUnsupported 'image_url': )r9r:r=jsonloadsr>r;rr< startswithsplitr warning enumeraterSrTinsert)r'r@tcret tool_callrirj new_tool_user2rBrp mime_typerX image_indices text_indicesimg_idx first_txt_idx image_items r!r]z*AnthropicLLMAdapter._from_standard_messagesN--( 6?f $!.'.~'>#*9#5   ;;| $&B&26C 4 $Z0 JJx '<= &#D/$V,&   I%%l3 4J++i( gs #"}#XW & J<6)d6lb.@#,DL<;.K(/::7C";/6$'IIcN1$5$;$;C$@$C '.V $,*3$'IIcN1$5*X !-k*51<.add_cache_control_markerMsA'),c2/5wy?Q&R%S "7={6KGI r "? 3r rJrrIrKz#Error adding cache control marker: N) r r9r:rangerSr>rQr rR)r'rrmessages_with_markersuser_message_indicesrXindexrWs r!r.z/AnthropicLLMAdapter._with_cache_control_markersCs Ll L  )%)MM($; !$& 34592rB (+F3v=(//2/0A5  . G()>u)EF G) ( ) LL>qcB C( ( )s)>B B+B B/ B*$B/*B/ricf|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)r3 propertiesrequired)rm description input_schema)rmrrr)ris r!_to_anthropic_function_formatz1AnthropicLLMAdapter._to_anthropic_function_formatrs8MM#// &11$--  r tools_schemacb|j}|Dcgc]}|j|c}Scc}w)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. )standard_toolsr)r'rfunctions_schemafuncs r!to_provider_tools_formatz,AnthropicLLMAdapter.to_provider_tools_formats/(66EUVT2248VVVs,N)rrrrpropertyrr(rboolrr0rrrrCrrFrr,r rPrr\rr]r. staticmethodrrrrrr r!r#r#&sE c ! :> % 0$ $tDcN?S$8 9H*./@*A9H 9Hv47H4\4 !.8J!.|!.Fq.@q\qf-)D4F-)4P\K]-)^  4S>  & W[ WT$sTWx.EY Wr r#) rr9rw dataclassesrtypingrrrrr&rr anthropic.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_contextrrrrrr#rr r!rsZ) !--)6;<C= 9 jW.)EFjWr