Spaces:
Sleeping
Sleeping
Melika Kheirieh
commited on
Commit
·
e6abe8c
1
Parent(s):
6f6e439
fix(planner): increase OpenAI timeout & add graceful timeout handling
Browse files- nl2sql/planner.py +1 -3
nl2sql/planner.py
CHANGED
|
@@ -125,9 +125,7 @@ class Planner:
|
|
| 125 |
else:
|
| 126 |
# Call with increased timeout
|
| 127 |
plan_text, pin, pout, cost = self.llm.plan(
|
| 128 |
-
user_query=user_query,
|
| 129 |
-
schema_preview=trimmed,
|
| 130 |
-
timeout=120, # Increase timeout for large schemas
|
| 131 |
)
|
| 132 |
self._plan_cache[key] = (plan_text, pin, pout, cost)
|
| 133 |
|
|
|
|
| 125 |
else:
|
| 126 |
# Call with increased timeout
|
| 127 |
plan_text, pin, pout, cost = self.llm.plan(
|
| 128 |
+
user_query=user_query, schema_preview=trimmed
|
|
|
|
|
|
|
| 129 |
)
|
| 130 |
self._plan_cache[key] = (plan_text, pin, pout, cost)
|
| 131 |
|