WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

2026-09-22 16:19:06 +08:00
commit f7fc992f57
3929 changed files with 1006607 additions and 0 deletions
@@ -0,0 +1,46 @@
+package tools
+
+import (
+	"encoding/json"
+	"strings"
+	"testing"
+
+	"github.com/Tencent/WeKnora/internal/types"
+)
+
+func TestLLMToolOutputsUseMarkdownImages(t *testing.T) {
+	imageInfo, err := json.Marshal([]types.ImageInfo{
+		{
+			URL:     "resource://AbCdEfGhIjKlMnOpQrStUv",
+			Caption: "目标说话人提取流程图",
+			OCRText: "输入\n输出",
+		},
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	chunk := &types.Chunk{
+		ID:         "chunk-1",
+		ChunkIndex: 0,
+		ChunkType:  types.ChunkTypeText,
+		Content:    "测试阶段主要流程",
+		ImageInfo:  string(imageInfo),
+	}
+
+	listOutput := (&ListKnowledgeChunksTool{}).buildOutput("knowledge-1", "测试文档", 1, 1, []*types.Chunk{chunk})
+	wikiOutput := enrichChunkContent(chunk)
+	for name, output := range map[string]string{
+		"list_knowledge_chunks": listOutput,
+		"wiki_read_source_doc":  wikiOutput,
+	} {
+		t.Run(name, func(t *testing.T) {
+			if !strings.Contains(output, "![目标说话人提取流程图](resource://AbCdEfGhIjKlMnOpQrStUv)") {
+				t.Fatalf("expected Markdown image in tool output:\n%s", output)
+			}
+			if strings.Contains(output, "<image") || strings.Contains(output, "<caption>") || strings.Contains(output, "<ocr_text>") {
+				t.Fatalf("tool output leaked legacy image XML:\n%s", output)
+			}
+		})
+	}
+}