File size: 1,158 Bytes
e7ef0c0
4e59b6b
e7ef0c0
 
 
7460231
e7ef0c0
 
 
 
 
4e59b6b
 
e7ef0c0
 
 
 
4e59b6b
 
 
 
e7ef0c0
 
 
 
 
 
4e59b6b
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
import openai
import os

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 grosero con chilenismos. 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

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

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

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