From 7e3f32b296639521747ec549282bf51b9c8a4860 Mon Sep 17 00:00:00 2001
From: "Kirill Igumenshchev (aider)" <kirill.igum@gmail.com>
Date: Sun, 18 May 2025 17:23:57 -0700
Subject: [PATCH] fix: update generate_humor_dataset.py to use new OpenAI
 Python client API

---
 generate_humor_dataset.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/generate_humor_dataset.py b/generate_humor_dataset.py
index 6ebfddb1..22a9afda 100644
--- a/generate_humor_dataset.py
+++ b/generate_humor_dataset.py
@@ -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}],
                 )