a
    ^g                  +   @  s   U d dl mZ ddlmZ ddlmZ dddddddddddddZded	< ddddddddddddd
d
ddddddddd
d
d
d
d
d
ddddddddddddddd*Zded< dddddZ	dddddZ
dS )    )annotations   )Encoding)get_encodingZ
o200k_baseZcl100k_base)zo1-zo3-zchatgpt-4o-zgpt-4o-zgpt-4-zgpt-3.5-turbo-zgpt-35-turbo-z	ft:gpt-4ozft:gpt-4zft:gpt-3.5-turbozft:davinci-002zft:babbage-002zdict[str, str]MODEL_PREFIX_TO_ENCODINGZ	p50k_baseZ	r50k_baseZ	p50k_editgpt2)*Zo1Zo3zgpt-4ozgpt-4zgpt-3.5-turbozgpt-3.5zgpt-35-turbozdavinci-002zbabbage-002ztext-embedding-ada-002ztext-embedding-3-smallztext-embedding-3-largeztext-davinci-003ztext-davinci-002ztext-davinci-001ztext-curie-001ztext-babbage-001ztext-ada-001ZdavinciZcurieZbabbageadazcode-davinci-002zcode-davinci-001zcode-cushman-002zcode-cushman-001zdavinci-codexzcushman-codexztext-davinci-edit-001zcode-davinci-edit-001ztext-similarity-davinci-001ztext-similarity-curie-001ztext-similarity-babbage-001ztext-similarity-ada-001ztext-search-davinci-doc-001ztext-search-curie-doc-001ztext-search-babbage-doc-001ztext-search-ada-doc-001zcode-search-babbage-code-001zcode-search-ada-code-001r   zgpt-2MODEL_TO_ENCODINGstr)
model_namereturnc                 C  sX   d}| t v rt |  }n$t D ]\}}| |r|  S q|du rTtd|  dd|S )zrReturns the name of the encoding used by a model.

    Raises a KeyError if the model name is not recognised.
    NzCould not automatically map z_ to a tokeniser. Please use `tiktoken.get_encoding` to explicitly get the tokeniser you expect.)r	   r   items
startswithKeyError)r   Zencoding_nameZmodel_prefixZmodel_encoding_name r   \/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/tiktoken/model.pyencoding_name_for_modelP   s    



r   r   c                 C  s   t t| S )zfReturns the encoding used by a model.

    Raises a KeyError if the model name is not recognised.
    )r   r   )r   r   r   r   encoding_for_modeli   s    r   N)
__future__r   corer   registryr   r   __annotations__r	   r   r   r   r   r   r   <module>   sx   8