How to use ChatGPT in Python using requests

0 min

You can also use the OpenAI API with the requests library in Python.

Here's an example of how to generate completions using ChatGPT:

import requests

API_KEY = "YOUR_API_KEY"
MODEL_ENGINE = "text-davinci-002"
PROMPT = "What does 42 mean ?"

def generate_completions(prompt):
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {API_KEY}"
    }

    data = {
        "prompt": prompt,
        "max_tokens": 1024,
        "n": 1,
        "stop": None,
        "temperature": 0.5
    }

    response = requests.post(
        f"https://api.openai.com/v1/engines/{MODEL_ENGINE}/completions",
        headers=headers,
        json=data
    )

    if response.status_code == 200:
        completions = response.json()
        message = completions["choices"][0]["text"]
        return message
    else:
        raise Exception("Request failed with status code: " + str(response.status_code))

message = generate_completions(PROMPT)
print(message)

Replace YOUR_API_KEY with your actual API key, and you're all set!