amalsp commited on
Commit
0d558b8
·
verified ·
1 Parent(s): 90878de

Improve error handling - check API key before client init

Browse files
Files changed (1) hide show
  1. main.py +9 -2
main.py CHANGED
@@ -152,14 +152,21 @@ async def process_query(request: QueryRequest):
152
  session_id = request.session_id or "demo-session"
153
 
154
  # Check if Groq is available
155
- if not groq_client:
 
156
  return QueryResponse(
157
  success=False,
158
  error="GROQ_API_KEY not configured. Please add it in HF Spaces Settings > Variables.",
159
  session_id=session_id
160
  )
161
 
162
- try:
 
 
 
 
 
 
163
  # Generate SQL using Groq
164
  sql, error = generate_sql_with_groq(request.question)
165
 
 
152
  session_id = request.session_id or "demo-session"
153
 
154
  # Check if Groq is available
155
+ # Check if Groq API key is configured
156
+ if not GROQ_API_KEY:
157
  return QueryResponse(
158
  success=False,
159
  error="GROQ_API_KEY not configured. Please add it in HF Spaces Settings > Variables.",
160
  session_id=session_id
161
  )
162
 
163
+ # Check if Groq client initialized successfully
164
+ if not groq_client:
165
+ return QueryResponse(
166
+ success=False,
167
+ error="Groq client initialization failed. Running in limited mode.",
168
+ session_id=session_id
169
+ )try:
170
  # Generate SQL using Groq
171
  sql, error = generate_sql_with_groq(request.question)
172