agstack 1.22.0__tar.gz → 1.22.1__tar.gz

{agstack-1.22.0 → agstack-1.22.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.22.0
+Version: 1.22.1
 Summary: Production-ready toolkit for building FastAPI and LLM applications
 Author-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>
 Maintainer-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>

{agstack-1.22.0 → agstack-1.22.1}/agstack/llm/flow/event.py

@@ -97,12 +97,12 @@ def run_error(*, message: str, code: str | None = None) -> dict[str, Any]:
 # ── Step ──
 
 
-def step_started(*, step_name: str) -> dict[str, Any]:
-    return _ev(EventType.STEP_STARTED, step_name=step_name)
+def step_started(*, step_name: str, step_id: str | None = None) -> dict[str, Any]:
+    return _ev(EventType.STEP_STARTED, step_name=step_name, step_id=step_id or str(uuid4()))
 
 
-def step_finished(*, step_name: str) -> dict[str, Any]:
-    return _ev(EventType.STEP_FINISHED, step_name=step_name)
+def step_finished(*, step_name: str, step_id: str) -> dict[str, Any]:
+    return _ev(EventType.STEP_FINISHED, step_name=step_name, step_id=step_id)
 
 
 # ── State ──

{agstack-1.22.0 → agstack-1.22.1}/agstack/llm/flow/flow.py

@@ -370,7 +370,8 @@ class Flow:
     async def stream(self, context: "FlowContext") -> AsyncIterator[dict[str, Any]]:
         """流式执行 Flow（输出 AG-UI 标准事件）"""
         context.trace.started_at = time.time()
-        yield event.step_started(step_name=f"flow:{self.name}")
+        flow_sid = str(uuid4())
+        yield event.step_started(step_name=f"flow:{self.name}", step_id=flow_sid)
 
         try:
             if not self.edges:
@@ -386,7 +387,7 @@ class Flow:
             context.trace.finished_at = time.time()
             context.trace.total_usage = context.usage
 
-        yield event.step_finished(step_name=f"flow:{self.name}")
+        yield event.step_finished(step_name=f"flow:{self.name}", step_id=flow_sid)
 
     async def _stream_sequential(self, context: "FlowContext") -> AsyncIterator[dict[str, Any]]:
         """顺序流式执行"""
@@ -437,7 +438,8 @@ class Flow:
                 context.trace.record_node_start(current_node_id, "message", inputs=msg_config)
 
                 # message 节点增加 STEP 事件
-                step_evt = event.step_started(step_name=f"message:{current_node_id}")
+                msg_sid = str(uuid4())
+                step_evt = event.step_started(step_name=f"message:{current_node_id}", step_id=msg_sid)
                 step_evt["_node_id"] = current_node_id
                 step_evt["_label"] = msg_config.get("label")
                 step_evt["_echo"] = msg_config.get("echo", True)
@@ -454,7 +456,7 @@ class Flow:
                 text = template.format_map(_SafeFormatDict(context.variables))
                 context.set_output(current_node_id, {"result": text})
 
-                fin_evt = event.step_finished(step_name=f"message:{current_node_id}")
+                fin_evt = event.step_finished(step_name=f"message:{current_node_id}", step_id=msg_sid)
                 fin_evt["_node_id"] = current_node_id
                 fin_evt["_label"] = msg_config.get("label")
                 fin_evt["_echo"] = msg_config.get("echo", True)
@@ -469,7 +471,8 @@ class Flow:
 
                 context.trace.record_node_start(current_node_id, "parallel", inputs=config)
 
-                step_evt = event.step_started(step_name=f"parallel:{current_node_id}")
+                parallel_sid = str(uuid4())
+                step_evt = event.step_started(step_name=f"parallel:{current_node_id}", step_id=parallel_sid)
                 step_evt["_node_id"] = current_node_id
                 step_evt["_label"] = None
                 step_evt["_echo"] = False
@@ -510,7 +513,7 @@ class Flow:
                         merged.update(branch_result)
                 context.set_output(current_node_id, merged)
 
-                fin_evt = event.step_finished(step_name=f"parallel:{current_node_id}")
+                fin_evt = event.step_finished(step_name=f"parallel:{current_node_id}", step_id=parallel_sid)
                 fin_evt["_node_id"] = current_node_id
                 fin_evt["_label"] = None
                 fin_evt["_echo"] = False
@@ -533,7 +536,8 @@ class Flow:
 
                 context.trace.record_node_start(current_node_id, "iteration", inputs=config)
 
-                step_evt = event.step_started(step_name=f"iteration:{current_node_id}")
+                iter_sid = str(uuid4())
+                step_evt = event.step_started(step_name=f"iteration:{current_node_id}", step_id=iter_sid)
                 step_evt["_node_id"] = current_node_id
                 step_evt["_label"] = None
                 step_evt["_echo"] = False
@@ -574,7 +578,7 @@ class Flow:
                 iteration_output = {"results": results}
                 context.set_output(current_node_id, iteration_output)
 
-                fin_evt = event.step_finished(step_name=f"iteration:{current_node_id}")
+                fin_evt = event.step_finished(step_name=f"iteration:{current_node_id}", step_id=iter_sid)
                 fin_evt["_node_id"] = current_node_id
                 fin_evt["_label"] = None
                 fin_evt["_echo"] = False

{agstack-1.22.0 → agstack-1.22.1}/agstack/llm/flow/nodes/agent_node.py

@@ -3,6 +3,7 @@
 """Agent 节点处理器 — 从 flow.py 提取"""
 
 from typing import TYPE_CHECKING, Any, AsyncIterator
+from uuid import uuid4
 
 from .. import event
 from ..exceptions import FlowError
@@ -46,8 +47,9 @@ class AgentNodeHandler(NodeHandler):
     async def stream(self, node: dict, context: "FlowContext", node_id: str) -> AsyncIterator[dict[str, Any]]:
         config = node.get("config", {})
         step_name = self.get_step_name(node, node_id)
+        sid = str(uuid4())
 
-        yield event.step_started(step_name=step_name)
+        yield event.step_started(step_name=step_name, step_id=sid)
         resolved = self.resolve_inputs(config, context)
         ag = self._create_agent(config, context)
         async for evt in ag.stream(context, inputs=resolved):
@@ -63,4 +65,4 @@ class AgentNodeHandler(NodeHandler):
         tool_calls = context.pop_execution_records()
         context.set_variable("_last_node_tool_calls", tool_calls)
 
-        yield event.step_finished(step_name=step_name)
+        yield event.step_finished(step_name=step_name, step_id=sid)

{agstack-1.22.0 → agstack-1.22.1}/agstack/llm/flow/nodes/base.py

@@ -3,6 +3,7 @@
 """NodeHandler 基类 — 所有执行类节点的公共接口"""
 
 from typing import TYPE_CHECKING, Any, AsyncIterator
+from uuid import uuid4
 
 from .. import event
 
@@ -45,7 +46,8 @@ class NodeHandler:
         需要流式输出的节点（如 agent, llm_chat）应覆盖此方法。
         """
         step_name = self.get_step_name(node, node_id)
-        yield event.step_started(step_name=step_name)
+        sid = str(uuid4())
+        yield event.step_started(step_name=step_name, step_id=sid)
         result = await self.execute(node, context)
         context.set_output(node_id, result)
-        yield event.step_finished(step_name=step_name)
+        yield event.step_finished(step_name=step_name, step_id=sid)

{agstack-1.22.0 → agstack-1.22.1}/agstack/llm/flow/nodes/echo_node.py

@@ -32,8 +32,9 @@ class EchoNodeHandler(NodeHandler):
     async def stream(self, node: dict, context: "FlowContext", node_id: str) -> AsyncIterator[dict[str, Any]]:
         config = node.get("config", {})
         step_name = self.get_step_name(node, node_id)
+        sid = str(uuid4())
 
-        yield event.step_started(step_name=step_name)
+        yield event.step_started(step_name=step_name, step_id=sid)
 
         resolved = self.resolve_inputs(config, context)
         content = resolved.get("content", "")
@@ -51,4 +52,4 @@ class EchoNodeHandler(NodeHandler):
             yield event.text_message_end(message_id=msg_id)
 
         context.set_output(node_id, {"result": content})
-        yield event.step_finished(step_name=step_name)
+        yield event.step_finished(step_name=step_name, step_id=sid)

{agstack-1.22.0 → agstack-1.22.1}/agstack/llm/flow/nodes/llm_chat_node.py

@@ -92,15 +92,17 @@ class LLMChatNodeHandler(NodeHandler):
         if not use_stream:
             # 非流式：走默认 execute 路径
             step_name = self.get_step_name(node, node_id)
-            yield event.step_started(step_name=step_name)
+            sid = str(uuid4())
+            yield event.step_started(step_name=step_name, step_id=sid)
             result = await self.execute(node, context)
             context.set_output(node_id, result)
-            yield event.step_finished(step_name=step_name)
+            yield event.step_finished(step_name=step_name, step_id=sid)
             return
 
         # 流式输出
         step_name = self.get_step_name(node, node_id)
-        yield event.step_started(step_name=step_name)
+        sid = str(uuid4())
+        yield event.step_started(step_name=step_name, step_id=sid)
 
         resolved_inputs = self.resolve_inputs(config, context)
         prompt_text = self._build_prompt(config.get("prompt", ""), resolved_inputs)
@@ -153,4 +155,4 @@ class LLMChatNodeHandler(NodeHandler):
         result_text = "".join(content_parts)
         context.set_output(node_id, {"result": result_text})
 
-        yield event.step_finished(step_name=step_name)
+        yield event.step_finished(step_name=step_name, step_id=sid)

{agstack-1.22.0 → agstack-1.22.1}/agstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.22.0
+Version: 1.22.1
 Summary: Production-ready toolkit for building FastAPI and LLM applications
 Author-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>
 Maintainer-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>

{agstack-1.22.0 → agstack-1.22.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agstack"
-version = "1.22.0"
+version = "1.22.1"
 description = "Production-ready toolkit for building FastAPI and LLM applications"
 readme = "README.md"
 license = "MIT"