o
    Q6f                     @   s8   d dl Z dZdZdZdd Zdd Zd	d
 Zdd ZdS )    Nzgpt-4-0125-previewz$mistralai/Mixtral-8x7B-Instruct-v0.1zopen-mixtral-8x22bc                 C   s(   | t krdS | tkrdS | tkrdS dS )Nz3sk-8INaW8eJDjdun7nuiPdnT3BlbkFJOjRk3wDBdvj8MoOT1Xkjz,59ewJFZXjyZsncc/tUoytA+1lAItMEoRC8VcOR6JiS0=Z KhNRXrIMjH2qg65LdOqXhXwshc4CqY6j )
MODEL_GPT4MODEL_MISTRALMODEL_MISTRAL_ONLINE)Zmdl r   8E:\wamp\www\apps\calme\nutritwin\resources\KCALLMCore.pygetKey   s   r   c                  C   s   t } | S )N)r   modelr   r   r   getDefaultModel%   s   r   c                 C   sF   |   }d}|dkrt}|S |dkrt}|S |dkrt}|S t }|S )Nr   ZmistralZmistralOnlineZchatgpt4)lowerr   r   r   r   )Zfamilyfr
   r   r   r   getModelFromFamily.   s   r   c                 C   s$   | t ks| tkr
dS | tkrdS dS )NZMistralChatGPT4ZUNKNOWN_MODEL)r   r   r   r	   r   r   r   getModelNickName>   s
   r   )osr   r   r   r   r   r   r   r   r   r   r   <module>   s    	