feat(collector): workers-ai 路改吃共用契約——免金鑰預設路的卡跟 BYOK 一樣完整(Arcrun#134)

InkStoneCo#44 ④ 只修了 gemma BYOK 路(JSON 契約+wikishape 機械組卡),
免金鑰預設路(多數用戶實際走的)仍吃雲端舊 prompt 的舊格式卡 ⇒ 走預設路
拿到次級知識庫。

修法=契約單一真相源:wikiExtractPrompt(原 gemmaPrompt 改名,因為它已是
兩條路共用)由 daemon 整段帶上雲(request prompt 欄位),雲端只當執行器回
模型原文(response output),解析(parseWikiExtractJSON)與組卡(BuildWikiDoc)
回到本 package 與 gemma 路同一段程式碼——同形不再靠「兩邊要一起改」的叮嚀。

版本歪斜兩向都有路:舊雲端忽略 prompt 回 card ⇒ fallback 走 legacy 落卡
(#60 前綴/不覆蓋保護原封不動,收端 lint 新舊雙軌);舊 daemon 不帶 prompt
⇒ 雲端 legacy 行為不變(對向修法在 Arcrun work/extract-wiki-json-0815)。

測試:+2——①request 的 prompt 必須是 wikiExtractPrompt 本人(不是手抄第二份)
②同原稿+同判斷走兩條路,.wiki/ 產物逐位元組相同(卡+00-INDEX+manifest)。
既有兩則(stub 回 card)自動變成舊雲端 fallback 的守衛。go test ./... 全綠。

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
2026-08-15 21:26:30 +08:00
parent 12c2d41886
commit 6c74b97718
6 changed files with 165 additions and 8 deletions
+122
View File
@@ -102,3 +102,125 @@ func TestExtractWithWorkersAI_NonVaultUnchanged(t *testing.T) {
t.Fatalf("非 vault 卡片路徑不對:%vwant [system-dev/wiki/cards/arcrun-note.md]", cards)
}
}
// ── Arcrun#134workers-ai 路與 gemma 路共用同一份契約 ─────────────────────────
//
// 修法=daemon 把 wikiExtractPrompt 整段帶去雲端(request `prompt` 欄位),
// 雲端只回模型原文(response `output`),解析與組卡回到本 package 與 gemma 路
// 同一段程式碼。⇒ 「兩條路的卡同形」不再是兩份 prompt 各自維持的巧合,
// 是同一份程式碼的必然。以下兩則就是這句話的機械守衛;
// 檔案上方兩則既有測試(stub 只回 `card`)則守住「舊雲端 fallback 不斷炊」。
// request 必帶 prompt,且必須就是 wikiExtractPrompt 本人——不是另一份手抄。
func TestExtractWithWorkersAI_SendsSharedPrompt(t *testing.T) {
root := t.TempDir()
const srcName = "報銷規則.md"
const srcBody = "# 報銷規則\n\n內文"
if err := os.WriteFile(filepath.Join(root, srcName), []byte(srcBody), 0o644); err != nil {
t.Fatal(err)
}
var gotPrompt string
url, closeFn := workersAIStub(t, func(w http.ResponseWriter, r *http.Request) {
var req map[string]string
_ = json.NewDecoder(r.Body).Decode(&req)
gotPrompt = req["prompt"]
_ = json.NewEncoder(w).Encode(map[string]any{
"success": true,
"output": cardFixture("報銷規則", "財務"),
})
})
defer closeFn()
if _, err := ExtractWithWorkersAI(url, "key123", root, srcName); err != nil {
t.Fatal(err)
}
srcText, err := ConvertToText(srcName, []byte(srcBody))
if err != nil {
t.Fatal(err)
}
if want := wikiExtractPrompt("報銷規則", srcText); gotPrompt != want {
t.Fatalf("送上雲的 prompt 不是共用那份 wikiExtractPromptlen got=%d want=%d", len(gotPrompt), len(want))
}
}
// wikiTreeOf 收齊 root 底下所有 .wiki/ 產物(相對路徑 → 內容)。
func wikiTreeOf(t *testing.T, root string) map[string]string {
t.Helper()
out := map[string]string{}
err := filepath.Walk(root, func(p string, info os.FileInfo, err error) error {
if err != nil || info.IsDir() {
return err
}
rel, _ := filepath.Rel(root, p)
if !strings.Contains(filepath.ToSlash(rel), ".wiki/") {
return nil
}
b, rerr := os.ReadFile(p)
if rerr != nil {
return rerr
}
out[filepath.ToSlash(rel)] = string(b)
return nil
})
if err != nil {
t.Fatal(err)
}
return out
}
// 🔴 本票的驗收本體:同一份原稿+同一份模型判斷,走 gemma 路與 workers-ai 路,
// `.wiki/` 產物(卡、00-INDEX、manifest)必須**逐位元組相同**。
func TestExtractWorkersAIAndGemmaProduceIdenticalWiki(t *testing.T) {
const srcName = "報銷規則.md"
const srcBody = "# 報銷規則\n\n機密內容 XYZZY"
fixture := cardFixture("報銷規則", "財務")
// gemma 路
rootG := t.TempDir()
if err := os.WriteFile(filepath.Join(rootG, srcName), []byte(srcBody), 0o644); err != nil {
t.Fatal(err)
}
undo := gemmaCardStub(t, fixture)
cardsG, errG := ExtractWithGemma("k-test", "", rootG, srcName)
undo()
if errG != nil {
t.Fatal(errG)
}
// workers-ai 路
rootW := t.TempDir()
if err := os.WriteFile(filepath.Join(rootW, srcName), []byte(srcBody), 0o644); err != nil {
t.Fatal(err)
}
url, closeFn := workersAIStub(t, func(w http.ResponseWriter, r *http.Request) {
_ = json.NewEncoder(w).Encode(map[string]any{"success": true, "output": fixture})
})
cardsW, errW := ExtractWithWorkersAI(url, "key123", rootW, srcName)
closeFn()
if errW != nil {
t.Fatal(errW)
}
if strings.Join(cardsG, "|") != strings.Join(cardsW, "|") {
t.Fatalf("兩條路回報的卡片清單不同:gemma=%v workers-ai=%v", cardsG, cardsW)
}
treeG, treeW := wikiTreeOf(t, rootG), wikiTreeOf(t, rootW)
if len(treeG) == 0 {
t.Fatal("gemma 路沒有產出任何 .wiki 檔案(測試前提壞了)")
}
for rel, want := range treeG {
got, ok := treeW[rel]
if !ok {
t.Errorf("workers-ai 路缺檔:%s", rel)
continue
}
if got != want {
t.Errorf("兩條路的 %s 內容不同(前 200 字):\n─ gemma ─\n%.200s\n─ workers-ai ─\n%.200s", rel, want, got)
}
}
for rel := range treeW {
if _, ok := treeG[rel]; !ok {
t.Errorf("workers-ai 路多出 gemma 路沒有的檔:%s", rel)
}
}
}