o
    3Ih                     @   s   zd dl Z W n ey   edw d dlmZ d dlmZ d dlmZ	 G dd de j
Z
G dd	 d	e jZG d
d de jZG dd de jZdS )    NzMPlease install the Anthropic SDK to use this feature: 'pip install anthropic')WrappedMessages)AsyncWrappedMessages)Clientc                       0   e Zd ZU dZeed< def fddZ  ZS )AnthropicBedrockj
    A wrapper around the Anthropic Bedrock SDK that automatically sends LLM usage events to PostHog.
    
_ph_clientposthog_clientc                    &   t  jdi | || _t| | _d S N super__init__r   r   messagesselfr	   kwargs	__class__r   h/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/posthog/ai/anthropic/anthropic_providers.pyr         zAnthropicBedrock.__init____name__
__module____qualname____doc__PostHogClient__annotations__r   __classcell__r   r   r   r   r         
 r   c                       r   )AsyncAnthropicBedrockr   r   r	   c                    r
   r   r   r   r   r   r   r   r   r   r   r   !   r   zAsyncAnthropicBedrock.__init__r   r   r   r   r   r!      r    r!   c                       r   )AnthropicVertexi
    A wrapper around the Anthropic Vertex SDK that automatically sends LLM usage events to PostHog.
    r   r	   c                    r
   r   r   r   r   r   r   r   .   r   zAnthropicVertex.__init__r   r   r   r   r   r#   '   r    r#   c                       r   )AsyncAnthropicVertexr$   r   r	   c                    r
   r   r"   r   r   r   r   r   ;   r   zAsyncAnthropicVertex.__init__r   r   r   r   r   r%   4   r    r%   )	anthropicImportErrorModuleNotFoundErrorposthog.ai.anthropic.anthropicr   $posthog.ai.anthropic.anthropic_asyncr   posthog.clientr   r   r   r!   r#   r%   r   r   r   r   <module>   s    