sycommon-python-lib 0.2.1b2__tar.gz → 0.2.1b3__tar.gz

{sycommon_python_lib-0.2.1b2 → sycommon_python_lib-0.2.1b3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sycommon-python-lib
-Version: 0.2.1b2
+Version: 0.2.1b3
 Summary: Add your description here
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
@@ -19,6 +19,7 @@ Requires-Dist: langchain-core>=1.2.20
 Requires-Dist: langchain-openai>=1.1.11
 Requires-Dist: langfuse>=4.0.1
 Requires-Dist: langgraph>=1.1.3
+Requires-Dist: ldap3>=2.9.1
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: mysql-connector-python>=9.6.0
 Requires-Dist: nacos-sdk-python<3.0,>=2.0.9
@@ -33,6 +34,7 @@ Requires-Dist: sqlalchemy[asyncio]>=2.0.48
 Requires-Dist: starlette[full]>=1.0.0
 Requires-Dist: tiktoken>=0.12.0
 Requires-Dist: uvicorn>=0.42.0
+Requires-Dist: wecom-aibot-python-sdk>=1.0.2
 
 # sycommon-python-lib
 

{sycommon_python_lib-0.2.1b2 → sycommon_python_lib-0.2.1b3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sycommon-python-lib"
-version = "0.2.1b2"
+version = "0.2.1b3"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -20,6 +20,7 @@ dependencies = [
     "langchain-openai>=1.1.11",
     "langfuse>=4.0.1",
     "langgraph>=1.1.3",
+    "ldap3>=2.9.1",
     "loguru>=0.7.3",
     "mysql-connector-python>=9.6.0",
     "nacos-sdk-python>=2.0.9,<3.0",
@@ -34,6 +35,7 @@ dependencies = [
     "starlette[full]>=1.0.0",
     "tiktoken>=0.12.0",
     "uvicorn>=0.42.0",
+    "wecom-aibot-python-sdk>=1.0.2",
 ]
 
 [tool.setuptools]

sycommon_python_lib-0.2.1b3/src/sycommon/agent/__init__.py

@@ -0,0 +1,150 @@
+"""
+Agent 模块
+
+提供数字员工和数字团队的高级封装方案，方便后续其他服务接入使用。
+
+核心组件:
+- DeepAgent: 数字员工（单 Agent）
+- MultiAgentTeam: 数字团队（多 Agent 协作）
+- SandboxBackend: 沙箱后端（用于代码执行）
+- ChatEvent: 聊天事件（前后端交互数据结构）
+
+使用示例:
+
+    # 1. 简单的数字员工
+    from sycommon.agent import create_deep_agent
+
+    agent = await create_deep_agent(user_id="user123")
+    async for event in agent.chat("帮我写一个 Python 脚本"):
+        yield event.to_sse()  # 直接转换为 SSE 格式
+
+    # 2. 自定义模型的数字员工
+    from sycommon.agent import create_deep_agent, AgentConfig
+
+    config = AgentConfig(
+        model_name="gpt-4o",
+        model_provider="openai",
+        system_prompt="你是一个后端工程师...",
+    )
+    agent = await create_deep_agent(user_id="user123", config=config)
+
+    # 3. 数字团队（多 Agent 协作）
+    from sycommon.agent import create_multi_agent_team, TeamMember
+
+    team = await create_multi_agent_team(
+        user_id="user123",
+        members=[
+            TeamMember(name="后端工程师", description="负责后端开发..."),
+            TeamMember(name="前端工程师", description="负责前端开发..."),
+        ],
+    )
+    async for event in team.chat("帮我开发一个 TODO 应用"):
+        yield event.to_sse()
+
+    # 4. 使用 ChatEventBuilder 构建事件
+    from sycommon.agent import ChatEventBuilder
+
+    event = ChatEventBuilder.ai_chunk("你好")
+    event = ChatEventBuilder.tool_call("execute", {"command": "ls"}, "call_123")
+
+    # 5. 通过 Services.plugins 接入
+    from fastapi import FastAPI
+    from sycommon.services import Services
+
+    app = Services.plugins(
+        app=FastAPI(),
+        deep_agent_service=True,  # 启用数字员工服务
+        # 或
+        multi_agent_service=True,  # 启用数字团队服务
+    )
+"""
+
+from sycommon.agent.sandbox import (
+    HTTPSandboxBackend,
+    SandboxPool,
+    SandboxInstance,
+    SandboxRecoveryManager,
+    areset_all_sandboxes,
+    aget_sandbox_status,
+)
+from sycommon.agent.deep_agent import (
+    AgentConfig,
+    DeepAgent,
+    create_deep_agent,
+)
+from sycommon.agent.multi_agent_team import (
+    TeamMember,
+    TeamConfig,
+    MultiAgentTeam,
+    create_multi_agent_team,
+)
+from sycommon.agent.agent_manager import (
+    AgentManager,
+    AgentInfo,
+    get_agent_manager,
+    create_agent,
+    create_team,
+    get_agent,
+    destroy_agent,
+)
+from sycommon.agent.chat_events import (
+    ChatEvent,
+    ChatEventBuilder,
+    EventType,
+    DEFAULT_AGENT_NAME,
+    is_multi_agent_mode,
+    # 便捷函数
+    progress_event,
+    ai_chunk_event,
+    tool_call_event,
+    tool_result_event,
+    agent_switch_event,
+    parallel_agents_event,
+    agent_complete_event,
+    done_event,
+    error_event,
+    cancelled_event,
+)
+
+__all__ = [
+    # 沙箱
+    "HTTPSandboxBackend",
+    "SandboxPool",
+    "SandboxInstance",
+    "SandboxRecoveryManager",
+    "areset_all_sandboxes",
+    "aget_sandbox_status",
+
+    # 数字员工
+    "AgentConfig",
+    "DeepAgent",
+    "create_deep_agent",
+
+    # 数字团队
+    "TeamMember",
+    "TeamConfig",
+    "MultiAgentTeam",
+    "create_multi_agent_team",
+
+    # Agent 管理
+    "AgentManager",
+    "AgentInfo",
+    "get_agent_manager",
+
+    # 聊天事件
+    "ChatEvent",
+    "ChatEventBuilder",
+    "EventType",
+    "DEFAULT_AGENT_NAME",
+    "is_multi_agent_mode",
+    "progress_event",
+    "ai_chunk_event",
+    "tool_call_event",
+    "tool_result_event",
+    "agent_switch_event",
+    "parallel_agents_event",
+    "agent_complete_event",
+    "done_event",
+    "error_event",
+    "cancelled_event",
+]

sycommon_python_lib-0.2.1b3/src/sycommon/agent/agent_manager.py

@@ -0,0 +1,398 @@
+"""
+Agent 管理器
+
+提供数字员工和数字团队的动态创建和管理能力。
+支持运行时动态创建、销毁、查询 Agent。
+
+使用示例:
+
+    from sycommon.agent import AgentManager, AgentConfig, TeamMember, TeamConfig
+
+    # 获取管理器实例
+    manager = AgentManager()
+
+    # 1. 创建单个数字员工
+    agent = await manager.create_agent(
+        user_id="user123",
+        config=AgentConfig(
+            model_name="gpt-4o",
+            system_prompt="你是后端工程师...",
+        ),
+    )
+
+    # 2. 创建数字团队
+    team = await manager.create_team(
+        user_id="user456",
+        members=[
+            TeamMember(name="后端工程师", description="负责后端开发"),
+            TeamMember(name="前端工程师", description="负责前端开发"),
+        ],
+        config=TeamConfig(model_name="gpt-4o"),
+    )
+
+    # 3. 获取已存在的 agent
+    agent = await manager.get_agent("user123")
+
+    # 4. 销毁 agent
+    await manager.destroy_agent("user123")
+
+    # 5. 列出所有 agent
+    agents = manager.list_agents()
+"""
+
+import asyncio
+from typing import Any, Dict, List, Optional, Union
+from pathlib import Path
+
+from pydantic import BaseModel
+from sycommon.logging.kafka_log import SYLogger
+from sycommon.agent.sandbox.http_sandbox_backend import HTTPSandboxBackend
+from sycommon.agent.deep_agent import DeepAgent, AgentConfig, create_deep_agent
+from sycommon.agent.multi_agent_team import MultiAgentTeam, TeamMember, TeamConfig, create_multi_agent_team
+from sycommon.config.Config import SingletonMeta
+
+
+class AgentInfo(BaseModel):
+    """Agent 信息（用于序列化返回）"""
+    user_id: str
+    type: str  # "agent" 或 "team"
+    config: Dict[str, Any] = {}
+    members: List[Dict[str, str]] = []  # 团队成员信息
+    sandbox_backend: Optional[str] = None
+
+
+class AgentManager(metaclass=SingletonMeta):
+    """
+    Agent 管理器
+
+    负责数字员工和数字团队的生命周期管理：
+    - 创建/销毁 Agent
+    - 会话管理
+    - 配置管理
+    """
+
+    def __init__(self):
+        self._agents: Dict[str, Union[DeepAgent, MultiAgentTeam]] = {}
+        self._agent_configs: Dict[str, AgentInfo] = {}
+        self._sandbox_backends: Dict[str, HTTPSandboxBackend] = {}
+        self._lock = asyncio.Lock()
+
+    async def create_agent(
+        self,
+        user_id: str,
+        config: Optional[AgentConfig] = None,
+        project_root: Optional[Path] = None,
+        force_recreate: bool = False,
+    ) -> DeepAgent:
+        """
+        创建或获取数字员工
+
+        Args:
+            user_id: 用户 ID
+            config: Agent 配置（AgentConfig 对象）
+            project_root: 项目根目录
+            force_recreate: 是否强制重新创建（即使已存在）
+
+        Returns:
+            DeepAgent: 数字员工实例
+        """
+        async with self._lock:
+            # 检查是否已存在
+            if not force_recreate and user_id in self._agents:
+                agent = self._agents[user_id]
+                if isinstance(agent, DeepAgent):
+                    return agent
+
+            # 使用默认配置
+            if config is None:
+                config = AgentConfig()
+
+            # 创建 agent
+            agent = await create_deep_agent(
+                user_id=user_id,
+                config=config,
+                project_root=project_root,
+            )
+
+            # 保存引用
+            self._agents[user_id] = agent
+            self._agent_configs[user_id] = AgentInfo(
+                user_id=user_id,
+                type="agent",
+                config=config.model_dump(exclude={'tools'}),  # 排除不可序列化的 tools
+                sandbox_backend=agent.sandbox_backend.base_url,
+            )
+            self._sandbox_backends[user_id] = agent.sandbox_backend
+
+            SYLogger.info(f"[AgentManager] 创建数字员工: {user_id}")
+            return agent
+
+    async def create_team(
+        self,
+        user_id: str,
+        members: List[TeamMember],
+        config: Optional[TeamConfig] = None,
+        project_root: Optional[Path] = None,
+        force_recreate: bool = False,
+    ) -> MultiAgentTeam:
+        """
+        创建或获取数字团队
+
+        Args:
+            user_id: 用户 ID
+            members: 团队成员列表（TeamMember 对象）
+            config: 团队配置（TeamConfig 对象）
+            project_root: 项目根目录
+            force_recreate: 是否强制重新创建（即使已存在）
+
+        Returns:
+            MultiAgentTeam: 数字团队实例
+        """
+        async with self._lock:
+            # 检查是否已存在
+            if not force_recreate and user_id in self._agents:
+                agent = self._agents[user_id]
+                if isinstance(agent, MultiAgentTeam):
+                    return agent
+
+            # 使用默认配置
+            if config is None:
+                config = TeamConfig()
+
+            # 创建团队
+            team = await create_multi_agent_team(
+                user_id=user_id,
+                members=members,
+                config=config,
+                project_root=project_root,
+            )
+
+            # 保存引用
+            self._agents[user_id] = team
+            self._agent_configs[user_id] = AgentInfo(
+                user_id=user_id,
+                type="team",
+                config=config.model_dump(exclude={'shared_tools'}),  # 排除不可序列化的 tools
+                members=[{"name": m.name, "description": m.description} for m in members],
+                sandbox_backend=team.sandbox_backend.base_url,
+            )
+            self._sandbox_backends[user_id] = team.sandbox_backend
+
+            SYLogger.info(f"[AgentManager] 创建数字团队: {user_id}, 成员: {[m.name for m in members]}")
+            return team
+
+    async def get_agent(self, user_id: str) -> Optional[Union[DeepAgent, MultiAgentTeam]]:
+        """
+        获取已存在的 Agent
+
+        Args:
+            user_id: 用户 ID
+
+        Returns:
+            DeepAgent | MultiAgentTeam | None: Agent 实例或 None
+        """
+        return self._agents.get(user_id)
+
+    async def get_or_create_agent(
+        self,
+        user_id: str,
+        config: Optional[AgentConfig] = None,
+        project_root: Optional[Path] = None,
+    ) -> DeepAgent:
+        """
+        获取或创建数字员工
+
+        Args:
+            user_id: 用户 ID
+            config: Agent 配置
+            project_root: 项目根目录
+
+        Returns:
+            DeepAgent: 数字员工实例
+        """
+        agent = await self.get_agent(user_id)
+        if agent and isinstance(agent, DeepAgent):
+            return agent
+        return await self.create_agent(user_id, config, project_root)
+
+    async def get_or_create_team(
+        self,
+        user_id: str,
+        members: List[TeamMember],
+        config: Optional[TeamConfig] = None,
+        project_root: Optional[Path] = None,
+    ) -> MultiAgentTeam:
+        """
+        获取或创建数字团队
+
+        Args:
+            user_id: 用户 ID
+            members: 团队成员列表
+            config: 团队配置
+            project_root: 项目根目录
+
+        Returns:
+            MultiAgentTeam: 数字团队实例
+        """
+        agent = await self.get_agent(user_id)
+        if agent and isinstance(agent, MultiAgentTeam):
+            return agent
+        return await self.create_team(user_id, members, config, project_root)
+
+    async def destroy_agent(self, user_id: str) -> bool:
+        """
+        销毁 Agent
+
+        Args:
+            user_id: 用户 ID
+
+        Returns:
+            bool: 是否成功销毁
+        """
+        async with self._lock:
+            if user_id not in self._agents:
+                return False
+
+            # 清理引用
+            del self._agents[user_id]
+            if user_id in self._agent_configs:
+                del self._agent_configs[user_id]
+            if user_id in self._sandbox_backends:
+                del self._sandbox_backends[user_id]
+
+            SYLogger.info(f"[AgentManager] 销毁 Agent: {user_id}")
+            return True
+
+    def list_agents(self) -> List[AgentInfo]:
+        """
+        列出所有 Agent
+
+        Returns:
+            List[AgentInfo]: Agent 信息列表
+        """
+        return list(self._agent_configs.values())
+
+    def get_agent_info(self, user_id: str) -> Optional[AgentInfo]:
+        """
+        获取 Agent 信息
+
+        Args:
+            user_id: 用户 ID
+
+        Returns:
+            AgentInfo | None: Agent 信息
+        """
+        return self._agent_configs.get(user_id)
+
+    def is_multi_agent(self, user_id: str) -> bool:
+        """
+        判断是否为多 Agent 模式
+
+        Args:
+            user_id: 用户 ID
+
+        Returns:
+            bool: True 表示数字团队，False 表示单个数字员工
+        """
+        info = self._agent_configs.get(user_id)
+        return info.type == "team" if info else False
+
+    @property
+    def count(self) -> int:
+        """当前 Agent 数量"""
+        return len(self._agents)
+
+    async def clear_all(self) -> int:
+        """
+        清理所有 Agent
+
+        Returns:
+            int: 清理的 Agent 数量
+        """
+        async with self._lock:
+            count = len(self._agents)
+            self._agents.clear()
+            self._agent_configs.clear()
+            self._sandbox_backends.clear()
+            SYLogger.info(f"[AgentManager] 清理所有 Agent: {count} 个")
+            return count
+
+
+# 全局便捷函数
+def get_agent_manager() -> AgentManager:
+    """获取 Agent 管理器实例"""
+    return AgentManager()
+
+
+async def create_agent(
+    user_id: str,
+    config: Optional[AgentConfig] = None,
+    project_root: Optional[Path] = None,
+    force_recreate: bool = False,
+) -> DeepAgent:
+    """
+    便捷函数：创建数字员工
+
+    Args:
+        user_id: 用户 ID
+        config: Agent 配置（AgentConfig 对象）
+        project_root: 项目根目录
+        force_recreate: 是否强制重新创建
+
+    Returns:
+        DeepAgent: 数字员工实例
+    """
+    manager = get_agent_manager()
+    return await manager.create_agent(user_id, config, project_root, force_recreate)
+
+
+async def create_team(
+    user_id: str,
+    members: List[TeamMember],
+    config: Optional[TeamConfig] = None,
+    project_root: Optional[Path] = None,
+    force_recreate: bool = False,
+) -> MultiAgentTeam:
+    """
+    便捷函数：创建数字团队
+
+    Args:
+        user_id: 用户 ID
+        members: 团队成员列表（TeamMember 对象）
+        config: 团队配置（TeamConfig 对象）
+        project_root: 项目根目录
+        force_recreate: 是否强制重新创建
+
+    Returns:
+        MultiAgentTeam: 数字团队实例
+    """
+    manager = get_agent_manager()
+    return await manager.create_team(user_id, members, config, project_root, force_recreate)
+
+
+async def get_agent(user_id: str) -> Optional[Union[DeepAgent, MultiAgentTeam]]:
+    """
+    便捷函数：获取 Agent
+
+    Args:
+        user_id: 用户 ID
+
+    Returns:
+        DeepAgent | MultiAgentTeam | None: Agent 实例
+    """
+    manager = get_agent_manager()
+    return await manager.get_agent(user_id)
+
+
+async def destroy_agent(user_id: str) -> bool:
+    """
+    便捷函数：销毁 Agent
+
+    Args:
+        user_id: 用户 ID
+
+    Returns:
+        bool: 是否成功
+    """
+    manager = get_agent_manager()
+    return await manager.destroy_agent(user_id)