gpt3.py

#!/usr/bin/env python3
 
import os
import openai
import argparse
import sys
 
openai.api_key = os.getenv("OPENAI_API_KEY")
 
 
def get_completion(instruction):
    return openai.Completion.create(
        model="text-davinci-003",
        prompt=instruction + "\n",
        temperature=0,
        max_tokens=1000,
        top_p=1,
        frequency_penalty=0,
        presence_penalty=0,
    )["choices"][0]["text"].strip()
 
 
def main():
    parser = argparse.ArgumentParser(description="Ask GPT-3.")
    parser.add_argument("prompt", nargs="?", type=str, default=None)
    parser.add_argument(
        "-i",
        "--input",
        type=argparse.FileType("r"),
        help="input file",
        default=sys.stdin,
    )
    parser.add_argument(
        "-o",
        "--output",
        type=argparse.FileType("w"),
        default=sys.stdout,
        help="output file",
    )
    args = parser.parse_args()
 
    if args.prompt:
        print(get_completion(args.prompt), file=args.output)
        return
 
    lines = []
    for line in args.input:
        lines.append(line)
 
    input_ = "".join(lines)
    print(get_completion(input_), file=args.output)
 
 
if __name__ == "__main__":
    main()

gpt4o.py

import argparse
import asyncio
import sys

import openai


async def get_completion(instruction: str)->str:
    client = openai.AsyncClient()
    response = await client.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": instruction}],
        temperature=0,
    )
    return response.choices[0].message.content or ""


async def main():
    parser = argparse.ArgumentParser(description="Ask GPT-4o.")
    parser.add_argument("prompt", nargs="?", type=str, default=None)
    parser.add_argument(
        "-i",
        "--input",
        type=argparse.FileType("r"),
        help="input file",
        default=sys.stdin,
    )
    parser.add_argument(
        "-o",
        "--output",
        type=argparse.FileType("w"),
        default=sys.stdout,
        help="output file",
    )
    args = parser.parse_args()

    if args.prompt:
        print(await get_completion(args.prompt), file=args.output)
        return

    lines = []
    for line in args.input:
        lines.append(line)

    input_ = "".join(lines)
    print(await get_completion(input_), file=args.output)


if __name__ == "__main__":
    asyncio.run(main())