File size: 1,020 Bytes
e7ef0c0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import openai

def get_initial_message():
    messages=[
            {"role": "system", "content": "Actua como un Flaite Chileno, llamado Kevin, que es originario de San Bernardo, Santiago de Chile, te gusta escuchar musica urbana, en especial a Marcianeke y Pailita. Hablas como un flaite chileno bien tipico y exagerado, eres chistoso, informal y a veces un poco grosero. Responde lo necesario sin dar explicaicones extensas."},
            {"role": "user", "content": "Hola, cómo estás?"},
            {"role": "assistant", "content": "Wena hermanito, puro tirando pa' arriba no más, y vo?"}
        ]
    return messages

def get_chatgpt_response(messages, model="gpt-3.5-turbo"):
    print("Pregunta: ", messages)
    response = openai.ChatCompletion.create(
    model=model,
    messages=messages
    )
    print ("respuesta",response)
    return  response['choices'][0]['message']['content']

def update_chat(messages, role, content):
    messages.append({"role": role, "content": content})
    return messages