weflow-cli 1.5.0 → 1.6.1

package/scripts/pipeline.py

@@ -25,7 +25,7 @@ def run_step(name: str, args: list[str]) -> bool:
     print(f'\n{"="*50}')
     print(f'  Step: {name}')
     print(f'{"="*50}')
-    result = subprocess.run([PYTHON] + args, cwd=os.path.dirname(SCRIPTS_DIR))
+    result = subprocess.run([PYTHON, '-u'] + args, cwd=os.path.dirname(SCRIPTS_DIR))
     if result.returncode != 0:
         print(f'\n[FAIL] {name} 失败 (exit={result.returncode})')
         return False
@@ -36,9 +36,9 @@ def main():
     sys.stdout.reconfigure(encoding='utf-8', errors='replace')
     import argparse
     parser = argparse.ArgumentParser(description='端到端公众号日报流水线')
-    parser.add_argument('--api-key', help='云端 API key（--engine 为 deepseek/claude 时需要）')
-    parser.add_argument('--engine', default='local',
-                        help='AI 引擎: local(自动检测·默认) / deepseek / claude / ollama')
+    parser.add_argument('--api-key', help='DeepSeek API key（也可通过环境变量 DEEPSEEK_API_KEY 或 ~/.weflow-cli/config.json 提供）')
+    parser.add_argument('--engine', default='deepseek',
+                        help='AI 引擎: deepseek(默认) / claude / ollama / local')
     parser.add_argument('--date', help='日期 YYYY-MM-DD')
     parser.add_argument('--interest', default='AI', help='兴趣主题（默认 AI）')
     parser.add_argument('--wiki-limit', type=int, default=20, help='概念编译数量（默认 20）')
@@ -47,43 +47,54 @@ def main():
     parser.add_argument('--skip-vault', action='store_true', help='跳过 Vault 同步')
     parser.add_argument('--skip-html', action='store_true', help='跳过 HTML 生成')
     parser.add_argument('--skip-ai-report', action='store_true', help='跳过 AI 深度阅读报告')
+    parser.add_argument('--no-ai', action='store_true', help='关闭所有 AI 调用，但保留抓取和本地输出')
     parser.add_argument('--ai-report-range', type=int, default=1, help='AI 报告覆盖最近 N 天（默认 1=仅当天）')
+    parser.add_argument('--source', action='append', default=[], metavar='NAME',
+                        help='仅处理指定公众号，可重复或用逗号分隔')
     args = parser.parse_args()
 
     sys.path.insert(0, SCRIPTS_DIR)
-    from _utils import load_config
+    from _utils import load_config, get_api_key
     config = load_config()
+    if str(config.get('dailyAiEnabled', 'true')).strip().lower() == 'false':
+        args.no_ai = True
 
     # api_key：本地/ollama 引擎不需要；云端需要
     api_key = args.api_key or ''
-    if args.engine in ('deepseek', 'claude') and not api_key:
-        api_key = os.environ.get('DEEPSEEK_API_KEY', '') or config.get('deepseekApiKey', '')
+    if args.engine in ('deepseek', 'claude') and not api_key and not args.no_ai:
+        api_key = os.environ.get('DEEPSEEK_API_KEY', '') or get_api_key(config)
         if not api_key:
             print(f'[ERROR] --engine {args.engine} 需要 API key。请通过 --api-key、'
                   f'环境变量 DEEPSEEK_API_KEY 或 ~/.weflow-cli/config.json 提供')
             sys.exit(1)
 
+    if api_key:
+        os.environ['DEEPSEEK_API_KEY'] = api_key
+
     started = time.time()
 
     # Step 1: biz_daily
-    step1_args = [os.path.join(SCRIPTS_DIR, 'biz_daily.py')]
-    if api_key:
-        step1_args += ['--api-key', api_key]
+    step1_args = [os.path.join(SCRIPTS_DIR, 'biz_daily.py'), '--engine', args.engine]
     if args.date:
         step1_args += ['--date', args.date]
+    for source in args.source:
+        step1_args += ['--source', source]
+    if args.no_ai:
+        step1_args += ['--no-ai']
     if not run_step('biz_daily — 抓取+摘要', step1_args):
         sys.exit(1)
 
     # Step 2: classify_daily（可选）
-    if not args.skip_classify:
+    if not args.skip_classify and not args.no_ai:
         step2_args = [
             os.path.join(SCRIPTS_DIR, 'classify_daily.py'),
-            '--api-key', api_key if api_key else 'local',
-            '--interest', args.interest,
+            '--engine', args.engine,
         ]
+        step2_args += ['--interest', args.interest]
         if args.date:
             step2_args.insert(1, args.date)
-        run_step('classify_daily — 后处理', step2_args)
+        if not run_step('classify_daily — 后处理', step2_args):
+            print('[WARN] classify_daily 失败，继续后续步骤')
 
     # Step 3: Vault sync（可选）
     if not args.skip_vault:
@@ -99,10 +110,9 @@ def main():
             print(f'\n  Vault 同步: {file_count} 个文件 → {vault_dir}')
 
     # Step 4: wiki compile（可选）
-    if not args.skip_wiki:
+    if not args.skip_wiki and not args.no_ai:
         step3_args = [
             os.path.join(SCRIPTS_DIR, 'compile_wiki.py'),
-            '--api-key', api_key if api_key else 'local',
             '--limit', str(args.wiki_limit),
         ]
         run_step('wiki compile — 概念编译', step3_args)
@@ -123,14 +133,12 @@ def main():
              [os.path.join(SCRIPTS_DIR, 'create_reading_notes.py'), '--date', date_str])
 
     # Step 8: AI 深度阅读报告
-    if not args.skip_ai_report:
+    if not args.skip_ai_report and not args.no_ai:
         step8_args = [
             os.path.join(SCRIPTS_DIR, 'generate_ai_report.py'),
             '--engine', args.engine,
             '--range', str(args.ai_report_range),
         ]
-        if api_key:
-            step8_args += ['--api-key', api_key]
         if args.date and args.ai_report_range == 1:
             step8_args += ['--date', args.date]
         run_step('generate_ai_report — AI 阅读日报', step8_args)