fleet-python 0.2.117__tar.gz → 0.2.118__tar.gz

{fleet_python-0.2.117/fleet_python.egg-info → fleet_python-0.2.118}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.117
+Version: 0.2.118
 Summary: Python SDK for Fleet environments
 Author-email: Fleet AI <nic@fleet.so>
 License: Apache-2.0

{fleet_python-0.2.117 → fleet_python-0.2.118}/fleet/__init__.py

@@ -40,6 +40,7 @@ from .verifiers import (
     TASK_SUCCESSFUL_SCORE,
     execute_verifier_local,
     LocalEnvironment,
+    diff_dbs,
 )
 
 # Import async verifiers (default verifier is async for modern usage)
@@ -78,7 +79,7 @@ from . import env
 from . import global_client as _global_client
 from ._async import global_client as _async_global_client
 
-__version__ = "0.2.117"
+__version__ = "0.2.118"
 
 __all__ = [
     # Core classes
@@ -118,6 +119,7 @@ __all__ = [
     "TASK_SUCCESSFUL_SCORE",
     "execute_verifier_local",
     "LocalEnvironment",
+    "diff_dbs",
     # Environment module
     "env",
     # Global client helpers

{fleet_python-0.2.117 → fleet_python-0.2.118}/fleet/_async/__init__.py

@@ -44,7 +44,7 @@ from ..types import VerifierFunction
 from .. import env
 from . import global_client as _async_global_client
 
-__version__ = "0.2.117"
+__version__ = "0.2.118"
 
 __all__ = [
     # Core classes

{fleet_python-0.2.117 → fleet_python-0.2.118}/fleet/_async/base.py

@@ -26,7 +26,7 @@ from .exceptions import (
 try:
     from .. import __version__
 except ImportError:
-    __version__ = "0.2.117"
+    __version__ = "0.2.118"
 
 logger = logging.getLogger(__name__)
 

{fleet_python-0.2.117 → fleet_python-0.2.118}/fleet/_async/client.py

@@ -871,6 +871,35 @@ class AsyncFleet:
             execute_verifier_local, verifier_func, seed_db, current_db, final_answer
         )
 
+    @staticmethod
+    async def diff_dbs(
+        seed_db: str,
+        current_db: str,
+        ignore_tables: Optional[set] = None,
+        ignore_table_fields: Optional[Dict[str, set]] = None,
+    ) -> Dict[str, Any]:
+        """Compute a structured diff between two local SQLite databases.
+
+        Returns the same format as the runner's ``/diff/structured`` endpoint.
+        No authentication or network access required.
+
+        Args:
+            seed_db: Path to the seed (before) SQLite database file.
+            current_db: Path to the current (after) SQLite database file.
+            ignore_tables: Optional set of table names to skip entirely.
+            ignore_table_fields: Optional mapping of ``{table: {field, ...}}``
+                to strip from the output.
+
+        Returns:
+            Dict with keys ``success``, ``diff``, and ``message``.
+        """
+        import asyncio
+        from ..verifiers.local_executor import diff_dbs
+
+        return await asyncio.to_thread(
+            diff_dbs, seed_db, current_db, ignore_tables, ignore_table_fields
+        )
+
     async def list_runs(
         self, profile_id: Optional[str] = None, status: Optional[str] = "active"
     ) -> List[Run]:

{fleet_python-0.2.117 → fleet_python-0.2.118}/fleet/base.py

@@ -27,7 +27,7 @@ from .exceptions import (
 try:
     from . import __version__
 except ImportError:
-    __version__ = "0.2.117"
+    __version__ = "0.2.118"
 
 logger = logging.getLogger(__name__)
 

{fleet_python-0.2.117 → fleet_python-0.2.118}/fleet/client.py

@@ -880,6 +880,32 @@ class Fleet:
 
         return execute_verifier_local(verifier_func, seed_db, current_db, final_answer)
 
+    @staticmethod
+    def diff_dbs(
+        seed_db: str,
+        current_db: str,
+        ignore_tables: Optional[set] = None,
+        ignore_table_fields: Optional[Dict[str, set]] = None,
+    ) -> Dict[str, Any]:
+        """Compute a structured diff between two local SQLite databases.
+
+        Returns the same format as the runner's ``/diff/structured`` endpoint.
+        No authentication or network access required.
+
+        Args:
+            seed_db: Path to the seed (before) SQLite database file.
+            current_db: Path to the current (after) SQLite database file.
+            ignore_tables: Optional set of table names to skip entirely.
+            ignore_table_fields: Optional mapping of ``{table: {field, ...}}``
+                to strip from the output.
+
+        Returns:
+            Dict with keys ``success``, ``diff``, and ``message``.
+        """
+        from .verifiers.local_executor import diff_dbs
+
+        return diff_dbs(seed_db, current_db, ignore_tables, ignore_table_fields)
+
     def list_runs(
         self, profile_id: Optional[str] = None, status: Optional[str] = "active"
     ) -> List[Run]:

{fleet_python-0.2.117 → fleet_python-0.2.118}/fleet/verifiers/__init__.py

@@ -6,7 +6,7 @@ from .verifier import (
     verifier,
     SyncVerifierFunction,
 )
-from .local_executor import execute_verifier_local, LocalEnvironment
+from .local_executor import execute_verifier_local, LocalEnvironment, diff_dbs
 
 __all__ = [
     "DatabaseSnapshot",
@@ -18,4 +18,5 @@ __all__ = [
     "SyncVerifierFunction",
     "execute_verifier_local",
     "LocalEnvironment",
+    "diff_dbs",
 ]

{fleet_python-0.2.117 → fleet_python-0.2.118}/fleet/verifiers/local_executor.py

@@ -1,7 +1,8 @@
-"""Local verifier execution against SQLite files.
+"""Local verifier execution and database diffing against SQLite files.
 
-Executes verifier function code directly against local SQLite database files,
-without requiring authentication or a remote runner API server.
+Executes verifier function code directly against local SQLite database files
+and computes structured diffs, without requiring authentication or a remote
+runner API server.
 """
 
 import inspect
@@ -245,3 +246,126 @@ def execute_verifier_local(
             "error": error_msg,
             "stdout": captured_stdout.getvalue(),
         }
+
+
+# ---------------------------------------------------------------------------
+#  Structured database diff (matches /diff/structured response format)
+# ---------------------------------------------------------------------------
+
+def diff_dbs(
+    seed_db: str,
+    current_db: str,
+    ignore_tables: Optional[set] = None,
+    ignore_table_fields: Optional[dict] = None,
+) -> Dict[str, Any]:
+    """Compute a structured diff between two SQLite databases locally.
+
+    Returns the exact same format as the runner's ``/diff/structured`` endpoint:
+
+    .. code-block:: python
+
+        {
+            "success": True,
+            "diff": {
+                "table_name": {
+                    "table_name": str,
+                    "primary_key": [str],
+                    "added_rows": [{"row_id": ..., "data": {...}}],
+                    "removed_rows": [{"row_id": ..., "data": {...}}],
+                    "modified_rows": [{"row_id": ..., "changes": {...}, "data": {...}}],
+                    "unchanged_count": int,
+                    "total_changes": int,
+                }
+            },
+            "message": str,
+        }
+
+    No authentication or network access required.
+
+    Args:
+        seed_db: Path to the seed (before) SQLite database file.
+        current_db: Path to the current (after) SQLite database file.
+        ignore_tables: Optional set of table names to skip entirely.
+        ignore_table_fields: Optional mapping of ``{table_name: {field, ...}}``
+            whose fields are stripped from the diff output.
+
+    Returns:
+        Dict matching the ``StructuredDiffResponse`` schema.
+    """
+    from .sql_differ import SQLiteDiffer
+
+    ignore_tables = ignore_tables or set()
+    ignore_table_fields = ignore_table_fields or {}
+
+    try:
+        differ = SQLiteDiffer(seed_db, current_db)
+        raw_diff = differ.diff_all_tables()
+
+        filtered_diff: Dict[str, Any] = {}
+        for table_name, table_diff in raw_diff.items():
+            if table_name in ignore_tables:
+                continue
+
+            # Skip tables that errored during diffing
+            if "error" in table_diff:
+                continue
+
+            ignored_fields = ignore_table_fields.get(table_name, set())
+
+            # Added rows
+            filtered_added = []
+            for row in table_diff.get("added_rows", []):
+                filtered_data = {
+                    k: v for k, v in row["data"].items() if k not in ignored_fields
+                }
+                filtered_added.append({"row_id": row["row_id"], "data": filtered_data})
+
+            # Removed rows
+            filtered_removed = []
+            for row in table_diff.get("removed_rows", []):
+                filtered_data = {
+                    k: v for k, v in row["data"].items() if k not in ignored_fields
+                }
+                filtered_removed.append({"row_id": row["row_id"], "data": filtered_data})
+
+            # Modified rows
+            filtered_modified = []
+            for row in table_diff.get("modified_rows", []):
+                filtered_changes = {
+                    k: v for k, v in row["changes"].items() if k not in ignored_fields
+                }
+                if filtered_changes:
+                    after_row = row.get("after_row", {})
+                    filtered_data = {
+                        k: v for k, v in after_row.items() if k not in ignored_fields
+                    }
+                    filtered_modified.append({
+                        "row_id": row["row_id"],
+                        "changes": filtered_changes,
+                        "data": filtered_data,
+                    })
+
+            total_changes = len(filtered_added) + len(filtered_removed) + len(filtered_modified)
+
+            filtered_diff[table_name] = {
+                "table_name": table_name,
+                "primary_key": table_diff.get("primary_key", []),
+                "added_rows": filtered_added,
+                "removed_rows": filtered_removed,
+                "modified_rows": filtered_modified,
+                "unchanged_count": table_diff.get("unchanged_count", 0),
+                "total_changes": total_changes,
+            }
+
+        return {
+            "success": True,
+            "diff": filtered_diff,
+            "message": "Structured diff generated successfully",
+        }
+
+    except Exception as e:
+        return {
+            "success": False,
+            "diff": {},
+            "message": f"Failed to generate structured diff: {e}",
+        }

{fleet_python-0.2.117 → fleet_python-0.2.118/fleet_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.117
+Version: 0.2.118
 Summary: Python SDK for Fleet environments
 Author-email: Fleet AI <nic@fleet.so>
 License: Apache-2.0

{fleet_python-0.2.117 → fleet_python-0.2.118}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "fleet-python"
 
-version = "0.2.117"
+version = "0.2.118"
 description = "Python SDK for Fleet environments"
 authors = [
     {name = "Fleet AI", email = "nic@fleet.so"},