feat: add database= parameter to sql engine and editor 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).
- HotdataMarimoEngine: add default_database= constructor param, pass to execute()
- SqlEditor: add database= constructor param, pass to both execute_sql calls
- ManagedDatabaseWriter: use description= kwarg matching ManagedDatabase v0.2.0 API
- Fix test_databases_marimo.py syntax error and update assertions

Author: eddietejeda

hotdata_marimo/databases.py

@@ -47,7 +47,7 @@ def databases_panel(client: HotdataClient):
             gap=1,
         )
     rows: list[dict[str, object]] = [
-        {"name": db.name, "id": db.id, "sql_prefix": f"{db.name}.{{schema}}.{{table}}"}
+        {"description": db.description or db.id, "id": db.id, "sql_prefix": f"{db.id}.{{schema}}.{{table}}"}
         for db in dbs
     ]
     return mo.vstack(
@@ -127,8 +127,8 @@ def _rebuild_database_pick(self) -> None:
                 message="(create one first)",
             )
             return
-        options = {db.name: db.name for db in dbs}
-        value = current if current in options else next(iter(options))
+        options = {db.description or db.id: db.id for db in dbs}
+        value = current if current in options.values() else next(iter(options.values()))
         self.database = mo.ui.dropdown(
             options=options,
             label="Database",
@@ -153,7 +153,7 @@ def _maybe_create(self) -> None:
         tables = _parse_table_names(self.tables.value)
         try:
             self._create_result = self._client.create_managed_database(
-                db_name,
+                description=db_name,
                 schema=schema,
                 tables=tables or None,
             )
@@ -209,7 +209,7 @@ def result_panel(self):
             db = self._create_result
             return mo.callout(
                 mo.md(
-                    f"Created **{db.name}** (`{db.id}`). "
+                    f"Created **{db.description or db.id}** (`{db.id}`). "
                     "Load parquet into a declared table below."
                 ),
                 kind="success",

hotdata_marimo/sql_editor.py

@@ -20,8 +20,10 @@ def __init__(
         default_sql: str = "",
         label: str = "SQL",
         run_label: str = "Run on Hotdata",
+        database: str | None = None,
     ) -> None:
         self._client = client
+        self._database = database
         self.sql = mo.ui.text_area(default_sql, label=label)
         self.run = mo.ui.button(
             value=0,
@@ -103,7 +105,7 @@ def _execute_or_cached(self) -> QueryResult | None:
                 title="Running on Hotdata",
                 subtitle="Re-running last query and waiting for results…",
             ):
-                result = self._client.execute_sql(self._cached_sql or "")
+                result = self._client.execute_sql(self._cached_sql or "", database=self._database)
             self._result_cache = result
             self._last_rerun_n = rerun_n
             return result
@@ -113,7 +115,7 @@ def _execute_or_cached(self) -> QueryResult | None:
                 title="Running on Hotdata",
                 subtitle="Executing query and waiting for results…",
             ):
-                result = self._client.execute_sql(sql_text)
+                result = self._client.execute_sql(sql_text, database=self._database)
             self._result_cache = result
             self._cached_sql = sql_text
             self._last_run_n = run_n
@@ -195,7 +197,8 @@ def sql_editor(
     default_sql: str = "",
     label: str = "SQL",
     run_label: str = "Run on Hotdata",
+    database: str | None = None,
 ) -> SqlEditor:
     return SqlEditor(
-        client, default_sql=default_sql, label=label, run_label=run_label
+        client, default_sql=default_sql, label=label, run_label=run_label, database=database
     )

hotdata_marimo/sql_engine.py

@@ -37,9 +37,12 @@ def __init__(
         self,
         connection: HotdataClient,
         engine_name: VariableName | None = None,
+        *,
+        default_database: str | None = None,
     ) -> None:
         super().__init__(connection, engine_name)
         self._connections_cache: list[Any] | None = None
+        self._default_database = default_database
 
     @property
     def source(self) -> str:
@@ -291,7 +294,7 @@ def get_table_details(
         )
 
     def execute(self, query: str) -> Any:
-        qr = self._connection.execute_sql(query)
+        qr = self._connection.execute_sql(query, database=self._default_database)
         fmt = self.sql_output_format()
 
         def to_polars() -> Any:

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.1",
   "hotdata>=0.2.0",
   "marimo>=0.10.0",
 ]

tests/test_databases_marimo.py

@@ -17,7 +17,7 @@ def test_databases_panel_empty_state(mock_client):
 
 def test_databases_panel_lists_managed_databases(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"),
     ]
     with patch("hotdata_marimo.databases.mo.vstack", return_value="panel"), patch(
         "hotdata_marimo.databases.mo.md", side_effect=lambda x: x
@@ -30,8 +30,8 @@ def test_managed_database_writer_creates_database(mock_client):
     mock_client.list_managed_databases.return_value = []
     mock_client.create_managed_database.return_value = ManagedDatabase(
         id="conn_new",
-        name="sales",
-        source_type="managed",
+        description="sales",
+        default_connection_id="conn_c1",
     )
     create = MagicMock()
     create.value = 1
@@ -71,7 +71,7 @@ def test_managed_database_writer_creates_database(mock_client):
         panel = writer.result_panel
 
     mock_client.create_managed_database.assert_called_once_with(
-        "sales",
+        description="sales",
         schema="public",
         tables=["orders", "customers"],
     )
@@ -80,7 +80,7 @@ def test_managed_database_writer_creates_database(mock_client):
 
 def test_managed_database_writer_loads_parquet(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"),
     ]
     mock_client.upload_parquet.return_value = "upl_1"
     mock_client.load_managed_table.return_value = LoadManagedTableResult(