airbyte-internal-ops 0.10.1__tar.gz → 0.10.2__tar.gz

{airbyte_internal_ops-0.10.1 → airbyte_internal_ops-0.10.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airbyte-internal-ops
-Version: 0.10.1
+Version: 0.10.2
 Summary: MCP and API interfaces that let the agents do the admin work
 Author-email: Aaron Steers <aj@airbyte.io>
 Keywords: admin,airbyte,api,mcp

{airbyte_internal_ops-0.10.1 → airbyte_internal_ops-0.10.2}/src/airbyte_ops_mcp/mcp/connector_rollout.py

@@ -843,25 +843,25 @@ def query_prod_rollout_monitoring_stats(
     )
 
     data = response.get("data", {})
-    actor_selection_info_data = data.get("actorSelectionInfo", {})
+    actor_selection_info_data = data.get("actor_selection_info", {})
     syncs_data = data.get("syncs", {})
 
     actor_selection_info = RolloutActorSelectionInfo(
-        num_actors=actor_selection_info_data.get("numActors", 0),
+        num_actors=actor_selection_info_data.get("num_actors", 0),
         num_pinned_to_connector_rollout=actor_selection_info_data.get(
-            "numPinnedToConnectorRollout", 0
+            "num_pinned_to_connector_rollout", 0
         ),
         num_actors_eligible_or_already_pinned=actor_selection_info_data.get(
-            "numActorsEligibleOrAlreadyPinned", 0
+            "num_actors_eligible_or_already_pinned", 0
         ),
     )
 
     actor_sync_stats = [
         RolloutActorSyncStats(
             actor_id=actor_id,
-            num_connections=sync_info.get("numConnections", 0),
-            num_succeeded=sync_info.get("numSucceeded", 0),
-            num_failed=sync_info.get("numFailed", 0),
+            num_connections=sync_info.get("num_connections", 0),
+            num_succeeded=sync_info.get("num_succeeded", 0),
+            num_failed=sync_info.get("num_failed", 0),
         )
         for actor_id, sync_info in syncs_data.items()
     ]

{airbyte_internal_ops-0.10.1 → airbyte_internal_ops-0.10.2}/tests/test_connector_rollout_monitoring.py

@@ -21,21 +21,21 @@ from airbyte_ops_mcp.mcp.connector_rollout import (
         pytest.param(
             {
                 "data": {
-                    "actorSelectionInfo": {
-                        "numActors": 100,
-                        "numPinnedToConnectorRollout": 10,
-                        "numActorsEligibleOrAlreadyPinned": 50,
+                    "actor_selection_info": {
+                        "num_actors": 100,
+                        "num_pinned_to_connector_rollout": 10,
+                        "num_actors_eligible_or_already_pinned": 50,
                     },
                     "syncs": {
                         "actor-1": {
-                            "numConnections": 5,
-                            "numSucceeded": 10,
-                            "numFailed": 2,
+                            "num_connections": 5,
+                            "num_succeeded": 10,
+                            "num_failed": 2,
                         },
                         "actor-2": {
-                            "numConnections": 3,
-                            "numSucceeded": 8,
-                            "numFailed": 0,
+                            "num_connections": 3,
+                            "num_succeeded": 8,
+                            "num_failed": 0,
                         },
                     },
                 }
@@ -61,10 +61,10 @@ from airbyte_ops_mcp.mcp.connector_rollout import (
         pytest.param(
             {
                 "data": {
-                    "actorSelectionInfo": {
-                        "numActors": 0,
-                        "numPinnedToConnectorRollout": 0,
-                        "numActorsEligibleOrAlreadyPinned": 0,
+                    "actor_selection_info": {
+                        "num_actors": 0,
+                        "num_pinned_to_connector_rollout": 0,
+                        "num_actors_eligible_or_already_pinned": 0,
                     },
                     "syncs": {},
                 }
@@ -78,7 +78,7 @@ from airbyte_ops_mcp.mcp.connector_rollout import (
             id="empty_rollout_no_actors",
         ),
         pytest.param(
-            {"data": {"actorSelectionInfo": {}, "syncs": {}}},
+            {"data": {"actor_selection_info": {}, "syncs": {}}},
             RolloutActorSelectionInfo(
                 num_actors=0,
                 num_pinned_to_connector_rollout=0,
@@ -136,7 +136,7 @@ def test_get_actor_sync_info_api_call() -> None:
         mock_response = MagicMock()
         mock_response.status_code = 200
         mock_response.json.return_value = {
-            "data": {"actorSelectionInfo": {}, "syncs": {}}
+            "data": {"actor_selection_info": {}, "syncs": {}}
         }
         mock_post.return_value = mock_response
 
@@ -154,7 +154,7 @@ def test_get_actor_sync_info_api_call() -> None:
         )
         assert call_args[1]["json"] == {"id": "test-rollout-id"}
         assert "Authorization" in call_args[1]["headers"]
-        assert result == {"data": {"actorSelectionInfo": {}, "syncs": {}}}
+        assert result == {"data": {"actor_selection_info": {}, "syncs": {}}}
 
 
 @pytest.mark.unit