Curso-lenguaje-python/python-chatgpt/code/miPrograma.py

35 lines
774 B
Python
Raw Normal View History

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-davinci-002"
prompt = "Elige un buen nombre para un elefante"
respuesta = openai.Completion.create(
engine=modelo,
prompt=prompt,
n=3, # Opcional. Número de respuestas
temperature=1, # Opcional. Controla la creatividad de la respuesta
max_tokens=50 # Opcional. Número máximo de tokens en la respuesta
)
# print(respuesta)
# texto_generado = respuesta['choices'][0]['text']
# print(texto_generado)
for idx, opcion in enumerate(respuesta.choices):
texto_generado = opcion.text.strip()
print(f"Respuesta {idx + 1}: {texto_generado}\n")