stageflow-framework 0.1.4__tar.gz → 0.1.6__tar.gz

{stageflow_framework-0.1.4 → stageflow_framework-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stageflow-framework
-Version: 0.1.4
+Version: 0.1.6
 Summary: StageFlow: Pipeline framework for stages
 Author: lethargy
 Requires-Python: >=3.10

{stageflow_framework-0.1.4 → stageflow_framework-0.1.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "stageflow-framework"
-version = "0.1.4"
+version = "0.1.6"
 description = "StageFlow: Pipeline framework for stages"
 readme = "README.md"
 authors = [

{stageflow_framework-0.1.4 → stageflow_framework-0.1.6}/stageflow/core/event.py

@@ -1,5 +1,6 @@
 from dataclasses import dataclass
 from datetime import datetime, UTC
+from .utils import schema_to_jsonable
 
 
 @dataclass
@@ -10,6 +11,15 @@ class InputSpec:
     default: str | int | float | bool | None = None
     payload_schema: object | None = None
 
+    def to_dict(self) -> dict:
+        return {
+            "type": self.type,
+            "description": self.description,
+            "required": self.required,
+            "default": self.default,
+            "payload_schema": schema_to_jsonable(self.payload_schema),
+        }
+
 
 @dataclass
 class EventSpec:
@@ -17,6 +27,13 @@ class EventSpec:
     description: str | None = None
     payload_schema: object | None = None
 
+    def to_dict(self) -> dict:
+        return {
+            "type": self.type,
+            "description": self.description,
+            "payload_schema": schema_to_jsonable(self.payload_schema),
+        }
+
 
 class Event:
     type: str

{stageflow_framework-0.1.4 → stageflow_framework-0.1.6}/stageflow/core/stage.py

@@ -158,8 +158,8 @@ class BaseStage:
         return {
             "stage_name": cls.stage_name,
             "skipable": cls.skipable,
-            "allowed_events": [e.__dict__ for e in cls.allowed_events],
-            "allowed_inputs": [i.__dict__ for i in cls.allowed_inputs],
+            "allowed_events": [e.to_dict() for e in cls.allowed_events],
+            "allowed_inputs": [i.to_dict() for i in cls.allowed_inputs],
             "category": cls.category,
             "description": parsed_description.get("description", "") if isinstance(parsed_description, dict) else "",
             "arguments": _normalize_fields(parsed_description.get("arguments", [])) if isinstance(parsed_description, dict) else [],

{stageflow_framework-0.1.4 → stageflow_framework-0.1.6}/stageflow/core/utils.py

@@ -60,3 +60,49 @@ def validate_schema(value: Any, schema: object, path: str = "payload"):
         return
     if not isinstance(value, expected):
         raise ValueError(f"{path} expected {schema}, got {type(value).__name__}")
+
+
+_TYPE_NAME_OVERRIDES = {
+    str: "str",
+    int: "int",
+    float: "float",
+    bool: "bool",
+    dict: "object",
+    list: "list",
+    type(None): "null",
+}
+
+
+def schema_to_jsonable(schema: object) -> object:
+    """Convert payload schema hint into a JSON-serializable structure."""
+    origin = get_origin(schema)
+    args = get_args(schema)
+
+    if schema is None:
+        return None
+
+    if schema is Any or schema is object:
+        return "any"
+
+    if origin is Union and args:
+        return {"anyOf": [schema_to_jsonable(arg) for arg in args]}
+
+    if origin is list and args:
+        return [schema_to_jsonable(args[0])]
+
+    if origin is dict and args:
+        return {"key": schema_to_jsonable(args[0]), "value": schema_to_jsonable(args[1])}
+
+    if isinstance(schema, dict):
+        return {k: schema_to_jsonable(v) for k, v in schema.items()}
+
+    if isinstance(schema, list):
+        return [schema_to_jsonable(v) for v in schema]
+
+    if isinstance(schema, tuple):
+        return [schema_to_jsonable(v) for v in schema]
+
+    if isinstance(schema, type):
+        return _TYPE_NAME_OVERRIDES.get(schema, getattr(schema, "__name__", str(schema)))
+
+    return str(schema)

{stageflow_framework-0.1.4 → stageflow_framework-0.1.6}/stageflow_framework.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stageflow-framework
-Version: 0.1.4
+Version: 0.1.6
 Summary: StageFlow: Pipeline framework for stages
 Author: lethargy
 Requires-Python: >=3.10

{stageflow_framework-0.1.4 → stageflow_framework-0.1.6}/tests/test_docs_schema.py

@@ -28,8 +28,20 @@ class DocStage(BaseStage):
         type: string
     """
 
-    allowed_events = [EventSpec(type="progress", description="Progress event")]
-    allowed_inputs = [InputSpec(type="user_input", description="User input")]
+    allowed_events = [
+        EventSpec(
+            type="progress",
+            description="Progress event",
+            payload_schema={"step": int, "tags": list[str]},
+        )
+    ]
+    allowed_inputs = [
+        InputSpec(
+            type="user_input",
+            description="User input",
+            payload_schema={"answer": str, "meta": {"id": int}, "tags": [str]},
+        )
+    ]
     category = "demo"
 
     async def run(self):
@@ -68,6 +80,11 @@ class DocsSchemaTests(unittest.TestCase):
 
         self.assertEqual(doc_spec["allowed_events"][0]["type"], "progress")
         self.assertEqual(doc_spec["allowed_inputs"][0]["type"], "user_input")
+        self.assertEqual(doc_spec["allowed_events"][0]["payload_schema"], {"step": "int", "tags": ["str"]})
+        self.assertEqual(
+            doc_spec["allowed_inputs"][0]["payload_schema"],
+            {"answer": "str", "meta": {"id": "int"}, "tags": ["str"]},
+        )
         self.assertEqual(doc_spec["category"], "demo")
 
     def test_generate_yaml_contains_description(self):