MCP: Update to @bytebase/dbhub@0.2.3

- New resource `db://schemas`
- New argument `schema` for prompt `generate_sql`
- New arguments `schema` and `table` for prompt `explain_db`

Author: amotl

framework/mcp/example_dbhub.py

@@ -14,7 +14,7 @@
     command="npx",
     args=[
         "-y",
-        "@bytebase/dbhub@0.1",
+        "@bytebase/dbhub@0.2.3",
         "--transport=stdio",
         #"--transport=sse",
         #"--port=8080",
@@ -43,18 +43,21 @@ async def run():
 
             # Validate database content.
             db = DatabaseAdapter("crate://crate@localhost:4200/")
-            db.run_sql("CREATE TABLE IF NOT EXISTS public.testdrive (id INT, data TEXT)")
-            db.run_sql("INSERT INTO public.testdrive (id, data) VALUES (42, 'Hotzenplotz')")
+            db.run_sql("CREATE TABLE IF NOT EXISTS testdrive.dbhub (id INT, data TEXT)")
+            db.run_sql("INSERT INTO testdrive.dbhub (id, data) VALUES (42, 'Hotzenplotz')")
             db.refresh_table("public.testdrive")
 
-            # Read a few resources.
-            # FIXME: Only works on schema=public, because the PostgreSQL adapter hard-codes `WHERE table_schema = 'public'`.
-            # https://github.com/bytebase/dbhub/blob/09424c8513c8c7bef7f66377b46a2b93a69a57d2/src/connectors/postgres/index.ts#L89-L107
-            await client.read_resource("db://tables")
+            # Read available resources.
+            await client.read_resource("db://schemas")
 
-            # Get a few prompts.
-            await client.get_prompt("generate_sql", arguments={"description": "Please enumerate the highest five mountains.", "dialect": "postgres"})
-            await client.get_prompt("explain_db", arguments={"target": "testdrive"})
+            # Invoke available prompts.
+            await client.get_prompt("generate_sql", arguments={
+                "description": "Please enumerate the highest five mountains.",
+                "dialect": "postgres",
+                "schema": "sys",
+            })
+            await client.get_prompt("explain_db", arguments={"schema": "testdrive"})
+            await client.get_prompt("explain_db", arguments={"schema": "testdrive", "table": "dbhub"})
 
 
 if __name__ == "__main__":

framework/mcp/test.py

@@ -108,9 +108,11 @@ def test_dbhub():
     assert b"Calling tool: list_connectors" in p.stdout
     assert b"dsn: postgres://postgres" in p.stdout
 
-    assert b"Reading resource: db://tables" in p.stdout
+    assert b"Reading resource: db://schemas" in p.stdout
+    assert b"- doc" in p.stdout
+    assert b"- sys" in p.stdout
     assert b"- testdrive" in p.stdout
 
     assert b"Getting prompt: explain_db" in p.stdout
-    assert b"Table: testdrive" in p.stdout
+    assert b"Table: dbhub in schema 'testdrive'" in p.stdout
     assert b"Structure:\\n- id (integer)\\n- data (text)" in p.stdout