feat: add database= parameter to execute_sql for managed database scoping

Pass database= to client.execute_sql() so queries are scoped to a
managed database via the X-Database-Id header (hotdata-runtime>=0.2.1).
Also updates ManagedDatabase constructor calls to use description= and
default_connection_id= fields introduced in hotdata-runtime v0.2.0.

Author: eddietejeda

hotdata_llamaindex/databases.py

@@ -16,9 +16,9 @@
 def list_managed_databases_json(client: HotdataClient) -> str:
     rows = [
         {
-            "name": db.name,
+            "description": db.description,
             "id": db.id,
-            "sql_prefix": f"{db.name}.{{schema}}.{{table}}",
+            "sql_prefix": f"{db.id}.{{schema}}.{{table}}",
         }
         for db in client.list_managed_databases()
     ]
@@ -32,7 +32,7 @@ def create_managed_database(
     schema: str = DEFAULT_SCHEMA,
     tables: list[str] | None = None,
 ) -> ManagedDatabase:
-    return client.create_managed_database(name, schema=schema, tables=tables)
+    return client.create_managed_database(description=name, schema=schema, tables=tables)
 
 
 def load_managed_table(
@@ -47,7 +47,7 @@ def load_managed_table(
 
 
 def managed_database_summary(db: ManagedDatabase) -> dict[str, str]:
-    return {"id": db.id, "name": db.name, "source_type": db.source_type}
+    return {"id": db.id, "description": db.description or db.id}
 
 
 def load_result_summary(result: LoadManagedTableResult) -> dict[str, Any]:

hotdata_llamaindex/tools.py

@@ -22,8 +22,14 @@ def result_rows_for_llm(result: QueryResult, *, max_rows: int = 20) -> list[dict
     return result.to_records(max_rows=max_rows)
 
 
-def execute_sql_json(client: HotdataClient, sql: str, *, max_rows: int = 100) -> str:
-    result = client.execute_sql(sql)
+def execute_sql_json(
+    client: HotdataClient,
+    sql: str,
+    *,
+    max_rows: int = 100,
+    database: str | None = None,
+) -> str:
+    result = client.execute_sql(sql, database=database)
     payload = {
         "metadata": result.metadata_dict(),
         "rows": result.to_records(max_rows=max_rows),
@@ -35,12 +41,13 @@ def make_hotdata_tools(
     client: HotdataClient,
     *,
     max_rows: int = 100,
+    database: str | None = None,
 ) -> list[FunctionTool]:
     """Return LlamaIndex tools for SQL and managed database workflows."""
 
     def hotdata_execute_sql(sql: str) -> str:
         """Run SQL against the Hotdata workspace and return JSON rows."""
-        return execute_sql_json(client, sql, max_rows=max_rows)
+        return execute_sql_json(client, sql, max_rows=max_rows, database=database)
 
     def hotdata_list_managed_databases() -> str:
         """List Hotdata-managed databases in the workspace."""

pyproject.toml

@@ -10,7 +10,7 @@ readme = "README.md"
 requires-python = ">=3.10"
 license = { text = "MIT" }
 dependencies = [
-  "hotdata-runtime>=0.1.1",
+  "hotdata-runtime>=0.2.0",
   "hotdata>=0.2.0",
   "llama-index-core>=0.12.0",
 ]

tests/test_tools.py

@@ -21,30 +21,35 @@ def test_execute_sql_json(mock_client, sample_result):
     payload = json.loads(execute_sql_json(mock_client, "select 1"))
     assert payload["metadata"]["row_count"] == 2
     assert payload["rows"] == [{"n": 1}, {"n": 2}]
-    mock_client.execute_sql.assert_called_once_with("select 1")
+    mock_client.execute_sql.assert_called_once_with("select 1", database=None)
+
+
+def test_execute_sql_json_with_database(mock_client, sample_result):
+    execute_sql_json(mock_client, "select 1", database="my_db")
+    mock_client.execute_sql.assert_called_once_with("select 1", database="my_db")
 
 
 def test_list_managed_databases_json(mock_client):
     mock_client.list_managed_databases.return_value = [
-        ManagedDatabase(id="c1", name="sales", source_type="managed"),
+        ManagedDatabase(id="c1", description="sales", default_connection_id="conn_c1"),
     ]
     payload = json.loads(list_managed_databases_json(mock_client))
-    assert payload[0]["name"] == "sales"
+    assert payload[0]["description"] == "sales"
 
 
 def test_create_managed_database_delegates(mock_client):
     mock_client.create_managed_database.return_value = ManagedDatabase(
         id="c1",
-        name="sales",
-        source_type="managed",
+        description="sales",
+        default_connection_id="conn_c1",
     )
     db = create_managed_database(mock_client, name="sales", tables=["orders"])
     mock_client.create_managed_database.assert_called_once_with(
-        "sales",
+        description="sales",
         schema="public",
         tables=["orders"],
     )
-    assert db.name == "sales"
+    assert db.description == "sales"
 
 
 def test_load_managed_table_delegates(mock_client):
@@ -73,8 +78,8 @@ def test_load_managed_table_delegates(mock_client):
 def test_make_hotdata_tools(mock_client, sample_result):
     mock_client.create_managed_database.return_value = ManagedDatabase(
         id="c1",
-        name="sales",
-        source_type="managed",
+        description="sales",
+        default_connection_id="conn_c1",
     )
     mock_client.load_managed_table.return_value = LoadManagedTableResult(
         connection_id="c1",