Method WrapModelCall

pkg/middleware/agent_memory.go:113–148

WrapModelCall 包装模型调用,注入记忆到 system prompt

(ctx context.Context, req *ModelRequest, handler ModelCallHandler)

// WrapModelCall 包装模型调用,注入记忆到 system prompt
func (m *AgentMemoryMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error) {
	// 如果还没加载记忆,先加载
	if !m.memoryLoaded {
		err := m.OnAgentStart(ctx, "default")
		if err != nil {
			amLog.Error(ctx, "failed to load memory", map[string]any{"error": err.Error()})
		}
	}

	// 保存原始 system prompt
	originalSystemPrompt := req.SystemPrompt

	// 构建记忆部分
	memorySection := fmt.Sprintf(m.systemPromptTemplate, m.memoryContent)

	// 注入到 system prompt 开头
	if originalSystemPrompt != "" {
		req.SystemPrompt = memorySection + "\n\n" + originalSystemPrompt
	} else {
		req.SystemPrompt = memorySection
	}

	// 追加长期记忆使用文档
	longTermMemoryPrompt := m.buildLongTermMemoryPrompt()
	req.SystemPrompt = req.SystemPrompt + "\n\n" + longTermMemoryPrompt

	amLog.Debug(ctx, "injected agent memory", map[string]any{"total_chars": len(req.SystemPrompt)})

	// 调用处理器
	resp, err := handler(ctx, req)

	// 恢复原始 system prompt,避免在重用请求对象时累积
	req.SystemPrompt = originalSystemPrompt

	return resp, err
}

// buildLongTermMemoryPrompt 构建长期记忆使用指南
func (m *AgentMemoryMiddleware) buildLongTermMemoryPrompt() string {