070d4c3459
后端： 1. 排程工具从 tools/ 根目录拆分为 tools/schedule 独立子包 - 12 个排程工具文件等价迁入 tools/schedule/，tools/ 根目录仅保留 registry.go 作为统一注册入口 - 所有依赖方（conv / model / node / prompt / service）import 统一切到 schedule 子包 2. Web 搜索工具链落地（tools/web 子包） - 新增 web_search（结构化检索）与 web_fetch（正文抓取）两个读工具，支持博查 API / mock 降级 - 启动流程按配置选择 provider，未识别类型自动降级为 mock，不阻断主流程 - 执行提示补齐 web 工具使用约束与返回值示例 - config.example.yaml 补齐 websearch 配置段 前端：无 仓库：无
82 lines
3.9 KiB
Go
82 lines
3.9 KiB
Go
package model
|
||
|
||
import (
|
||
"context"
|
||
|
||
schedule "github.com/LoveLosita/smartflow/backend/newAgent/tools/schedule"
|
||
)
|
||
|
||
// 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"`
|
||
ScheduleState *schedule.ScheduleState `json:"schedule_state,omitempty"`
|
||
OriginalScheduleState *schedule.ScheduleState `json:"original_schedule_state,omitempty"`
|
||
}
|
||
|
||
// 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) (*schedule.ScheduleState, error)
|
||
// LoadTaskClassMetas 只加载指定任务类的约束元数据，供 Plan 节点提前消费。
|
||
LoadTaskClassMetas(ctx context.Context, userID int, taskClassIDs []int) ([]schedule.TaskClassMeta, error)
|
||
}
|
||
|
||
// ScopedScheduleStateProvider 定义“按本轮任务类范围加载 ScheduleState”的可选增强接口。
|
||
//
|
||
// 设计说明：
|
||
// 1. 负责：在 Execute / RoughBuild 首次加载状态时，把 DayMapping、TaskClasses 与 pending 任务限定在本轮 task_class_ids 相关窗口；
|
||
// 2. 不负责：改变既有 ScheduleStateProvider 的基础能力，老实现仍可只实现 LoadScheduleState；
|
||
// 3. 兜底策略：若调用方拿到的 provider 不实现该接口，则回退到全量 LoadScheduleState，再走工具层 scope 裁剪。
|
||
type ScopedScheduleStateProvider interface {
|
||
LoadScheduleStateForTaskClasses(ctx context.Context, userID int, taskClassIDs []int) (*schedule.ScheduleState, error)
|
||
}
|
||
|
||
// SchedulePersistor 定义持久化 ScheduleState 变更的接口。
|
||
// 由 Service 层或 DAO 层实现，注入到 AgentGraphDeps 中。
|
||
// 使用接口而非具体 DAO 类型，避免 model → dao 的循环依赖。
|
||
type SchedulePersistor interface {
|
||
PersistScheduleChanges(ctx context.Context, original, modified *schedule.ScheduleState, userID int) error
|
||
}
|