lda.chat/lda-wf
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: describe structured foreach context

Browse Source
This commit is contained in:
lda
2026-09-05 00:16:25 +07:00 Verified
parent f6e5190696
commit 77537ff35a
6 changed files with 525 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/wf_core/analysis/__init__.py
+6
View File
@@ -4,6 +4,9 @@ from .context_scopes import (
ContextFieldAvailability,
context_analysis_warnings,
context_fields_by_node,
context_schema_for_node,
context_schemas_by_node,
root_context_schema,
)
from .control_regions import (
ControlRegionAnalysis,
@@ -22,4 +25,7 @@ __all__ = [
"analyze_control_regions",
"context_analysis_warnings",
"context_fields_by_node",
"context_schema_for_node",
"context_schemas_by_node",
"root_context_schema",
]
src/wf_core/analysis/context_scopes.py
+239 -49
View File
@@ -6,14 +6,20 @@ from dataclasses import dataclass
from typing import Literal
from wf_core.analysis.control_regions import (
ControlRegionAnalysis,
ForeachOwnerStack,
analyze_control_regions,
)
from wf_core.context_contracts import (
FOREACH_CONTEXT_KEY,
LOOP_INDEX_CONTEXT_KEY,
LOOP_ITEM_CONTEXT_KEY,
STANDARD_CONTEXT_FIELDS,
ContextFieldContract,
ContextSchema,
foreach_context_fields,
foreach_entry_schema,
structured_foreach_contract,
)
from wf_core.models.steps import ForeachNode
from wf_core.models.workflow import Edge, Workflow
@@ -65,17 +71,19 @@ class _Warnings:
def context_fields_by_node(
workflow: Workflow,
*,
control_regions: ControlRegionAnalysis | None = None,
) -> dict[str, tuple[ContextFieldAvailability, ...]]:
"""Return runtime context contracts for every reachable graph node.
This is an abstract execution-frame analysis keyed by static control
region: each node use belongs to exactly one foreach-owner stack, and
that stack decides which foreach aliases the node exposes. A node
that stack decides which foreach entries the node exposes. A node
reachable under two stacks is a region conflict and receives no foreach
fields. The traversal still memoizes node id and owner stack so cyclic
graphs terminate.
"""
return _analyze(workflow).fields_by_node
return _analyze(workflow, control_regions=control_regions).fields_by_node
def context_analysis_warnings(workflow: Workflow) -> tuple[str, ...]:
@@ -83,11 +91,69 @@ def context_analysis_warnings(workflow: Workflow) -> tuple[str, ...]:
return _analyze(workflow).warnings
def _analyze(workflow: Workflow) -> _ContextAnalysis:
def context_schemas_by_node(
workflow: Workflow,
*,
control_regions: ControlRegionAnalysis | None = None,
) -> dict[str, ContextSchema]:
"""Return schemas for all unambiguous, reachable program locations.
A conflicted or unreachable node has no generated per-node schema;
callers treat that absence as invalid, not as root context. Schemas are
composed from the same owner-stack analysis as field contracts, not from
a second graph traversal.
"""
analysis = control_regions or analyze_control_regions(workflow)
foreach_nodes = {
node.id: node for node in workflow.nodes if isinstance(node, ForeachNode)
}
schemas: dict[str, ContextSchema] = {}
for node_id, stack in analysis.owner_stack_by_node.items():
# Conflicted nodes are already dropped from owner_stack_by_node by the
# control-region analysis, so every entry here is unambiguous.
schemas[node_id] = _context_schema_for_stack(
workflow, foreach_nodes, analysis.owner_stack_by_node, stack
)
return schemas
def context_schema_for_node(workflow: Workflow, node_id: str) -> ContextSchema:
"""Return the complete graph-visible context object schema at one node."""
schemas = context_schemas_by_node(workflow)
try:
return schemas[node_id]
except KeyError as exc:
raise KeyError(f"no context schema for node {node_id!r}") from exc
def root_context_schema() -> ContextSchema:
"""Return standard fields plus an empty structured foreach map."""
properties: dict[str, ContextSchema] = {
field.name: deepcopy(field.schema) for field in STANDARD_CONTEXT_FIELDS
}
properties[FOREACH_CONTEXT_KEY] = {
"type": "object",
"properties": {},
"required": [],
"additionalProperties": False,
}
return {
"type": "object",
"properties": properties,
"required": sorted(properties),
"additionalProperties": False,
}
def _analyze(
workflow: Workflow,
*,
control_regions: ControlRegionAnalysis | None = None,
) -> _ContextAnalysis:
"""Derive context contracts from static foreach control regions.
Each unambiguous node use has exactly one owner stack; its active foreach
is the final stack item. A canonical return edge pops the stack, so the
Each unambiguous node use has exactly one owner stack; all active entries
in that stack are exposed. A canonical return edge pops the stack, so the
controller itself stays in the outer context. Conflicted nodes receive no
foreach fields.
"""
@@ -118,7 +184,7 @@ def _analyze(workflow: Workflow) -> _ContextAnalysis:
warnings.add(f"workflow start targets missing node {workflow.start!r}")
return _ContextAnalysis({}, tuple(warnings.values))
analysis = analyze_control_regions(workflow)
analysis = control_regions or analyze_control_regions(workflow)
for issue in analysis.issues:
warnings.add(
f"control region {issue.kind.value} at {issue.path}: {issue.message}"
@@ -126,12 +192,11 @@ def _analyze(workflow: Workflow) -> _ContextAnalysis:
fields_by_node: dict[str, tuple[ContextFieldAvailability, ...]] = {}
for node_id, stack in analysis.owner_stack_by_node.items():
active_foreach_id = stack[-1] if stack else None
fields_by_node[node_id] = _available_fields(
workflow,
foreach_nodes,
analysis.owner_stack_by_node,
active_foreach_id,
stack,
)
return _ContextAnalysis(fields_by_node, tuple(warnings.values))
@@ -140,30 +205,67 @@ def _available_fields(
workflow: Workflow,
foreach_nodes: Mapping[str, ForeachNode],
owner_stack_by_node: Mapping[str, ForeachOwnerStack],
active_scope: FrameScope,
stack: ForeachOwnerStack,
) -> tuple[ContextFieldAvailability, ...]:
"""Return contracts for one static owner stack; all are guaranteed.
A single node use has one control region, so foreach fields are either
present (inside a body) or absent (outside). Conditional availability is
not used to represent multiple owner stacks: a node reached under two
stacks is a region conflict and receives no foreach fields at all.
present (inside bodies) or absent (outside). Every owner in the stack
contributes one required ``foreach.<id>`` entry, every active configured
alias, and ``loop_item``/``loop_index`` from the final owner only.
Conditional availability is not used to represent multiple owner stacks:
a node reached under two stacks is a region conflict and receives no
foreach fields at all.
"""
contracts = list(STANDARD_CONTEXT_FIELDS)
if active_scope is not None:
foreach = foreach_nodes.get(active_scope)
if foreach is not None:
contracts.extend(
foreach_context_fields(
foreach.as_,
_foreach_item_schema(
workflow,
foreach,
foreach_nodes,
owner_stack_by_node,
),
)
if not stack:
return tuple(
ContextFieldAvailability(
contract=ContextFieldContract(
contract.name,
deepcopy(contract.schema),
contract.description,
),
availability="available",
)
for contract in contracts
)
entry_schemas: dict[str, ContextSchema] = {}
for owner_id in stack:
foreach = foreach_nodes.get(owner_id)
if foreach is None:
continue
entry_schemas[owner_id] = foreach_entry_schema(
owner_id,
_foreach_item_schema(
workflow,
foreach,
foreach_nodes,
owner_stack_by_node,
),
)
if entry_schemas:
contracts.append(structured_foreach_contract(entry_schemas))
# Aliases for every active owner, plus innermost loop keys.
for owner_id in stack:
foreach = foreach_nodes.get(owner_id)
if foreach is None:
continue
item_schema = _foreach_item_schema(
workflow, foreach, foreach_nodes, owner_stack_by_node
)
# Only the innermost owner contributes loop_item/loop_index; every
# owner contributes its configured alias when unambiguous. Alias
# collision handling is a validation concern; here we expose what the
# stack declares.
if owner_id == stack[-1]:
contracts.extend(foreach_context_fields(foreach.as_, item_schema))
elif foreach.as_:
# Outer aliases are exposed without re-emitting loop keys.
outer_fields = foreach_context_fields(foreach.as_, item_schema)
for field in outer_fields:
if field.name not in (LOOP_ITEM_CONTEXT_KEY, LOOP_INDEX_CONTEXT_KEY):
contracts.append(field)
return tuple(
ContextFieldAvailability(
contract=ContextFieldContract(
@@ -177,6 +279,70 @@ def _available_fields(
)
def _context_schema_for_stack(
workflow: Workflow,
foreach_nodes: Mapping[str, ForeachNode],
owner_stack_by_node: Mapping[str, ForeachOwnerStack],
stack: ForeachOwnerStack,
) -> ContextSchema:
"""Build the complete graph-visible context object schema for one stack."""
properties: dict[str, ContextSchema] = {
field.name: deepcopy(field.schema) for field in STANDARD_CONTEXT_FIELDS
}
entry_schemas: dict[str, ContextSchema] = {}
for owner_id in stack:
foreach = foreach_nodes.get(owner_id)
if foreach is None:
continue
entry_schemas[owner_id] = foreach_entry_schema(
owner_id,
_foreach_item_schema(
workflow, foreach, foreach_nodes, owner_stack_by_node
),
)
properties[FOREACH_CONTEXT_KEY] = {
"type": "object",
"properties": entry_schemas,
"required": sorted(entry_schemas),
"additionalProperties": False,
}
required: list[str] = [
field.name for field in STANDARD_CONTEXT_FIELDS
] + [FOREACH_CONTEXT_KEY]
if stack:
innermost = foreach_nodes.get(stack[-1])
if innermost is not None:
inner_item = _foreach_item_schema(
workflow, innermost, foreach_nodes, owner_stack_by_node
)
properties[LOOP_ITEM_CONTEXT_KEY] = deepcopy(inner_item)
properties[LOOP_INDEX_CONTEXT_KEY] = {"type": "integer"}
required.extend([LOOP_ITEM_CONTEXT_KEY, LOOP_INDEX_CONTEXT_KEY])
for owner_id in stack:
foreach = foreach_nodes.get(owner_id)
if foreach is not None and foreach.as_:
properties[foreach.as_] = deepcopy(
_foreach_item_schema(
workflow, foreach, foreach_nodes, owner_stack_by_node
)
)
required.append(foreach.as_)
# Deduplicate while preserving order; aliases could theoretically repeat
# a standard name, but validation rejects those collisions.
seen: set[str] = set()
deduped_required: list[str] = []
for name in required:
if name not in seen:
seen.add(name)
deduped_required.append(name)
return {
"type": "object",
"properties": properties,
"required": sorted(deduped_required),
"additionalProperties": False,
}
def _foreach_item_schema(
workflow: Workflow,
foreach: ForeachNode,
@@ -185,18 +351,17 @@ def _foreach_item_schema(
) -> ContextSchema:
"""Resolve one controller's item schema in its own static context.
An inner foreach may declare ``over="context.outer_item"``; the lookup
uses the controller's own owner stack, not the inner body stack.
An inner foreach may declare ``over="context.foreach.outer.item.children"``;
the lookup uses the controller's own owner stack, not the inner body stack.
"""
controller_stack = owner_stack_by_node.get(foreach.id)
if controller_stack is None:
return {}
controller_scope: FrameScope = controller_stack[-1] if controller_stack else None
source_schema = _schema_at_path(
workflow,
foreach.over.root,
foreach.over.parts,
controller_scope,
controller_stack,
foreach_nodes,
owner_stack_by_node,
)
@@ -228,7 +393,7 @@ def _schema_at_path(
workflow: Workflow,
root: str,
parts: tuple[str, ...],
active_scope: FrameScope,
stack: ForeachOwnerStack,
foreach_nodes: Mapping[str, ForeachNode],
owner_stack_by_node: Mapping[str, ForeachOwnerStack],
) -> Mapping[str, object] | None:
@@ -236,7 +401,7 @@ def _schema_at_path(
schema_document = _schema_document(
workflow,
root,
active_scope=active_scope,
stack=stack,
foreach_nodes=foreach_nodes,
owner_stack_by_node=owner_stack_by_node,
)
@@ -260,6 +425,7 @@ def _schema_document(
workflow: Workflow,
root: str,
*,
stack: ForeachOwnerStack | None = None,
active_scope: FrameScope = None,
foreach_nodes: Mapping[str, ForeachNode] | None = None,
owner_stack_by_node: Mapping[str, ForeachOwnerStack] | None = None,
@@ -269,30 +435,54 @@ def _schema_document(
if root == "state":
return workflow.state_schema.model_dump(mode="json", exclude_none=True)
if root == "context":
# Prefer the full owner stack when available; fall back to the legacy
# single active scope for callers that have not migrated yet.
resolved_stack: ForeachOwnerStack = ()
if stack is not None:
resolved_stack = stack
elif active_scope is not None:
resolved_stack = (active_scope,)
current: dict[str, object] = {
field.name: field.schema for field in STANDARD_CONTEXT_FIELDS
}
if (
active_scope is not None
and foreach_nodes is not None
foreach_nodes is not None
and owner_stack_by_node is not None
):
foreach = foreach_nodes.get(active_scope)
if foreach is not None:
current.update(
{
field.name: field.schema
for field in foreach_context_fields(
foreach.as_,
_foreach_item_schema(
workflow,
foreach,
foreach_nodes,
owner_stack_by_node,
),
)
}
)
entry_schemas: dict[str, object] = {}
for owner_id in resolved_stack:
foreach = foreach_nodes.get(owner_id)
if foreach is not None:
entry_schemas[owner_id] = foreach_entry_schema(
owner_id,
_foreach_item_schema(
workflow,
foreach,
foreach_nodes,
owner_stack_by_node,
),
)
current[FOREACH_CONTEXT_KEY] = {
"type": "object",
"properties": entry_schemas,
"required": sorted(entry_schemas),
"additionalProperties": False,
}
for owner_id in resolved_stack:
foreach = foreach_nodes.get(owner_id)
if foreach is not None:
item_schema = _foreach_item_schema(
workflow,
foreach,
foreach_nodes,
owner_stack_by_node,
)
for field in foreach_context_fields(foreach.as_, item_schema):
# Innermost loop keys win; outer aliases accumulate.
# Validation owns collision diagnostics.
is_innermost = bool(resolved_stack) and owner_id == resolved_stack[-1]
if field.name not in current or is_innermost:
current[field.name] = field.schema
return {"type": "object", "properties": current}
return {}
src/wf_core/context_contracts.py
+48
View File
@@ -1,5 +1,6 @@
from __future__ import annotations
from collections.abc import Mapping
from copy import deepcopy
from dataclasses import dataclass
from typing import Any
@@ -87,3 +88,50 @@ def foreach_context_fields(
)
)
return tuple(fields)
def foreach_entry_schema(node_id: str, item_schema: ContextSchema) -> ContextSchema:
"""Return the JSON schema for one structured ``foreach.<id>`` entry.
Identity fields disclose the dynamic activation/frame/scope/lineage when
advanced runtime code needs them; ``item`` carries the inferred
collection item schema.
"""
return {
"type": "object",
"properties": {
"node_id": {"const": node_id},
"activation_id": {"type": "string"},
"frame_id": {"type": "string"},
"scope_id": {"type": "string"},
"lineage_id": {"type": "string"},
"index": {"type": "integer"},
"item": deepcopy(item_schema),
},
"required": [
"node_id",
"activation_id",
"frame_id",
"scope_id",
"lineage_id",
"index",
"item",
],
"additionalProperties": False,
}
def structured_foreach_contract(
entry_schemas: Mapping[str, ContextSchema],
) -> ContextFieldContract:
"""Return the ``foreach`` map contract for one static owner stack."""
return ContextFieldContract(
FOREACH_CONTEXT_KEY,
{
"type": "object",
"properties": dict(entry_schemas),
"required": sorted(entry_schemas),
"additionalProperties": False,
},
"Structured foreach context by static node id",
)