o
    ˜º¥g‚  ã                   @   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 )ÚAnthropicBedrockúj
    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   ú_/var/www/visachat/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   )ÚAnthropicVertexúi
    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   9   r   zAsyncAnthropicVertex.__init__r   r   r   r   r   r%   2   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    ÿ