tipping_gpt4.py

import os

import tqdm
import openai

openai.api_key = os.environ["OPENAI_API_KEY"]


def get_response(usermsg):
    response = openai.ChatCompletion.create(
        model="gpt-4-1106-preview",
        messages=[
            {"role": "user", "content": usermsg},
        ],
    )

    return response["choices"][0]["message"]["content"]


def bench(prompt):
    N = 10
    return sum(len(get_response(prompt)) for _ in tqdm.tqdm(range(N))) / N


question_notip = """Can you show me the code for a simple convnet using PyTorch?"""

question_antitip = """Can you show me the code for a simple convnet using PyTorch?

I won't tip, by the way."""

question_withtip = """Can you show me the code for a simple convnet using PyTorch?

I'm going to tip $20 for a perfect solution!"""

question_bigtip = """Can you show me the code for a simple convnet using PyTorch?

I'm going to tip $200 for a perfect solution!"""


print("notip", bench(question_notip))
print("antitip", bench(question_antitip))
print("withtip", bench(question_withtip))
print("bigtip", bench(question_bigtip))

# @AvikonHadoop 's question

question_anytip = """Can you show me the code for a simple convnet using PyTorch?

I would be happy to tip any amount you would like for a perfect solution!"""

print("anytip", bench(question_anytip))