agstack 1.17.1__tar.gz → 1.18.0__tar.gz

{agstack-1.17.1 → agstack-1.18.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.17.1
+Version: 1.18.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.17.1 → agstack-1.18.0}/agstack/llm/flow/agent.py

@@ -88,6 +88,10 @@ class Agent:
     ) -> AsyncIterator[dict[str, Any]]:
         """流式执行 Agent，输出 AG-UI 标准事件"""
 
+        # 注入 agent_call_id 供 tool 审计关联
+        agent_call_id = str(uuid4())
+        context.set_variable("_agent_call_id", agent_call_id)
+
         # 输入来源：优先 inputs 参数，回退到 context.variables
         user_input = ""
         if inputs:
@@ -230,6 +234,7 @@ class Agent:
                 context.set_output(self.name, {"result": assistant_content})
                 # AG-UI: TEXT_MESSAGE_END
                 yield event.text_message_end(message_id=msg_id)
+                context.set_variable("_agent_call_id", None)
                 return
 
             # 执行工具调用
@@ -266,5 +271,10 @@ class Agent:
                 # AG-UI: TOOL_CALL_RESULT
                 yield event.tool_call_result(tool_call_id=tool_call["id"], content=result_content)
 
+                # 可观测性：yield tool 执行记录
+                for record in context.pop_execution_records():
+                    record["tool_call_id"] = tool_call["id"]
+                    yield event.custom(name="tool_execution", value=record)
+
             # 更新消息列表，继续下一轮
             messages = [self.get_system_message()] + context.history + context.get_messages(self.name)

{agstack-1.17.1 → agstack-1.18.0}/agstack/llm/flow/context.py

@@ -151,3 +151,9 @@ class FlowContext:
         if task_id is None:
             return self.execution_records
         return [r for r in self.execution_records if r.get("task_id") == task_id]
+
+    def pop_execution_records(self) -> list[dict[str, Any]]:
+        """取出并清空待发射的执行记录"""
+        records = self.execution_records
+        self.execution_records = []
+        return records

{agstack-1.17.1 → agstack-1.18.0}/agstack/llm/flow/nodes/base.py

@@ -48,4 +48,9 @@ class NodeHandler:
         yield event.step_started(step_name=step_name)
         result = await self.execute(node, context)
         context.set_output(node_id, result)
+
+        # 可观测性：drain tool 执行记录
+        for record in context.pop_execution_records():
+            yield event.custom(name="tool_execution", value=record)
+
         yield event.step_finished(step_name=step_name)

{agstack-1.17.1 → agstack-1.18.0}/agstack/llm/flow/tool.py

@@ -2,7 +2,9 @@
 
 """工具定义和执行"""
 
+import json
 import logging
+import time
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any, Callable
 
@@ -54,18 +56,38 @@ class Tool:
         self.echo = echo
 
     async def execute_async(self, context: "FlowContext", inputs: dict[str, Any] | None = None) -> ToolResult:
-        """异步执行工具"""
+        """异步执行工具（包含计时和可观测性记录）"""
+        args = inputs or {}
+        _t0 = time.perf_counter()
+        result = await self._execute(context, args)
+        _duration_ms = int((time.perf_counter() - _t0) * 1000)
+
+        result_content = json.dumps(result.result) if result.success else json.dumps({"error": result.error})
+        context.execution_records.append({
+            "agent_call_id": context.get_variable("_agent_call_id"),
+            "tool_name": self.name,
+            "tool_args": args,
+            "success": result.success,
+            "result": result_content,
+            "error": result.error,
+            "duration_ms": _duration_ms,
+        })
+
+        return result
+
+    async def _execute(self, context: "FlowContext", inputs: dict[str, Any]) -> ToolResult:
+        """实际执行逻辑，子类应覆写此方法"""
         try:
-            result = self.function(context, inputs or {})
+            result = self.function(context, inputs)
             if hasattr(result, "__await__"):
                 result = await result
 
-            return ToolResult(name=self.name, arguments=inputs or {}, result=result, success=True)
+            return ToolResult(name=self.name, arguments=inputs, result=result, success=True)
         except Exception as e:
             logger.warning("Tool %s failed: %s", self.name, e, exc_info=True)
             return ToolResult(
                 name=self.name,
-                arguments=inputs or {},
+                arguments=inputs,
                 result={},
                 success=False,
                 error=str(e),

{agstack-1.17.1 → agstack-1.18.0}/agstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.17.1
+Version: 1.18.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.17.1 → agstack-1.18.0}/pyproject.toml

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