e06284d0b0
后端： - ♻️ 将 `taskquery` 模块迁移至 `agent2`，并完成与 `agent2` 业务链路及整体结构的正式接入 前端： - 🧱 已完成基础框架搭建，并完成了登录、注册、主页等页面并对接了对应接口；但整体功能实现仍在完善中
103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
package agentmodel
|
||
|
||
import (
|
||
"time"
|
||
|
||
agentshared "github.com/LoveLosita/smartflow/backend/agent2/shared"
|
||
)
|
||
|
||
const (
|
||
// QuickNoteDatetimeMinuteLayout 是随口记链路统一使用的分钟级时间格式。
|
||
QuickNoteDatetimeMinuteLayout = "2006-01-02 15:04"
|
||
// QuickNoteTimezoneName 是随口记时间解析与展示优先使用的时区。
|
||
QuickNoteTimezoneName = "Asia/Shanghai"
|
||
|
||
QuickNotePriorityImportantUrgent = TaskPriorityImportantUrgent
|
||
QuickNotePriorityImportantNotUrgent = TaskPriorityImportantNotUrgent
|
||
QuickNotePrioritySimpleNotImportant = TaskPrioritySimpleNotImportant
|
||
QuickNotePriorityComplexNotImportant = TaskPriorityComplexNotImportant
|
||
)
|
||
|
||
// QuickNoteState 是随口记图在节点间流转的完整状态。
|
||
//
|
||
// 职责边界：
|
||
// 1. 负责保存意图识别、任务提取、工具重试和最终回复所需状态。
|
||
// 2. 不负责图编排，也不直接映射数据库任务实体。
|
||
type QuickNoteState struct {
|
||
TraceID string
|
||
UserID int
|
||
ConversationID string
|
||
RequestNow time.Time
|
||
RequestNowText string
|
||
UserInput string
|
||
|
||
IsQuickNoteIntent bool
|
||
IntentJudgeReason string
|
||
|
||
ExtractedTitle string
|
||
ExtractedDeadline *time.Time
|
||
ExtractedDeadlineText string
|
||
ExtractedUrgencyThreshold *time.Time
|
||
ExtractedPriority int
|
||
ExtractedBanter string
|
||
PlannedBySingleCall bool
|
||
ExtractedPriorityReason string
|
||
DeadlineValidationError string
|
||
|
||
ToolAttemptCount int
|
||
MaxToolRetry int
|
||
LastToolError string
|
||
PersistedTaskID int
|
||
Persisted bool
|
||
AssistantReply string
|
||
}
|
||
|
||
// NewQuickNoteState 负责创建随口记图的初始状态。
|
||
//
|
||
// 输入输出语义：
|
||
// 1. RequestNow 与 RequestNowText 会在创建时同步写入，保证整条链路共用同一时间基准。
|
||
// 2. MaxToolRetry 默认给 3，避免上层未配置时完全失去重试能力。
|
||
func NewQuickNoteState(traceID string, userID int, conversationID, userInput string) *QuickNoteState {
|
||
requestNow := agentshared.NowToMinute()
|
||
return &QuickNoteState{
|
||
TraceID: traceID,
|
||
UserID: userID,
|
||
ConversationID: conversationID,
|
||
RequestNow: requestNow,
|
||
RequestNowText: agentshared.FormatMinute(requestNow),
|
||
UserInput: userInput,
|
||
MaxToolRetry: 3,
|
||
}
|
||
}
|
||
|
||
// CanRetryTool 返回当前是否还允许再次调用持久化工具。
|
||
//
|
||
// 输入输出语义：
|
||
// 1. true 表示“尚未达到最大重试次数”，调用方仍可继续重试。
|
||
// 2. false 表示必须收口，避免无限重试。
|
||
func (s *QuickNoteState) CanRetryTool() bool {
|
||
return s.ToolAttemptCount < s.MaxToolRetry
|
||
}
|
||
|
||
// RecordToolError 记录一次工具失败，并推进重试计数。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只更新与工具失败相关的状态。
|
||
// 2. 不决定是否继续重试，是否重试由节点分支逻辑判断。
|
||
func (s *QuickNoteState) RecordToolError(errMsg string) {
|
||
s.ToolAttemptCount++
|
||
s.LastToolError = errMsg
|
||
}
|
||
|
||
// RecordToolSuccess 记录一次工具成功结果。
|
||
//
|
||
// 输入输出语义：
|
||
// 1. taskID 必须是持久化后的真实任务 ID。
|
||
// 2. 成功后会清空 LastToolError，表示当前链路已进入稳定态。
|
||
func (s *QuickNoteState) RecordToolSuccess(taskID int) {
|
||
s.ToolAttemptCount++
|
||
s.PersistedTaskID = taskID
|
||
s.Persisted = true
|
||
s.LastToolError = ""
|
||
}
|