weflow-cli 1.5.0 → 1.6.0

package/scripts/_batch_link_sources.py

@@ -1,73 +1,73 @@
-"""Batch add local_source links to existing V2 notes."""
-from pathlib import Path
-import re
-
-vault = Path('output/wechat-vault')
-lit_dir = vault / '002_Literature' / 'WeChat'
-biz = Path('output/biz-daily')
-
-updated = 0
-for date_dir in sorted(lit_dir.iterdir()):
-    if not date_dir.is_dir():
-        continue
-    date_str = date_dir.name
-    for note_file in date_dir.glob('*.md'):
-        content = note_file.read_text(encoding='utf-8')
-        if 'local_source:' in content:
-            continue
-
-        title_match = re.search(r'title:\s*"(.+?)"', content)
-        source_match = re.search(r'source:\s*"(.+?)"', content)
-        if not title_match:
-            continue
-
-        title = title_match.group(1)
-        source = source_match.group(1) if source_match else ''
-
-        # Find matching biz-daily article
-        found = False
-        for topic_dir in biz.glob(f'{date_str}/*'):
-            if not topic_dir.is_dir():
-                continue
-            for article in topic_dir.glob('*.md'):
-                if article.name == 'README.md':
-                    continue
-                try:
-                    ac = article.read_text(encoding='utf-8')
-                except Exception:
-                    continue
-                if f'title: "{title}"' in ac or f"title: '{title}'" in ac:
-                    local_path = 'file:///' + str(article.resolve()).replace('\\', '/')
-
-                    # Add local_source in frontmatter
-                    content = content.replace(
-                        'source_type: wechat-article\n',
-                        f'source_type: wechat-article\nlocal_source: "{local_path}"\n'
-                    )
-
-                    # Update callout: rename 链接→网页, add 本地
-                    if '📂 打开源文件' not in content:
-                        # Replace the link line in callout
-                        content = content.replace(
-                            '- **链接**: [',
-                            '- **网页**: [微信原文]('
-                        )
-                        # Find where to insert local line (after 网页 line)
-                        lines = content.split('\n')
-                        for i, line in enumerate(lines):
-                            if line.strip().startswith('- **网页**: ['):
-                                # Insert local file line after this
-                                lines.insert(i + 1, f'> - **本地**: [📂 打开源文件]({local_path})')
-                                break
-                        content = '\n'.join(lines)
-
-                    note_file.write_text(content, encoding='utf-8')
-                    updated += 1
-                    found = True
-                    break
-            if found:
-                break
-
-    print(f'  {date_str}: {updated} total')
-
-print(f'\nBatch done: {updated} notes updated')
+"""Batch add local_source links to existing V2 notes."""
+from pathlib import Path
+import re
+
+vault = Path('output/wechat-vault')
+lit_dir = vault / '002_Literature' / 'WeChat'
+biz = Path('output/biz-daily')
+
+updated = 0
+for date_dir in sorted(lit_dir.iterdir()):
+    if not date_dir.is_dir():
+        continue
+    date_str = date_dir.name
+    for note_file in date_dir.glob('*.md'):
+        content = note_file.read_text(encoding='utf-8')
+        if 'local_source:' in content:
+            continue
+
+        title_match = re.search(r'title:\s*"(.+?)"', content)
+        source_match = re.search(r'source:\s*"(.+?)"', content)
+        if not title_match:
+            continue
+
+        title = title_match.group(1)
+        source = source_match.group(1) if source_match else ''
+
+        # Find matching biz-daily article
+        found = False
+        for topic_dir in biz.glob(f'{date_str}/*'):
+            if not topic_dir.is_dir():
+                continue
+            for article in topic_dir.glob('*.md'):
+                if article.name == 'README.md':
+                    continue
+                try:
+                    ac = article.read_text(encoding='utf-8')
+                except Exception:
+                    continue
+                if f'title: "{title}"' in ac or f"title: '{title}'" in ac:
+                    local_path = 'file:///' + str(article.resolve()).replace('\\', '/')
+
+                    # Add local_source in frontmatter
+                    content = content.replace(
+                        'source_type: wechat-article\n',
+                        f'source_type: wechat-article\nlocal_source: "{local_path}"\n'
+                    )
+
+                    # Update callout: rename 链接→网页, add 本地
+                    if '📂 打开源文件' not in content:
+                        # Replace the link line in callout
+                        content = content.replace(
+                            '- **链接**: [',
+                            '- **网页**: [微信原文]('
+                        )
+                        # Find where to insert local line (after 网页 line)
+                        lines = content.split('\n')
+                        for i, line in enumerate(lines):
+                            if line.strip().startswith('- **网页**: ['):
+                                # Insert local file line after this
+                                lines.insert(i + 1, f'> - **本地**: [📂 打开源文件]({local_path})')
+                                break
+                        content = '\n'.join(lines)
+
+                    note_file.write_text(content, encoding='utf-8')
+                    updated += 1
+                    found = True
+                    break
+            if found:
+                break
+
+    print(f'  {date_str}: {updated} total')
+
+print(f'\nBatch done: {updated} notes updated')

package/scripts/_utils.py

@@ -237,6 +237,7 @@ def create_engine(engine_type: str, api_key: str = '') -> AIEngine:
     """工厂函数：根据类型创建 AI 引擎。
 
     引擎类型: local(自动检测) / deepseek / claude / ollama
+    配置 aiBaseUrl 时, deepseek 引擎走自定义 OpenAI 兼容端点 (中转站)。
     """
     t = engine_type.lower()
     if t == 'local':
@@ -251,6 +252,9 @@ def create_engine(engine_type: str, api_key: str = '') -> AIEngine:
             )
         return engine
     elif t == 'deepseek':
+        base_url, model = _custom_endpoint()
+        if base_url:
+            return AIEngine(api_key, base_url, model)
         return DeepSeekEngine(api_key)
     elif t == 'claude':
         return ClaudeEngine(api_key)
@@ -261,11 +265,29 @@ def create_engine(engine_type: str, api_key: str = '') -> AIEngine:
     )
 
 
+def _custom_endpoint():
+    """读取 config.json 的 aiBaseUrl/aiModel; 未配置返回 (None, None)。"""
+    try:
+        with open(CONFIG_PATH, encoding='utf-8') as f:
+            cfg = json.load(f)
+        base_url = (cfg.get('aiBaseUrl') or '').strip().rstrip('/')
+        model = (cfg.get('aiModel') or '').strip() or 'deepseek-chat'
+        return (base_url or None), (model if base_url else None)
+    except Exception:
+        return None, None
+
+
 def call_deepseek(prompt: str, api_key: str, max_tokens=2000, timeout=60) -> str:
     """向后兼容：调用 DeepSeek API。"""
     return DeepSeekEngine(api_key, timeout=timeout).chat(prompt, max_tokens=max_tokens)
 
 
+def call_ai(prompt: str, engine_type: str = 'local', api_key: str = '', max_tokens=2000) -> str:
+    """通用 AI 调用：支持 local/deepseek/claude/ollama 引擎。"""
+    engine = create_engine(engine_type, api_key)
+    return engine.chat(prompt, max_tokens=max_tokens)
+
+
 # ======================================================================
 # 缓存机制 — 避免重跑 pipeline 时重复调用 API
 # ======================================================================
@@ -411,6 +433,13 @@ def get_db_config(config=None):
     }
 
 
+def get_api_key(config=None) -> str:
+    """读取 AI API key (自动解密 lock: 前缀)。"""
+    if config is None:
+        config = load_config()
+    return decrypt_lock(config.get('deepseekApiKey', ''))
+
+
 # ======================================================================
 # Markdown / Frontmatter
 # ======================================================================