auto-coder 1.0.0__py3-none-any.whl → 2.0.1__py3-none-any.whl

autocoder/sdk/cli/options.py

@@ -1,19 +1,33 @@
 """
 CLI 选项定义模块
 
-定义命令行接口的选项和结果数据结构。
+定义命令行接口的选项数据结构。
 """
 
+import os
+from pathlib import Path
 from dataclasses import dataclass, field
 from typing import Optional, Dict, Any
 
 from ..constants import DEFAULT_MODEL
+from ..models.responses import CLIResult  # 从 models.responses 导入，避免重复定义
+from autocoder.common.international import get_message_with_format, get_message
+
+
+def get_default_async_workdir() -> str:
+    """获取默认的异步运行器工作目录，支持多平台"""
+    home_dir = Path.home()
+    return str(home_dir / ".auto-coder" / "async_agent")
 
 
 @dataclass
 class CLIOptions:
     """CLI选项数据类，用于配置命令行工具的行为。"""
     
+    # 命令选项
+    command: Optional[str] = None  # 子命令名称，如 'config'
+    config_args: list = field(default_factory=list)  # config 命令的参数列表
+    
     # 会话选项
     continue_session: bool = False  # 继续最近的对话
     resume_session: Optional[str] = None  # 恢复特定会话的ID
@@ -27,47 +41,106 @@ class CLIOptions:
     verbose: bool = False  # 是否输出详细信息
     
     # 高级选项
-    max_turns: int = 3  # 最大对话轮数
+    max_turns: int = 10000  # 最大对话轮数
     system_prompt: Optional[str] = None  # 系统提示
+    system_prompt_path: Optional[str] = None  # 系统提示文件路径
     allowed_tools: list = field(default_factory=list)  # 允许使用的工具列表
     permission_mode: str = "manual"  # 权限模式，可选值: manual, acceptEdits
     model: Optional[str] = None  # 模型名称，如 gpt-4, gpt-3.5-turbo 等
+    include_rules: bool = False  # 是否包含规则文件
     pr: bool = False  # 是否创建 PR
+    is_sub_agent: bool = False  # 是否为子代理模式，避免conversation冲突
+    loop: int = 1  # 循环执行次数，默认为1
+    loop_keep_conversation: bool = False  # 循环时是否保持对话连续性
+    loop_additional_prompt: Optional[str] = None  # 循环时的额外提示，可自定义
+    include_libs: list = field(default_factory=list)  # 要包含的LLM friendly packages列表
+    
+    # Async Agent Runner 选项
+    async_mode: bool = False  # 是否启用异步代理运行器模式
+    split_mode: str = "h1"  # 分割模式: h1, h2, h3, any, delimiter
+    delimiter: str = "==="  # 自定义分隔符（当 split_mode=delimiter 时使用）
+    min_level: int = 1  # 最小标题级别（当 split_mode=any 时使用）
+    max_level: int = 3  # 最大标题级别（当 split_mode=any 时使用）
+    workdir: str = field(default_factory=get_default_async_workdir)  # 工作目录
+    from_branch: str = ""  # 基础分支（为空时自动检测）
+    bg_mode: bool = False  # 后台运行模式
+    task_prefix: str = ""  # 任务名称前缀（异步模式下使用）
+    worktree_name: Optional[str] = None  # 指定的 worktree 名称（为空时自动生成）
     
     def validate(self) -> None:
         """验证选项的有效性。"""
         # 验证输出格式
         valid_formats = ["text", "json", "stream-json"]
         if self.output_format not in valid_formats:
-            raise ValueError(f"无效的输出格式: {self.output_format}，有效值为: {', '.join(valid_formats)}")
+            raise ValueError(get_message_with_format("invalid_output_format", 
+                                                   format=self.output_format, 
+                                                   valid_formats=', '.join(valid_formats)))
+        
+        # stream-json 格式建议开启verbose以获得实时输出
+        # 但不强制要求，JSON格式可以在非verbose模式下正常工作
         
         # 验证输入格式
         if self.input_format not in valid_formats:
-            raise ValueError(f"无效的输入格式: {self.input_format}，有效值为: {', '.join(valid_formats)}")
+            raise ValueError(get_message_with_format("invalid_input_format", 
+                                                   format=self.input_format, 
+                                                   valid_formats=', '.join(valid_formats)))
         
         # 验证权限模式
         valid_permission_modes = ["manual", "acceptEdits"]
         if self.permission_mode not in valid_permission_modes:
-            raise ValueError(f"无效的权限模式: {self.permission_mode}，有效值为: {', '.join(valid_permission_modes)}")
-        
-        # 验证最大对话轮数（-1表示不限制）
-        if self.max_turns <= 0 and self.max_turns != -1:
-            raise ValueError("max_turns必须为正数或-1（表示不限制）")
+            raise ValueError(get_message_with_format("invalid_permission_mode", 
+                                                   mode=self.permission_mode, 
+                                                   valid_modes=', '.join(valid_permission_modes)))                
         
         # 验证会话选项的互斥性
         if self.continue_session and self.resume_session:
-            raise ValueError("continue_session和resume_session不能同时设置")
+            raise ValueError(get_message_with_format("session_options_conflict"))
         
-        # 验证模型参数必须存在
-        if not self.model or not self.model.strip():
-            raise ValueError("model参数是必需的，请使用 --model 参数指定要使用的模型")
-
+        # 验证 system_prompt 和 system_prompt_path 的互斥性
+        if self.system_prompt and self.system_prompt_path:
+            raise ValueError("只能配置 --system-prompt 或 --system-prompt-path 中的一个，不能同时配置两个")
+        
+        # 验证异步模式相关参数
+        if self.async_mode:
+            # 验证分割模式
+            valid_split_modes = ["h1", "h2", "h3", "any", "delimiter"]
+            if self.split_mode not in valid_split_modes:
+                raise ValueError(get_message_with_format("invalid_split_mode", 
+                                                       mode=self.split_mode, 
+                                                       valid_modes=', '.join(valid_split_modes)))
+            
+            # 验证标题级别范围
+            if self.split_mode == "any":
+                if self.min_level < 1 or self.max_level < 1:
+                    raise ValueError(get_message("heading_level_must_be_positive"))
+                if self.min_level > self.max_level:
+                    raise ValueError(get_message("min_level_cannot_exceed_max"))
+            
+            # 验证分隔符
+            if self.split_mode == "delimiter" and not self.delimiter.strip():
+                raise ValueError(get_message("delimiter_required_for_delimiter_mode"))
+        
+        # 自动选择模型或验证指定的模型（config 命令时跳过）
+        if self.command != "config":
+            if not self.model or not self.model.strip():
+                self.model = self._auto_select_model()
+                if not self.model:
+                    raise ValueError(get_message_with_format("model_auto_select_failed"))
 
-@dataclass
-class CLIResult:
-    """CLI执行结果数据类，用于返回命令执行的结果。"""
-    
-    success: bool  # 是否成功执行
-    output: str = ""  # 输出内容
-    error: Optional[str] = None  # 错误信息
-    debug_info: Optional[Dict[str, Any]] = None  # 调试信息
+    def _auto_select_model(self) -> Optional[str]:
+        """自动选择第一个设置了API key的模型"""
+        try:
+            from autocoder.common.llms import LLMManager
+            
+            manager = LLMManager()
+            all_models = manager.get_all_models()
+            
+            # 遍历所有模型，找到第一个有API key的模型
+            for model_name, model in all_models.items():
+                if manager.has_key(model_name):
+                    return model_name
+            
+            return None
+        except Exception:
+            # 如果导入或初始化失败，返回 None
+            return None

autocoder/sdk/constants.py

@@ -6,47 +6,154 @@ Auto-Coder SDK 常量定义
 定义SDK中使用的各种常量，包括默认值、配置选项、错误码等。
 """
 
+from enum import Enum, auto
+from autocoder.common.international import get_message
+
 # 版本信息
-SDK_VERSION = "0.1.0"
+SDK_VERSION = "1.7.4"
 
 # 默认配置
-DEFAULT_MAX_TURNS = 3
+DEFAULT_MAX_TURNS = 10000
 DEFAULT_OUTPUT_FORMAT = "text"
 DEFAULT_PERMISSION_MODE = "manual"
 DEFAULT_SESSION_TIMEOUT = 3600  # 1小时
 DEFAULT_MODEL = "deepseek/v3"  # 默认使用的模型
 
-# 支持的输出格式
-OUTPUT_FORMATS = {
-    "text": "纯文本格式",
-    "json": "JSON格式",
-    "stream-json": "流式JSON格式"
-}
-
-# 支持的输入格式
-INPUT_FORMATS = {
-    "text": "纯文本格式",
-    "json": "JSON格式",
-    "stream-json": "流式JSON格式"
-}
-
-# 权限模式
-PERMISSION_MODES = {
-    "manual": "手动确认每个操作",
-    "acceptedits": "自动接受文件编辑",
-    "acceptall": "自动接受所有操作"
-}
 
-# 支持的工具
-ALLOWED_TOOLS = [
-    "Read",       # 文件读取
-    "Write",      # 文件写入
-    "Bash",       # 命令执行
-    "Search",     # 文件搜索
-    "Index",      # 索引操作
-    "Chat",       # 对话
-    "Design"      # 设计
-]
+class OutputFormat(Enum):
+    """输出格式枚举"""
+    TEXT = "text"
+    JSON = "json"
+    STREAM_JSON = "stream-json"
+    
+    @classmethod
+    def values(cls):
+        return [e.value for e in cls]
+
+
+class InputFormat(Enum):
+    """输入格式枚举"""
+    TEXT = "text"
+    JSON = "json"
+    STREAM_JSON = "stream-json"
+    
+    @classmethod
+    def values(cls):
+        return [e.value for e in cls]
+
+
+class PermissionMode(Enum):
+    """权限模式枚举"""
+    MANUAL = "manual"
+    ACCEPT_EDITS = "acceptEdits"
+    ACCEPT_ALL = "acceptAll"
+    
+    @classmethod
+    def values(cls):
+        return [e.value for e in cls]
+
+
+class AllowedTool(Enum):
+    """允许的工具枚举"""
+    READ = "Read"
+    WRITE = "Write" 
+    BASH = "Bash"
+    SEARCH = "Search"
+    INDEX = "Index"
+    CHAT = "Chat"
+    DESIGN = "Design"
+    
+    @classmethod
+    def values(cls):
+        return [e.value for e in cls]
+
+
+class EventType(Enum):
+    """事件类型枚举"""
+    START = "start"
+    CONTENT = "content"
+    END = "end"
+    ERROR = "error"
+    LLM_THINKING = "llm_thinking"
+    LLM_OUTPUT = "llm_output"
+    TOOL_CALL = "tool_call"
+    TOOL_RESULT = "tool_result"
+    COMPLETION = "completion"
+    TOKEN_USAGE = "token_usage"
+    WINDOW_CHANGE = "window_change"
+    CONVERSATION_ID = "conversation_id"
+    FILE_MODIFIED = "file_modified"
+    FILE_CREATED = "file_created"
+    FILE_DELETED = "file_deleted"
+    PLAN_MODE_RESPOND = "plan_mode_respond"
+    
+    @classmethod
+    def values(cls):
+        return [e.value for e in cls]
+
+
+class ErrorCode(Enum):
+    """错误码枚举"""
+    SDK_ERROR = "SDK_ERROR"
+    SESSION_NOT_FOUND = "SESSION_NOT_FOUND"
+    INVALID_OPTIONS = "INVALID_OPTIONS"
+    BRIDGE_ERROR = "BRIDGE_ERROR"
+    CLI_ERROR = "CLI_ERROR"
+    VALIDATION_ERROR = "VALIDATION_ERROR"
+    TIMEOUT_ERROR = "TIMEOUT_ERROR"
+    NETWORK_ERROR = "NETWORK_ERROR"
+    
+    @classmethod
+    def values(cls):
+        return [e.value for e in cls]
+
+
+# 支持的输出格式（向后兼容）
+def _get_output_formats():
+    """动态获取输出格式描述，支持国际化"""
+    try:
+        return {
+            OutputFormat.TEXT.value: get_message("format_description_text"),
+            OutputFormat.JSON.value: get_message("format_description_json"),
+            OutputFormat.STREAM_JSON.value: get_message("format_description_stream_json")
+        }
+    except:
+        # 回退到硬编码描述
+        return {
+            OutputFormat.TEXT.value: "Plain text format",
+            OutputFormat.JSON.value: "JSON format",
+            OutputFormat.STREAM_JSON.value: "Streaming JSON format"
+        }
+
+OUTPUT_FORMATS = _get_output_formats()
+
+# 支持的输入格式（向后兼容）
+INPUT_FORMATS = OUTPUT_FORMATS
+
+# 权限模式（向后兼容）
+def _get_permission_modes():
+    """动态获取权限模式描述，支持国际化"""
+    try:
+        return {
+            PermissionMode.MANUAL.value: get_message("permission_description_manual"),
+            PermissionMode.ACCEPT_EDITS.value: get_message("permission_description_accept_edits"),
+            PermissionMode.ACCEPT_ALL.value: get_message("permission_description_accept_all")
+        }
+    except:
+        # 回退到硬编码描述
+        return {
+            PermissionMode.MANUAL.value: "Manually confirm each operation",
+            PermissionMode.ACCEPT_EDITS.value: "Automatically accept file edits",
+            PermissionMode.ACCEPT_ALL.value: "Automatically accept all operations"
+        }
+
+PERMISSION_MODES = _get_permission_modes()
+
+# 支持的工具（向后兼容）
+ALLOWED_TOOLS = [tool.value for tool in AllowedTool]
+
+# 错误码（向后兼容）
+ERROR_CODES = {code.value: code.name for code in ErrorCode}
 
 # CLI相关常量
 CLI_COMMAND_NAME = "auto-coder.run"
@@ -76,25 +183,6 @@ LOG_LEVELS = {
     "CRITICAL": 50
 }
 
-# 错误码
-ERROR_CODES = {
-    "SDK_ERROR": "SDK通用错误",
-    "SESSION_NOT_FOUND": "会话未找到",
-    "INVALID_OPTIONS": "无效选项",
-    "BRIDGE_ERROR": "桥接层错误",
-    "CLI_ERROR": "CLI错误",
-    "VALIDATION_ERROR": "验证错误",
-    "TIMEOUT_ERROR": "超时错误",
-    "NETWORK_ERROR": "网络错误"
-}
-
-# 消息类型
-MESSAGE_ROLES = {
-    "user": "用户消息",
-    "assistant": "助手消息",
-    "system": "系统消息"
-}
-
 # 流式输出标记
 STREAM_START_MARKER = "<<STREAM_START>>"
 STREAM_END_MARKER = "<<STREAM_END>>"