Melika Kheirieh commited on
Commit
e6abe8c
·
1 Parent(s): 6f6e439

fix(planner): increase OpenAI timeout & add graceful timeout handling

Browse files
Files changed (1) hide show
  1. 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