e653d63bc1
OSS restructuring for open-core model: - Rename package from agentserver/ to xml_pipeline/ - Update all imports (44 Python files, 31 docs/configs) - Update pyproject.toml for OSS distribution (v0.3.0) - Move prompt_toolkit from core to optional [console] extra - Remove auth/server/lsp from core optional deps (-> Nextra) New console example in examples/console/: - Self-contained demo with handlers and config - Uses prompt_toolkit (optional, falls back to input()) - No password auth, no TUI, no LSP — just the basics - Shows how to use xml-pipeline as a library Import changes: - from agentserver.* -> from xml_pipeline.* - CLI entry points updated: xml_pipeline.cli:main Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
"""
|
|
routing_resolution.py — Resolve routing based on derived root tag.
|
|
|
|
This step computes the root tag from the deserialized payload and looks it up
|
|
in a routing table (root_tag → list[Listener]).
|
|
|
|
NOTE: The StreamPump has routing built-in via _route_step(). This standalone
|
|
step is provided for custom pipeline configurations or testing.
|
|
|
|
Usage:
|
|
routing_step = make_routing_step(routing_table)
|
|
state = await routing_step(state)
|
|
|
|
Part of AgentServer v2.1 message pump.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Dict, List, Callable, Awaitable, TYPE_CHECKING
|
|
|
|
from xml_pipeline.message_bus.message_state import MessageState
|
|
|
|
if TYPE_CHECKING:
|
|
from xml_pipeline.message_bus.stream_pump import Listener
|
|
|
|
|
|
def make_routing_step(
|
|
routing_table: Dict[str, List["Listener"]]
|
|
) -> Callable[[MessageState], Awaitable[MessageState]]:
|
|
"""
|
|
Factory: create a routing step with a specific routing table.
|
|
|
|
The routing table maps root tags to lists of listeners:
|
|
{"agent.payload": [listener1, listener2], ...}
|
|
"""
|
|
|
|
async def routing_resolution_step(state: MessageState) -> MessageState:
|
|
"""
|
|
Resolve which listener(s) should handle this payload.
|
|
|
|
Root tag = f"{from_id.lower()}.{payload_class_name.lower()}"
|
|
|
|
Supports:
|
|
- Normal unique routing (one listener)
|
|
- Broadcast (multiple listeners if same root tag)
|
|
|
|
If no match → error, falls to system pipeline.
|
|
"""
|
|
if state.payload is None:
|
|
state.error = "routing_resolution_step: no deserialized payload"
|
|
return state
|
|
|
|
if state.to_id is None:
|
|
state.error = "routing_resolution_step: missing to_id"
|
|
return state
|
|
|
|
payload_class_name = type(state.payload).__name__.lower()
|
|
root_tag = f"{state.to_id.lower()}.{payload_class_name}"
|
|
|
|
targets = routing_table.get(root_tag)
|
|
|
|
if not targets:
|
|
state.error = f"routing_resolution_step: unknown root tag '{root_tag}'"
|
|
return state
|
|
|
|
state.target_listeners = targets
|
|
return state
|
|
|
|
routing_resolution_step.__name__ = "routing_resolution_step"
|
|
return routing_resolution_step
|