
    ڵ1i                     F    d dl Z d dlmZ d dlmZ  e         G d d      Zy)    N)OpenAI)load_dotenvc            	       ,    e Zd Zd ZddedededefdZy)		AIServicec                 L    t        t        j                  d            | _        y)z6Initialize OpenAI client with API key from environmentOPENAI_API_KEY)api_keyN)r   osgetenvclient)selfs    7/var/www/html/drjob-dev/drjob-ai/services/ai_service.py__init__zAIService.__init__
   s    II./
    questionsystem_prompt
max_tokensreturnc                 L   d}|r	|dz   |z   }n|}	 | j                   j                  j                  j                  t	        j
                  d      d|dd|dg|d      }|j                  d	   j                  j                  S # t        $ r}d
t        |       cY d }~S d }~ww xY w)NzYou are a helpful assistant. OPENAI_API_MODELsystem)rolecontentusergffffff?)modelmessagesr   temperaturer   zError: )r   chatcompletionscreater
   r   choicesmessager   	Exceptionstr)r   r   r   r   base_system_promptfinal_system_promptresponsees           r   ai_responsezAIService.ai_response   s     < "4s":]"J"4	&{{''33::ii 23%2EF#9 & ; H ##A&..666 	&SVH%%	&s   A0B 	B#BB#B#N) i  )__name__
__module____qualname__r   r%   intr*    r   r   r   r   	   s)    
&C & &c &]` &r   r   )r
   openair   dotenvr   r   r0   r   r   <module>r3      s    	   !& !&r   