agstack 1.11.0__tar.gz → 1.13.0__tar.gz

{agstack-1.11.0 → agstack-1.13.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.11.0
+Version: 1.13.0
 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.11.0 → agstack-1.13.0}/agstack/llm/flow/agent.py

@@ -30,7 +30,9 @@ class Agent:
         temperature: float = 0.7,
         max_tokens: int | None = None,
         max_turns: int = 10,
-        display_name: str | None = None,
+        *,
+        label: str | None = None,
+        echo: bool = False,
     ):
         """初始化 Agent
 
@@ -41,7 +43,8 @@ class Agent:
         :param temperature: 温度参数
         :param max_tokens: 最大 token 数
         :param max_turns: 最大轮次
-        :param display_name: 面向用户的展示名称（用于前端脱敏展示）
+        :param label: 面向用户的展示名称（控制 STEP 进度事件可见性）
+        :param echo: 是否转发 TEXT_MESSAGE 给用户
         """
         self.name = name
         self.instructions = instructions or f"You are {name}, a helpful AI assistant."
@@ -50,7 +53,8 @@ class Agent:
         self.temperature = temperature
         self.max_tokens = max_tokens
         self.max_turns = max_turns
-        self.display_name = display_name
+        self.label = label
+        self.echo = echo
 
     def get_system_message(self) -> dict[str, Any]:
         """获取系统消息"""

{agstack-1.11.0 → agstack-1.13.0}/agstack/llm/flow/nodes/__init__.py

@@ -5,6 +5,7 @@
 from .agent_node import AgentNodeHandler
 from .base import NodeHandler
 from .detect_node import DetectNodeHandler
+from .echo_node import EchoNodeHandler
 from .llm_chat_node import LLMChatNodeHandler
 from .llm_embed_node import LLMEmbedNodeHandler
 from .llm_rerank_node import LLMRerankNodeHandler
@@ -23,6 +24,7 @@ builtin_handlers: list[NodeHandler] = [
     LLMEmbedNodeHandler(),
     LLMRerankNodeHandler(),
     DetectNodeHandler(),
+    EchoNodeHandler(),
     SwitchNodeHandler(),
     SubflowNodeHandler(),
 ]

agstack-1.13.0/agstack/llm/flow/nodes/echo_node.py

@@ -0,0 +1,54 @@
+#  Copyright (c) 2020-2026 XtraVisions, All rights reserved.
+
+"""Echo 节点 — 将 context 中的文本作为 TEXT_MESSAGE 事件流式输出（不调用 LLM）"""
+
+from typing import TYPE_CHECKING, Any, AsyncIterator
+from uuid import uuid4
+
+from agstack.llm.flow import event
+from agstack.llm.flow.nodes.base import NodeHandler
+
+
+if TYPE_CHECKING:
+    from agstack.llm.flow.context import FlowContext
+
+
+class EchoNodeHandler(NodeHandler):
+    """Echo 节点：从 context 读取文本并流式输出为 TEXT_MESSAGE 事件
+
+    用于将内部 agent 的结果展示给用户，不产生额外 LLM 调用。
+    """
+
+    node_type = "echo"
+
+    async def execute(self, node: dict, context: "FlowContext") -> Any:
+        config = node.get("config", {})
+        resolved = self.resolve_inputs(config, context)
+        content = resolved.get("content", "")
+        if isinstance(content, dict):
+            content = content.get("result", "")
+        return {"result": content}
+
+    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)
+
+        yield event.step_started(step_name=step_name)
+
+        resolved = self.resolve_inputs(config, context)
+        content = resolved.get("content", "")
+        if isinstance(content, dict):
+            content = content.get("result", "")
+
+        if content:
+            msg_id = context.message_id or str(uuid4())
+            yield event.text_message_start(message_id=msg_id, role="assistant")
+
+            chunk_size = 80
+            for i in range(0, len(content), chunk_size):
+                yield event.text_message_content(message_id=msg_id, delta=content[i : i + chunk_size])
+
+            yield event.text_message_end(message_id=msg_id)
+
+        context.set_output(node_id, {"result": content})
+        yield event.step_finished(step_name=step_name)

{agstack-1.11.0 → agstack-1.13.0}/agstack/llm/flow/registry.py

@@ -19,8 +19,10 @@ class FlowRegistry:
         self._flows: dict[str, Any] = {}
         self._node_handlers: dict[str, Any] = {}
         self._builtins_loaded = False
-        self._tool_display_names: dict[str, str] = {}
-        self._agent_display_names: dict[str, str] = {}
+        self._tool_labels: dict[str, str] = {}
+        self._tool_echo: dict[str, bool] = {}
+        self._agent_labels: dict[str, str] = {}
+        self._agent_echo: dict[str, bool] = {}
 
     def _ensure_builtins(self) -> None:
         """延迟加载内置 node handlers（避免循环导入）"""
@@ -34,17 +36,23 @@ class FlowRegistry:
 
     # ── 注册 ──
 
-    def register_tool(self, name: str, tool_class, display_name: str | None = None) -> None:
+    def register_tool(self, name: str, tool_class, *, label: str | None = None, echo: bool = False) -> None:
         """注册工具工厂/类/实例"""
         self._tools[name] = tool_class
-        if display_name is not None:
-            self._tool_display_names[name] = display_name
-
-    def register_agent(self, name: str, agent_class: type[Agent], display_name: str | None = None) -> None:
+        if label is not None:
+            self._tool_labels[name] = label
+        if echo:
+            self._tool_echo[name] = True
+
+    def register_agent(
+        self, name: str, agent_class: type[Agent], *, label: str | None = None, echo: bool = False
+    ) -> None:
         """注册 Agent 类型"""
         self._agents[name] = agent_class
-        if display_name is not None:
-            self._agent_display_names[name] = display_name
+        if label is not None:
+            self._agent_labels[name] = label
+        if echo:
+            self._agent_echo[name] = True
 
     def register_flow(self, name: str, flow_class: type) -> None:
         """注册 Flow 类型"""
@@ -93,23 +101,21 @@ class FlowRegistry:
 
     # ── 查询 ──
 
-    def get_tool_display_name(self, name: str) -> str | None:
-        """获取工具的展示名称（用于前端脱敏展示）"""
-        if name in self._tool_display_names:
-            return self._tool_display_names[name]
-        component = self._tools.get(name)
-        if component and hasattr(component, "display_name"):
-            return component.display_name
-        return None
-
-    def get_agent_display_name(self, name: str) -> str | None:
-        """获取 Agent 的展示名称（用于前端脱敏展示）"""
-        if name in self._agent_display_names:
-            return self._agent_display_names[name]
-        component = self._agents.get(name)
-        if component and hasattr(component, "display_name"):
-            return getattr(component, "display_name", None)
-        return None
+    def get_tool_label(self, name: str) -> str | None:
+        """获取工具的展示名称"""
+        return self._tool_labels.get(name)
+
+    def get_tool_echo(self, name: str) -> bool:
+        """获取工具是否转发 TEXT_MESSAGE（label 隐含 echo）"""
+        return name in self._tool_labels or self._tool_echo.get(name, False)
+
+    def get_agent_label(self, name: str) -> str | None:
+        """获取 Agent 的展示名称"""
+        return self._agent_labels.get(name)
+
+    def get_agent_echo(self, name: str) -> bool:
+        """获取 Agent 是否转发 TEXT_MESSAGE（label 隐含 echo）"""
+        return name in self._agent_labels or self._agent_echo.get(name, False)
 
     def get_tool_class(self, name: str) -> Any | None:
         """获取工具工厂/类"""

{agstack-1.11.0 → agstack-1.13.0}/agstack/llm/flow/tool.py

@@ -33,7 +33,9 @@ class Tool:
         description: str,
         function: Callable,
         parameters: dict[str, Any] | None = None,
-        display_name: str | None = None,
+        *,
+        label: str | None = None,
+        echo: bool = False,
     ):
         """初始化工具
 
@@ -41,13 +43,15 @@ class Tool:
         :param description: 工具描述
         :param function: 工具函数，签名 fn(context, inputs) -> dict[str, Any]
         :param parameters: JSON Schema 参数定义（用于 LLM 调用）
-        :param display_name: 面向用户的展示名称（用于前端脱敏展示）
+        :param label: 面向用户的展示名称（控制 STEP/TOOL_CALL 进度事件可见性）
+        :param echo: 是否转发 TEXT_MESSAGE 给用户
         """
         self.name = name
         self.description = description
         self.function = function
         self.parameters = parameters or {"type": "object", "properties": {}, "required": []}
-        self.display_name = display_name
+        self.label = label
+        self.echo = echo
 
     async def execute_async(self, context: "FlowContext", inputs: dict[str, Any] | None = None) -> ToolResult:
         """异步执行工具"""

{agstack-1.11.0 → agstack-1.13.0}/agstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.11.0
+Version: 1.13.0
 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.11.0 → agstack-1.13.0}/agstack.egg-info/SOURCES.txt

@@ -47,6 +47,7 @@ agstack/llm/flow/nodes/__init__.py
 agstack/llm/flow/nodes/agent_node.py
 agstack/llm/flow/nodes/base.py
 agstack/llm/flow/nodes/detect_node.py
+agstack/llm/flow/nodes/echo_node.py
 agstack/llm/flow/nodes/llm_chat_node.py
 agstack/llm/flow/nodes/llm_embed_node.py
 agstack/llm/flow/nodes/llm_rerank_node.py

{agstack-1.11.0 → agstack-1.13.0}/pyproject.toml

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