Add workspace_health_lines for notebook UI status

Author: eddietejeda

hotdata_core_notebook/__init__.py

@@ -1,6 +1,7 @@
 """Hotdata client and shared models for notebook integrations."""
 
 from hotdata_core_notebook.client import HotdataClient, from_env
+from hotdata_core_notebook.health import workspace_health_lines
 from hotdata_core_notebook.env import (
     default_api_key,
     default_host,
@@ -15,6 +16,7 @@
 __all__ = [
     "HotdataClient",
     "QueryResult",
+    "workspace_health_lines",
     "default_api_key",
     "default_host",
     "default_session_id",

hotdata_core_notebook/health.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+
+from hotdata.exceptions import ApiException
+
+from hotdata_core_notebook.client import HotdataClient
+
+
+def workspace_health_lines(client: HotdataClient) -> tuple[bool, list[str]]:
+    """Return ``(ok, parts)`` where ``parts`` are short markdown fragments.
+
+    On failure, ``ok`` is False and ``parts`` is a single-element list with the error text.
+    """
+    try:
+        listing = client.connections().list_connections()
+        n = len(listing.connections)
+        lines = [
+            "**API** reachable",
+            f"**workspace** `{client.workspace_id}`",
+            f"**connections** {n}",
+        ]
+        if client.session_id:
+            lines.append(f"**sandbox** `{client.session_id}`")
+        return True, lines
+    except ApiException as e:
+        return False, [e.reason or str(e)]

tests/test_health.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+from unittest.mock import patch
+
+from hotdata.exceptions import ApiException
+
+from hotdata_core_notebook.client import HotdataClient
+from hotdata_core_notebook.health import workspace_health_lines
+
+
+def test_workspace_health_ok():
+    client = HotdataClient("k", "ws", host="https://api.hotdata.dev")
+    listing = type("L", (), {"connections": [object()]})()
+
+    class FakeConnectionsApi:
+        def list_connections(self):
+            return listing
+
+    with patch.object(client, "connections", return_value=FakeConnectionsApi()):
+        ok, parts = workspace_health_lines(client)
+    assert ok is True
+    assert any("reachable" in p for p in parts)
+
+
+def test_workspace_health_api_error():
+    client = HotdataClient("k", "ws", host="https://api.hotdata.dev")
+
+    class Boom:
+        def list_connections(self):
+            raise ApiException(status=500, reason="nope")
+
+    with patch.object(client, "connections", return_value=Boom()):
+        ok, parts = workspace_health_lines(client)
+    assert ok is False
+    assert parts == ["nope"]