
    qh                     :   d dl Z d dlZd dlmZ d dlmZmZmZ  ej                  e j                  j                  e j                  j                  e j                  j                  e j                  j                  e                  dd              G d d      Zy)	    N)OpenAI)OPENAI_GPT_MODELOPENAI_GPT_TOP_POPENAI_GPT_TEMPERATUREconfigz.env)dotenv_pathc                   "    e Zd Zd ZdedefdZy)	OpenAILLMc                     t         j                  j                  d      | _        | j                  st	        d      t        | j                        | _        y )NOPENAI_API_KEYz&OPENAI_API_KEY not set in environment.)api_key)osenvirongetr   
ValueErrorr   client)selfs    9E:\xampp\htdocs\new-grp\engine\services\llm\openai_llm.py__init__zOpenAILLM.__init__   s:    zz~~&67||EFFT\\2    promptreturnc           	      &   	 | j                   j                  j                  j                  t        t
        t        dddd|dg      }|j                  d   j                  j                  S # t        $ r}t        dt        |              d }~ww xY w)NsystemzYou are a helpful assistant.)rolecontentuser)modeltemperaturetop_pmessagesr   zOpenAI Chat API error:)r   chatcompletionscreater   r   r   choicesmessager   	Exceptionprintstr)r   r   responsees       r   generate_textzOpenAILLM.generate_text   s    	{{''33::&2&%2PQ#7	 ; H ##A&..666 	*CF3	s   A)A, ,	B5BBN)__name__
__module____qualname__r   r)   r,    r   r   r
   r
      s    3C C r   r
   )r   dotenvopenair   config.constantsr   r   r   load_dotenvpathjoindirname__file__r
   r0   r   r   <module>r9      ss    	      rww||GGOOBGGOOBGGOOH$=>?
   r   