agstack 1.10.2__tar.gz → 1.12.0__tar.gz

{agstack-1.10.2 → agstack-1.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.10.2
+Version: 1.12.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.10.2 → agstack-1.12.0}/agstack/llm/flow/agent.py

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

{agstack-1.10.2 → agstack-1.12.0}/agstack/llm/flow/registry.py

@@ -19,6 +19,10 @@ class FlowRegistry:
         self._flows: dict[str, Any] = {}
         self._node_handlers: dict[str, Any] = {}
         self._builtins_loaded = False
+        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（避免循环导入）"""
@@ -32,13 +36,23 @@ class FlowRegistry:
 
     # ── 注册 ──
 
-    def register_tool(self, name: str, tool_class) -> None:
+    def register_tool(self, name: str, tool_class, *, label: str | None = None, echo: bool = False) -> None:
         """注册工具工厂/类/实例"""
         self._tools[name] = tool_class
-
-    def register_agent(self, name: str, agent_class: type[Agent]) -> 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 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 类型"""
@@ -87,6 +101,22 @@ class FlowRegistry:
 
     # ── 查询 ──
 
+    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:
         """获取工具工厂/类"""
         return self._tools.get(name)

{agstack-1.10.2 → agstack-1.12.0}/agstack/llm/flow/tool.py

@@ -33,6 +33,9 @@ class Tool:
         description: str,
         function: Callable,
         parameters: dict[str, Any] | None = None,
+        *,
+        label: str | None = None,
+        echo: bool = False,
     ):
         """初始化工具
 
@@ -40,11 +43,15 @@ class Tool:
         :param description: 工具描述
         :param function: 工具函数，签名 fn(context, inputs) -> dict[str, Any]
         :param parameters: JSON Schema 参数定义（用于 LLM 调用）
+        :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.label = label
+        self.echo = echo
 
     async def execute_async(self, context: "FlowContext", inputs: dict[str, Any] | None = None) -> ToolResult:
         """异步执行工具"""

{agstack-1.10.2 → agstack-1.12.0}/agstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.10.2
+Version: 1.12.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.10.2 → agstack-1.12.0}/pyproject.toml

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