how/script/data_processing/analyze_first_step.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
第一步分析脚本

基于过滤后的 how 解构结果，分析哪些点最有可能是创作者的第一步（创作起点）。

输入：intermediate/filtered_results/ 中的过滤结果
输出：第一步分析结果
"""

import asyncio
import json
from pathlib import Path
from typing import Dict, List
import sys

# 添加项目根目录到路径
project_root = Path(__file__).parent.parent.parent
sys.path.insert(0, str(project_root))

from agents import Agent, Runner, ModelSettings, trace
from agents.tracing.create import custom_span
from lib.client import get_model
from lib.my_trace import set_trace_smith as set_trace
from script.data_processing.path_config import PathConfig

# 模型配置
MODEL_NAME = "google/gemini-3-pro-preview"
# MODEL_NAME = 'anthropic/claude-sonnet-4.5'

agent = Agent(
    name="First Step Analyzer",
    model=get_model(MODEL_NAME),
    model_settings=ModelSettings(
        temperature=0.0,
        max_tokens=65536,
    ),
    tools=[],
)


def extract_points_from_filtered_result(how_result: Dict) -> List[Dict]:
    """
    从过滤后的 how 解构结果中提取所有点的信息

    Args:
        how_result: how解构结果

    Returns:
        点信息列表，不包含原始分类标签
    """
    points = []

    for point_type in ["灵感点", "关键点", "目的点"]:
        point_list_key = f"{point_type}列表"
        point_list = how_result.get(point_list_key, [])

        for point in point_list:
            point_name = point.get("名称", "")
            point_desc = point.get("描述", "")

            # 检查是否有匹配到人设特征
            has_match = False
            matched_feature = None
            similarity = None

            # 遍历 how 步骤列表中的特征
            for step in point.get("how步骤列表", []):
                for feature in step.get("特征列表", []):
                    match_results = feature.get("匹配结果", [])
                    if match_results:  # 如果有匹配结果（top1）
                        has_match = True
                        match = match_results[0]
                        matched_feature = match.get("人设特征名称", "")
                        similarity = match.get("匹配结果", {}).get("相似度", 0)
                        break
                if has_match:
                    break

            point_info = {
                "名称": point_name,
                "描述": point_desc,
                "是否匹配到已有人设": has_match
            }

            if has_match:
                point_info["匹配的人设特征"] = matched_feature
                point_info["相似度"] = similarity

            points.append(point_info)

    return points


def build_prompt(points: List[Dict]) -> str:
    """
    构建分析 prompt

    Args:
        points: 点信息列表

    Returns:
        prompt 文本
    """
    # 构建点的描述文本
    points_text = []
    for i, point in enumerate(points, 1):
        text = f"{i}. {point['名称']}\n   {point['描述']}"

        if point['是否匹配到已有人设']:
            text += f"\n   [已匹配] 匹配到人设特征: {point['匹配的人设特征']} (相似度: {point['相似度']:.2f})"
        else:
            text += "\n   [未匹配] 未匹配到已有人设特征"

        points_text.append(text)

    points_section = "\n\n".join(points_text)

    return f'''
以下是一个内容创作的解构结果。这些点已经被分析和分类，但这个分类是分析维度，不代表真实的创作顺序。

请判断：在这些点中，哪些最有可能是创作者的"第一步"（创作起点）？

## 判断标准

**起点特征**：
- 最先触发创作、不依赖其他点的节点
- 可能是外部事件、时事热点、商业需求等
- 起点可能有多个

**参考信息**：
- **已匹配到人设的点**：来源于创作者已有的人设/风格/习惯
- **未匹配的点**：可能来自外部触发、人设推导、或新尝试

## 待分析的点

{points_section}

## 输出要求

以 JSON 格式输出：
{{
  "推理过程": "详细说明判断逻辑...",
  "第一步候选": [
    {{
      "点名称": "...",
      "第一步概率": 0.95,  // 0-1之间的数值
      "推理依据": "...",
      "来源分析": "外部触发/人设延伸/商业驱动/其他"
    }}
  ]
}}

注意：
1. 只输出最有可能是第一步的点（通常1-3个）
2. 按第一步概率降序排列
3. 不要被点的呈现顺序影响判断
'''.strip()


async def analyze_post(post_data: Dict) -> Dict:
    """
    分析单个帖子

    Args:
        post_data: 帖子数据（包含 how解构结果）

    Returns:
        分析结果
    """
    post_id = post_data.get("帖子id", "")
    how_result = post_data.get("how解构结果", {})

    # 提取所有点的信息
    points = extract_points_from_filtered_result(how_result)

    if not points:
        return {
            "帖子id": post_id,
            "模型": MODEL_NAME,
            "输入": {"点列表": []},
            "输出": None,
            "错误": "没有可分析的点"
        }

    # 构建 prompt
    prompt = build_prompt(points)

    # 使用 custom_span 标识单个帖子的分析流程
    with custom_span(
        name=f"分析第一步 - 帖子 {post_id}",
        data={
            "帖子id": post_id,
            "点数量": len(points),
            "模型": MODEL_NAME
        }
    ):
        # 调用 agent
        result = await Runner.run(agent, input=prompt)
        output = result.final_output

    # 解析 JSON
    try:
        if "```json" in output:
            json_start = output.find("```json") + 7
            json_end = output.find("```", json_start)
            json_str = output[json_start:json_end].strip()
        elif "{" in output and "}" in output:
            json_start = output.find("{")
            json_end = output.rfind("}") + 1
            json_str = output[json_start:json_end]
        else:
            json_str = output

        analysis_result = json.loads(json_str)

        return {
            "帖子id": post_id,
            "模型": MODEL_NAME,
            "输入": {
                "点列表": points,
                "prompt": prompt
            },
            "输出": analysis_result
        }
    except Exception as e:
        return {
            "帖子id": post_id,
            "模型": MODEL_NAME,
            "输入": {
                "点列表": points,
                "prompt": prompt
            },
            "输出": None,
            "错误": str(e),
            "原始输出": output
        }


async def main(current_time: str = None, log_url: str = None):
    """主函数

    Args:
        current_time: 当前时间戳（从外部传入）
        log_url: 日志链接（从外部传入）
    """
    # 使用路径配置
    config = PathConfig()

    # 确保输出目录存在
    config.ensure_dirs()

    # 获取路径
    input_dir = config.intermediate_dir / "filtered_results"
    output_dir = config.intermediate_dir / "first_step_analysis"

    # 确保输出目录存在
    output_dir.mkdir(parents=True, exist_ok=True)

    print(f"账号: {config.account_name}")
    print(f"输入目录: {input_dir}")
    print(f"输出目录: {output_dir}")
    print(f"使用模型: {MODEL_NAME}")
    if log_url:
        print(f"Trace URL: {log_url}")
    print()

    # 读取所有过滤后的文件
    input_files = list(input_dir.glob("*_filtered.json"))

    if not input_files:
        print(f"错误: 在 {input_dir} 中没有找到任何 *_filtered.json 文件")
        return

    print(f"找到 {len(input_files)} 个文件待分析\n")

    # 批量分析
    results = []
    for i, input_file in enumerate(input_files, 1):
        print(f"[{i}/{len(input_files)}] 分析文件: {input_file.name}")

        # 读取文件
        with open(input_file, "r", encoding="utf-8") as f:
            post_data = json.load(f)

        # 分析
        result = await analyze_post(post_data)
        results.append(result)

        # 立即保存单个帖子的结果
        post_id = result.get("帖子id", "unknown")
        single_output_file = output_dir / f"{post_id}_first_step.json"

        single_result = {
            "元数据": {
                "current_time": current_time,
                "log_url": log_url,
                "model": MODEL_NAME
            },
            "帖子id": post_id,
            "分析结果": result
        }

        with open(single_output_file, "w", encoding="utf-8") as f:
            json.dump(single_result, f, ensure_ascii=False, indent=2)

        # 显示结果
        output = result.get("输出", {})
        if output:
            first_steps = output.get("第一步候选", [])
            print(f"  第一步候选:")
            for step in first_steps:
                print(f"    - {step.get('点名称', 'N/A')} (概率: {step.get('第一步概率', 0):.2f})")
            print(f"  ✓ 已保存: {single_output_file.name}")
        else:
            print(f"  分析失败: {result.get('错误', 'N/A')}")
        print()

    print(f"✓ 所有分析完成，结果已保存到: {output_dir}")
    if log_url:
        print(f"Trace: {log_url}")

    # 打印汇总
    print("\n" + "=" * 80)
    print("分析汇总")
    print("=" * 80)
    for result in results:
        post_id = result["帖子id"]
        output = result.get("输出", {})
        if output:
            first_steps = output.get("第一步候选", [])
            print(f"\n帖子 {post_id}:")
            for step in first_steps:
                print(f"  - {step.get('点名称', 'N/A')} ({step.get('来源分析', 'N/A')}, 概率: {step.get('第一步概率', 0):.2f})")
        else:
            print(f"\n帖子 {post_id}: 分析失败")


if __name__ == "__main__":
    # 设置 trace
    current_time, log_url = set_trace()

    # 使用 trace 上下文包裹整个执行流程
    with trace("第一步分析"):
        asyncio.run(main(current_time, log_url))