Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
12 |
|
13 |
# =========================
|
14 |
# Configuration and Setup
|
@@ -83,13 +83,13 @@ async def parse_query(nl_query):
|
|
83 |
]
|
84 |
|
85 |
try:
|
86 |
-
response =
|
87 |
model="gpt-4",
|
88 |
messages=messages,
|
89 |
temperature=0,
|
90 |
max_tokens=150,
|
91 |
)
|
92 |
-
sql_query = response
|
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}"
|