o
    ?Gh                     @   sX   d dl mZ d dlZd dlZd dlmZ d dlZedZedZ	dd Z
dd	 ZdS )
    )load_dotenvN)OpenAIEMBEDDING_MODEL	GPT_MODELc                  C   s&   t   td} | stdt| dS )NOPENAI_API_KEYz0API Key is not set in the environment variables.api_key)r   osgetenv
ValueErrorr   r    r   =/var/www/html/tg_maintain_bot/openai_config/openai_connect.pyget_openai_client   s
   

r   c                 C   s0   t  }|jjjd| dgtdd}|jd jjS )Nuser)rolecontentr   )messagesmodeltemperature)r   chatcompletionscreater   choicesmessager   )queryclientresponser   r   r   openai_chat   s   r   )dotenvr   r	   sysopenair   tiktokenr
   r   r   r   r   r   r   r   r   <module>   s    

