gT ddlZddlmZddlZddlZddlmZm Z m Z ddl m Z m Z mZmZddlmZGddej&ZGd d eZy#e$r edwxYw) N) AsyncMessageszMPlease install the Anthropic SDK to use this feature: 'pip install anthropic')AnyDictOptional)call_llm_and_track_usage_asyncget_model_paramsmerge_system_promptwith_privacy_mode)Clientc4eZdZUdZeed<deffd ZxZS)AsyncAnthropiczi An async wrapper around the Anthropic SDK that automatically sends LLM usage events to PostHog. _ph_clientposthog_clientc Rt|di|||_t||_y)z Args: posthog_client: PostHog client for tracking usage **kwargs: Additional arguments passed to the Anthropic client N)super__init__rAsyncWrappedMessagesmessages)selfrkwargs __class__s Y/var/www/openai/venv/lib/python3.12/site-packages/posthog/ai/anthropic/anthropic_async.pyrzAsyncAnthropic.__init__s( "6"(,T2 )__name__ __module__ __qualname____doc__ PostHogClient__annotations__r __classcell__rs@rr r s!3}33rr ceZdZUeed< ddeedeedeeeefde deeeefdef fd Z ddeedeedeeeefde deeeefdef d Z deedeedeeeefde deeeefdef fd Z deedeedeeeefde deeeefdeeefd eee fd ed efdZxZS)r_clientposthog_distinct_idposthog_trace_idposthog_propertiesposthog_privacy_modeposthog_groupsrc :K|tj}|jddr|j|||||fi|d{St ||j j d|||||j jt|$f i|d{S7P7w)a8 Create a message using Anthropic's API while tracking usage in PostHog. Args: posthog_distinct_id: Optional ID to associate with the usage event posthog_trace_id: Optional trace UUID for linking events posthog_properties: Optional dictionary of extra properties to include in the event posthog_privacy_mode: Whether to redact sensitive information in tracking posthog_groups: Optional group analytics properties **kwargs: Arguments passed to Anthropic's messages.create NstreamF anthropic) uuiduuid4get_create_streamingrr$rbase_urlrcreate)rr%r&r'r(r)rrs rr2zAsyncWrappedMessages.create$s(  ##zz|  ::h &///# "$    4  LL # #     LL ! ! GN       s%ABBA BBBBcxK|tj}|j|||||fi|d{S7w)N)r-r.r0)rr%r&r'r(r)rs rr+zAsyncWrappedMessages.streamRsR  ##zz| +T++           s 1:8:c  Ktj ddd gt did{  f d}|S7w)Nr input_tokens output_tokensc ` K 23d{}t|dr2|jr&dDcic]}|t|j|dc}t|dr'|jrj |j|7{cc}w6 t j }|z }dj } j || d{7y#t j }|z }dj } j || d{7wxYww)Nusager5rcontent)hasattrr9getattrr:appendtimejoin_capture_streaming_event)eventkend_timelatencyoutputaccumulated_contentrr%r)r(r'r&responser start_time usage_statss r generatorz9AsyncWrappedMessages._create_streaming..generatorusF #+  %ug.5;;&'&wu{{Aq99&' ui0U]]+225==AK  '$,  99;"Z/!4533'$&("     99;"Z/!4533'$&("   saD.CBBBCB  r`sgo1 &&uu23Y--3$i=i;o m nnos AA