Alibrown commited on
Commit
c9d4ca5
·
verified ·
1 Parent(s): e1ae95a

Update app/app.py

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