@jungtz/wiki-router 1.0.1 → 1.0.2

package/README.md

@@ -67,6 +67,7 @@ console.log(ctx)
 | `outputDir` | `string` | ✅ | Wiki 輸出目錄 |
 | `modelId` | `string` | | Wiki 生成模型，格式 `provider/model` |
 | `routerModelId` | `string` | | 路由選擇模型，預設同 `modelId` |
+| `timeout` | `number` | | LLM 對話逾時毫秒數，預設 `300000` (5 分鐘) |
 
 
 ### `wiki.build()`
@@ -117,6 +118,29 @@ my-project/
 │   └── ...
 ```
 
+## 互動預覽
+
+內建互動式 TUI，可在終端直接問答測試 Wiki 路由效果：
+
+```bash
+npm run preview
+
+# 自訂路徑
+npm run preview -- --knowledge ./my-knowledge --output ./my-wiki
+```
+
+支援指令：`:list` 查看維基頁面、`:build` 重建維基、`:files` 顯示檔案、`:quit` 離開。
+
+## 發布腳本
+
+```bash
+npm run bump:patch   # 版號 patch +1 並自動 git commit + tag
+npm run bump:minor   # 版號 minor +1
+npm run bump:major   # 版號 major +1
+npm run push:version # 推送 commit 與 tag 至 remote
+npm run release      # bump:patch + push:version
+```
+
 ## 錯誤處理
 
 Wiki Router 採用靜默失敗策略：若模型設定錯誤或 provider 無法連線，只會輸出警告並回傳空內容，不會中斷主流程。

package/dist/index.cjs

@@ -266,6 +266,7 @@ function createWikiRouter(config) {
     outputDir,
     modelId,
     routerModelId,
+    timeout = 300000,
   } = config;
 
   if (!router) throw new Error('[wiki-router] router is required')
@@ -275,14 +276,44 @@ function createWikiRouter(config) {
   const resolvedKnowledgeDir = path.resolve(knowledgeDir);
   const resolvedOutputDir = path.resolve(outputDir);
 
+  let modelsFetched = false;
+
+  /**
+   * 預先獲取本地模型清單（仿照 models.js 的 getModelsList 邏輯）
+   * 確保 switchModel 能識別動態發現的 local 模型
+   */
+  async function ensureModels() {
+    if (modelsFetched) return
+    try {
+      const { models } = await router.getModels();
+      const localProviders = Object.keys(router.config.providers).filter(
+        p => router.config.providers[p].type === 'local'
+      );
+      localProviders.forEach(pName => {
+        const pModels = models.filter(m => m.provider === pName);
+        if (pModels.length > 0) {
+          router.config.providers[pName].models = pModels.map(m => ({
+            model: m.model,
+            description: m.description
+          }));
+        }
+      });
+      modelsFetched = true;
+    } catch (e) {
+      console.warn('[Wiki] Failed to fetch models:', e.message);
+    }
+  }
+
   /**
    * 向 LLM 發送對話並取得完整回應
    * @param {{ role: string, content: string }[]} messages
    * @param {string} [overrideModelId]
+   * @param {number} [overrideTimeout]
    * @returns {Promise<string|null>}
    */
-  async function chat(messages, overrideModelId) {
+  async function chat(messages, overrideModelId, overrideTimeout) {
     const wikiModelId = overrideModelId || modelId || process.env.WIKI_MODEL || 'ollama-local/gemma4:31b';
+    const effectiveTimeout = overrideTimeout || timeout;
 
     try {
       let provider;
@@ -296,10 +327,17 @@ function createWikiRouter(config) {
 
       // 使用 AI Router 進行對話
       if (router && provider) {
+        await ensureModels();
         await router.switchModel(provider, modelName);
         let content = '';
         const stream = router.chat(messages);
+
+        const deadline = Date.now() + effectiveTimeout;
         for await (const chunk of stream) {
+          if (Date.now() > deadline) {
+            console.warn(`[Wiki Chat] Timeout after ${effectiveTimeout}ms`);
+            break
+          }
           if (chunk.type === 'content') content += chunk.content;
         }
         return content

package/dist/index.d.ts

@@ -9,6 +9,7 @@
  * @property {string} outputDir - 輸出 Wiki 檔案的目錄路徑 (required)
  * @property {string} [modelId] - Wiki 生成使用的模型 ID，格式: 'provider/model'
  * @property {string} [routerModelId] - 路由選擇使用的模型 ID，預設同 modelId
+ * @property {number} [timeout=300000] - 每次 LLM 對話的逾時毫秒數
  */
 
 /**

package/dist/index.mjs

@@ -264,6 +264,7 @@ function createWikiRouter(config) {
     outputDir,
     modelId,
     routerModelId,
+    timeout = 300000,
   } = config;
 
   if (!router) throw new Error('[wiki-router] router is required')
@@ -273,14 +274,44 @@ function createWikiRouter(config) {
   const resolvedKnowledgeDir = path.resolve(knowledgeDir);
   const resolvedOutputDir = path.resolve(outputDir);
 
+  let modelsFetched = false;
+
+  /**
+   * 預先獲取本地模型清單（仿照 models.js 的 getModelsList 邏輯）
+   * 確保 switchModel 能識別動態發現的 local 模型
+   */
+  async function ensureModels() {
+    if (modelsFetched) return
+    try {
+      const { models } = await router.getModels();
+      const localProviders = Object.keys(router.config.providers).filter(
+        p => router.config.providers[p].type === 'local'
+      );
+      localProviders.forEach(pName => {
+        const pModels = models.filter(m => m.provider === pName);
+        if (pModels.length > 0) {
+          router.config.providers[pName].models = pModels.map(m => ({
+            model: m.model,
+            description: m.description
+          }));
+        }
+      });
+      modelsFetched = true;
+    } catch (e) {
+      console.warn('[Wiki] Failed to fetch models:', e.message);
+    }
+  }
+
   /**
    * 向 LLM 發送對話並取得完整回應
    * @param {{ role: string, content: string }[]} messages
    * @param {string} [overrideModelId]
+   * @param {number} [overrideTimeout]
    * @returns {Promise<string|null>}
    */
-  async function chat(messages, overrideModelId) {
+  async function chat(messages, overrideModelId, overrideTimeout) {
     const wikiModelId = overrideModelId || modelId || process.env.WIKI_MODEL || 'ollama-local/gemma4:31b';
+    const effectiveTimeout = overrideTimeout || timeout;
 
     try {
       let provider;
@@ -294,10 +325,17 @@ function createWikiRouter(config) {
 
       // 使用 AI Router 進行對話
       if (router && provider) {
+        await ensureModels();
         await router.switchModel(provider, modelName);
         let content = '';
         const stream = router.chat(messages);
+
+        const deadline = Date.now() + effectiveTimeout;
         for await (const chunk of stream) {
+          if (Date.now() > deadline) {
+            console.warn(`[Wiki Chat] Timeout after ${effectiveTimeout}ms`);
+            break
+          }
           if (chunk.type === 'content') content += chunk.content;
         }
         return content

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jungtz/wiki-router",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "LLM Wiki 知識庫路由引擎 - 將結構化知識轉為 Markdown 維基，智慧路由查詢",
   "main": "dist/index.cjs",
   "module": "dist/index.mjs",
@@ -33,6 +33,7 @@
     "build:prod": "cross-env NODE_ENV=production rollup -c",
     "dev": "rollup -c -w",
     "test": "node --test",
+    "preview": "node scripts/preview.js",
     "lint": "eslint src/",
     "lint:fix": "eslint src/ --fix",
     "format": "prettier --write src/",
@@ -55,7 +56,6 @@
     "ai"
   ],
   "license": "MIT",
-  "dependencies": {},
   "peerDependencies": {
     "@jungtz/ai-router": "^1.0.0"
   },
@@ -65,6 +65,7 @@
     }
   },
   "devDependencies": {
+    "@jungtz/ai-router": "^1.0.23",
     "@rollup/plugin-node-resolve": "^15.2.3",
     "cross-env": "^10.1.0",
     "eslint": "^8.57.0",

package/src/types.d.ts

@@ -9,6 +9,7 @@
  * @property {string} outputDir - 輸出 Wiki 檔案的目錄路徑 (required)
  * @property {string} [modelId] - Wiki 生成使用的模型 ID，格式: 'provider/model'
  * @property {string} [routerModelId] - 路由選擇使用的模型 ID，預設同 modelId
+ * @property {number} [timeout=300000] - 每次 LLM 對話的逾時毫秒數
  */
 
 /**