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

internal/agent/image_requirement.go

@@ -0,0 +1,48 @@
+package agent
+
+import (
+	"strings"
+
+	"github.com/Tencent/WeKnora/internal/models/chat"
+	"github.com/Tencent/WeKnora/internal/searchutil"
+	"github.com/Tencent/WeKnora/internal/types"
+)
+
+const agentRetrievedImageRequirementMarker = "## Retrieved Image Output Requirement"
+
+const agentRetrievedImageSystemRequirement = `
+
+## Retrieved Image Output Requirement
+Retrieved tool results for this turn contain Markdown images. Treat images attached to retrieved passages as relevant by default.
+- Unless the user explicitly requests text-only output, or every retrieved image is clearly unrelated, the final answer MUST include at least one relevant Markdown image copied verbatim from the tool results.
+- Preserve the complete Markdown image syntax and URL exactly; never invent, shorten, normalize, or replace the URL.
+- Use ASCII half-width parentheses exactly as ![alt](url); never use full-width （ or ）.
+- Place each image immediately after the paragraph it supports.
+- When multiple retrieved images support different sections, distribute them across those sections instead of stopping after the first image.
+- Before finishing, silently verify that the answer contains a Markdown image whenever this requirement applies.`
+
+func stepContainsMarkdownImage(step types.AgentStep) bool {
+	for _, toolCall := range step.ToolCalls {
+		if toolCall.Result != nil &&
+			toolCall.Result.Success &&
+			searchutil.MarkdownImageRegex.MatchString(toolCall.Result.Output) {
+			return true
+		}
+	}
+	return false
+}
+
+func appendAgentRetrievedImageRequirement(messages []chat.Message) []chat.Message {
+	for _, message := range messages {
+		if strings.Contains(message.Content, agentRetrievedImageRequirementMarker) {
+			return messages
+		}
+	}
+	// Append after the current prefix instead of editing the system prompt.
+	// Provider prompt caches are prefix matches: mutating the system block
+	// invalidates the tools and the whole transcript for every later round.
+	return append(messages, chat.Message{
+		Role:    "user",
+		Content: strings.TrimSpace(agentRetrievedImageSystemRequirement),
+	})
+}