Initial Commit

gpt4o_infer.py

@@ -1,10 +1,10 @@
-import openai
 import os
+import openai
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 def call_gpt4o(prompt):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4o",
         messages=[
             {"role": "system", "content": "You are a helpful assistant that converts natural language into SQL queries using the provided schema."},
@@ -13,4 +13,4 @@ def call_gpt4o(prompt):
         temperature=0.2,
         max_tokens=512
     )
-    return response['choices'][0]['message']['content']
+    return response.choices[0].message.content

requirements.txt

@@ -1,2 +1,2 @@
 gradio
-openai
+openai>=1.0.0