illusion-code 0.1.0__py3-none-any.whl

illusion/state/app_state.py

@@ -0,0 +1,67 @@
+"""
+应用状态模块
+===========
+
+本模块定义 IllusionCode 应用状态数据模型。
+
+主要功能：
+    - 定义共享的UI/会话状态数据结构
+    - 支持状态属性的不可变更新
+
+类说明：
+    - AppState: 应用状态数据类
+
+使用示例：
+    >>> from illusion.state import AppState
+    >>> state = AppState(model="claude-3-5-sonnet-20241022", permission_mode="default")
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+
+@dataclass
+class AppState:
+    """共享的可变UI/会话状态数据类
+    
+    Attributes:
+        model: 当前使用的模型名称
+        permission_mode: 权限模式 (default/plan/bypassPermissions 等)
+        ui_language: UI语言 (默认 zh-CN)
+        cwd: 当前工作目录
+        provider: API提供者名称
+        auth_status: 认证状态
+        base_url: API基础URL
+        fast_mode: 是否启用快速模式
+        effort: 推理 Effort 级别 (low/medium/high)
+        passes: 推理通过次数
+        mcp_connected: 已连接的MCP服务器数量
+        mcp_failed: 失败的MCP服务器数量
+        bridge_sessions: 活跃的桥接会话数量
+        output_style: 输出样式名称
+        show_thinking: 是否显示思考过程
+        phase: 会话阶段 (idle/thinking/tool_executing)
+        team_context: 当前会话的团队上下文（若已创建团队）
+    """
+
+    model: str  # 模型名称
+    permission_mode: str  # 权限模式
+    ui_language: str = "zh-CN"  # UI语言
+    cwd: str = "."  # 当前工作目录
+    provider: str = "unknown"  # API提供者
+    auth_status: str = "missing"  # 认证状态
+    base_url: str = ""  # API基础URL
+    fast_mode: bool = False  # 快速模式标志
+    effort: str = "medium"  # 推理 Effort 级别
+    passes: int = 1  # 推理通过次数
+    mcp_connected: int = 0  # 已连接的MCP服务器数量
+    mcp_failed: int = 0  # 失败的MCP服务器数量
+    bridge_sessions: int = 0  # 活跃的桥接会话数量
+    output_style: str = "default"  # 输出样式名称
+    show_thinking: bool = True  # 是否显示思考过程
+    phase: str = "idle"  # 会话阶段: idle / thinking / tool_executing
+    session_id: str = ""  # 当前会话 ID
+    context_window: int = 0  # 上下文窗口大小（tokens）
+    context_tokens: int = 0  # 当前已用 tokens（估算）
+    team_context: dict[str, object] | None = None  # 团队上下文

illusion/state/store.py

@@ -0,0 +1,93 @@
+"""
+状态存储模块
+===========
+
+本模块实现可观察的应用状态存储容器。
+
+主要功能：
+    - 提供响应式状态管理
+    - 支持状态变更监听和通知
+
+类说明：
+    - AppStateStore: 应用状态存储容器
+    - Listener: 状态监听器类型
+
+使用示例：
+    >>> from illusion.state import AppState, AppStateStore
+    >>> store = AppStateStore(AppState(model="claude", permission_mode="default", theme="default"))
+    >>> state = store.get()
+"""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from dataclasses import replace
+
+from illusion.state.app_state import AppState
+
+
+# 状态监听器类型别名：接收AppState并返回None的Callable
+Listener = Callable[[AppState], None]
+
+
+class AppStateStore:
+    """可观察的应用状态存储容器
+    
+    提供 immutable 风格的状态更新，通过监听器模式通知状态变更。
+    
+    Attributes:
+        _state: 当前的应用状态快照
+        _listeners: 注册的状态监听器列表
+    
+    Example:
+        >>> store = AppStateStore(initial_state)
+        >>> store.subscribe(lambda state: print(f"State changed: {state.model}"))
+        >>> store.set(model="new-model")  # 触发所有监听器
+    """
+
+    def __init__(self, initial_state: AppState) -> None:
+        """初始化状态存储容器
+        
+        Args:
+            initial_state: 初始的应用状态
+        """
+        self._state = initial_state  # 内部状态存储
+        self._listeners: list[Listener] = []  # 监听器列表初始化
+
+    def get(self) -> AppState:
+        """获取当前状态的快照
+        
+        Returns:
+            AppState: 当前应用状态的副本
+        """
+        return self._state  # 返回当前状态
+
+    def set(self, **updates) -> AppState:
+        """更新状态并通知所有监听器
+        
+        Keyword Args:
+            updates: 要更新的状态属性
+        Returns:
+            AppState: 更新后的新状态
+        """
+        self._state = replace(self._state, **updates)  # 使用dataclasses.replace创建新状态
+        for listener in list(self._listeners):  # 通知所有监听器
+            listener(self._state)
+        return self._state  # 返回更新后的状态
+
+    def subscribe(self, listener: Listener) -> Callable[[], None]:
+        """注册状态监听器并返回取消订阅的回调函数
+        
+        Args:
+            listener: 状态变更时调用的监听器函数
+        Returns:
+            Callable[[], None]: 用于取消订阅的回调函数
+        """
+        self._listeners.append(listener)  # 添加监听器
+
+        def _unsubscribe() -> None:
+            """取消订阅的内部函数"""
+            if listener in self._listeners:  # 检查监听器是否存在
+                self._listeners.remove(listener)  # 移除监听器
+
+        return _unsubscribe  # 返回取消订阅函数

illusion/swarm/__init__.py

@@ -0,0 +1,71 @@
+"""
+Agent 派发后端抽象模块
+=====================
+
+本模块提供代理执行的后端抽象功能。
+
+主要组件：
+    - agent_executor: 代理执行器核心模块
+    - in_process: 进程内执行后端
+    - subprocess_backend: 子进程执行后端
+    - types: 类型定义
+    - worktree: Git worktree 管理
+
+使用示例：
+    >>> from illusion.swarm import InProcessBackend, SubprocessBackend
+    >>> from illusion.swarm.agent_executor import run_agent_in_process
+"""
+
+from __future__ import annotations
+
+# 导入核心组件
+from illusion.swarm.agent_executor import (
+    AgentAbortController,
+    AgentExecutionContext,
+    AgentResult,
+    AgentSpawnConfig,
+    TaskNotification,
+    TeammateMessage,
+    format_task_notification,
+    get_active_agent,
+    get_active_agent_by_name,
+    get_agent_context,
+    list_active_agents,
+    parse_task_notification,
+    resolve_agent_tools,
+    run_agent_in_process,
+    run_agent_subprocess,
+)
+from illusion.swarm.in_process import InProcessBackend
+from illusion.swarm.subprocess_backend import SubprocessBackend
+from illusion.swarm.types import (
+    BackendType,
+    SpawnResult,
+    TeammateExecutor,
+    TeammateSpawnConfig,
+)
+
+# 导出列表：定义公开 API
+__all__ = [
+    "AgentAbortController",
+    "AgentExecutionContext",
+    "AgentResult",
+    "AgentSpawnConfig",
+    "BackendType",
+    "InProcessBackend",
+    "SpawnResult",
+    "SubprocessBackend",
+    "TaskNotification",
+    "TeammateExecutor",
+    "TeammateMessage",
+    "TeammateSpawnConfig",
+    "format_task_notification",
+    "get_active_agent",
+    "get_active_agent_by_name",
+    "get_agent_context",
+    "list_active_agents",
+    "parse_task_notification",
+    "resolve_agent_tools",
+    "run_agent_in_process",
+    "run_agent_subprocess",
+]