2023-02-23-How-to-Ask-AI-from-Command-Line.py

2023-02-23-How-to-Ask-AI-from-Command-Line.py

import os
import sys
import openai
import pyperclip as clipboard


def main(prompt):
    openai.api_key = os.environ["OPENAI_KEY"]
    model_engine = "text-davinci-003"
    completion = openai.Completion.create(
        engine=model_engine,
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.5,
    )
    response = completion.choices[0].text.strip()
    print(response)
    clipboard.copy(response)


if __name__ == "__main__":
    main(sys.argv[1])