NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-26 17:13:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: update generate_humor_dataset.py to use new OpenAI Python client API

Browse source
This commit is contained in:
Kirill Igumenshchev (aider) 2025-05-18 17:23:57 -07:00
parent aa8a1ed416
commit 7e3f32b296
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
generate_humor_dataset.py
View file

@ -1,11 +1,11 @@
import os
import json
from dotenv import load_dotenv
import openai
from openai import OpenAI
def main():
load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
comedians = [
"Norm Macdonald",
@ -31,7 +31,7 @@ def main():
f"What’s the best local LLM model to generate {fmt} jokes "
f"in the style of {comedian}? Please explain your reasoning step by step."
)
response = openai.ChatCompletion.create(
response = client.chat.completions.create(
model=model_name,
messages=[{"role": "user", "content": question}],
)