
    Zi                         	 d dl Z n# e$ r  ed          w xY wd dlmZ d dlmZ d dlmZ	  G d de j
                  Z
 G d d	e j                  Z G d
 de j                  Z G d de j                  ZdS )    NzMPlease install the Anthropic SDK to use this feature: 'pip install anthropic')WrappedMessages)AsyncWrappedMessages)Clientc                   4     e Zd ZU dZeed<   def fdZ xZS )AnthropicBedrockj
    A wrapper around the Anthropic Bedrock SDK that automatically sends LLM usage events to PostHog.
    
_ph_clientposthog_clientc                 p     t                      j        di | || _        t          |           | _        d S N super__init__r	   r   messagesselfr
   kwargs	__class__s      c/root/projects/butler/venv/lib/python3.11/site-packages/posthog/ai/anthropic/anthropic_providers.pyr   zAnthropicBedrock.__init__   :    ""6"""('--    __name__
__module____qualname____doc__PostHogClient__annotations__r   __classcell__r   s   @r   r   r      ^           .} . . . . . . . . . .r   r   c                   4     e Zd ZU dZeed<   def fdZ xZS )AsyncAnthropicBedrockr   r	   r
   c                 p     t                      j        di | || _        t          |           | _        d S r   r   r   r	   r   r   r   s      r   r   zAsyncAnthropicBedrock.__init__!   :    ""6"""(,T22r   r   r!   s   @r   r$   r$      ^           3} 3 3 3 3 3 3 3 3 3 3r   r$   c                   4     e Zd ZU dZeed<   def fdZ xZS )AnthropicVertexi
    A wrapper around the Anthropic Vertex SDK that automatically sends LLM usage events to PostHog.
    r	   r
   c                 p     t                      j        di | || _        t          |           | _        d S r   r   r   s      r   r   zAnthropicVertex.__init__.   r   r   r   r!   s   @r   r*   r*   '   r"   r   r*   c                   4     e Zd ZU dZeed<   def fdZ xZS )AsyncAnthropicVertexr+   r	   r
   c                 p     t                      j        di | || _        t          |           | _        d S r   r&   r   s      r   r   zAsyncAnthropicVertex.__init__;   r'   r   r   r!   s   @r   r.   r.   4   r(   r   r.   )	anthropicImportErrorModuleNotFoundErrorposthog.ai.anthropic.anthropicr   $posthog.ai.anthropic.anthropic_asyncr   posthog.clientr   r   r   r$   r*   r.   r   r   r   <module>r6      sB     

W  
 ; : : : : : E E E E E E 2 2 2 2 2 2
. 
. 
. 
. 
.y1 
. 
. 
.
3 
3 
3 
3 
3I; 
3 
3 
3
. 
. 
. 
. 
.i/ 
. 
. 
.
3 
3 
3 
3 
399 
3 
3 
3 
3 
3s    