agstack 1.10.2__tar.gz → 1.11.0__tar.gz

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

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

@@ -30,6 +30,7 @@ class Agent:
         temperature: float = 0.7,
         max_tokens: int | None = None,
         max_turns: int = 10,
+        display_name: str | None = None,
     ):
         """初始化 Agent
 
@@ -40,6 +41,7 @@ class Agent:
         :param temperature: 温度参数
         :param max_tokens: 最大 token 数
         :param max_turns: 最大轮次
+        :param display_name: 面向用户的展示名称（用于前端脱敏展示）
         """
         self.name = name
         self.instructions = instructions or f"You are {name}, a helpful AI assistant."
@@ -48,6 +50,7 @@ class Agent:
         self.temperature = temperature
         self.max_tokens = max_tokens
         self.max_turns = max_turns
+        self.display_name = display_name
 
     def get_system_message(self) -> dict[str, Any]:
         """获取系统消息"""

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

@@ -19,6 +19,8 @@ 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] = {}
 
     def _ensure_builtins(self) -> None:
         """延迟加载内置 node handlers（避免循环导入）"""
@@ -32,13 +34,17 @@ class FlowRegistry:
 
     # ── 注册 ──
 
-    def register_tool(self, name: str, tool_class) -> None:
+    def register_tool(self, name: str, tool_class, display_name: str | None = None) -> 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]) -> None:
+    def register_agent(self, name: str, agent_class: type[Agent], display_name: str | None = None) -> None:
         """注册 Agent 类型"""
         self._agents[name] = agent_class
+        if display_name is not None:
+            self._agent_display_names[name] = display_name
 
     def register_flow(self, name: str, flow_class: type) -> None:
         """注册 Flow 类型"""
@@ -87,6 +93,24 @@ 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_class(self, name: str) -> Any | None:
         """获取工具工厂/类"""
         return self._tools.get(name)

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

@@ -33,6 +33,7 @@ class Tool:
         description: str,
         function: Callable,
         parameters: dict[str, Any] | None = None,
+        display_name: str | None = None,
     ):
         """初始化工具
 
@@ -40,11 +41,13 @@ class Tool:
         :param description: 工具描述
         :param function: 工具函数，签名 fn(context, inputs) -> dict[str, Any]
         :param parameters: JSON Schema 参数定义（用于 LLM 调用）
+        :param display_name: 面向用户的展示名称（用于前端脱敏展示）
         """
         self.name = name
         self.description = description
         self.function = function
         self.parameters = parameters or {"type": "object", "properties": {}, "required": []}
+        self.display_name = display_name
 
     async def execute_async(self, context: "FlowContext", inputs: dict[str, Any] | None = None) -> ToolResult:
         """异步执行工具"""

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

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

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