smartmate/backend/newAgent/model/state_store.go

package model

import (
	"context"

	newagenttools "github.com/LoveLosita/smartflow/backend/newAgent/tools"
)

// AgentStateSnapshot 是需要持久化的 agent 运行态最小快照。
//
// 设计说明：
// 1. 只保存恢复执行所需的 RuntimeState 和 ConversationContext；
// 2. 不保存 Request（每轮请求级，天然不跨连接）；
// 3. 不保存 Deps（依赖注入，每次由 Service 层重建）；
// 4. 不保存 ToolSchemas（每次请求由 Service 层重新注入）。
type AgentStateSnapshot struct {
	RuntimeState        *AgentRuntimeState   `json:"runtime_state"`
	ConversationContext *ConversationContext `json:"conversation_context"`
}

// AgentStateStore 定义 agent 状态持久化的最小接口。
//
// 职责边界：
// 1. 只负责"存 / 取 / 删"三个原子操作；
// 2. 不负责序列化细节（由实现层决定 JSON / protobuf）；
// 3. 不负责业务级状态校验，校验仍在 node / graph 层完成。
//
// 实现层：
// 1. dao/cache.go 上的 CacheDAO 隐式实现该接口（Go duck typing）；
// 2. newAgent 包不直接 import dao，由 Service 层在组装 Deps 时注入。
type AgentStateStore interface {
	// Save 序列化并保存一份 agent 状态快照。
	//
	// 语义：
	// 1. 同一 conversationID 被覆盖写入，保证 Redis 里始终只有最新快照；
	// 2. 实现层应设 TTL，避免已完成的任务快照永不清理。
	Save(ctx context.Context, conversationID string, snapshot *AgentStateSnapshot) error

	// Load 读取并反序列化 agent 状态快照。
	//
	// 返回值语义：
	// 1. (snapshot, true, nil)：命中快照，正常返回；
	// 2. (nil, false, nil)：未命中，不是错误，调用方应走新建对话路径；
	// 3. (nil, false, error)：真正的存储层错误。
	Load(ctx context.Context, conversationID string) (*AgentStateSnapshot, bool, error)

	// Delete 删除指定会话的 agent 状态快照。
	//
	// 语义：
	// 1. 删除是幂等的，key 不存在也视为成功；
	// 2. 典型调用时机：Deliver 节点任务完成后清理。
	Delete(ctx context.Context, conversationID string) error
}

// ScheduleStateProvider 定义加载 ScheduleState 的接口。
// 由 DAO 层或 Service 层实现，注入到 AgentGraphDeps 中。
// 使用接口而非具体 DAO 类型，避免 model → dao 的循环依赖。
type ScheduleStateProvider interface {
	LoadScheduleState(ctx context.Context, userID int) (*newagenttools.ScheduleState, error)
	// LoadTaskClassMetas 只加载指定任务类的约束元数据，供 Plan 节点提前消费。
	LoadTaskClassMetas(ctx context.Context, userID int, taskClassIDs []int) ([]newagenttools.TaskClassMeta, error)
}

// SchedulePersistor 定义持久化 ScheduleState 变更的接口。
// 由 Service 层或 DAO 层实现，注入到 AgentGraphDeps 中。
// 使用接口而非具体 DAO 类型，避免 model → dao 的循环依赖。
type SchedulePersistor interface {
	PersistScheduleChanges(ctx context.Context, original, modified *newagenttools.ScheduleState, userID int) error
}