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

internal/agent/tools/tool_policy_coverage_test.go

@@ -0,0 +1,66 @@
+package tools
+
+import (
+	"testing"
+
+	"github.com/Tencent/WeKnora/internal/modelcontext"
+)
+
+// Every built-in tool must declare a model-handle policy. Without an entry,
+// modelcontext treats the tool's output like dynamic MCP output: durable chunk
+// IDs, knowledge IDs and KB IDs are passed through to the model verbatim,
+// which the source protocol prompt explicitly forbids. An empty policy is a
+// valid answer ("compact known IDs, render nothing structured"); silence is
+// not, because a newly added tool would opt out by accident.
+func TestEveryBuiltInToolDeclaresAModelHandlePolicy(t *testing.T) {
+	builtIns := []string{
+		ToolThinking,
+		ToolTodoWrite,
+		ToolGrepChunks,
+		ToolKnowledgeSearch,
+		ToolListKnowledgeChunks,
+		ToolQueryKnowledgeGraph,
+		ToolGetDocumentInfo,
+		ToolSearchConversations,
+		ToolSearchMemory,
+		ToolDatabaseQuery,
+		ToolDataAnalysis,
+		ToolDataSchema,
+		ToolWebSearch,
+		ToolWebFetch,
+		ToolExecuteSkillScript,
+		ToolReadSkill,
+		ToolListSandboxFiles,
+		ToolReadSandboxFile,
+		ToolWriteSandboxFile,
+		ToolEditSandboxFile,
+		ToolWriteSkillFile,
+		ToolEditSkillFile,
+		ToolShellExec,
+		ToolWikiReadPage,
+		ToolWikiWritePage,
+		ToolWikiReplaceText,
+		ToolWikiRenamePage,
+		ToolWikiDeletePage,
+		ToolWikiSearch,
+		ToolWikiReadSourceDoc,
+		ToolWikiFlagIssue,
+		ToolWikiReadIssue,
+		ToolWikiUpdateIssue,
+	}
+	for _, name := range builtIns {
+		if !modelcontext.HasToolPolicy(name) {
+			t.Errorf("built-in tool %q has no entry in modelcontext.toolHandlePolicies", name)
+		}
+	}
+}
+
+// AvailableToolDefinitions drives the settings UI, so anything listed there is
+// a built-in the Agent can register and must also carry a policy.
+func TestAvailableToolDefinitionsDeclareModelHandlePolicies(t *testing.T) {
+	for _, definition := range AvailableToolDefinitions() {
+		if !modelcontext.HasToolPolicy(definition.Name) {
+			t.Errorf("tool %q is exposed to the UI but has no model-handle policy", definition.Name)
+		}
+	}
+}