Agent/examples/mini_restore/config.py

"""
项目配置
"""

from agent.core.runner import KnowledgeConfig, RunConfig


# ===== Agent 运行配置 =====

RUN_CONFIG = RunConfig(
    model="qwen3.5-plus",
    temperature=0.3,
    max_iterations=200,
    tool_groups=["core", "browser", "content", "knowledge", "toolhub", "feishu", "im"],

    extra_llm_params={"extra_body": {"enable_thinking": True}},

    agent_type="main",

    name="工具调研测试（KM 通信）",

    knowledge=KnowledgeConfig(
        enable_extraction=False,
        enable_completion_extraction=False,
        enable_injection=False,
        owner="sunlit.howard@gmail.com",
        default_tags={"project": "new_search"},
        default_scopes=["org:cybertogether"],
    )
)


# ===== 任务配置 =====

OUTPUT_DIR = "examples/new_search/outputs"


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

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

# ===== Knowledge Manager 配置 =====
KNOWLEDGE_MANAGER_ENABLED = True
KNOWLEDGE_MANAGER_CONTACT_ID = "knowledge_manager"