
    j                     N    d dl mZmZmZ ddlmZ  G d dej                        Zy)    )DictAnyOptional   )base_original_provider_usagec                        e Zd ZU dZeed<   	 eed<   	 dZee   ed<   	 dZee   ed<   	 de	de
e	ef   f fd	Zed
e
e	ef   dd fd       Z xZS )BedrockUsagez2Bedrock calls token usage data. Updated 04.08.2025inputTokensoutputTokensNcacheReadInputTokenscacheWriteInputTokensparent_key_prefixreturnc                 "    t         |   |      S )N)superto_backend_compatible_flat_dict)selfr   	__class__s     u/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/llm_usage/bedrock_usage.pyr   z,BedrockUsage.to_backend_compatible_flat_dict   s    w67HII    usagec                      | di |S )N r   )clsr   s     r   from_original_usage_dictz%BedrockUsage.from_original_usage_dict   s    |U|r   )__name__
__module____qualname____doc__int__annotations__r   r   r   strr   r   classmethodr   r   __classcell__)r   s   @r   r	   r	      s    <56*.(3-.D+/8C=/EJ JcSVh J T#s(^   r   r	   N)typingr   r   r    r   BaseOriginalProviderUsager	   r   r   r   <module>r(      s!    & & */II r   