openai_demo.py

import openai
import wandb
openai.api_key = "sk-..."  # supply your API key however you choose

# https://platform.openai.com/docs/models

tbl = wandb.Table("my_table", columns=["input", "output", "temperature"])

query = "hello world"
systemp_prompts = "you are friendly"


completion = openai.ChatCompletion.create(
    model="gpt-4", 
    temperature=0.7, 
    messages=[
            {"role": "system", "content": system_prompts}
            {"role": "user", "content": query}
            # {"role": "assistant", "content": "hello"}
            ]
    )

output = completion.choices[0].message.content # extract token counts from "completion"

tbl.add_data(query, output, temperature)
wandb.log({"tbl": tbl})