
    6Fi                     6   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	        
                    e j	                            e j	                            e j	                            e                              dd                      G d d          ZdS )	    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dS )	OpenAILLMc                     t           j                            d          | _        | j        st	          d          t          | j                  | _        d S )NOPENAI_API_KEYz&OPENAI_API_KEY not set in environment.)api_key)osenvirongetr   
ValueErrorr   client)selfs    9D:\xampp\htdocs\new-grp\engine\services\llm\openai_llm.py__init__zOpenAILLM.__init__   sJ    z~~&677| 	GEFFFT\222    promptreturnc           	         	 | 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::&2&%2PQQ#77	 ;  H #A&.66 	 	 	*CFF333	s   AA 
B"B  BN)__name__
__module____qualname__r   r)   r,    r   r   r
   r
      sE        3 3 3C C      r   r
   )r   dotenvopenair   config.constantsr   r   r   load_dotenvpathjoindirname__file__r
   r0   r   r   <module>r9      s    				                  rw||GOOBGOOBGOOH$=$=>>??
                r   