f4ef6fb256
🐛 fix(agent/schedulerefine): 修复复合微调分支链路问题，并将 MinContextSwitch 重构为固定坑位重排语义 - 🔧 修复 `schedulerefine` 复合路由中参数透传不完整、缺少 deterministic objective 时错误降级，以及“复合工具执行成功”与“终审通过”语义混淆的问题 - ✅ 保证新的独立复合分支能够正确执行、正确出站，并统一交由 `hard_check` 裁决最终结果 - 🔍 排查时发现 `MinContextSwitch` 上游 `context_tag` 存在整体退化为 `General` 的风险，影响MinContextSwitch - 🛡️ 为 `MinContextSwitch` 增加兜底策略：当标签整体退化时，按任务名关键词推断学科分组，避免分组能力失效 - ♻️ 将 `MinContextSwitch` 从“整周重新寻找新坑位”调整为“坑位不变，任务顺序改变” - 🎯 将落地方式从顺序 `BatchMove` 改为固定坑位原子重写，避免出现远距离跳位、跨天错迁、异常嵌入课位及循环换位冲突 - 🧹 修复 `hard_check` 在 `MinContextSwitch` 成功后仍执行 `origin_rank` 顺序归位、并导致逆序终审误判的问题 - 🚦 命中该分支后跳过顺序归位与顺序硬校验，避免 `summary` / `hard_check` 将有效重排结果误判为失败 📈 当前连续微调规划涉及的全部功能已可以稳定运行；下一步将继续扩展能力边界，并进一步优化 `schedule_plan` 流程 ♻️ refactor: 重整 agent2 架构，并迁移 quicknote/chat 新链路，目前还剩3个模块未迁移，后续迁移完成后会删除原agent并将此目录命名为agent - 🏗️ 明确 `agent2` 采用“统一分层目录 + 文件分层 + 依赖注入”的重构方案，不再沿用模块目录多层嵌套结构 - 🧩 完善 `agent2` 基础骨架，统一收口 `entrance` / `router` / `llm` / `stream` / `shared` / `model` / `prompt` / `node` / `graph` 等层级职责 - 🚚 将通用路由能力迁移至 `agent2/router`，沉淀统一的 `Action`、`RoutingDecision`、控制码解析，以及 `Dispatcher` / `Resolver` 抽象 - 💬 将普通聊天链路迁移至 `agent2/chat`，复用 `stream` 的 OpenAI 兼容输出协议与 LLM usage 聚合能力 - 📝 将 `quicknote` 链路迁移到 `agent2` 新结构，拆分为 `model` / `prompt` / `llm` / `node` / `graph` 多层实现，替换对旧 `agent/quicknote` 的直接依赖 - 🔌 调整 `agentsvc` 对 `agent2` 的引用，普通聊天、通用分流与 `quicknote` 全部切换到新链路 - ✂️ 去除 graph 内部 `runner` 转接层，改为由 node 层直接持有请求级依赖，并向 graph 暴露节点方法 - 🧹 合并 `graph/quicknote` 与 `graph/quicknote_run`，删除冗余骨架文件，收敛为单一 `quicknote graph` 文件 - 📚 新增 `agent2`《通用能力接入文档》，明确公共能力边界、接入方式以及 graph/node 协作约定 - 📝 更新 `AGENTS.md`，要求后续扩展 `agent2` 通用能力时必须同步维护接入文档 ♻️ refactor: 删除了现Agent目录内Chat模块的两条冗余Prompt
378 lines
12 KiB
Go
378 lines
12 KiB
Go
package schedulerefine
|
||
|
||
import (
|
||
"sort"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
)
|
||
|
||
const (
|
||
// 固定业务时区，避免“今天/明天”在容器默认时区下偏移。
|
||
timezoneName = "Asia/Shanghai"
|
||
// 统一分钟级时间文本格式。
|
||
datetimeLayout = "2006-01-02 15:04"
|
||
|
||
// 预算默认值。
|
||
defaultPlanMax = 2
|
||
defaultExecuteMax = 24
|
||
defaultPerTaskBudget = 4
|
||
defaultReplanMax = 2
|
||
defaultCompositeRetry = 2
|
||
defaultRepairReserve = 1
|
||
)
|
||
|
||
// RefineContract 表示本轮微调意图契约。
|
||
type RefineContract struct {
|
||
Intent string `json:"intent"`
|
||
Strategy string `json:"strategy"`
|
||
HardRequirements []string `json:"hard_requirements"`
|
||
HardAssertions []RefineAssertion `json:"hard_assertions,omitempty"`
|
||
KeepRelativeOrder bool `json:"keep_relative_order"`
|
||
OrderScope string `json:"order_scope"`
|
||
}
|
||
|
||
// RefineAssertion 表示可由后端直接判定的结构化硬断言。
|
||
//
|
||
// 字段说明：
|
||
// 1. Metric：断言指标名，例如 source_move_ratio_percent；
|
||
// 2. Operator：比较操作符，支持 == / <= / >= / between；
|
||
// 3. Value/Min/Max：阈值；
|
||
// 4. Week/TargetWeek：可选周次上下文。
|
||
type RefineAssertion struct {
|
||
Metric string `json:"metric"`
|
||
Operator string `json:"operator"`
|
||
Value int `json:"value,omitempty"`
|
||
Min int `json:"min,omitempty"`
|
||
Max int `json:"max,omitempty"`
|
||
Week int `json:"week,omitempty"`
|
||
TargetWeek int `json:"target_week,omitempty"`
|
||
}
|
||
|
||
// HardCheckReport 表示终审硬校验结果。
|
||
type HardCheckReport struct {
|
||
PhysicsPassed bool `json:"physics_passed"`
|
||
PhysicsIssues []string `json:"physics_issues,omitempty"`
|
||
|
||
IntentPassed bool `json:"intent_passed"`
|
||
IntentReason string `json:"intent_reason,omitempty"`
|
||
IntentUnmet []string `json:"intent_unmet,omitempty"`
|
||
|
||
OrderPassed bool `json:"order_passed"`
|
||
OrderIssues []string `json:"order_issues,omitempty"`
|
||
|
||
RepairTried bool `json:"repair_tried"`
|
||
}
|
||
|
||
// ReactRoundObservation 记录每轮 ReAct 的关键观察。
|
||
type ReactRoundObservation struct {
|
||
Round int `json:"round"`
|
||
GoalCheck string `json:"goal_check,omitempty"`
|
||
Decision string `json:"decision,omitempty"`
|
||
ToolName string `json:"tool_name,omitempty"`
|
||
ToolParams map[string]any `json:"tool_params,omitempty"`
|
||
ToolSuccess bool `json:"tool_success"`
|
||
ToolErrorCode string `json:"tool_error_code,omitempty"`
|
||
ToolResult string `json:"tool_result,omitempty"`
|
||
Reflect string `json:"reflect,omitempty"`
|
||
}
|
||
|
||
// PlannerPlan 表示 Planner 生成的阶段执行计划。
|
||
type PlannerPlan struct {
|
||
Summary string `json:"summary"`
|
||
Steps []string `json:"steps,omitempty"`
|
||
}
|
||
|
||
// RefineSlicePlan 表示切片节点输出。
|
||
type RefineSlicePlan struct {
|
||
WeekFilter []int `json:"week_filter,omitempty"`
|
||
SourceDays []int `json:"source_days,omitempty"`
|
||
TargetDays []int `json:"target_days,omitempty"`
|
||
ExcludeSections []int `json:"exclude_sections,omitempty"`
|
||
Reason string `json:"reason,omitempty"`
|
||
}
|
||
|
||
// RefineObjective 表示“可执行且可校验”的目标约束。
|
||
//
|
||
// 设计说明：
|
||
// 1. 由 contract/slice 从自然语言编译得到；
|
||
// 2. 执行阶段（done 收口）与终审阶段（hard_check）共用同一份约束；
|
||
// 3. 避免“执行逻辑与终审逻辑各说各话”。
|
||
type RefineObjective struct {
|
||
Mode string `json:"mode,omitempty"` // none | move_all | move_ratio
|
||
|
||
SourceWeeks []int `json:"source_weeks,omitempty"`
|
||
TargetWeeks []int `json:"target_weeks,omitempty"`
|
||
SourceDays []int `json:"source_days,omitempty"`
|
||
TargetDays []int `json:"target_days,omitempty"`
|
||
|
||
ExcludeSections []int `json:"exclude_sections,omitempty"`
|
||
|
||
BaselineSourceTaskCount int `json:"baseline_source_task_count,omitempty"`
|
||
RequiredMoveMin int `json:"required_move_min,omitempty"`
|
||
RequiredMoveMax int `json:"required_move_max,omitempty"`
|
||
|
||
Reason string `json:"reason,omitempty"`
|
||
}
|
||
|
||
// ScheduleRefineState 是连续微调图的统一状态。
|
||
type ScheduleRefineState struct {
|
||
// 1) 请求上下文
|
||
TraceID string
|
||
UserID int
|
||
ConversationID string
|
||
UserMessage string
|
||
RequestNow time.Time
|
||
RequestNowText string
|
||
|
||
// 2) 继承自预览快照的数据
|
||
TaskClassIDs []int
|
||
Constraints []string
|
||
// InitialHybridEntries 保存本轮微调开始前的基线，用于终审做“前后对比”。
|
||
// 说明：
|
||
// 1. 只读语义，不参与执行期改写；
|
||
// 2. 终审可基于它判断“来源任务是否真正迁移到目标区域”。
|
||
InitialHybridEntries []model.HybridScheduleEntry
|
||
HybridEntries []model.HybridScheduleEntry
|
||
AllocatedItems []model.TaskClassItem
|
||
CandidatePlans []model.UserWeekSchedule
|
||
|
||
// 3) 本轮执行状态
|
||
UserIntent string
|
||
Contract RefineContract
|
||
|
||
PlanMax int
|
||
PerTaskBudget int
|
||
ExecuteMax int
|
||
ReplanMax int
|
||
// CompositeRetryMax 表示复合路由失败后的最大重试次数（不含首次尝试）。
|
||
CompositeRetryMax int
|
||
|
||
PlanUsed int
|
||
ReplanUsed int
|
||
|
||
MaxRounds int
|
||
RepairReserve int
|
||
RoundUsed int
|
||
ActionLogs []string
|
||
|
||
ConsecutiveFailures int
|
||
ThinkingBoostArmed bool
|
||
ObservationHistory []ReactRoundObservation
|
||
|
||
CurrentPlan PlannerPlan
|
||
BatchMoveAllowed bool
|
||
// DisableCompositeTools=true 表示已进入 ReAct 兜底，禁止再调用复合工具。
|
||
DisableCompositeTools bool
|
||
// CompositeRouteTried 标记是否尝试过“复合批处理路由”。
|
||
CompositeRouteTried bool
|
||
// CompositeRouteSucceeded 标记复合批处理路由是否已完成“复合分支出站”。
|
||
//
|
||
// 说明：
|
||
// 1. true 表示当前链路可以跳过 ReAct 兜底，直接进入 hard_check；
|
||
// 2. 它不等价于“终审已通过”，终审是否通过仍以后续 HardCheck 结果为准；
|
||
// 3. 这样区分是为了避免“复合工具已成功执行，但业务目标要等终审裁决”时被误判为失败。
|
||
CompositeRouteSucceeded bool
|
||
TaskActionUsed map[int]int
|
||
EntriesVersion int
|
||
SeenSlotQueries map[string]struct{}
|
||
|
||
// RequiredCompositeTool 表示本轮策略要求“必须至少成功一次”的复合工具。
|
||
// 取值约定："" | "SpreadEven" | "MinContextSwitch"。
|
||
RequiredCompositeTool string
|
||
// CompositeToolCalled 记录复合工具是否至少调用过一次（不区分成功失败）。
|
||
CompositeToolCalled map[string]bool
|
||
// CompositeToolSuccess 记录复合工具是否至少成功过一次。
|
||
CompositeToolSuccess map[string]bool
|
||
|
||
SlicePlan RefineSlicePlan
|
||
Objective RefineObjective
|
||
WorksetTaskIDs []int
|
||
WorksetCursor int
|
||
CurrentTaskID int
|
||
CurrentTaskAttempt int
|
||
|
||
LastFailedCallSignature string
|
||
OriginOrderMap map[int]int
|
||
|
||
// 4) 终审状态
|
||
HardCheck HardCheckReport
|
||
|
||
// 5) 最终输出
|
||
FinalSummary string
|
||
Completed bool
|
||
}
|
||
|
||
// NewScheduleRefineState 基于上一版预览快照初始化状态。
|
||
//
|
||
// 职责边界：
|
||
// 1. 负责初始化预算、上下文字段与可变状态容器；
|
||
// 2. 负责拷贝 preview 数据，避免跨请求引用污染；
|
||
// 3. 不负责做任何调度动作。
|
||
func NewScheduleRefineState(traceID string, userID int, conversationID string, userMessage string, preview *model.SchedulePlanPreviewCache) *ScheduleRefineState {
|
||
now := nowToMinute()
|
||
st := &ScheduleRefineState{
|
||
TraceID: strings.TrimSpace(traceID),
|
||
UserID: userID,
|
||
ConversationID: strings.TrimSpace(conversationID),
|
||
UserMessage: strings.TrimSpace(userMessage),
|
||
RequestNow: now,
|
||
RequestNowText: now.In(loadLocation()).Format(datetimeLayout),
|
||
PlanMax: defaultPlanMax,
|
||
PerTaskBudget: defaultPerTaskBudget,
|
||
ExecuteMax: defaultExecuteMax,
|
||
ReplanMax: defaultReplanMax,
|
||
CompositeRetryMax: defaultCompositeRetry,
|
||
RepairReserve: defaultRepairReserve,
|
||
MaxRounds: defaultExecuteMax + defaultRepairReserve,
|
||
ActionLogs: make([]string, 0, 32),
|
||
ObservationHistory: make([]ReactRoundObservation, 0, 24),
|
||
TaskActionUsed: make(map[int]int),
|
||
SeenSlotQueries: make(map[string]struct{}),
|
||
OriginOrderMap: make(map[int]int),
|
||
CompositeToolCalled: map[string]bool{
|
||
"SpreadEven": false,
|
||
"MinContextSwitch": false,
|
||
},
|
||
CompositeToolSuccess: map[string]bool{
|
||
"SpreadEven": false,
|
||
"MinContextSwitch": false,
|
||
},
|
||
CurrentPlan: PlannerPlan{
|
||
Summary: "初始化完成，等待 Planner 生成执行计划。",
|
||
},
|
||
SlicePlan: RefineSlicePlan{
|
||
Reason: "尚未切片",
|
||
},
|
||
}
|
||
if preview == nil {
|
||
return st
|
||
}
|
||
|
||
st.TaskClassIDs = append([]int(nil), preview.TaskClassIDs...)
|
||
st.InitialHybridEntries = cloneHybridEntries(preview.HybridEntries)
|
||
st.HybridEntries = cloneHybridEntries(preview.HybridEntries)
|
||
st.AllocatedItems = cloneTaskClassItems(preview.AllocatedItems)
|
||
st.CandidatePlans = cloneWeekSchedules(preview.CandidatePlans)
|
||
st.OriginOrderMap = buildOriginOrderMap(st.HybridEntries)
|
||
return st
|
||
}
|
||
|
||
func loadLocation() *time.Location {
|
||
loc, err := time.LoadLocation(timezoneName)
|
||
if err != nil {
|
||
return time.Local
|
||
}
|
||
return loc
|
||
}
|
||
|
||
func nowToMinute() time.Time {
|
||
return time.Now().In(loadLocation()).Truncate(time.Minute)
|
||
}
|
||
|
||
func cloneHybridEntries(src []model.HybridScheduleEntry) []model.HybridScheduleEntry {
|
||
if len(src) == 0 {
|
||
return nil
|
||
}
|
||
dst := make([]model.HybridScheduleEntry, len(src))
|
||
copy(dst, src)
|
||
return dst
|
||
}
|
||
|
||
func cloneTaskClassItems(src []model.TaskClassItem) []model.TaskClassItem {
|
||
if len(src) == 0 {
|
||
return nil
|
||
}
|
||
dst := make([]model.TaskClassItem, 0, len(src))
|
||
for _, item := range src {
|
||
copied := item
|
||
if item.CategoryID != nil {
|
||
v := *item.CategoryID
|
||
copied.CategoryID = &v
|
||
}
|
||
if item.Order != nil {
|
||
v := *item.Order
|
||
copied.Order = &v
|
||
}
|
||
if item.Content != nil {
|
||
v := *item.Content
|
||
copied.Content = &v
|
||
}
|
||
if item.Status != nil {
|
||
v := *item.Status
|
||
copied.Status = &v
|
||
}
|
||
if item.EmbeddedTime != nil {
|
||
t := *item.EmbeddedTime
|
||
copied.EmbeddedTime = &t
|
||
}
|
||
dst = append(dst, copied)
|
||
}
|
||
return dst
|
||
}
|
||
|
||
func cloneWeekSchedules(src []model.UserWeekSchedule) []model.UserWeekSchedule {
|
||
if len(src) == 0 {
|
||
return nil
|
||
}
|
||
dst := make([]model.UserWeekSchedule, 0, len(src))
|
||
for _, week := range src {
|
||
eventsCopy := make([]model.WeeklyEventBrief, len(week.Events))
|
||
copy(eventsCopy, week.Events)
|
||
dst = append(dst, model.UserWeekSchedule{
|
||
Week: week.Week,
|
||
Events: eventsCopy,
|
||
})
|
||
}
|
||
return dst
|
||
}
|
||
|
||
// buildOriginOrderMap 构建 suggested 任务的初始顺序基线（task_item_id -> rank）。
|
||
func buildOriginOrderMap(entries []model.HybridScheduleEntry) map[int]int {
|
||
orderMap := make(map[int]int)
|
||
if len(entries) == 0 {
|
||
return orderMap
|
||
}
|
||
suggested := make([]model.HybridScheduleEntry, 0, len(entries))
|
||
for _, entry := range entries {
|
||
if isMovableSuggestedTask(entry) {
|
||
suggested = append(suggested, entry)
|
||
}
|
||
}
|
||
sort.SliceStable(suggested, func(i, j int) bool {
|
||
left := suggested[i]
|
||
right := suggested[j]
|
||
if left.Week != right.Week {
|
||
return left.Week < right.Week
|
||
}
|
||
if left.DayOfWeek != right.DayOfWeek {
|
||
return left.DayOfWeek < right.DayOfWeek
|
||
}
|
||
if left.SectionFrom != right.SectionFrom {
|
||
return left.SectionFrom < right.SectionFrom
|
||
}
|
||
if left.SectionTo != right.SectionTo {
|
||
return left.SectionTo < right.SectionTo
|
||
}
|
||
return left.TaskItemID < right.TaskItemID
|
||
})
|
||
for i, entry := range suggested {
|
||
orderMap[entry.TaskItemID] = i + 1
|
||
}
|
||
return orderMap
|
||
}
|
||
|
||
// FinalHardCheckPassed 判断“最终终审”是否整体通过。
|
||
//
|
||
// 职责边界：
|
||
// 1. 负责聚合 physics/order/intent 三类硬校验结果，给服务层与总结阶段统一复用；
|
||
// 2. 不负责触发终审，也不负责推导修复动作；
|
||
// 3. nil state 视为未通过，避免上层把缺失结果误判为成功。
|
||
func FinalHardCheckPassed(st *ScheduleRefineState) bool {
|
||
if st == nil {
|
||
return false
|
||
}
|
||
return st.HardCheck.PhysicsPassed && st.HardCheck.OrderPassed && st.HardCheck.IntentPassed
|
||
}
|