ChileanGPT / utils.py
Matias Stager
Inicio de Repositorio ChileanGPT
e7ef0c0
raw
history blame
1.02 kB
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