Spaces:
Running
Running
Update app/app.py
Browse files- app/app.py +6 -0
app/app.py
CHANGED
|
@@ -96,6 +96,12 @@ async def api_endpoint():
|
|
| 96 |
if tool_name == "health_check":
|
| 97 |
return jsonify({"result": {"status": "ok"}})
|
| 98 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 99 |
# rename 'provider' → 'provider_name' for tools.run()
|
| 100 |
if "provider" in params:
|
| 101 |
params["provider_name"] = params.pop("provider")
|
|
|
|
| 96 |
if tool_name == "health_check":
|
| 97 |
return jsonify({"result": {"status": "ok"}})
|
| 98 |
|
| 99 |
+
# db_query — handled by db_sync directly, not tools.run()
|
| 100 |
+
if tool_name == "db_query":
|
| 101 |
+
sql = params.get("sql", "")
|
| 102 |
+
result = await db_sync.query(sql)
|
| 103 |
+
return jsonify({"result": result})
|
| 104 |
+
|
| 105 |
# rename 'provider' → 'provider_name' for tools.run()
|
| 106 |
if "provider" in params:
|
| 107 |
params["provider_name"] = params.pop("provider")
|