Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -74,12 +74,12 @@ def parse_query(nl_query):
|
|
74 |
|
75 |
try:
|
76 |
response = openai.chat.completions.create(
|
77 |
-
model="gpt-
|
78 |
messages=messages,
|
79 |
temperature=0,
|
80 |
max_tokens=150,
|
81 |
)
|
82 |
-
sql_query = response.choices[0].message
|
83 |
return sql_query
|
84 |
except Exception as e:
|
85 |
return f"Error generating SQL query: {e}"
|
|
|
74 |
|
75 |
try:
|
76 |
response = openai.chat.completions.create(
|
77 |
+
model="gpt-4o-mini",
|
78 |
messages=messages,
|
79 |
temperature=0,
|
80 |
max_tokens=150,
|
81 |
)
|
82 |
+
sql_query = response.choices[0].message.content.strip()
|
83 |
return sql_query
|
84 |
except Exception as e:
|
85 |
return f"Error generating SQL query: {e}"
|