Bound connector source previews

src/govoplan_connectors/backend/datasource_origins.py

@@ -7,6 +7,7 @@ from govoplan_core.core.datasources import (
     DatasourceOrigin,
     DatasourceOriginReadRequest,
     DatasourceOriginReadResult,
+    DatasourceUnavailableError,
     DatasourceValidationError,
 )
 from govoplan_core.core.tabular_sources import (
@@ -15,6 +16,7 @@ from govoplan_core.core.tabular_sources import (
     TabularSourceAccessError,
     TabularSourceError,
     TabularSourceNotFoundError,
+    TabularSourceUnavailableError,
     TabularSourceValidationError,
 )
 from govoplan_connectors.backend.tabular_sources import SqlTabularSourceProvider
@@ -79,6 +81,8 @@ class ConnectorDatasourceOriginProvider:
                     offset=request.offset,
                     columns=request.columns,
                     expected_fingerprint=request.expected_fingerprint,
+                    max_bytes=request.max_bytes,
+                    timeout_ms=request.timeout_ms,
                 ),
             )
         except TabularSourceError as exc:
@@ -88,6 +92,12 @@ class ConnectorDatasourceOriginProvider:
             rows=result.rows,
             total_rows=result.total_rows,
             truncated=result.truncated,
+            returned_bytes=result.returned_bytes,
+            elapsed_ms=result.elapsed_ms,
+            effective_row_limit=result.effective_row_limit,
+            effective_byte_limit=result.effective_byte_limit,
+            effective_timeout_ms=result.effective_timeout_ms,
+            diagnostics=result.diagnostics,
         )
 
 
@@ -116,6 +126,9 @@ def _origin(source: TabularSource) -> DatasourceOrigin:
         updated_at=source.updated_at,
         capabilities=source.capabilities,
         metadata=dict(source.metadata),
+        source_mode=source.source_mode,
+        pushdown=source.pushdown,
+        health=source.health,
     )
 
 
@@ -124,6 +137,8 @@ def _datasource_error(exc: TabularSourceError):
         return DatasourceAccessError(str(exc))
     if isinstance(exc, TabularSourceNotFoundError):
         return DatasourceNotFoundError(str(exc))
+    if isinstance(exc, TabularSourceUnavailableError):
+        return DatasourceUnavailableError(str(exc))
     if isinstance(exc, TabularSourceValidationError):
         return DatasourceValidationError(str(exc))
     return DatasourceValidationError(str(exc))