Agent/examples/process/config.py

"""
项目配置
"""

from agent.core.runner import KnowledgeConfig, RunConfig


# ===== Stage 1：调研配置 =====

RESEARCH_RUN_CONFIG = RunConfig(
    model="qwen3.5-plus",
    temperature=0.3,
    max_iterations=1000,
    extra_llm_params={"extra_body": {"enable_thinking": True}},
    agent_type="main",
    name="工具调研与文档生成",
    knowledge=KnowledgeConfig(
        enable_extraction=False,
        enable_completion_extraction=True,
        enable_injection=False,
        owner="sunlit.howard@gmail.com",
        default_tags={"project": "tool_research", "domain": "ai_agent"},
        default_scopes=["org:cybertogether"],
        default_search_types=["tool", "guide"],
        default_search_owner="sunlit.howard@gmail.com"
    )
)

# ===== Stage 2：分析配置 =====

ANALYSIS_RUN_CONFIG = RunConfig(
    model="anthropic/claude-sonnet-4-6",
    temperature=0.3,
    max_iterations=500,
    agent_type="coordinator",
    name="工作流分析 Pipeline",
)

ANALYSIS_MODEL = "anthropic/claude-sonnet-4-6"

# ===== 输出目录 =====

RESEARCH_OUTPUT_DIR = "examples/tool_research_v2/output"            # Stage 1 输出根目录
ANALYSIS_OUTPUT_DIR = "examples/tool_research_v2/output/analysis"   # Stage 2 输出目录


# ===== 基础设施配置 =====

SKILLS_DIR = "./skills"
TRACE_STORE_PATH = ".trace"
DEBUG = True
LOG_LEVEL = "INFO"
LOG_FILE = None

# ===== 浏览器配置 =====
BROWSER_TYPE = "local"
HEADLESS = False

# ===== IM 配置 =====
IM_ENABLED = True
IM_CONTACT_ID = "agent_research"
IM_SERVER_URL = "ws://43.106.118.91:8105"
IM_WINDOW_MODE = True
IM_NOTIFY_INTERVAL = 10.0