Regresar

Cómo conectar al API de ChatGPT con Python

por Jorge, 21-Mar-2023

Antes de empezar, asegúrate de tener una cuenta en OpenAI y haber creado una API Key en la plataforma. Si no tienes una cuenta, puedes crear una gratuita en https://beta.openai.com/signup/ y seguir las instrucciones para generar una API Key.

Una vez que tengas tu API Key, sigue los siguientes pasos:

Instalar la librería openai de Python. Puedes hacerlo mediante el siguiente comando en tu terminal:

pip install openai

Importar la librería openai y configurar la conexión con tu API Key:


import openai

openai.api_key = "tu_api_key"

Llamar al modelo de lenguaje que quieras utilizar. En este caso, usaremos el modelo davinci, que es el más avanzado disponible en la plataforma. También puedes probar otros modelos, como curie, babbage, ada, entre otros.


model_engine = "davinci"

Finalmente, puedes hacer una solicitud al API para generar una respuesta a partir de una entrada de texto:


prompt = "Hola, ¿cómo estás?"
response = openai.Completion.create(
    engine=model_engine,
    prompt=prompt,
    max_tokens=60,
    n=1,
    stop=None,
    temperature=0.7,
)

print(response.choices[0].text)

En este ejemplo, estamos generando una respuesta a partir de la entrada de texto "Hola, ¿cómo estás?". max_tokens indica la longitud máxima de la respuesta generada.


response = openai.Completion.create(
  engine="text-davinci-002",
  prompt="Hola, ¿cómo estás?",
  max_tokens=50
)

Finalmente, puedes imprimir la respuesta generada mediante el siguiente código:


print(response.choices[0].text)

Este código imprimirá la respuesta generada por el modelo de OpenAI a partir de la entrada de texto proporcionada.