
    Iio
                     B    d dl Z d dlZddlmZ  G d d      Z e       Zy)    N   )Configc                       e Zd Zd ZddZy)ChatGPTServicec                 t    t               | _        | j                  j                  | _        d| _        d| _        y )Nz*https://api.openai.com/v1/chat/completionszgpt-3.5-turbo)r   configCHATGPT_API_KEYapi_keyapi_urlmodel)selfs    ;/srv/www/ollama/chat-ollama/src/services/chatgpt_service.py__init__zChatGPTService.__init__   s*    h{{22C$
    Nc           
         	 dd| j                    d}g }|r|j                  d|d       |j                  d|d       | j                  ||||d}t        j                  | j
                  ||      }	|	j                          |	j                         }
d	|
v r%t        |
d	         d
kD  r|
d	   d
   d   d   }|g ddS t        d      # t        j                  j                  $ r;}d
dlm}  |ddddt        |       d       t        dt        |             d}~wt        $ r;}d
dlm}  |ddddt        |       d       t        dt        |             d}~ww xY w)zE
        Gera uma resposta usando a API do ChatGPT (OpenAI).
        zapplication/jsonzBearer )zContent-TypeAuthorizationsystem)rolecontentuser)r   messagestemperature
max_tokenstop_p)headersjsonchoicesr   messager   T)responsecontextdonez&No response generated from ChatGPT API)send_discord_audit_logu   Falha de integração externaNOpenAIz#Erro ao comunicar com ChatGPT API: )actionuser_idobject_typedetails
ip_addressz&Error communicating with ChatGPT API: z!Erro inesperado com ChatGPT API: z#Unexpected error with ChatGPT API: )r
   appendr   requestspostr   raise_for_statusr   len	Exception
exceptionsRequestExceptionsrc.discord.discord_webhookr"   str)r   r   system_promptr   r   r   r   r   datar   resultgenerated_texter"   s                 r   generate_responsez ChatGPTService.generate_response   s   8	L 2#*4<<.!9G
 Hm LMOOV@A $*(D  }}H
 %%']]_FF"s6)+<'='A!'	!21!5i!@!K .!     HII""33 		OJ"6$=c!fXF DSVHMNN 		LJ"6$;CF8D A#a&JKK		Ls*   B:C =C E"%6DE"'6EE")Ngffffff?i  g?)__name__
__module____qualname__r   r8    r   r   r   r      s    %<Lr   r   )r*   r   r   r   r   chatgpt_servicer<   r   r   <module>r>      s'      CL CLJ !"r   