LeonceNsh commited on
Commit
24b6a6d
·
verified ·
1 Parent(s): d9a0200

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,7 +8,7 @@ import plotly.express as px
8
  import os
9
 
10
  # Set OpenAI API key
11
- openai.api_key = os.getenv("OPENAI_API_KEY")
12
 
13
  # =========================
14
  # Configuration and Setup
@@ -83,13 +83,13 @@ async def parse_query(nl_query):
83
  ]
84
 
85
  try:
86
- response = await openai.ChatCompletion.acreate(
87
  model="gpt-4",
88
  messages=messages,
89
  temperature=0,
90
  max_tokens=150,
91
  )
92
- sql_query = response['choices'][0]['message']['content'].strip()
93
  return sql_query
94
  except Exception as e:
95
  return f"Error generating SQL query: {e}"
 
8
  import os
9
 
10
  # Set OpenAI API key
11
+ client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
12
 
13
  # =========================
14
  # Configuration and Setup
 
83
  ]
84
 
85
  try:
86
+ response = openai.ChatCompletion.create(
87
  model="gpt-4",
88
  messages=messages,
89
  temperature=0,
90
  max_tokens=150,
91
  )
92
+ sql_query = response.choices[0].message.content.strip()
93
  return sql_query
94
  except Exception as e:
95
  return f"Error generating SQL query: {e}"