From 133f5add4675ecbf17be71207a4b3f1a84f6d3c9 Mon Sep 17 00:00:00 2001 From: richblack Date: Tue, 4 Aug 2026 16:29:56 +0800 Subject: [PATCH] =?UTF-8?q?t182=EF=BC=9A=E8=80=81=20config=20=E6=8A=B9?= =?UTF-8?q?=E9=99=A4=E6=88=90=20Workers=20AI=EF=BC=8B=E9=9B=B2=E7=AB=AF?= =?UTF-8?q?=E6=8E=A2=E6=B8=AC=E2=80=94=E2=80=94leo=20=E5=AF=A6=E6=B8=AC?= =?UTF-8?q?=E3=80=8C=E7=84=A1=E5=8D=A1=E3=80=8D=E7=9A=84=E7=9C=9F=E5=9B=A0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit leo 08-04 實測回報（更新 v0.15.5 後仍顯示 Gemini／丟 PDF 無卡）， 查出來不是 Workers AI 不通，是**根本沒走到那條路**。三顆真 bug： ① 抹除只做一半（leo：「就要抹除改成用 Workers AI，如果保持 Gemini 它不會改掉，那就是失敗的」） t181 只在 RunDirectOnce 改頂層記憶體值，但 - makeAccountSubConfig 會把**帳號層**舊值蓋回來（t126「帳號層優先」） - 完全**沒寫回檔案** ⇒ 托盤是另一個行程，讀檔還是念 Gemini leo 的 config 正是頂層＋兩個帳號各留 "gemma"、無 explicit ⇒ 畫面顯示 Gemini、萃取也真的跑 Gemini。 修：遷移移到 LoadDirectConfig（每次啟動必經），逐層抹除 + 寫回檔案。 金鑰保留（Gemini 是選配不是廢除）。 ② 驗證器擋掉自己的新預設 `extractor 只能是 claude / gemma` ⇒ config 一旦寫成 workers-ai， daemon 直接載入失敗起不來。v0.15.5 已帶著這顆出貨。 ③ 托盤標籤照 config 舊字串念（t178 的通則版） t178 只把 claude 這**一個**殘留值導向 Gemini，殘留 gemma 一樣脫鉤。 改成與 direct.go 同一條判準：沒 explicit 就一律念「雲端 AI」。 ＋ t182 雲端探測（leo 指定設計：「會去掃一次看雲端是否裝好，沒裝好就顯示 workers AI 還沒通，一旦通了就顯示可用」）：ProbeWorkersAI 逐帳號探 /portal/daemon/extract（送空 text，不燒 LLM 額度），結果寫進 status.json 由托盤「狀態：」講白話。**不做靜默退回 Gemini**——那會讓用戶永遠 不知道自己雲端沒更新。 實測證據（非 mock，打真實例）： - youlin 實例萃真卡：3.67s，產出完整知識卡（一句話定義/要點/關鍵實體/關聯） - 連測三次：3.51s / 3.34s / 3.59s（Gemini 實測 16.87s ⇒ 快約 5 倍） - 兩個實例都已有 /portal/daemon/extract（皆回 400＝route 存在） - 拿 leo 真 config 複本跑遷移：三層全抹成 workers-ai、金鑰留著、重讀仍是 workers-ai 測試：新增 direct_t182_test.go（抹除全層/寫回磁碟/主動選過不動/合法值/冪等）； 既有三則因預設變更而失效的斷言已更新（t108 二進位不出機的契約未放寬，仍綠）。 全套綠。 Co-Authored-By: Claude Opus 5 (1M context) --- cmd/arcrun-tray/main.go | 15 +++- direct.go | 78 ++++++++++++++++++++- direct_extract_test.go | 22 ++++-- direct_multi_test.go | 4 ++ direct_t182_test.go | 147 ++++++++++++++++++++++++++++++++++++++++ probe_workersai.go | 87 ++++++++++++++++++++++++ sync_status.go | 5 ++ 7 files changed, 348 insertions(+), 10 deletions(-) create mode 100644 direct_t182_test.go create mode 100644 probe_workersai.go diff --git a/cmd/arcrun-tray/main.go b/cmd/arcrun-tray/main.go index 90d5e98..c3673c7 100644 --- a/cmd/arcrun-tray/main.go +++ b/cmd/arcrun-tray/main.go @@ -724,7 +724,18 @@ func removeAccountWatchFolder(cfg *directConfig, accIdx int, p string) { // accountEngineLabel 依帳號或全局預設 extractor 產生「· 引擎名」後綴（t126 托盤可見性）。 // 帳號層有值時優先；兩者皆空回空字串。 -func accountEngineLabel(acc accountCfg, defaultExtractor string) string { +// +// 🔴 t182：`explicit` ＝使用者有沒有在「AI 設定…」**主動選過**引擎。 +// 沒主動選過 ⇒ 一律念「雲端 AI」（回空字串），不管 config 裡殘留什麼舊值。 +// 這是 t178 那個病的**通則版**：t178 只把 `claude` 這一個殘留值導向 Gemini， +// 但殘留 `gemma` 一樣會脫鉤——leo 08-04 實撞：更新到 v0.15.5 後托盤**兩個帳號 +// 都還是顯示 Gemini**，因為他 config 的帳號層留著 extractor="gemma"、explicit 沒設。 +// 判準必須與 `direct.go` 的預設邏輯**同一條**（沒 explicit 就是 workers-ai）， +// 否則又回到「畫面說 A、實際跑 B」。 +func accountEngineLabel(acc accountCfg, defaultExtractor string, explicit bool) string { + if !explicit { + return "" + } extractor := strings.TrimSpace(acc.Extractor) if extractor == "" { extractor = strings.TrimSpace(defaultExtractor) @@ -1067,7 +1078,7 @@ func main() { accHost := shortCypherHost(acc.CypherURL) // 帳號標題（不可點）；t126 附引擎名（如「geek6688 · Gemini」） - headerItem := fyne.NewMenuItem("◉ "+accountDisplayName(acc)+accountEngineLabel(acc, cfg.Extractor), nil) + headerItem := fyne.NewMenuItem("◉ "+accountDisplayName(acc)+accountEngineLabel(acc, cfg.Extractor, cfg.ExtractorExplicit), nil) headerItem.Disabled = true items = append(items, headerItem) diff --git a/direct.go b/direct.go index b1e832b..362ed0d 100644 --- a/direct.go +++ b/direct.go @@ -185,6 +185,34 @@ func LoadDirectConfig(path string) (*DirectConfig, error) { }} } + // 🔴 t182 遷移（leo 2026-08-04 拍板，**這條要在 t126 複製之前跑**）： + // + // 「如果是我的 config 保持舊的，那新版裝上就要檢查，因為已經是 default worker AI， + // **就要抹除改成用 Workers AI**，如果保持 Gemini 它不會改掉，**那就是失敗的**」 + // + // 舊版沒有 `extractor_explicit` 這個欄位 ⇒ 老 config 一律是「沒有主動選過」， + // 但裡頭留著 extractor="gemma"（頂層＋每個帳號各一份，t126 複製過去的）。 + // 只在記憶體裡改預設不夠——**沒寫回檔案**，托盤下次讀 config 還是念 Gemini， + // 萃取也照舊走 Gemini（leo 08-04 實測：更新到 v0.15.5 後兩個帳號仍顯示 Gemini、 + // 丟 PDF 進去產不出卡，因為根本沒走到 Workers AI 這條路）。 + // + // ⇒ 沒 explicit 就把**每一層**的舊值抹掉改成 workers-ai，並回寫檔案（見下方 saveDirectConfig）。 + // 金鑰不動：之後他想選 Gemini，貼過的金鑰還在（leo：Gemini 變選配，不是廢除）。 + // 冪等：抹完即與新版一致，重跑不再改動。 + migrated := false + if !c.ExtractorExplicit { + if strings.TrimSpace(c.Extractor) != "" && c.Extractor != "workers-ai" { + c.Extractor = "workers-ai" + migrated = true + } + for i := range c.Accounts { + if strings.TrimSpace(c.Accounts[i].Extractor) != "" && c.Accounts[i].Extractor != "workers-ai" { + c.Accounts[i].Extractor = "workers-ai" + migrated = true + } + } + } + // t126 遷移：把頂層金鑰複製到每個沒有金鑰的帳號（複製非搬移，頂層保留當預設；冪等）。 // 帳號已有自己的值（非空）→ 不覆蓋，讓帳號層設定永遠優先。 for i := range c.Accounts { @@ -219,6 +247,17 @@ func LoadDirectConfig(path string) (*DirectConfig, error) { return nil, fmt.Errorf("config 缺必填欄位：%s", strings.Join(missing, ", ")) } + // 🔴 t182：把抹除**寫回檔案**（leo：「新版裝上就要檢查……就要抹除改成用 Workers AI」）。 + // 只改記憶體不算抹除——托盤是**另一個行程**、自己讀同一份 config.json， + // 不寫回去它就繼續念「Gemini」，畫面與實際行為又脫鉤（t178 那個病）。 + // 放在驗證通過之後：確定這份 config 是好的才回寫，不把半殘結構蓋掉使用者的檔。 + // 寫失敗不擋啟動（唯讀目錄等）——記憶體裡已是 workers-ai，這輪行為仍正確。 + if migrated { + if err := saveDirectConfig(path, &c); err != nil { + fmt.Fprintf(os.Stderr, "⚠ 舊設定已改用雲端 AI，但寫回 config 失敗（不影響本次執行）：%v\n", err) + } + } + // 補各帳號預設值 for i := range c.Accounts { if c.Accounts[i].APIKey == "" { @@ -245,8 +284,12 @@ func LoadDirectConfig(path string) (*DirectConfig, error) { if c.LLMModel == "" { c.LLMModel = defaultLLMModel } - if c.Extractor != "" && c.Extractor != "claude" && c.Extractor != "gemma" { - return nil, fmt.Errorf("extractor 只能是 claude / gemma（或留空走舊制），got %q", c.Extractor) + // t182：`workers-ai` 是**現在的預設**，必須列入合法值。 + // （原本漏了它 ⇒ 一旦 config 寫成 workers-ai，載入直接失敗、daemon 起不來。 + // `claude` 保留在合法清單只為**讀得進舊 config**——RunDirectOnce 會把它正規化成 + // gemma（t176），不是還支援 claude。） + if c.Extractor != "" && c.Extractor != "workers-ai" && c.Extractor != "claude" && c.Extractor != "gemma" { + return nil, fmt.Errorf("extractor 只能是 workers-ai / gemma（或留空走預設），got %q", c.Extractor) } if c.PollSec <= 0 { c.PollSec = 5 @@ -457,6 +500,15 @@ func RunDirectOnce(cfg *DirectConfig, dryRun bool) ([]DirectResult, int, *Trigge // Gemini 仍是選配（leo：「不需要推廣，特定人告訴他怎麼做就好」）。 if !cfg.ExtractorExplicit { cfg.Extractor = "workers-ai" + // 🔴 t182（leo 08-04 實撞：更新到 v0.15.5 後托盤**兩個帳號都還是顯示 Gemini**）： + // 只清機器層不夠——`makeAccountSubConfig` 會把**帳號層**的 extractor 蓋回來 + // （t126 的「帳號層優先」規則，direct.go:414）。leo 的 config 正是這樣： + // 頂層 extractor="gemma"、每個帳號也各自 "gemma"，全都沒有 explicit + // ⇒ 頂層被改成 workers-ai，跑起來仍逐帳號走 gemma，畫面也照舊念 Gemini。 + // 沒有主動選過就是沒有主動選過，**每一層都要清**，否則預設等於沒改。 + for i := range cfg.Accounts { + cfg.Accounts[i].Extractor = "workers-ai" + } } if cfg.Extractor == "workers-ai" { // 這條路要的不是金鑰，是「連得上自己的知識庫」——沒連線才是真的沒法做事。 @@ -506,6 +558,22 @@ func RunDirectOnce(cfg *DirectConfig, dryRun bool) ([]DirectResult, int, *Trigge CloudCheckOK: cloudOK, } + // 🔴 t182（leo 08-04）：「會去**掃一次**看雲端是否裝好，**沒裝好就顯示 workers AI + // 還沒通，一旦通了就顯示可用**」。 + // 每個帳號各掃各的——雲端實例是**逐帳號**各自更新的（leo 自己就有兩個帳號、 + // 更新進度不同步），一個帳號通了不代表另一個也通。 + // 只在走 workers-ai 這條路時掃；選了 Gemini 的人不需要知道這件事。 + if accCfg.Extractor == "workers-ai" { + state := ProbeWorkersAI(accCfg.CypherURL, accCfg.APIKey) + accSt.CloudAIReady = state.Ready + accSt.CloudAINote = state.Note + if !state.Ready && state.Note != "" { + // 浮到頂層讓托盤「狀態：」那行直接看得到，不必展開帳號才發現。 + extractorOK = false + extractorError = state.Note + } + } + multi := len(accCfg.Folders()) > 1 for _, root := range accCfg.Folders() { r, e, p := runDirectOnceRoot(accCfg, root, dryRun) @@ -672,6 +740,12 @@ func runDirectOnceRoot(cfg *DirectConfig, root string, dryRun bool) ([]DirectRes case "workers-ai": // t181（leo 08-04 最優先）：走自己雲端實例的 Workers AI ⇒ **免金鑰**。 // 用戶不必去 Google 申請、也不受 Google 帳號被 flag 影響。 + // 🔴 t182：雲端還沒更新時**不在這裡默默退回 Gemini**。 + // leo 08-04 指定的設計是「掃一次、把狀態講出來」： + // 「沒裝好就顯示 workers AI 還沒通，一旦通了就顯示可用」 + // ⇒ 探測在 RunDirectOnce（ProbeWorkersAI），結果寫進 status.json， + // 托盤那行「狀態：」直接告訴用戶該做什麼。 + // 靜默退回會讓用戶**永遠不知道自己的雲端還沒更新**——正是要避免的黑箱。 cards, xerr = ExtractWithWorkersAI(cfg.CypherURL, cfg.APIKey, absRoot, ev.Path) case "gemma": cards, xerr = ExtractWithGemma(cfg.GeminiAPIKey, cfg.LLMModel, absRoot, ev.Path) diff --git a/direct_extract_test.go b/direct_extract_test.go index a211444..b0fac59 100644 --- a/direct_extract_test.go +++ b/direct_extract_test.go @@ -279,6 +279,15 @@ func TestMultiAccountInheritsExtractor(t *testing.T) { } // t108 Test C：extractor 空時，非 .md/.txt 檔禁止直送——標 failed 且絕不打任何 ingest 端點。 +// +// 🔴 t182 更新（leo 08-04 起 workers-ai 成為預設）：`extractor:""` 已**不再**代表 +// 「舊制直送」——LoadDirectConfig/RunDirectOnce 會把它正規化成 workers-ai。 +// 要測「舊制直送模式擋二進位」，必須把 config 逼進那條路：這裡用 ExtractorExplicit +// 明示、且不給任何可用引擎，才是真正的「無萃取器」狀態。 +// +// ⚠️ 本測試守的契約沒變、也不准放寬：**原始二進位永遠不出用戶的電腦**。 +// workers-ai 路一樣守——它送的是 ConvertToText 之後的純文字（extract_workersai.go）， +// 不是 PDF 位元組本身。 func TestExtractorEmptyBlocksNonTextDirect(t *testing.T) { root := t.TempDir() if err := os.WriteFile(filepath.Join(root, "report.pdf"), []byte("%PDF-1.4 機密原文"), 0o644); err != nil { @@ -296,9 +305,11 @@ func TestExtractorEmptyBlocksNonTextDirect(t *testing.T) { WatchFolders: []string{root}, Manifest: filepath.Join(t.TempDir(), "m.json"), CypherURL: srv.URL, Namespace: "demo", APIKey: "demo", - Library: "kb", - Extractor: "", // 舊制直送模式 - IngestWF: "rag_ingest_direct", + Library: "kb", + // t182：明示「使用者選過、但沒有可用引擎」＝真正的無萃取器狀態。 + // （不能只寫 Extractor:""——那現在會被正規化成 workers-ai，測不到這條防禦閘。） + Extractor: "gemma", ExtractorExplicit: true, GeminiAPIKey: "", + IngestWF: "rag_ingest_direct", RemovedWF: "rag_takedown_direct", MaxRemoved: DefaultMaxRemovedRatio, } @@ -312,10 +323,9 @@ func TestExtractorEmptyBlocksNonTextDirect(t *testing.T) { if len(results) != 1 || results[0].Status != "failed" { t.Fatalf("results=%+v", results) } - // t181：預設改走 workers-ai（免金鑰）後，「沒設 extractor」不再是失敗原因 - // ⇒ 這隻壞 PDF 會停在**轉檔層**（更早、更正確的把關點）。 + // t182：這裡是「選了 Gemini 卻沒有金鑰」⇒ 停在萃取層、誠實報缺什麼。 // 本測真正要守的契約沒變、也仍然綠：**PDF 不得被直送上雲**（上面的 serverCalled）。 - if !strings.Contains(results[0].Error, "轉檔失敗") { + if !strings.Contains(results[0].Error, "gemini_api_key") { t.Errorf("錯誤訊息不符：%q", results[0].Error) } } diff --git a/direct_multi_test.go b/direct_multi_test.go index b6163ef..715cbf1 100644 --- a/direct_multi_test.go +++ b/direct_multi_test.go @@ -692,6 +692,7 @@ func TestLoadDirectConfigMigratesCopyKeyToAccounts(t *testing.T) { p := writeDirectConfig(t, dir, map[string]any{ "manifest": filepath.Join(dir, "m.json"), "extractor": "gemma", + "extractor_explicit": true, // t182：主動選過 ⇒ 不被抹成 workers-ai "gemini_api_key": "top-level-key", "llm_model": "model-x", "accounts": []map[string]any{ @@ -717,6 +718,8 @@ func TestLoadDirectConfigMigratesCopyKeyToAccounts(t *testing.T) { if cfg.Accounts[0].GeminiAPIKey != "top-level-key" { t.Errorf("③ account A 應複製頂層 key，got %q", cfg.Accounts[0].GeminiAPIKey) } + // t182：加 extractor_explicit（見下方 writeDirectConfig）＝使用者主動選過 Gemini + // ⇒ 不觸發 workers-ai 抹除，t126 的「複製頂層 extractor」行為照舊可測。 if cfg.Accounts[0].Extractor != "gemma" { t.Errorf("③ account A 應複製頂層 extractor，got %q", cfg.Accounts[0].Extractor) } @@ -739,6 +742,7 @@ func TestLoadDirectConfigMigrationIdempotent(t *testing.T) { p := writeDirectConfig(t, dir, map[string]any{ "manifest": filepath.Join(dir, "m.json"), "extractor": "gemma", + "extractor_explicit": true, // t182：主動選過 ⇒ 不被抹成 workers-ai "gemini_api_key": "top-key", "accounts": []map[string]any{ { diff --git a/direct_t182_test.go b/direct_t182_test.go new file mode 100644 index 0000000..5e7f473 --- /dev/null +++ b/direct_t182_test.go @@ -0,0 +1,147 @@ +// direct_t182_test.go — t182：老 config 必須被**抹除**成 Workers AI（leo 2026-08-04）。 +// +// 「如果是我的 config 保持舊的，那新版裝上就要檢查，因為已經是 default worker AI， +// **就要抹除改成用 Workers AI**，如果保持 Gemini 它不會改掉，**那就是失敗的**」 +// +// 這組測試守的是 leo 08-04 實撞的三個真 bug： +// 1. 更新到 v0.15.5 後托盤**兩個帳號都還顯示 Gemini**（帳號層舊值沒清） +// 2. 丟 PDF 進去**產不出卡**（因為根本沒走到 Workers AI，還在跑 Gemini） +// 3. `workers-ai` 不在合法值清單裡 ⇒ 一旦寫進 config，daemon 直接起不來 +package main + +import ( + "encoding/json" + "os" + "path/filepath" + "testing" +) + +// t182①：老 config（extractor=gemma、帳號層也是 gemma、無 explicit） +// → 每一層都要被抹成 workers-ai，而且**要寫回檔案**。 +func TestT182ErasesLegacyExtractorAllLayers(t *testing.T) { + dir := t.TempDir() + p := writeDirectConfig(t, dir, map[string]any{ + "manifest": filepath.Join(dir, "m.json"), + "extractor": "gemma", // 舊值：頂層 + "gemini_api_key": "old-key", + // 注意：**沒有** extractor_explicit ⇒ 使用者從沒主動選過 + "accounts": []map[string]any{ + { + "cypher_url": "https://a.example", "namespace": "nsA", + "watch_folders": []string{"/tmp/a"}, + "extractor": "gemma", // 舊值：帳號層（leo 的 config 正是這樣） + }, + { + "cypher_url": "https://b.example", "namespace": "nsB", + "watch_folders": []string{"/tmp/b"}, + "extractor": "claude", // 更舊的殘留值 + }, + }, + }) + + cfg, err := LoadDirectConfig(p) + if err != nil { + t.Fatalf("LoadDirectConfig: %v", err) + } + if cfg.Extractor != "workers-ai" { + t.Errorf("① 頂層沒被抹除：got %q want workers-ai", cfg.Extractor) + } + for i, a := range cfg.Accounts { + if a.Extractor != "workers-ai" { + t.Errorf("① 帳號[%d] 沒被抹除：got %q want workers-ai（leo 實撞：兩個帳號都還顯示 Gemini）", i, a.Extractor) + } + } + + // 金鑰要留著——Gemini 只是變選配，不是廢除（leo：「客戶說他要用 Gemini」） + if cfg.GeminiAPIKey != "old-key" { + t.Errorf("① Gemini 金鑰不該被清掉：got %q", cfg.GeminiAPIKey) + } + + // 真的寫回檔案了嗎？（只改記憶體 ⇒ 托盤是另一個行程，還是會念 Gemini） + raw, err := os.ReadFile(p) + if err != nil { + t.Fatal(err) + } + var onDisk struct { + Extractor string `json:"extractor"` + Accounts []struct { + Extractor string `json:"extractor"` + } `json:"accounts"` + } + if err := json.Unmarshal(raw, &onDisk); err != nil { + t.Fatal(err) + } + if onDisk.Extractor != "workers-ai" { + t.Errorf("① **沒寫回檔案**：磁碟上頂層仍為 %q ⇒ 托盤下次讀還是念 Gemini", onDisk.Extractor) + } + for i, a := range onDisk.Accounts { + if a.Extractor != "workers-ai" { + t.Errorf("① **沒寫回檔案**：磁碟上帳號[%d] 仍為 %q", i, a.Extractor) + } + } +} + +// t182②：主動選過 Gemini 的人不被動到（Gemini 是選配、不是廢除）。 +func TestT182KeepsExplicitGeminiChoice(t *testing.T) { + dir := t.TempDir() + p := writeDirectConfig(t, dir, map[string]any{ + "manifest": filepath.Join(dir, "m.json"), + "extractor": "gemma", + "extractor_explicit": true, // ← 使用者在「AI 設定…」主動選過 + "gemini_api_key": "my-key", + "accounts": []map[string]any{ + {"cypher_url": "https://a.example", "namespace": "nsA", "watch_folders": []string{"/tmp/a"}}, + }, + }) + cfg, err := LoadDirectConfig(p) + if err != nil { + t.Fatalf("LoadDirectConfig: %v", err) + } + if cfg.Extractor != "gemma" { + t.Errorf("② 主動選過 Gemini 不該被抹掉：got %q", cfg.Extractor) + } +} + +// t182③：`workers-ai` 必須是合法值——否則寫進 config 後 daemon 直接起不來。 +// （v0.15.5 就踩到這顆：預設改了、驗證器沒跟上。） +func TestT182WorkersAIIsValidExtractor(t *testing.T) { + dir := t.TempDir() + p := writeDirectConfig(t, dir, map[string]any{ + "manifest": filepath.Join(dir, "m.json"), + "extractor": "workers-ai", + "extractor_explicit": true, + "accounts": []map[string]any{ + {"cypher_url": "https://a.example", "namespace": "nsA", "watch_folders": []string{"/tmp/a"}}, + }, + }) + if _, err := LoadDirectConfig(p); err != nil { + t.Fatalf("③ workers-ai 應為合法值，卻載入失敗：%v", err) + } +} + +// t182④：冪等——已是 workers-ai 再載一次不應改動任何東西。 +func TestT182MigrationIdempotent(t *testing.T) { + dir := t.TempDir() + p := writeDirectConfig(t, dir, map[string]any{ + "manifest": filepath.Join(dir, "m.json"), + "extractor": "workers-ai", + "accounts": []map[string]any{ + {"cypher_url": "https://a.example", "namespace": "nsA", + "watch_folders": []string{"/tmp/a"}, "extractor": "workers-ai"}, + }, + }) + before, err := os.ReadFile(p) + if err != nil { + t.Fatal(err) + } + if _, err := LoadDirectConfig(p); err != nil { + t.Fatal(err) + } + after, err := os.ReadFile(p) + if err != nil { + t.Fatal(err) + } + if string(before) != string(after) { + t.Error("④ 已是 workers-ai 不該再改寫檔案（非冪等＝每次啟動都寫一次磁碟）") + } +} diff --git a/probe_workersai.go b/probe_workersai.go new file mode 100644 index 0000000..5bd3cbe --- /dev/null +++ b/probe_workersai.go @@ -0,0 +1,87 @@ +// probe_workersai.go — 探測「這個帳號的雲端知識庫，Workers AI 萃取通了沒」（t182）。 +// +// 🔴 為什麼要探測（leo 2026-08-04 兩句話定的設計）： +// +// ① 「順序不可反這件事**門檻太高**了，用戶一定**先看到 daemon 有更新**， +// 不會想到更新網頁」 +// ② 「你可以設計如果它先做了地端就要求去做雲端，**會去掃一次看雲端是否裝好**， +// **沒裝好就顯示 workers AI 還沒通，一旦通了就顯示可用**」 +// +// 背景：Workers AI 萃取要**兩端都新**——daemon 有新版（會打 /portal/daemon/extract） +// 且雲端實例也有新版（有這條 route）。daemon 會自動更新、雲端實例要用戶自己重跑安裝器 +// ⇒ 中間必然存在一段「daemon 新、雲端舊」的空窗期。 +// +// 舊設計把這段空窗丟給用戶自己想通（「你順序做錯了」），那是**我們的設計沒容錯**。 +// 新設計：daemon 主動掃、把狀態講出來—— +// +// 雲端還沒更新 → 托盤直接顯示「雲端 AI 還沒通 ⇒ 去 portal 按『立即更新』」 +// 雲端已更新 → 顯示「雲端 AI 可用」，什麼都不必做 +// +// ⚠️ 這是**被動探測**，不是輪詢：只在同步時（RunDirectOnce 每輪一次）順手打一次 +// HEAD-ish 的探測請求，且結果快取在 status.json 給托盤讀。不掛任何排程。 +package main + +import ( + "bytes" + "encoding/json" + "fmt" + "net/http" + "strings" + "time" +) + +// probeHTTP 探測用短逾時：只是要知道 route 在不在，不等 LLM 生成。 +var probeHTTP = &http.Client{Timeout: 20 * time.Second} + +// CloudAIState 是「某個雲端實例的 Workers AI 萃取能力」探測結果。 +type CloudAIState struct { + Ready bool `json:"ready"` // true＝那端有 /portal/daemon/extract（雲端 AI 可用） + Note string `json:"note,omitempty"` // 給用戶看的白話（Ready=false 時說「還沒通、該做什麼」） +} + +// ProbeWorkersAI 探測指定實例是否支援雲端萃取。 +// +// 手法：送一個**刻意留空 text** 的請求。 +// - 舊版實例沒有這條 route ⇒ 404 ⇒ 還沒通。 +// - 新版實例有 route，會因為 text 是空的回 400（參數錯）——但 400 正好證明**route 存在** +// ⇒ 算通。這樣探測不會真的燒 LLM 額度，也不會產生垃圾卡片。 +// - 401 代表 route 在、只是金鑰不對 ⇒ route 存在，同樣算「雲端有這功能」， +// 金鑰問題由既有的連線流程去報，不混在這裡講。 +func ProbeWorkersAI(cypherURL, apiKey string) CloudAIState { + base := strings.TrimSuffix(strings.TrimSpace(cypherURL), "/") + if base == "" { + return CloudAIState{Ready: false, Note: "還沒連上知識庫"} + } + + body, _ := json.Marshal(map[string]string{"page_name": "", "text": ""}) + req, err := http.NewRequest(http.MethodPost, base+"/portal/daemon/extract", bytes.NewReader(body)) + if err != nil { + return CloudAIState{Ready: false, Note: "雲端 AI 狀態查不到"} + } + req.Header.Set("Content-Type", "application/json") + req.Header.Set("X-Arcrun-API-Key", strings.TrimSpace(apiKey)) + + resp, err := probeHTTP.Do(req) + if err != nil { + // 連不上（離線／網路問題）≠ 雲端沒裝。講「查不到」而不是「還沒通」， + // 免得把網路問題誤報成「你沒更新」讓用戶白跑一趟。 + return CloudAIState{Ready: false, Note: "連不上你的知識庫，雲端 AI 狀態查不到"} + } + defer resp.Body.Close() + + switch { + case resp.StatusCode == http.StatusNotFound: + return CloudAIState{ + Ready: false, + Note: "雲端 AI 還沒通 ⇒ 你的知識庫是舊版，請到 portal 按「立即更新」重裝一次", + } + case resp.StatusCode < 500: + // 200/400/401… 都代表這條 route 存在＝雲端有這個功能。 + return CloudAIState{Ready: true} + default: + return CloudAIState{ + Ready: false, + Note: fmt.Sprintf("雲端 AI 暫時有狀況（HTTP %d），稍後會自動再試", resp.StatusCode), + } + } +} diff --git a/sync_status.go b/sync_status.go index 236b541..f529694 100644 --- a/sync_status.go +++ b/sync_status.go @@ -16,6 +16,11 @@ type AccountSyncStatus struct { CloudCheckOK bool `json:"cloud_check_ok"` ExtractedOK int `json:"extracted_ok"` ExtractFailed int `json:"extract_failed"` + // t182（leo 08-04：「沒裝好就顯示 workers AI 還沒通，一旦通了就顯示可用」）： + // 這個帳號的雲端實例有沒有 /portal/daemon/extract。**逐帳號**各自記—— + // 用戶可能有多個實例、更新進度不同步。只在走 workers-ai 這條路時探測。 + CloudAIReady bool `json:"cloud_ai_ready"` + CloudAINote string `json:"cloud_ai_note,omitempty"` // 還沒通時的白話說明（含該做什麼） } // SyncStatus 彙總每輪同步的萃取結果，持久化至 ~/.arcrun-rag/status.json。