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:
2026-08-04 16:29:56 +08:00
parent cea6108924
commit 133f5add46
7 changed files with 348 additions and 10 deletions
+13 -2
View File
@@ -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)
+76 -2
View File
@@ -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
// 萃取也照舊走 Geminileo 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 會把它正規化成
// gemmat176),不是還支援 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"
// 🔴 t182leo 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,
}
// 🔴 t182leo 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":
// t181leo 08-04 最優先):走自己雲端實例的 Workers AI ⇒ **免金鑰**。
// 用戶不必去 Google 申請、也不受 Google 帳號被 flag 影響。
// 🔴 t182:雲端還沒更新時**不在這裡默默退回 Gemini**。
// leo 08-04 指定的設計是「掃一次、把狀態講出來」:
// 「沒裝好就顯示 workers AI 還沒通,一旦通了就顯示可用」
// ⇒ 探測在 RunDirectOnceProbeWorkersAI),結果寫進 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
View File
@@ -279,6 +279,15 @@ func TestMultiAccountInheritsExtractor(t *testing.T) {
}
// t108 Test Cextractor 空時,非 .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)
}
}
+4
View File
@@ -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 應複製頂層 keygot %q", cfg.Accounts[0].GeminiAPIKey)
}
// t182:加 extractor_explicit(見下方 writeDirectConfig)=使用者主動選過 Gemini
// ⇒ 不觸發 workers-ai 抹除,t126 的「複製頂層 extractor」行為照舊可測。
if cfg.Accounts[0].Extractor != "gemma" {
t.Errorf("③ account A 應複製頂層 extractorgot %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{
{
+147
View File
@@ -0,0 +1,147 @@
// direct_t182_test.go — t182:老 config 必須被**抹除**成 Workers AIleo 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①:老 configextractor=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-aileo 實撞:兩個帳號都還顯示 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 不該再改寫檔案(非冪等=每次啟動都寫一次磁碟)")
}
}
+87
View File
@@ -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),
}
}
}
+5
View File
@@ -16,6 +16,11 @@ type AccountSyncStatus struct {
CloudCheckOK bool `json:"cloud_check_ok"`
ExtractedOK int `json:"extracted_ok"`
ExtractFailed int `json:"extract_failed"`
// t182leo 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。