elksmmx 9 часов назад
Родитель
Сommit
8fd3ff177a
2 измененных файлов с 18 добавлено и 16 удалено
  1. 9 9
      examples/feature_extract/config.py
  2. 9 7
      examples/feature_extract/run.py

+ 9 - 9
examples/feature_extract/config.py

@@ -16,7 +16,7 @@ RUN_CONFIG = RunConfig(
     max_iterations=1000,
 
     # 任务名称
-    name="内容制作",
+    name="内容解构",
 
     # 知识管理配置
     knowledge=KnowledgeConfig(
@@ -32,19 +32,19 @@ RUN_CONFIG = RunConfig(
         enable_injection=True,
 
         # 默认字段(保存/搜索时自动注入)
-        owner="sunlit.howard@gmail.com",  # 所有者(空则尝试从 git config user.email 获取,再空则用 agent:{agent_id})
-        default_tags={"project": "production"},  # 默认 tags(会与工具调用参数合并)
+        owner="srt_feature_extract_1",  # 所有者(空则尝试从 git config user.email 获取,再空则用 agent:{agent_id})
+        default_tags={"project": "feature_extract"},  # 默认 tags(会与工具调用参数合并)
         default_scopes=["org:cybertogether"],  # 默认 scopes
-        default_search_types=["strategy", "tool","plan"],  # 默认搜索类型过滤
-        default_search_owner="sunlit.howard@gmail.com"  # 默认搜索 owner 过滤(空则不过滤)
+        default_search_types=["tool", "usecase", "definition"],  # 默认搜索类型过滤
+        default_search_owner="srt_feature_extract_1"  # 默认搜索 owner 过滤(空则不过滤)
     )
 )
 
 
 # ===== 任务配置 =====
 
-INPUT_DIR = "examples/plan/huahua"       # 输入素材目录
-OUTPUT_ID = "output_huahua"                   # 输出目录 ID,输出保存在 examples/plan/outputs/{OUTPUT_ID}/
+INPUT_DIR = "examples/feature_extract/huahua"        # 输入素材目录
+OUTPUT_DIR = "examples/feature_extract/huahua/output"  # 输出目录,直接保存在输入目录内
 
 
 # ===== 基础设施配置 =====
@@ -56,6 +56,6 @@ LOG_LEVEL = "INFO"
 LOG_FILE = None  # 设置为文件路径可以同时输出到文件
 
 # ===== 浏览器配置 =====
-# 可选值: "cloud" (云浏览器) 或 "local" (本地浏览器)
-BROWSER_TYPE = "cloud"
+# 可选值: "cloud" (云浏览器) 或 "local" (本地浏览器) 或 "container" (容器浏览器,支持预配置账户)
+BROWSER_TYPE = "container"
 HEADLESS = True

+ 9 - 7
examples/feature_extract/run.py

@@ -41,7 +41,7 @@ from agent.utils import setup_logging
 from agent.tools.builtin.browser.baseClass import init_browser_session, kill_browser_session
 
 # 导入项目配置
-from config import RUN_CONFIG, SKILLS_DIR, TRACE_STORE_PATH, DEBUG, LOG_LEVEL, LOG_FILE, BROWSER_TYPE, HEADLESS, INPUT_DIR, OUTPUT_ID
+from config import RUN_CONFIG, SKILLS_DIR, TRACE_STORE_PATH, DEBUG, LOG_LEVEL, LOG_FILE, BROWSER_TYPE, HEADLESS, INPUT_DIR, OUTPUT_DIR
 
 
 async def main():
@@ -57,8 +57,8 @@ async def main():
     base_dir = Path(__file__).parent
     project_root = base_dir.parent.parent
     prompt_path = base_dir / "requirement.prompt"
-    output_dir = base_dir / "output"
-    output_dir.mkdir(exist_ok=True)
+    output_dir = Path(OUTPUT_DIR)
+    output_dir.mkdir(parents=True, exist_ok=True)
 
     # 1. 配置日志
     setup_logging(level=LOG_LEVEL, file=LOG_FILE)
@@ -80,8 +80,8 @@ async def main():
     # 4. 构建任务消息
     print("4. 构建任务消息...")
     print(f"   - 输入目录: {INPUT_DIR}")
-    print(f"   - 输出 ID: {OUTPUT_ID}")
-    messages = prompt.build_messages(input_dir=INPUT_DIR, output_id=OUTPUT_ID)
+    print(f"   - 输出目录: {OUTPUT_DIR}")
+    messages = prompt.build_messages(input_dir=INPUT_DIR, output_dir=OUTPUT_DIR)
 
     # 5. 初始化浏览器
     import platform
@@ -90,12 +90,14 @@ async def main():
         actual_browser_type = "cloud"
         print("⚠️ Windows 平台检测到本地浏览器配置,自动切换为云浏览器模式")
 
-    browser_mode_name = "云浏览器" if actual_browser_type == "cloud" else "本地浏览器"
+    browser_mode_names = {"cloud": "云浏览器", "local": "本地浏览器", "container": "容器浏览器"}
+    browser_mode_name = browser_mode_names.get(actual_browser_type, actual_browser_type)
     print(f"5. 正在初始化{browser_mode_name}...")
     await init_browser_session(
         browser_type=actual_browser_type,
         headless=HEADLESS,
-        url="about:blank"
+        url="https://www.google.com/",
+        profile_name=""
     )
     print(f"   ✅ {browser_mode_name}初始化完成\n")