agstack 1.7.0__tar.gz → 1.8.0__tar.gz

{agstack-1.7.0 → agstack-1.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.7.0
+Version: 1.8.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>
@@ -27,7 +27,7 @@ Requires-Dist: fastapi>=0.133.1
 Requires-Dist: jwcrypto>=1.5.6
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: nebula3-python>=3.8.3
-Requires-Dist: openai>=2.26.0
+Requires-Dist: openai>=2.28.0
 Requires-Dist: passlib[bcrypt]>=1.7.4
 Requires-Dist: pycasbin>=2.8.0
 Requires-Dist: pydantic>=2.12.4
@@ -35,7 +35,7 @@ Requires-Dist: python-multipart>=0.0.20
 Requires-Dist: requests>=2.32.5
 Requires-Dist: RestrictedPython>=7.0
 Requires-Dist: sqlalchemy[asyncio]>=2.0.48
-Requires-Dist: sqlobjects>=1.4.0
+Requires-Dist: sqlobjects>=1.5.0
 Requires-Dist: tiktoken>=0.12.0
 Requires-Dist: uvicorn>=0.41.0
 Dynamic: license-file

{agstack-1.7.0 → agstack-1.8.0}/agstack/llm/flow/__init__.py

@@ -18,13 +18,15 @@ from .exceptions import (
 from .factory import create_agent, create_tool
 from .flow import Flow
 from .loader import FlowLoader
-from .nodes import NodeHandler, register_node_handler
+from .nodes import NodeHandler
 from .records import Record, Status
 from .registry import registry
 from .state import FlowState
 from .tool import Tool, ToolResult
 
 
+register_node_handler = registry.register_node_handler
+
 __all__ = [
     # 核心抽象
     "Tool",

{agstack-1.7.0 → agstack-1.8.0}/agstack/llm/flow/flow.py

@@ -46,12 +46,9 @@ class Flow:
     _node_handlers: dict[str, "NodeHandler"] = field(default_factory=dict, init=False, repr=False)
 
     def __post_init__(self) -> None:
-        from .nodes import _global_node_handlers, builtin_handlers
+        from .registry import registry
 
-        for handler in builtin_handlers:
-            self._node_handlers[handler.node_type] = handler
-        # 全局注册的自定义 handler 可覆盖内置
-        self._node_handlers.update(_global_node_handlers)
+        self._node_handlers = dict(registry.get_all_node_handlers())
 
     # ── 重试策略 ──
 

{agstack-1.7.0 → agstack-1.8.0}/agstack/llm/flow/nodes/__init__.py

@@ -23,17 +23,7 @@ builtin_handlers: list[NodeHandler] = [
     DetectNodeHandler(),
 ]
 
-# 全局自定义节点注册
-_global_node_handlers: dict[str, NodeHandler] = {}
-
-
-def register_node_handler(node_type: str, handler: NodeHandler) -> None:
-    """注册自定义节点处理器（全局，所有 Flow 实例共享）"""
-    _global_node_handlers[node_type] = handler
-
-
 __all__ = [
     "NodeHandler",
     "builtin_handlers",
-    "register_node_handler",
 ]

agstack-1.8.0/agstack/llm/flow/registry.py

@@ -0,0 +1,139 @@
+#  Copyright (c) 2020-2026 XtraVisions, All rights reserved.
+
+"""Flow 系统注册中心 — 自包含的精简组件注册与创建"""
+
+from __future__ import annotations
+
+from typing import Any, cast
+
+from .agent import Agent
+from .tool import Tool
+
+
+class FlowRegistry:
+    """Flow 系统统一注册中心"""
+
+    def __init__(self):
+        self._tools: dict[str, Any] = {}
+        self._agents: dict[str, Any] = {}
+        self._flows: dict[str, Any] = {}
+        self._node_handlers: dict[str, Any] = {}
+        self._builtins_loaded = False
+
+    def _ensure_builtins(self) -> None:
+        """延迟加载内置 node handlers（避免循环导入）"""
+        if self._builtins_loaded:
+            return
+        self._builtins_loaded = True
+        from .nodes import builtin_handlers
+
+        for handler in builtin_handlers:
+            self._node_handlers.setdefault(handler.node_type, handler)
+
+    # ── 注册 ──
+
+    def register_tool(self, name: str, tool_class) -> None:
+        """注册工具工厂/类/实例"""
+        self._tools[name] = tool_class
+
+    def register_agent(self, name: str, agent_class: type[Agent]) -> None:
+        """注册 Agent 类型"""
+        self._agents[name] = agent_class
+
+    def register_flow(self, name: str, flow_class: type) -> None:
+        """注册 Flow 类型"""
+        self._flows[name] = flow_class
+
+    def register_node_handler(self, node_type: str, handler) -> None:
+        """注册自定义节点处理器（所有 Flow 实例共享）"""
+        self._node_handlers[node_type] = handler
+
+    # ── 创建实例 ──
+
+    def create_tool(self, name: str, **kwargs) -> Tool | None:
+        """创建工具实例"""
+        component = self._tools.get(name)
+        if component is None:
+            return None
+        # 已实例化的 Tool 对象直接返回
+        if hasattr(component, "execute_async"):
+            return cast(Tool, component)
+        # 类或工厂函数：实例化
+        if callable(component):
+            return cast(Tool, component(**kwargs) if kwargs else component())
+        return cast(Tool, component)
+
+    def create_agent(self, name: str, **kwargs) -> Agent | None:
+        """创建 Agent 实例"""
+        component = self._agents.get(name)
+        if component is None:
+            return None
+        if callable(component):
+            return cast(Agent, component(**kwargs))
+        return cast(Agent, component)
+
+    def create_flow(self, name: str, **kwargs) -> Any | None:
+        """创建 Flow 实例"""
+        component = self._flows.get(name)
+        if component is None:
+            return None
+        if callable(component):
+            return component(**kwargs)
+        return component
+
+    def create_tools(self, names: list[str]) -> list[Tool]:
+        """批量创建工具"""
+        return [tool for name in names if (tool := self.create_tool(name))]
+
+    # ── 查询 ──
+
+    def get_tool_class(self, name: str) -> Any | None:
+        """获取工具工厂/类"""
+        return self._tools.get(name)
+
+    def get_agent_class(self, name: str) -> type[Agent] | None:
+        """获取 Agent 类型"""
+        return self._agents.get(name)
+
+    def get_flow_class(self, name: str) -> type | None:
+        """获取 Flow 类型"""
+        return self._flows.get(name)
+
+    def get_node_handler(self, node_type: str):
+        """获取节点处理器"""
+        self._ensure_builtins()
+        return self._node_handlers.get(node_type)
+
+    def get_all_node_handlers(self) -> dict[str, Any]:
+        """获取所有节点处理器（内置 + 自定义）"""
+        self._ensure_builtins()
+        return dict(self._node_handlers)
+
+    # ── 列表 ──
+
+    def list_tools(self) -> list[str]:
+        return list(self._tools.keys())
+
+    def list_agents(self) -> list[str]:
+        return list(self._agents.keys())
+
+    def list_flows(self) -> list[str]:
+        return list(self._flows.keys())
+
+    def get_all_info(self) -> dict[str, list[str]]:
+        """获取所有组件信息"""
+        self._ensure_builtins()
+        info: dict[str, list[str]] = {}
+        if self._tools:
+            info["tool"] = self.list_tools()
+        if self._agents:
+            info["agent"] = self.list_agents()
+        if self._flows:
+            info["flow"] = self.list_flows()
+        if self._node_handlers:
+            info["node_handler"] = list(self._node_handlers.keys())
+        return info
+
+
+# Flow 系统专用实例
+registry = FlowRegistry()

{agstack-1.7.0 → agstack-1.8.0}/agstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.7.0
+Version: 1.8.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>
@@ -27,7 +27,7 @@ Requires-Dist: fastapi>=0.133.1
 Requires-Dist: jwcrypto>=1.5.6
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: nebula3-python>=3.8.3
-Requires-Dist: openai>=2.26.0
+Requires-Dist: openai>=2.28.0
 Requires-Dist: passlib[bcrypt]>=1.7.4
 Requires-Dist: pycasbin>=2.8.0
 Requires-Dist: pydantic>=2.12.4
@@ -35,7 +35,7 @@ Requires-Dist: python-multipart>=0.0.20
 Requires-Dist: requests>=2.32.5
 Requires-Dist: RestrictedPython>=7.0
 Requires-Dist: sqlalchemy[asyncio]>=2.0.48
-Requires-Dist: sqlobjects>=1.4.0
+Requires-Dist: sqlobjects>=1.5.0
 Requires-Dist: tiktoken>=0.12.0
 Requires-Dist: uvicorn>=0.41.0
 Dynamic: license-file

{agstack-1.7.0 → agstack-1.8.0}/agstack.egg-info/SOURCES.txt

@@ -6,7 +6,6 @@ agstack/contexts.py
 agstack/decorators.py
 agstack/events.py
 agstack/exceptions.py
-agstack/registry.py
 agstack/schema.py
 agstack/status.py
 agstack.egg-info/PKG-INFO

{agstack-1.7.0 → agstack-1.8.0}/agstack.egg-info/requires.txt

@@ -5,7 +5,7 @@ fastapi>=0.133.1
 jwcrypto>=1.5.6
 loguru>=0.7.3
 nebula3-python>=3.8.3
-openai>=2.26.0
+openai>=2.28.0
 passlib[bcrypt]>=1.7.4
 pycasbin>=2.8.0
 pydantic>=2.12.4
@@ -13,6 +13,6 @@ python-multipart>=0.0.20
 requests>=2.32.5
 RestrictedPython>=7.0
 sqlalchemy[asyncio]>=2.0.48
-sqlobjects>=1.4.0
+sqlobjects>=1.5.0
 tiktoken>=0.12.0
 uvicorn>=0.41.0

{agstack-1.7.0 → agstack-1.8.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agstack"
-version = "1.7.0"
+version = "1.8.0"
 description = "Production-ready toolkit for building FastAPI and LLM applications"
 readme = "README.md"
 license = "MIT"
@@ -46,7 +46,7 @@ dependencies = [
     "jwcrypto>=1.5.6",
     "loguru>=0.7.3",
     "nebula3-python>=3.8.3",
-    "openai>=2.26.0",
+    "openai>=2.28.0",
     "passlib[bcrypt]>=1.7.4",
     "pycasbin>=2.8.0",
     "pydantic>=2.12.4",
@@ -54,7 +54,7 @@ dependencies = [
     "requests>=2.32.5",
     "RestrictedPython>=7.0",
     "sqlalchemy[asyncio]>=2.0.48",
-    "sqlobjects>=1.4.0",
+    "sqlobjects>=1.5.0",
     "tiktoken>=0.12.0",
     "uvicorn>=0.41.0",
 ]

agstack-1.7.0/agstack/llm/flow/registry.py

@@ -1,81 +0,0 @@
-#  Copyright (c) 2020-2026 XtraVisions, All rights reserved.
-
-"""Flow 系统注册中心 - 提供类型安全的组件注册和创建接口
-
-此模块是对全局注册中心的类型安全封装，为 Flow 系统提供：
-- 强类型的返回值（Tool、Agent、Flow 等）
-- Flow 系统的统一命名空间
-- 方便的批量操作和查询方法
-
-推荐在 Flow 模块内部使用此 registry 实例。
-"""
-
-from ...registry import registry as global_registry
-from .agent import Agent
-from .tool import Tool
-
-
-class FlowRegistry:
-    """统一注册中心 - Flow 系统适配器"""
-
-    def __init__(self):
-        self.global_registry = global_registry
-
-    def register_tool(self, name: str, tool_factory) -> None:
-        """注册工具工厂"""
-        self.global_registry.register_tool(name, tool_factory)
-
-    def register_agent(self, name: str, agent_class: type[Agent]) -> None:
-        """注册 Agent 类型"""
-        self.global_registry.register_agent(name, agent_class)
-
-    def register_flow(self, name: str, flow_class: type) -> None:
-        """注册 Flow 类型"""
-        self.global_registry.register_flow(name, flow_class)
-
-    def create_tool(self, name: str) -> Tool | None:
-        """创建工具实例"""
-        return self.global_registry.create_tool(name)
-
-    def create_agent(self, name: str, **kwargs) -> Agent | None:
-        """创建 Agent 实例"""
-        return self.global_registry.create_agent(name, **kwargs)
-
-    def create_flow(self, name: str, **kwargs):
-        """创建 Flow 实例"""
-        return self.global_registry.create_flow(name, **kwargs)
-
-    def create_tools(self, names: list[str]) -> list[Tool]:
-        """批量创建工具"""
-        return self.global_registry.create_tools(names)
-
-    def get_tool_class(self, name: str):
-        """获取工具工厂（兼容性）"""
-        manifest = self.global_registry.get("tool", name)
-        return manifest.component if manifest else None
-
-    def get_agent_class(self, name: str) -> type[Agent] | None:
-        """获取 Agent 类型"""
-        manifest = self.global_registry.get("agent", name)
-        return manifest.component if manifest else None
-
-    def get_flow_class(self, name: str) -> type | None:
-        """获取 Flow 类型"""
-        manifest = self.global_registry.get("flow", name)
-        return manifest.component if manifest else None
-
-    def list_tools(self) -> list[str]:
-        """列出所有工具"""
-        return self.global_registry.list("tool")
-
-    def list_agents(self) -> list[str]:
-        """列出所有 Agent"""
-        return self.global_registry.list("agent")
-
-    def list_flows(self) -> list[str]:
-        """列出所有 Flow"""
-        return self.global_registry.list("flow")
-
-
-# Flow 系统专用实例
-registry = FlowRegistry()

agstack-1.7.0/agstack/registry.py

@@ -1,187 +0,0 @@
-#  Copyright (c) 2020-2025 XtraVisions, All rights reserved.
-
-"""全局统一注册中心"""
-
-from __future__ import annotations
-
-from collections import defaultdict
-from dataclasses import dataclass, field
-from typing import Any
-
-
-@dataclass
-class ComponentManifest:
-    """组件清单"""
-
-    name: str  # 组件唯一标识
-    type: str  # 组件类型 (tool/agent/flow/router/preprocessor)
-    component: Any  # 实际组件对象
-    metadata: dict[str, Any] = field(default_factory=dict)  # 扩展元数据
-    dependencies: list[str] = field(default_factory=list)  # 依赖关系
-    version: str = "1.0.0"  # 版本信息
-
-
-class Registry:
-    """统一注册中心"""
-
-    def __init__(self):
-        self._components: dict[str, dict[str, ComponentManifest]] = defaultdict(dict)
-
-    # 底层通用接口
-    def register(self, manifest: ComponentManifest) -> None:
-        """注册组件"""
-        self._components[manifest.type][manifest.name] = manifest
-
-    def get(self, component_type: str, name: str) -> ComponentManifest | None:
-        """获取组件清单"""
-        return self._components[component_type].get(name)
-
-    def list(self, component_type: str) -> list[str]:
-        """列出指定类型的所有组件名称"""
-        return list(self._components[component_type].keys())
-
-    # 快捷注册接口
-    def register_tool(
-        self,
-        name: str,
-        tool_class: Any,
-        metadata: dict[str, Any] | None = None,
-        dependencies: list[str] | None = None,  # noqa
-        version: str = "1.0.0",
-    ) -> None:
-        """注册工具"""
-        manifest = ComponentManifest(
-            name=name,
-            type="tool",
-            component=tool_class,
-            metadata=metadata or {},
-            dependencies=dependencies or [],
-            version=version,
-        )
-        self.register(manifest)
-
-    def register_agent(
-        self,
-        name: str,
-        agent_class: Any,
-        metadata: dict[str, Any] | None = None,
-        dependencies: list[str] | None = None,  # noqa
-        version: str = "1.0.0",
-    ) -> None:
-        """注册代理"""
-        manifest = ComponentManifest(
-            name=name,
-            type="agent",
-            component=agent_class,
-            metadata=metadata or {},
-            dependencies=dependencies or [],
-            version=version,
-        )
-        self.register(manifest)
-
-    def register_flow(
-        self,
-        name: str,
-        flow_class: Any,
-        metadata: dict[str, Any] | None = None,
-        dependencies: list[str] | None = None,  # noqa
-        version: str = "1.0.0",
-    ) -> None:
-        """注册工作流"""
-        manifest = ComponentManifest(
-            name=name,
-            type="flow",
-            component=flow_class,
-            metadata=metadata or {},
-            dependencies=dependencies or [],
-            version=version,
-        )
-        self.register(manifest)
-
-    def register_router(
-        self,
-        name: str,
-        router: Any,
-        metadata: dict[str, Any] | None = None,
-        dependencies: list[str] | None = None,  # noqa
-        version: str = "1.0.0",
-    ) -> None:
-        """注册路由"""
-        manifest = ComponentManifest(
-            name=name,
-            type="router",
-            component=router,
-            metadata=metadata or {},
-            dependencies=dependencies or [],
-            version=version,
-        )
-        self.register(manifest)
-
-    def register_preprocessor(
-        self,
-        name: str,
-        processor_class: Any,
-        metadata: dict[str, Any] | None = None,
-        dependencies: list[str] | None = None,  # noqa
-        version: str = "1.0.0",
-    ) -> None:
-        """注册预处理器"""
-        manifest = ComponentManifest(
-            name=name,
-            type="preprocessor",
-            component=processor_class,
-            metadata=metadata or {},
-            dependencies=dependencies or [],
-            version=version,
-        )
-        self.register(manifest)
-
-    # 创建实例接口
-    def create_tool(self, name: str, **kwargs) -> Any | None:
-        """创建工具实例"""
-        manifest = self.get("tool", name)
-        if not manifest:
-            return None
-
-        component = manifest.component
-        # 已实例化的 Tool 对象直接返回
-        if hasattr(component, "execute_async"):
-            return component
-        # 类或工厂函数：实例化
-        if callable(component):
-            return component(**kwargs) if kwargs else component()
-        return component
-
-    def create_agent(self, name: str, **kwargs) -> Any | None:
-        """创建代理实例"""
-        manifest = self.get("agent", name)
-        if not manifest:
-            return None
-
-        component = manifest.component
-        if callable(component):
-            return component(**kwargs)
-        return component
-
-    def create_flow(self, name: str, **kwargs) -> Any | None:
-        """创建工作流实例"""
-        manifest = self.get("flow", name)
-        if not manifest:
-            return None
-
-        component = manifest.component
-        if callable(component):
-            return component(**kwargs)
-        return component
-
-    def create_tools(self, names: list[str]) -> list[Any]:  # noqa
-        """批量创建工具"""
-        return [tool for name in names if (tool := self.create_tool(name))]
-
-    def get_all_info(self) -> dict[str, list[str]]:  # noqa
-        """获取所有组件信息"""
-        return {component_type: list(components.keys()) for component_type, components in self._components.items()}
-
-
-# 全局注册中心实例
-registry = Registry()