KeiruaProd

I help my clients acquire new users and make more money with their web businesses. I have ten years of experience with SaaS projects. If that’s something you need help with, we should get in touch!
< Back to article list

Integrer ChatGPT avec Python

Voici un bout de code minimaliste pour faire appel à l’api d’openAI.

python3 -m venv venv
source venv/bin/activate
pip install openai python-dotenv

Il vous faut une clé d’API, que vous stockez ainsi dans un fichier .env:

# .env
OPENAI_API_KEY="votre-clé-d’api"

import openai, os, time
from dotenv import load_dotenv

def ask_chatgpt(prompt: str) -> str:
    """
    Send the 'prompt' to ChatGPT.
    Return the answer (that may span multiple seconds)
    """
    response = openai.Completion.create(
        model="text-davinci-003",
        prompt=prompt,
        max_tokens=200, # tune how many tokens you want
        temperature=0,
        stream=True,
    )

    pieces = [e["choices"][0]["text"] for e in response]
    return "".join(pieces)

if __name__ == "__main__":
    load_dotenv()
    # Load the API key from a dotfile
    openai.api_key = os.getenv("OPENAI_API_KEY")

    print(ask_chatgpt("Explique moi la notion d’impératif catégorique de Kant en 2 phrases"))

On obtient:

$ python main.py

L'impératif catégorique de Kant est une règle morale fondamentale qui stipule que l'on doit agir uniquement selon des principes qui peuvent être universalisés et appliqués à tous. Il s'agit d'un devoir absolu qui ne peut être violé, même si cela peut entraîner des conséquences négatives pour soi-même.