Agent/examples/piaoquan_demand/config.py

"""
项目配置

定义项目的运行配置。
"""

from agent.core.runner import RunConfig

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

RUN_CONFIG = RunConfig(
    # 模型配置
    model="claude-opus-4-6",
    temperature=0.3,
    max_iterations=1000,

    # 任务名称
    name="Needs Agent"
)

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

TRACE_STORE_PATH = ".trace"
DEBUG = True
LOG_LEVEL = "INFO"
LOG_FILE = None  # 设置为文件路径可以同时输出到文件

# 仅启用这些自定义工具（名字必须与 @tool 函数名一致）
ENABLED_TOOLS = ["think_and_plan", "get_category_tree", "get_frequent_itemsets", "get_itemset_detail",
                 "get_post_elements", "search_elements", "get_element_category_chain", "get_category_detail",
                 "search_categories", "get_category_elements", "get_category_co_occurrences",
                 "get_element_co_occurrences",
                 "get_weight_score_topn", "get_weight_score_by_name",
                 "create_demand_item", "create_demand_items",
                 "write_execution_summary"]