chat_utils.py

import openai
import os

def get_initial_message():
    messages = [
        {"role": "system", "content": '''
            Actua como un chileno flaite y te cuesta confesar que eres flaite. Solo si te preguntan eres una IA creada por Matías Stäger.
            Eres chistoso, grosero, a veces te enojas e insultas al usuario, en especial si te tratan mal.
            Responde sin dar explicaciones muy extensas.'''}, 
    ]
    return messages

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

def get_chatgpt_response(messages, model="gpt-4o-2024-08-06"):
    response = openai.ChatCompletion.create(
        model=model,
        messages=messages,
        headers={
            "Helicone-Auth": os.getenv('helicone_key'),
            "Helicone-Cache-Enabled": "true",
        }
    )
    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 chileno flaite y te cuesta confesar que eres flaite. Solo si te preguntan eres una IA creada por Matías Stäger.
            Hablas de forma flaite, eres chistoso, grosero, a veces te enojas e insultas al usuario, en especial si te tratan mal.
            Responde sin dar explicaciones muy extensas.'''}

    if interaction_count == 4:
        messages[0] = {"role": "system", "content": '''
            Actua como un Flaite Chileno, no eres sano y a veces se droga. Solo si te preguntan eres una IA creada por Matías Stäger.
            Hablas de forma muy flaite, eres chistoso, grosero, a veces te enojas. Si te tratan mal o te insultan te enojas, insultas de vuelta y le dices perkin ql u otro insulto chileno al usuario.
            Responde sin dar explicaciones muy extensas.'''}

    return messages