abstractgateway 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

abstractgateway/hosts/visualflow_host.py

@@ -6,7 +6,7 @@ from dataclasses import dataclass
 from pathlib import Path
 from typing import Any, Dict, Optional
 
-from abstractruntime import Runtime
+from abstractruntime import Runtime, persist_workflow_snapshot
 
 
 logger = logging.getLogger(__name__)
@@ -129,6 +129,7 @@ class VisualFlowGatewayHost:
         input_data: Dict[str, Any],
         actor_id: str = "gateway",
         bundle_id: Optional[str] = None,
+        session_id: Optional[str] = None,
     ) -> str:
         # bundle_id is ignored for VisualFlow sources (kept for API compatibility with bundle mode).
         del bundle_id
@@ -155,7 +156,34 @@ class VisualFlowGatewayHost:
             artifact_store=self._artifact_store,
             tool_executor=tool_executor,
         )
-        return str(vis_runner.start(data, actor_id=actor_id))
+        sid = str(session_id).strip() if isinstance(session_id, str) and session_id.strip() else None
+        run_id = str(vis_runner.start(data, actor_id=actor_id, session_id=sid))
+
+        # Persist a workflow snapshot for reproducible replay (best-effort).
+        try:
+            flow_dict = None
+            try:
+                to_dump = getattr(visual_flow, "model_dump", None)
+                if callable(to_dump):
+                    flow_dict = to_dump()
+                else:
+                    flow_dict = dict(visual_flow) if isinstance(visual_flow, dict) else None
+            except Exception:
+                flow_dict = None
+            if isinstance(flow_dict, dict):
+                snapshot = {"kind": "visualflow_json", "flow_id": fid, "visualflow": flow_dict}
+                persist_workflow_snapshot(
+                    run_store=self._run_store,
+                    artifact_store=self._artifact_store,
+                    run_id=str(run_id),
+                    workflow_id=str(fid),
+                    snapshot=snapshot,
+                    format="visualflow_json",
+                )
+        except Exception:
+            pass
+
+        return run_id
 
     def runtime_and_workflow_for_run(self, run_id: str) -> tuple[Runtime, Any]:
         """Return (runtime, workflow_spec) for the given run, ensuring derived VisualFlow workflows are registered."""
@@ -210,4 +238,3 @@ class VisualFlowGatewayHost:
         if wf is None:
             raise KeyError(f"Workflow '{workflow_id}' not registered")
         return (runtime, wf)
-

abstractgateway/integrations/__init__.py

@@ -0,0 +1,2 @@
+"""Optional external integrations for AbstractGateway."""
+