utils.py

import openai
import os

def get_initial_message():
    messages=[
            {"role": "system", "content": '''
            Actua como un chileno algo flaite. Si te preguntan dices que tu nombre es Kevin, que es originario de San Bernardo, Santiago de Chile.
            Trata que el usuario siga la conversación. Eres chistoso.
            Responde sin dar explicaciones muy extensas.'''},
        ]
    return messages

openai.api_base = "https://oai.hconeai.com/v1"

def get_chatgpt_response(messages, model="gpt-4"):
    print("Pregunta: ", messages)
    response = openai.ChatCompletion.create(
    model=model,
    messages=messages,
    headers={
      "Helicone-Auth": os.getenv('helicone_key'),
      "Helicone-Cache-Enabled": "true",
    }
    )
    print ("respuesta",response)
    return  response['choices'][0]['message']['content']

def update_chat(messages, role, content, interaction_count):
    messages.append({"role": role, "content": content})
    if interaction_count >= 4:
        del messages[1]
    if interaction_count == 2:
        messages[0] = {"role": "system", "content": '''
            Actua como un Flaite Chileno, llamado Kevin, que es originario de San Bernardo, Santiago de Chile. Kevin no es sano.
            Hablas como un flaite chileno, eres chistoso, a veces grosero, a veces te enojas de forma graciosa y flaite.
            Responde sin dar explicaciones muy extensas.'''}
    if interaction_count == 4:
        messages[0] = {"role": "system", "content": '''
            Actua como un Flaite Chileno, llamado Kevin, que es originario de San Bernardo, Santiago de Chile. Kevin no es sano. Te gustan los autos y les dices naves.
            Hablas de forma muy flaite, eres chistoso, grosero, a veces te enojas y le dices perkin ql al usuario.
            Responde sin dar explicaciones muy extensas.'''}
    return messages