
    Zg                     F    d dl Z dZdZdZdZdZd Zd Zd	 Zd
 Z	d Z
d ZdS )    Nzgpt-4o-2024-11-20z$mistralai/Mixtral-8x7B-Instruct-v0.1zmistral-large-2411zpixtral-large-2411zclaude-3-5-sonnet-20240620c                     | t           k    rdS | t          k    rdS | t          k    s| t          k    rdS | t          k    rdS dS )Nz3sk-8INaW8eJDjdun7nuiPdnT3BlbkFJOjRk3wDBdvj8MoOT1Xkjz,59ewJFZXjyZsncc/tUoytA+1lAItMEoRC8VcOR6JiS0= KhNRXrIMjH2qg65LdOqXhXwshc4CqY6jzl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 )
MODEL_GPT4MODEL_MISTRALMODEL_MISTRAL_ONLINEMODEL_PIXTRAL_ONLINEMODEL_ANTHRO)mdls    3/home/debian/html/nutritwin/resources/KCALLMCore.pygetKeyr       s[     jDD 
		==	$	$	$/C(C(C11			}}2    c                 @    d}| t           k    rd}n| t          k    rd}|S )Nr   i   i  )r   r   )r   dims     r   getSignatureSizer   2   s/    
C
"""	
		Jr   c                      t           } | S )N)r   models    r   getDefaultModelr   =   s     !E Lr   c                 V    d}| t           k    rd}n| t          k    s| t          k    rd}|S )Nr   ztext-embedding-3-largezmistral-embed)r   r   r   )r   r   s     r   getModelForEmbeddingr   G   s9    E
j(	$	$	$}(<(<Lr   c                     |                                  }d}|dk    rt          }n8|dk    rt          }n*|dk    rt          }n|dk    rt          }nt                      }|S )Nr   mistralmistralOnlinechatgpt4	anthropic)lowerr   r   r   r
   r   )familyfr   s      r   getModelFromFamilyr    S   sh    AEI~~	
o		$	
j	
k		!!Lr   c                     | t           k    s| t          k    s| t          k    rdS | t          k    rdS | t          k    rdS dS )NMistralChatGPT4	AnthropicUNKNOWN_MODEL)r   r   r	   r   r
   r   s    r   getModelNickNamer&   e   sM    *>!>!>%K_B_B_y	*		z	,		{?r   )osr   r   r   r	   r
   r   r   r   r   r    r&    r   r   <module>r)      s    				 !
 7 ,  ,  ,  $    	 	 	  $	 	 	 	 	r   