Cyrene/backend/ai-core/internal/context/builder.go

package context

import (
    "context"
    "fmt"
    "log"

    "github.com/yourname/cyrene-ai/ai-core/internal/memory"
    "github.com/yourname/cyrene-ai/ai-core/internal/model"
    "github.com/yourname/cyrene-ai/ai-core/internal/persona"
)

// Builder 对话上下文构建器
type Builder struct{}

type BuildParams struct {
    UserID       string
    SessionID    string
    UserMessage  string
    Persona      *persona.PersonaConfig
    Memories     []memory.MemoryEntry
    HistoryLimit int
}

// Build 构建发送给LLM的完整消息列表
func (b *Builder) Build(ctx context.Context, params BuildParams) ([]model.LLMMessage, error) {
    messages := []model.LLMMessage{}

    // 1. 系统消息 —— 昔涟的人格Prompt
    systemPrompt := params.Persona.BuildSystemPrompt(
        params.UserID, // 后续可替换为真实用户名
        1,             // 初始好感度
    )
    messages = append(messages, model.LLMMessage{
        Role:    "system",
        Content: systemPrompt,
    })

    // 2. 记忆注入 —— 相关记忆以系统消息形式注入
    if len(params.Memories) > 0 {
        memoryPrompt := "【以下是关于开拓者的一些重要记忆，请在合适的时机自然地提及】\n"
        for _, m := range params.Memories {
            memoryPrompt += fmt.Sprintf("- %s\n", m.Content)
        }
        messages = append(messages, model.LLMMessage{
            Role:    "system",
            Content: memoryPrompt,
        })
    }

    // 3. 历史对话
    history, err := b.loadHistory(ctx, params.SessionID, params.HistoryLimit)
    if err == nil {
        messages = append(messages, history...)
    }

    // 4. 当前用户消息
    messages = append(messages, model.LLMMessage{
        Role:    "user",
        Content: params.UserMessage,
    })

    return messages, nil
}

// loadHistory 加载会话历史 (MVP阶段返回空，后续对接数据库)
func (b *Builder) loadHistory(_ context.Context, sessionID string, limit int) ([]model.LLMMessage, error) {
    log.Printf("[context] 加载会话 %s 历史 (限制 %d 条) - 暂未实现持久化", sessionID, limit)
    return nil, nil
}