
    	*h                         d dl mZ d dlZd dlZd dlmZ d dlZ ej        d          Z ej        d          Z	d Z
d ZdS )    )load_dotenvN)OpenAIEMBEDDING_MODEL	GPT_MODELc                      t                       t          j        d          } | st          d          t	          |           S )NOPENAI_API_KEYz0API Key is not set in the environment variables.api_key)r   osgetenv
ValueErrorr   r	   s    M/Users/gualin/Documents/wsky_tg_auto_maintain/openai_config/openai_connect.pyget_openai_clientr      sD    MMMi())G MKLLL'""""    c                     t                      }|j        j                            d| dgt          d          }|j        d         j        j        S )Nuser)rolecontentr   )messagesmodeltemperature)r   chatcompletionscreater   choicesmessager   )queryclientresponses      r   openai_chatr       sZ      F{&--..
  .  H A&..r   )dotenvr   r   sysopenair   tiktokenr   r   r   r   r     r   r   <module>r&      s          				 



       ")-..BIk""	# # #
/ 
/ 
/ 
/ 
/r   