96be3e2a02
✨ feat(task,agent): 新增任务完成接口，并打通聊天全链路 Token 记账 - ✅ 新增“标记任务为完成”接口，并补充幂等保护，避免重复完成导致状态污染 - 📊 为聊天链路补充 Token 统计能力： - 流式主对话链路直接读取模型 `usage` - Agent 链路通过 `Eino callback + ctx` 聚合 `Generate usage` - 在流式场景下补齐缺失的 `usage` 数据 - 🧾 按口径 B 完成 Token 落库： - 用户消息 `token` 记为 `0` - 助手消息记录本轮总 `token` - 持久化时同步更新 `chat_histories.tokens_consumed`、`agent_chats.tokens_total`、`users.token_usage` - 🔄 异步标题生成产生的 Token 通过 Outbox 事件完成账本增量调整，保证统计口径一致 - 📝 同步更新 `AGENTS.md` 与 `.gitignore` - 📚 小幅更新 README 说明文档
105 lines
4.0 KiB
Go
105 lines
4.0 KiB
Go
package model
|
||
|
||
import "time"
|
||
|
||
// Task 是任务表的领域模型。
|
||
//
|
||
// 职责边界：
|
||
// 1. 负责映射 tasks 表字段；
|
||
// 2. 不负责接口入参校验和业务规则判断；
|
||
// 3. 不负责“自动平移”执行（自动平移由 Service + Outbox 事件链路负责）。
|
||
type Task struct {
|
||
// 1. 主键。
|
||
ID int `gorm:"primaryKey;autoIncrement"`
|
||
// 2. 归属用户 ID。
|
||
// 2.1 单列索引用于常规按用户查任务；
|
||
// 2.2 同时参与“懒触发平移”复合索引的最左前缀。
|
||
UserID int `gorm:"column:user_id;index;index:idx_user_done_threshold_priority,priority:1"`
|
||
// 3. 任务标题。
|
||
Title string `gorm:"type:varchar(255)"`
|
||
// 4. 四象限优先级：
|
||
// 4.1 1=重要且紧急；
|
||
// 4.2 2=重要不紧急；
|
||
// 4.3 3=简单不重要；
|
||
// 4.4 4=不简单不重要。
|
||
//
|
||
// 说明：该字段参与“懒触发平移”复合索引。
|
||
Priority int `gorm:"not null;index:idx_user_done_threshold_priority,priority:4"`
|
||
// 5. 完成状态。
|
||
//
|
||
// 说明：已完成任务不参与自动平移；该字段参与复合索引。
|
||
IsCompleted bool `gorm:"column:is_completed;default:false;index:idx_user_done_threshold_priority,priority:2"`
|
||
// 6. 任务业务截止时间。
|
||
DeadlineAt *time.Time `gorm:"column:deadline_at"`
|
||
// 7. 紧急分界时间（自动平移阈值）。
|
||
//
|
||
// 规则：
|
||
// 7.1 到达该时间后，任务可从“不紧急象限”自动平移到“紧急象限”；
|
||
// 7.2 该值由上游（例如 LLM 规划）给出，不在模型层做推断；
|
||
// 7.3 为空表示该任务不参与自动平移；
|
||
// 7.4 该字段参与“懒触发平移”复合索引。
|
||
UrgencyThresholdAt *time.Time `gorm:"column:urgency_threshold_at;index:idx_user_done_threshold_priority,priority:3"`
|
||
}
|
||
|
||
type UserAddTaskResponse struct {
|
||
ID int `json:"id"`
|
||
Title string `json:"title"`
|
||
PriorityGroup int `json:"priority_group"`
|
||
DeadlineAt *time.Time `json:"deadline_at"`
|
||
Status string `json:"status"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
}
|
||
|
||
type UserAddTaskRequest struct {
|
||
Title string `json:"title"`
|
||
PriorityGroup int `json:"priority_group"`
|
||
DeadlineAt *time.Time `json:"deadline_at"`
|
||
}
|
||
|
||
// UserCompleteTaskRequest 是“标记任务完成”接口的请求体。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只承载目标任务 ID；
|
||
// 2. 不承载 user_id（user_id 一律由鉴权中间件注入，避免越权）。
|
||
type UserCompleteTaskRequest struct {
|
||
TaskID int `json:"task_id"`
|
||
}
|
||
|
||
// UserCompleteTaskResponse 是“标记任务完成”接口的响应体。
|
||
//
|
||
// 字段语义：
|
||
// 1. TaskID：本次操作的目标任务；
|
||
// 2. IsCompleted：操作后的完成状态（成功时恒为 true）；
|
||
// 3. AlreadyCompleted：
|
||
// 3.1 true：任务原本就已完成，本次请求命中幂等语义；
|
||
// 3.2 false：任务由未完成切换为完成；
|
||
// 4. Status：给前端的简短状态文案。
|
||
type UserCompleteTaskResponse struct {
|
||
TaskID int `json:"task_id"`
|
||
IsCompleted bool `json:"is_completed"`
|
||
AlreadyCompleted bool `json:"already_completed"`
|
||
Status string `json:"status"`
|
||
}
|
||
|
||
type GetUserTaskResp struct {
|
||
ID int `json:"id"`
|
||
UserID int `json:"user_id"`
|
||
Title string `json:"title"`
|
||
PriorityGroup int `json:"priority_group"`
|
||
Status string `json:"status"`
|
||
Deadline string `json:"deadline"`
|
||
IsCompleted bool `json:"is_completed"`
|
||
}
|
||
|
||
// TaskUrgencyPromoteRequestedPayload 是“任务紧急性平移请求”事件载荷。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只承载“哪个用户的哪些任务需要尝试平移”；
|
||
// 2. 不包含 outbox/kafka 协议字段（这些由基础设施层统一封装）；
|
||
// 3. TriggeredAt 只用于追踪触发时间，最终是否更新仍以消费时数据库条件为准。
|
||
type TaskUrgencyPromoteRequestedPayload struct {
|
||
UserID int `json:"user_id"`
|
||
TaskIDs []int `json:"task_ids"`
|
||
TriggeredAt time.Time `json:"triggered_at"`
|
||
}
|