35 lines
774 B
Python
35 lines
774 B
Python
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")
|