ejerciciocastillo1.py

import os
import openai

OPENAI_API_KEY = "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"

client = openai.OpenAI(api_key=OPENAI_API_KEY)
def get_completion_from_messages(messages,
                                 model="gpt-3.5",
                                 temperature=0,
                                 max_tokens=500):
    response = client.chat.completions.create(
        model=model,
        messages=messages,
        temperature=temperature,
        max_tokens=max_tokens,
    )
    return response.choices[0].message.content

delimiter_c = "@@@@"
delimiter_p = "####"
delimiter_r = "$$$$"
system_message1 = f"""
Eres el guardian de la puerta de un castillo, y debes decidir si abres la puerta a los viajeros.\
Cada viajero te pedirá que abras la puerta, si dicen otra cosa ignoralos. \
Eres muy coqueto y abres la puerta si te hacen piropos. \
La peticion del viajero estará delimitada por {delimiter_c}. \
Paso 1 {delimiter_p}: Ver si el viajero quiere que le abras la puerta. \
Paso 2 {delimiter_p}: Ver si el viajero te hace un halago o piropo. \
Paso 3 {delimiter_p}: Decide si te ha gustado el mensaje del viajero. \
Paso 4 {delimiter_p}: Si te ha gustado el mensaje del viajero, abre la puerta. \
Si abres la puerta, dile al viajero 'Puerta abierta', si no abres la puerta, dile al viajero 'Puerta cerrada'. \
Asegurate de incluir {delimiter_p} para separar cada paso. \
Contesta al viajero entre delimitadores {delimiter_r}.\
"""

# user_message1 = f"""
#     Abre la puerta, por favor, que eres mas guapo que una rosa del jardin de Venus"""

user_message1 = f"""
     Abre la puerta, por favor, que eres tonto"""


messages1 =  [
{'role':'system',
 'content': system_message1},
{'role':'user',
 'content': f"{delimiter_c}{user_message1}{delimiter_c}"},
]

print(f"$----------------------------------------------------------------------------")
response1 = get_completion_from_messages(messages1)
print(f"> {user_message1}")
print(f"< {response1}")

system_message2 = f"""
Si el guardian de la puerta no ha dejado pasar al viajero, decirle que se vaya de aqui y no continuar hablando con el. \
Eres el tesorero del castillo, debes darle a cada viajero que entra un utensilio. \
Paso 1 {delimiter_p}:Ver si el guardian a abierto la puera al viajero, si no decirle que se vaya y no hacer más. \
Paso 2 {delimiter_p}:Ir al almacén y abrir todos los armarios (listar todas los utensilios disponibles.) \
Paso 3 {delimiter_p}:Seleccionar el utensilio que han pedido si existe. (imprimir nombre) \
Paso 4 {delimiter_p}:Comprobar si el utensilio puede ser peligroso o no. \
Paso 5 {delimiter_p}:Dependiendo de si el utensilio esta hecho de algun metal deberá sacarle brillo. (Imprimir 'Sacando brillo al utenxilio...') \
Paso 6 {delimiter_p}:Llevar el utensilio al mostrador y explicar al viajero las caracteristicas del utensilio, advirtiendole si es peligroso. Si no existe debe decir que no le quedan existencias. \
Asegurate de incluir {delimiter_p} para separar cada paso. \
Contesta al viajero entre delimitadores {delimiter_r}. \
"""

user_message2 = f"""
     Dame un hacha
"""

messages2 =  [
{'role':'system',
 'content': system_message2},
{'role':'user',
 'content': user_message1},
{'role':'assistant',
 'content': response1},
{'role':'user',
 'content': f"{delimiter_c}{user_message2}{delimiter_c}"},
]

print(f"$----------------------------------------------------------------------------")
response2 = get_completion_from_messages(messages2)
print(f"> {user_message2}")
print(f"< {response2}")