059b25872a
🧠 agent智能编排：删除了落库相关逻辑。再次重申：agent智能编排旨在为用户预览排程结果，实际的落库由用户决定，并通过按钮触发常规接口进行落库。目前仅保留 ReAct 精排循环链路（待改进）。 📄 修改了 ReAct 智能精排决策文档相关内容。 🔄 undo：当前 agent 智能排程逻辑待改进。
62 lines
2.0 KiB
Go
62 lines
2.0 KiB
Go
package scheduleplan
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"strconv"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
)
|
||
|
||
// SchedulePlanToolDeps 描述"智能排程工具包"需要的外部依赖。
|
||
//
|
||
// 设计目标：
|
||
// 1) 通过函数注入把 agent 包与 service/dao 解耦，避免循环依赖；
|
||
// 2) 每个函数对应一个可独立 mock 的业务能力；
|
||
// 3) 后续可按需扩展（如局部修补、任务类自动生成等）。
|
||
type SchedulePlanToolDeps struct {
|
||
// SmartPlanningRaw 调用粗排算法，同时返回展示结构和已分配的任务项。
|
||
// 返回值：
|
||
// - []UserWeekSchedule：展示型结构，供 SSE 阶段推送给前端预览；
|
||
// - []TaskClassItem：已分配的任务项（EmbeddedTime 已回填），供 ReAct 精排使用。
|
||
SmartPlanningRaw func(ctx context.Context, userID, taskClassID int) ([]model.UserWeekSchedule, []model.TaskClassItem, error)
|
||
|
||
// HybridScheduleWithPlan 构建混合日程（既有日程 + 粗排建议），供 ReAct 精排使用。
|
||
HybridScheduleWithPlan func(ctx context.Context, userID, taskClassID int) ([]model.HybridScheduleEntry, []model.TaskClassItem, error)
|
||
}
|
||
|
||
// validate 校验依赖完整性，缺失任意一个都无法完成排程链路。
|
||
func (d SchedulePlanToolDeps) validate() error {
|
||
if d.SmartPlanningRaw == nil {
|
||
return errors.New("schedule plan tool deps: SmartPlanningRaw is nil")
|
||
}
|
||
if d.HybridScheduleWithPlan == nil {
|
||
return errors.New("schedule plan tool deps: HybridScheduleWithPlan is nil")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// ExtraInt 从 extra map 中安全提取整数值。
|
||
//
|
||
// 兼容策略：
|
||
// 1) JSON 数字默认解析为 float64，做 int 转换；
|
||
// 2) 兼容字符串形式（如 "42"），用 Atoi 解析；
|
||
// 3) 其余类型返回 false，由调用方决定后续处理。
|
||
func ExtraInt(extra map[string]any, key string) (int, bool) {
|
||
v, ok := extra[key]
|
||
if !ok {
|
||
return 0, false
|
||
}
|
||
switch n := v.(type) {
|
||
case float64:
|
||
return int(n), true
|
||
case int:
|
||
return n, true
|
||
case string:
|
||
i, err := strconv.Atoi(n)
|
||
return i, err == nil
|
||
default:
|
||
return 0, false
|
||
}
|
||
}
|