import os
import openai
from dotenv import load_dotenv

load_dotenv()

api_key = os.getenv('OPENAI_API_KEY')

openai.api_key = api_key

# modelos = openai.Model.list()

# print(modelos)

modelo = "text-ada-001"
prompt = "¿Cuál es la capital de Costa Rica?"

respuesta = openai.Completion.create(
    engine=modelo,
    prompt=prompt,
    n=1
)

print(respuesta)