t182:老 config 抹除成 Workers AI+雲端探測——leo 實測「無卡」的真因
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) <noreply@anthropic.com>
This commit is contained in:
+13
-2
@@ -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)
|
||||
|
||||
|
||||
@@ -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)
|
||||
|
||||
+16
-6
@@ -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)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -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{
|
||||
{
|
||||
|
||||
@@ -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 不該再改寫檔案(非冪等=每次啟動都寫一次磁碟)")
|
||||
}
|
||||
}
|
||||
@@ -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),
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -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。
|
||||
|
||||
Reference in New Issue
Block a user