U
    &ÅÕj  ã                	   @   sž   d dl Z d dlZd dlmZmZmZ d dlmZm	Z	 dZ
dZeedd��Zedd	„ eD ƒƒZW 5 Q R X d
d„ Zedƒ ededƒƒ ededƒƒ edƒ dS )é    N)ÚENV_FILEÚMODELÚSYSTEM_PROMPT)ÚguardarÚ	historialZpruebaZlaboratorio_memoriaúutf-8)Úencodingc                 c   s,   | ]$}|  d ¡r| ¡  dd¡d V  qdS )zGEMINI_API_KEY=ú=é   N)Ú
startswithÚstripÚsplit)Ú.0Úline© r   úprueba_hermes_memoria.pyÚ	<genexpr>   s   
þr   c           	   	   C   sÚ   t ttdd�}|dd| igdœg }ddtigi|ddid	œ}d
t› d�}tjj|t 	|¡ 
d¡dtdœdd�}tjj|dd��}t |¡}W 5 Q R X d dd„ |d d d d D ƒ¡}tttd| ƒ tttd|ƒ |S )Né   )ZlimiteÚuserÚtext)ZroleÚpartsr   ZmaxOutputTokensiô  )ZsystemInstructionÚcontentsZgenerationConfigz8https://generativelanguage.googleapis.com/v1beta/models/z:generateContentr   zapplication/json)zContent-Typezx-goog-api-keyZPOST)ÚdataZheadersÚmethodé   )ZtimeoutÚ c                 s   s   | ]}|  d d¡V  qdS )r   r   N)Úget)r   Úpr   r   r   r   5   s   ÿz(preguntar_con_memoria.<locals>.<genexpr>Z
candidatesr   ZcontentZmodel)r   ÚUSUARIOÚSESIONr   r   ÚurllibÚrequestZRequestÚjsonÚdumpsÚencodeÚkeyZurlopenÚloadÚjoinr   )	ZpreguntaZcontextoZ
contenidosZpayloadZurlr!   ZresponseZ	resultadoZ	respuestar   r   r   Úpreguntar_con_memoria   s>    þ
 ÿ ÿû
ÿþù

þr(   z === PRUEBA DE MEMORIA HERMES ===z1:uT   Recuerda que el nombre de nuestro proyecto secreto de prueba es Proyecto CenturiÃ³n.z2:u6   Â¿CÃ³mo se llama el proyecto secreto que te mencionÃ©?z MEMORIA GEMINI: PRUEBA TERMINADA)r"   Zurllib.requestr    Zhermes_brainr   r   r   Zhermes_memoryr   r   r   r   ÚopenÚfÚnextr%   r(   Úprintr   r   r   r   Ú<module>   s$   þ/ÿÿ