Spaces:
Sleeping
Sleeping
Improve error handling - check API key before client init
Browse files
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
|
|
|
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|