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

autocoder/workflow_agents/utils.py

@@ -0,0 +1,434 @@
+"""
+Workflow Agents 工具函数
+
+提供模板渲染、条件评估、JSONPath 处理等辅助功能。
+"""
+
+import re
+import json
+from typing import Any, Dict, Optional
+from loguru import logger
+
+from autocoder.workflow_agents.types import (
+    WhenConfig,
+    RegexCondition,
+    JsonPathCondition,
+    OutputConfig,
+)
+from autocoder.workflow_agents.exceptions import (
+    WorkflowTemplateError,
+    WorkflowConditionError,
+    WorkflowOutputExtractionError,
+)
+
+# 常量定义
+TEMPLATE_PREFIX = "${"
+TEMPLATE_SUFFIX = "}"
+ATTEMPT_RESULT_VAR = "attempt_result"
+FORMAT_JSON = "json"
+FORMAT_TEXT = "text"
+
+
+def render_template(template: Any, context: Dict[str, Any]) -> Any:
+    """
+    渲染模板字符串
+
+    支持的模板语法：
+    - ${vars.key} - 访问全局变量
+    - ${steps.stepId.outputs.key} - 访问步骤输出
+    - ${attempt_result} - 访问上一次的 attempt 结果
+    - \\$ - 转义，输出字面的 $
+
+    支持在字符串中嵌入多个变量，例如：
+    "基于文件 ${steps.step1.outputs.files} 执行 ${vars.action}"
+
+    Args:
+        template: 模板（字符串或其他类型）
+        context: 上下文数据，包含 vars、steps 等
+
+    Returns:
+        渲染后的值
+    """
+    if not isinstance(template, str):
+        return template
+
+    # 如果没有任何模板标记，直接返回
+    if "${" not in template and "\\$" not in template:
+        return template
+
+    # 正则表达式：匹配非转义的 ${...}
+    # (?<!\\) - 负向后查找，确保前面不是反斜杠
+    # \$\{([^}]+)\} - 匹配 ${...}
+    pattern = r"(?<!\\)\$\{([^}]+)\}"
+
+    def replace_var(match):
+        expr = match.group(1).strip()
+        try:
+            value = _resolve_expression(expr, context)
+            # 如果值是 None，返回空字符串
+            return str(value) if value is not None else ""
+        except (KeyError, IndexError, TypeError) as e:
+            # 提供更友好的错误信息
+            context_keys = _get_context_keys(context)
+            logger.warning(f"模板表达式无法解析: ${{{expr}}}, 可用键: {context_keys}")
+            # 抛出友好的异常
+            raise WorkflowTemplateError(
+                template=template, expression=expr, context_keys=context_keys
+            ) from e
+
+    # 替换所有变量
+    result = re.sub(pattern, replace_var, template)
+
+    # 处理转义：将 \$ 替换为 $
+    result = result.replace(r"\$", "$")
+
+    return result
+
+
+def _resolve_expression(expr: str, context: Dict[str, Any]) -> Any:
+    """
+    解析模板表达式
+
+    Args:
+        expr: 表达式内容（去除 ${ } 后的部分）
+        context: 上下文数据
+
+    Returns:
+        解析后的值
+
+    Raises:
+        KeyError, IndexError, TypeError: 解析失败时抛出
+    """
+    parts = expr.split(".")
+
+    # ${vars.key}
+    if parts[0] == "vars" and len(parts) >= 2:
+        vars_dict = context.get("vars", {})
+        if parts[1] not in vars_dict:
+            raise KeyError(f"Variable not found: vars.{parts[1]}")
+        return vars_dict[parts[1]]
+
+    # ${steps.stepId.outputs.key}
+    if parts[0] == "steps" and len(parts) >= 4 and parts[2] == "outputs":
+        step_id = parts[1]
+        key = parts[3]
+        steps_dict = context.get("steps", {})
+        if step_id not in steps_dict:
+            raise KeyError(f"Step not found: {step_id}")
+        outputs_dict = steps_dict[step_id].get("outputs", {})
+        if key not in outputs_dict:
+            raise KeyError(f"Output not found: {step_id}.outputs.{key}")
+        return outputs_dict[key]
+
+    # ${attempt_result}
+    if expr == ATTEMPT_RESULT_VAR:
+        # attempt_result 可能为 None（初始状态），这是合法的
+        return context.get("_last_attempt_result")
+
+    # 不支持的表达式
+    raise KeyError(f"不支持的表达式: {expr}")
+
+
+def evaluate_condition(
+    when_config: Optional[WhenConfig],
+    attempt_result: Optional[str],
+    context: Dict[str, Any],
+) -> bool:
+    """
+    评估条件是否满足
+
+    Args:
+        when_config: 条件配置
+        attempt_result: AttemptCompletion 结果字符串
+        context: 上下文数据
+
+    Returns:
+        条件是否满足
+    """
+    if when_config is None:
+        return True
+
+    # 正则条件
+    if when_config.regex is not None:
+        return evaluate_regex_condition(when_config.regex, attempt_result, context)
+
+    # JSONPath 条件
+    if when_config.jsonpath is not None:
+        return evaluate_jsonpath_condition(
+            when_config.jsonpath, attempt_result, context
+        )
+
+    return True
+
+
+def _get_input_string(
+    input_template: str, context: Dict[str, Any], fallback: Optional[str]
+) -> str:
+    """
+    获取输入字符串（公共逻辑）
+
+    Args:
+        input_template: 输入模板
+        context: 上下文数据
+        fallback: 回退值（通常是 attempt_result）
+
+    Returns:
+        解析后的输入字符串
+    """
+    if input_template:
+        return str(render_template(input_template, context))
+    return fallback or ""
+
+
+def evaluate_regex_condition(
+    regex_config: RegexCondition, attempt_result: Optional[str], context: Dict[str, Any]
+) -> bool:
+    """
+    评估正则条件
+
+    Args:
+        regex_config: 正则条件配置
+        attempt_result: AttemptCompletion 结果字符串
+        context: 上下文数据
+
+    Returns:
+        条件是否满足
+    """
+    input_str = _get_input_string(regex_config.input, context, attempt_result)
+    pattern = regex_config.pattern
+
+    flags = 0
+    if regex_config.flags and regex_config.flags.lower() == "i":
+        flags |= re.IGNORECASE
+
+    match = re.search(pattern, input_str, flags)
+    return match is not None
+
+
+def evaluate_jsonpath_condition(
+    jsonpath_config: JsonPathCondition,
+    attempt_result: Optional[str],
+    context: Dict[str, Any],
+) -> bool:
+    """
+    评估 JSONPath 条件
+
+    Args:
+        jsonpath_config: JSONPath 条件配置
+        attempt_result: AttemptCompletion 结果字符串
+        context: 上下文数据
+
+    Returns:
+        条件是否满足
+    """
+    input_str = _get_input_string(jsonpath_config.input, context, attempt_result)
+
+    try:
+        data = json.loads(input_str)
+    except Exception:
+        return False
+
+    path = jsonpath_config.path
+    value = extract_jsonpath_value(data, path)
+
+    # exists 检查
+    if jsonpath_config.exists is not None:
+        return (value is not None) == jsonpath_config.exists
+
+    # equals 检查
+    if jsonpath_config.equals is not None:
+        return value == jsonpath_config.equals
+
+    # contains 检查
+    if jsonpath_config.contains is not None and isinstance(value, str):
+        return jsonpath_config.contains in value
+
+    # 默认：检查值是否存在且非空
+    return bool(value)
+
+
+def extract_jsonpath_value(data: Any, path: str) -> Any:
+    """
+    从 JSON 数据中提取 JSONPath 路径的值
+
+    简化实现，仅支持顶层访问：$.key
+
+    Args:
+        data: JSON 数据
+        path: JSONPath 路径
+
+    Returns:
+        提取的值，如果路径不存在则返回 None
+    """
+    if not isinstance(data, dict):
+        return None
+
+    if path.startswith("$."):
+        key = path[2:]
+        return data.get(key)
+
+    return None
+
+
+def extract_outputs(
+    outputs_map: Dict[str, Any],
+    attempt_result: str,
+    attempt_format: str,
+    default_jsonpaths: Dict[str, str],
+    context: Dict[str, Any],
+) -> Dict[str, Any]:
+    """
+    从 AttemptCompletion 结果中提取输出
+
+    Args:
+        outputs_map: 输出映射配置
+        attempt_result: AttemptCompletion 结果字符串
+        attempt_format: 格式（json | text）
+        default_jsonpaths: 默认 JSONPath 映射
+        context: 上下文数据
+
+    Returns:
+        提取的输出字典
+    """
+    if not outputs_map:
+        return {"attempt_raw": attempt_result}
+
+    # 如果格式是 JSON，尝试解析
+    parsed_json = _try_parse_json(attempt_result, attempt_format)
+
+    # 提取每个输出
+    extracted: Dict[str, Any] = {}
+    for key, spec in outputs_map.items():
+        value = _extract_single_output(
+            key, spec, attempt_result, parsed_json, default_jsonpaths
+        )
+        extracted[key] = value
+
+    # 确保总是有 attempt_raw
+    extracted.setdefault("attempt_raw", attempt_result)
+    return extracted
+
+
+def _try_parse_json(
+    attempt_result: str, attempt_format: str
+) -> Optional[Dict[str, Any]]:
+    """
+    尝试将 attempt_result 解析为 JSON
+
+    Args:
+        attempt_result: AttemptCompletion 结果字符串
+        attempt_format: 格式（json | text）
+
+    Returns:
+        解析后的 JSON 字典，如果失败或格式不是 json 则返回 None
+    """
+    if attempt_format != FORMAT_JSON:
+        return None
+
+    try:
+        return json.loads(attempt_result)
+    except Exception:
+        return None
+
+
+def _extract_single_output(
+    key: str,
+    spec: Any,
+    attempt_result: str,
+    parsed_json: Optional[Dict[str, Any]],
+    default_jsonpaths: Dict[str, str],
+) -> Any:
+    """
+    提取单个输出值
+
+    Args:
+        key: 输出键名
+        spec: 输出配置（可能是字符串或 OutputConfig）
+        attempt_result: 原始结果字符串
+        parsed_json: 解析后的 JSON（如果有）
+        default_jsonpaths: 默认 JSONPath 映射
+
+    Returns:
+        提取的值
+    """
+    # 直接透传 attempt_result（兼容旧代码）
+    template_value = f"{TEMPLATE_PREFIX}{ATTEMPT_RESULT_VAR}{TEMPLATE_SUFFIX}"
+    if isinstance(spec, str) and spec == template_value:
+        return attempt_result
+
+    # 不是 OutputConfig 对象，返回 None
+    if not isinstance(spec, OutputConfig):
+        return None
+
+    # 模板字符串提取（新增支持）
+    if spec.template is not None:
+        # 如果是 ${attempt_result}，直接返回
+        if spec.template == template_value:
+            return attempt_result
+        # 否则作为普通值返回（可能是其他模板）
+        return spec.template
+
+    # JSONPath 提取
+    if spec.jsonpath is not None and parsed_json is not None:
+        path = spec.jsonpath if spec.jsonpath else default_jsonpaths.get(key, "")
+        return extract_jsonpath_value(parsed_json, path)
+
+    # 正则提取
+    if spec.regex is not None:
+        return _extract_by_regex(spec.regex, spec.regex_group, attempt_result)
+
+    return None
+
+
+def _extract_by_regex(pattern: str, group: Optional[int], text: str) -> Optional[str]:
+    """
+    使用正则表达式提取文本
+
+    Args:
+        pattern: 正则表达式模式
+        group: 捕获组索引（None 表示整个匹配）
+        text: 待提取的文本
+
+    Returns:
+        提取的文本，如果未匹配则返回 None
+    """
+    match = re.search(pattern, text)
+    if not match:
+        return None
+
+    if group is None:
+        return match.group(0)
+    return match.group(group)
+
+
+def _get_context_keys(context: Dict[str, Any]) -> list:
+    """
+    获取上下文中所有可用的键（用于错误提示）
+
+    Args:
+        context: 上下文数据
+
+    Returns:
+        可用键的列表
+    """
+    keys = []
+
+    # vars.*
+    if "vars" in context and isinstance(context["vars"], dict):
+        for var_key in context["vars"].keys():
+            keys.append(f"vars.{var_key}")
+
+    # steps.*.outputs.*
+    if "steps" in context and isinstance(context["steps"], dict):
+        for step_id, step_data in context["steps"].items():
+            if isinstance(step_data, dict) and "outputs" in step_data:
+                outputs = step_data["outputs"]
+                if isinstance(outputs, dict):
+                    for output_key in outputs.keys():
+                        keys.append(f"steps.{step_id}.outputs.{output_key}")
+
+    # attempt_result
+    if "_last_attempt_result" in context:
+        keys.append("attempt_result")
+
+    return keys

autocoder/workflow_agents/workflow_manager.py

@@ -0,0 +1,211 @@
+"""
+Workflow Manager - 管理 workflow 文件的查找和加载
+
+提供优先级搜索路径支持，类似 AgentManager。
+"""
+
+from pathlib import Path
+from typing import List, Optional, Dict, Any
+from loguru import logger
+
+from autocoder.common.priority_directory_finder import (
+    PriorityDirectoryFinder,
+    FinderConfig,
+    SearchStrategy,
+    ValidationMode,
+    create_file_filter,
+)
+
+# 常量定义
+WORKFLOW_DIR_NAME = ".autocoderworkflow"
+WORKFLOW_EXTENSIONS = [".yaml", ".yml"]
+
+
+class WorkflowManager:
+    """
+    Workflow 文件管理器
+
+    支持从多个目录按优先级查找 workflow YAML 文件：
+    1. .autocoderworkflow (项目级，最高优先级)
+    2. .auto-coder/.autocoderworkflow (项目级)
+    3. ~/.auto-coder/.autocoderworkflow (全局级)
+    """
+
+    def __init__(self, project_root: str):
+        """
+        初始化 WorkflowManager
+
+        Args:
+            project_root: 项目根目录
+        """
+        self.project_root = Path(project_root)
+        self.workflow_directories: List[str] = []
+        self._discover_directories()
+
+    def _get_workflow_search_paths(self) -> List[Path]:
+        """
+        获取 workflow 搜索路径列表（按优先级排序）
+
+        Returns:
+            搜索路径列表
+        """
+        return [
+            self.project_root / WORKFLOW_DIR_NAME,  # 项目级（最高）
+            self.project_root / ".auto-coder" / WORKFLOW_DIR_NAME,  # 项目级
+            Path.home() / ".auto-coder" / WORKFLOW_DIR_NAME,  # 全局级
+        ]
+
+    def _discover_directories(self) -> None:
+        """使用优先级目录查找器发现 workflow 目录"""
+        try:
+            # 创建文件过滤器
+            yaml_filter = create_file_filter(
+                extensions=WORKFLOW_EXTENSIONS, recursive=False
+            )
+
+            # 创建查找器配置，使用 MERGE_ALL 策略
+            config = FinderConfig(strategy=SearchStrategy.MERGE_ALL)
+
+            # 按优先级添加目录
+            config.add_directory(
+                path=str(self.project_root / WORKFLOW_DIR_NAME),
+                priority=1,
+                validation_mode=ValidationMode.HAS_SPECIFIC_FILES,
+                file_filter=yaml_filter,
+                description="项目级 workflow 目录",
+            )
+            config.add_directory(
+                path=str(self.project_root / ".auto-coder" / WORKFLOW_DIR_NAME),
+                priority=2,
+                validation_mode=ValidationMode.HAS_SPECIFIC_FILES,
+                file_filter=yaml_filter,
+                description="项目 .auto-coder workflow 目录",
+            )
+            config.add_directory(
+                path=f"~/.auto-coder/{WORKFLOW_DIR_NAME}",
+                priority=3,
+                validation_mode=ValidationMode.HAS_SPECIFIC_FILES,
+                file_filter=yaml_filter,
+                description="全局 workflow 目录",
+            )
+
+            # 执行查找
+            finder = PriorityDirectoryFinder(config)
+            result = finder.find_directories()
+
+            if result.success and result.selected_directories:
+                logger.debug(
+                    f"使用优先级查找器找到 {len(result.selected_directories)} 个 workflow 目录"
+                )
+                self.workflow_directories = result.selected_directories
+            else:
+                logger.debug("优先级查找器未找到包含 workflow 文件的目录")
+                self.workflow_directories = []
+
+        except Exception as e:
+            logger.warning(f"使用优先级查找器发现 workflow 目录时出错: {e}")
+            # 回退到传统方法
+            self._discover_directories_fallback()
+
+    def _discover_directories_fallback(self) -> None:
+        """回退到传统的目录发现方法"""
+        logger.debug("回退到传统的 workflow 目录查找方法")
+        search_paths = self._get_workflow_search_paths()
+
+        self.workflow_directories = []
+        for path in search_paths:
+            if path.exists() and path.is_dir():
+                # 检查是否有 .yaml 或 .yml 文件
+                yaml_files = list(path.glob("*.yaml")) + list(path.glob("*.yml"))
+                if yaml_files:
+                    self.workflow_directories.append(str(path))
+
+    def find_workflow(self, workflow_name: str) -> Optional[str]:
+        """
+        按优先级查找 workflow 文件
+
+        支持的文件名格式：
+        - workflow_name.yaml
+        - workflow_name.yml
+
+        Args:
+            workflow_name: workflow 名称（不含扩展名）
+
+        Returns:
+            找到的 workflow 文件绝对路径，如果未找到则返回 None
+        """
+        # 按优先级顺序查找
+        for directory in self.workflow_directories:
+            found_path = self._find_workflow_in_directory(
+                Path(directory), workflow_name
+            )
+            if found_path:
+                logger.info(f"找到 workflow: {found_path}")
+                return found_path
+
+        logger.warning(f"未找到 workflow: {workflow_name}")
+        return None
+
+    def _find_workflow_in_directory(
+        self, dir_path: Path, workflow_name: str
+    ) -> Optional[str]:
+        """
+        在指定目录中查找 workflow 文件
+
+        Args:
+            dir_path: 目录路径
+            workflow_name: workflow 名称
+
+        Returns:
+            找到的文件路径，否则返回 None
+        """
+        for ext in WORKFLOW_EXTENSIONS:
+            workflow_path = dir_path / f"{workflow_name}{ext}"
+            if workflow_path.exists():
+                return str(workflow_path)
+        return None
+
+    def list_workflows(self) -> Dict[str, str]:
+        """
+        列出所有可用的 workflow
+
+        Returns:
+            字典，key 为 workflow 名称（不含扩展名），value 为文件路径
+        """
+        workflows: Dict[str, str] = {}
+
+        # 从低优先级到高优先级遍历，高优先级会覆盖低优先级
+        for directory in reversed(self.workflow_directories):
+            dir_path = Path(directory)
+
+            # 查找所有 .yaml 和 .yml 文件
+            for yaml_file in list(dir_path.glob("*.yaml")) + list(
+                dir_path.glob("*.yml")
+            ):
+                workflow_name = yaml_file.stem
+                workflows[workflow_name] = str(yaml_file)
+
+        return workflows
+
+    def get_workflow_directories(self) -> List[str]:
+        """
+        获取所有 workflow 目录路径
+
+        Returns:
+            workflow 目录路径列表，按优先级排序
+        """
+        return self.workflow_directories.copy()
+
+    def to_dict(self) -> Dict[str, Any]:
+        """
+        转换为字典表示
+
+        Returns:
+            包含搜索路径和可用 workflow 的字典
+        """
+        return {
+            "project_root": str(self.project_root),
+            "search_paths": [str(p) for p in self._get_workflow_search_paths()],
+            "workflow_directories": self.workflow_directories,
+            "workflows": self.list_workflows(),
+        }