File size: 1,942 Bytes
b7ae72f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
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