模块介绍 / 今日规划 / 下班日报 / 团队摘要 生成成功后没有上报,也几乎不写「AI」日志。
流结束时的 usage 用了非阻塞发送,通道一满就被丢掉;很多服务商默认还不在流里带 usage。 现在每次真正打到模型的调用都会: 在本地日志(分类 AI,消息 AI 调用完成)写下 scene / prompt / completion / cached 登录后同步上报到后台日汇总
This commit is contained in:
158
ai_tasks.go
158
ai_tasks.go
@@ -45,8 +45,12 @@ func (a *App) AIGenerateTasks(kind, text string) ([]AITaskDraft, error) {
|
||||
if e := a.checkAIPolicy(); e != nil {
|
||||
return nil, e
|
||||
}
|
||||
locale := "zh-CN"
|
||||
if st, se := a.store.Settings(); se == nil && st.Locale != "" {
|
||||
locale = st.Locale
|
||||
}
|
||||
msgs := []ai.Message{
|
||||
{Role: "system", Content: aiTaskPrompt(kind)},
|
||||
{Role: "system", Content: aiTaskPrompt(kind, locale)},
|
||||
{Role: "user", Content: text},
|
||||
}
|
||||
ctx, cancel := context.WithTimeout(a.ctx, 2*time.Minute)
|
||||
@@ -68,9 +72,7 @@ func (a *App) AIGenerateTasks(kind, text string) ([]AITaskDraft, error) {
|
||||
}
|
||||
sb.WriteString(chunk.Content)
|
||||
}
|
||||
if usage != nil {
|
||||
go a.reportAIUsage(provider.Name(), usage.PromptTokens, usage.CompletionTokens, usage.Estimated)
|
||||
}
|
||||
a.recordAIUsage(provider.Name(), "generate:"+kind, usage, msgs, sb.String())
|
||||
drafts, pe := parseAITaskDrafts(kind, sb.String())
|
||||
if pe != nil {
|
||||
a.store.Log("warning", "AI", "AI 任务生成解析失败", sb.String())
|
||||
@@ -80,20 +82,90 @@ func (a *App) AIGenerateTasks(kind, text string) ([]AITaskDraft, error) {
|
||||
return drafts, nil
|
||||
}
|
||||
|
||||
func aiTaskPrompt(kind string) string {
|
||||
now := time.Now()
|
||||
weekdays := []string{"日", "一", "二", "三", "四", "五", "六"}
|
||||
head := fmt.Sprintf("今天是 %s(星期%s),当前时间 %s。\n", now.Format("2006-01-02"), weekdays[int(now.Weekday())], now.Format("15:04"))
|
||||
func aiTaskPrompt(kind, locale string) string {
|
||||
head := aiCalendarContext(locale)
|
||||
if kind == "ticket" {
|
||||
return head + `你是研发工单拆解助手。把用户的一句话拆成 1~10 条工单。
|
||||
return head + `你是研发工单拆解助手。把用户输入拆成 1~10 条工单。
|
||||
只输出 JSON 数组,禁止任何解释、markdown 围栏或多余文字。每个元素:
|
||||
{"title":"简短标题(<=40字)","description":"补充细节,可为空","type":"feature|bug|task|improvement","priority":"low|medium|high","startAt":"YYYY-MM-DD","dueAt":"YYYY-MM-DD"}
|
||||
规则:标题用与用户输入相同的语言;startAt 默认今天;dueAt 不得早于 startAt,用户未提及工期时按任务量合理估算(1~7 天);能从用户话中推断出的日期(如“周五前”“下周”)必须转换为具体日期。`
|
||||
日期规则(必须遵守):
|
||||
1. 按用户输入的语言理解相对日期(中文:今天/明天/后天/大后天/本周X/下周X/这周五/下周一/月底/下个月;英文:today/tomorrow/this Friday/next Monday/end of week 等),对照上面日历换成具体 YYYY-MM-DD。
|
||||
2. 用户说的不一定是今天:若整段话指向某一天或某一周,所有条目的 startAt/dueAt 都落在那个时间范围内,不要一律填今天。
|
||||
3. 一句话里提到多天(如“周三做A,周五做B”“这周每天…”),每条工单各自写对应的 startAt 和 dueAt,不要全部同一天。
|
||||
4. 每条都必须写 dueAt。用户给了截止就用;只给了开始日则按工作量估 1~5 天;完全没提日期则 startAt=今天、dueAt 按工作量估。
|
||||
5. dueAt 不得早于 startAt。标题语言与用户输入一致。
|
||||
6. 用户提到节假日/黄金周/放假/调休(国庆、十一、春节、过年、中秋、端午、元旦、五一、清明、圣诞、感恩节等),必须对照上面节假日表换成具体 YYYY-MM-DD,禁止自己换算农历。「X前」= 节日前一天;「X期间」= 表中放假区间(没有则用当天);「X后」= 假期结束后一天。`
|
||||
}
|
||||
return head + `你是待办事项拆解助手。把用户的一句话拆成 1~10 条待办。
|
||||
return head + `你是待办事项拆解助手。把用户输入拆成 1~10 条待办。
|
||||
只输出 JSON 数组,禁止任何解释、markdown 围栏或多余文字。每个元素:
|
||||
{"title":"简短标题(<=40字)","content":"补充细节,可为空","priority":"low|medium|high","dueAt":"YYYY-MM-DDTHH:MM 或空字符串"}
|
||||
规则:标题用与用户输入相同的语言;只有用户明确或可合理推断截止时间时才填 dueAt(如“明天下午”→ 明天 18:00),否则留空字符串。`
|
||||
{"title":"简短标题(<=40字)","content":"补充细节,可为空","priority":"low|medium|high","dueAt":"YYYY-MM-DDTHH:MM"}
|
||||
日期规则(必须遵守):
|
||||
1. 按用户输入的语言理解相对日期和时间(中文:今天/明天/后天/本周X/下周X/上午/下午/晚上/点钟;英文:today/tomorrow/this Friday/next week/afternoon 等),对照上面日历换成具体 YYYY-MM-DDTHH:MM。
|
||||
2. 用户说的不一定是今天:若整段话指向某一天或某一周,所有条目的 dueAt 都落在那个时间,不要一律填今天。
|
||||
3. 一句话里提到多天(如“周一买菜,周三交报告”),每条待办各自写对应的 dueAt。
|
||||
4. 每条都必须写 dueAt,不要留空。用户给了时刻就用;只给了日期则:上午→10:00、下午→18:00、晚上→21:00、未提时段→18:00。完全没提日期则按紧急度估:高优今天 18:00,中优明天 18:00,低优 3 天后 18:00。
|
||||
5. 标题语言与用户输入一致。
|
||||
6. 用户提到节假日/黄金周/放假(国庆、十一、春节、过年、中秋、端午、元旦、五一、清明、圣诞、感恩节等),必须对照上面节假日表换成具体日期,禁止自己换算农历。「X前」= 节日前一天 18:00;「X期间」落在放假区间;「X后」从假期结束后一天起。`
|
||||
}
|
||||
|
||||
// aiCalendarContext 给模型一份「今天 + 本周/下周对照表」,避免相对日期猜错,也不默认全是今天。
|
||||
func aiCalendarContext(locale string) string {
|
||||
now := time.Now()
|
||||
en := strings.HasPrefix(strings.ToLower(locale), "en")
|
||||
wdCN := []string{"日", "一", "二", "三", "四", "五", "六"}
|
||||
wdEN := []string{"Sunday", "Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday"}
|
||||
weekday := func(t time.Time) string {
|
||||
if en {
|
||||
return wdEN[int(t.Weekday())]
|
||||
}
|
||||
return "星期" + wdCN[int(t.Weekday())]
|
||||
}
|
||||
// 本周一:Go Weekday 周日=0
|
||||
offset := int(now.Weekday())
|
||||
if offset == 0 {
|
||||
offset = 7
|
||||
}
|
||||
monday := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location()).AddDate(0, 0, 1-offset)
|
||||
var b strings.Builder
|
||||
if en {
|
||||
fmt.Fprintf(&b, "Today is %s (%s), current time %s. Timezone is the user's local time.\n", now.Format("2006-01-02"), weekday(now), now.Format("15:04"))
|
||||
b.WriteString("This week:\n")
|
||||
} else {
|
||||
fmt.Fprintf(&b, "今天是 %s(%s),当前时间 %s。时区按用户本机。\n", now.Format("2006-01-02"), weekday(now), now.Format("15:04"))
|
||||
b.WriteString("本周对照:\n")
|
||||
}
|
||||
for i := 0; i < 7; i++ {
|
||||
d := monday.AddDate(0, 0, i)
|
||||
mark := ""
|
||||
if d.Format("2006-01-02") == now.Format("2006-01-02") {
|
||||
if en {
|
||||
mark = " ← today"
|
||||
} else {
|
||||
mark = " ← 今天"
|
||||
}
|
||||
}
|
||||
fmt.Fprintf(&b, "- %s %s%s\n", weekday(d), d.Format("2006-01-02"), mark)
|
||||
}
|
||||
if en {
|
||||
b.WriteString("Next week:\n")
|
||||
} else {
|
||||
b.WriteString("下周对照:\n")
|
||||
}
|
||||
for i := 0; i < 7; i++ {
|
||||
d := monday.AddDate(0, 0, 7+i)
|
||||
fmt.Fprintf(&b, "- %s %s\n", weekday(d), d.Format("2006-01-02"))
|
||||
}
|
||||
monthEnd := time.Date(now.Year(), now.Month()+1, 0, 0, 0, 0, 0, now.Location())
|
||||
nextMonth := time.Date(now.Year(), now.Month()+1, 1, 0, 0, 0, 0, now.Location())
|
||||
if en {
|
||||
fmt.Fprintf(&b, "End of this month: %s. First day of next month: %s.\n", monthEnd.Format("2006-01-02"), nextMonth.Format("2006-01-02"))
|
||||
b.WriteString("Convert relative phrases using this calendar. Do not default every item to today unless the user said today or gave no date at all.\n")
|
||||
} else {
|
||||
fmt.Fprintf(&b, "本月最后一天:%s。下月 1 日:%s。\n", monthEnd.Format("2006-01-02"), nextMonth.Format("2006-01-02"))
|
||||
b.WriteString("相对日期必须对照上面日历换算。用户没说今天时,不要把每条都填成今天。\n")
|
||||
}
|
||||
b.WriteString(formatHolidayBlock(now, en))
|
||||
return b.String()
|
||||
}
|
||||
|
||||
// parseAITaskDrafts 从模型输出中提取 JSON 数组并规范化字段。
|
||||
@@ -128,11 +200,21 @@ func parseAITaskDrafts(kind, raw string) ([]AITaskDraft, error) {
|
||||
d.Type = "task"
|
||||
}
|
||||
d.Content = ""
|
||||
if !validDateStr(d.StartAt) {
|
||||
// 优先保留模型按用户语义写出的日期;缺了才补,避免把「下周三」整批改成今天。
|
||||
d.StartAt = dateOnly(d.StartAt)
|
||||
d.DueAt = dateOnly(d.DueAt)
|
||||
if d.StartAt == "" && d.DueAt != "" {
|
||||
d.StartAt = d.DueAt
|
||||
}
|
||||
if d.StartAt == "" {
|
||||
d.StartAt = today
|
||||
}
|
||||
if !validDateStr(d.DueAt) {
|
||||
d.DueAt = time.Now().AddDate(0, 0, 3).Format("2006-01-02")
|
||||
if d.DueAt == "" {
|
||||
if t, e := time.Parse("2006-01-02", d.StartAt); e == nil {
|
||||
d.DueAt = t.AddDate(0, 0, 2).Format("2006-01-02")
|
||||
} else {
|
||||
d.DueAt = time.Now().AddDate(0, 0, 2).Format("2006-01-02")
|
||||
}
|
||||
}
|
||||
if d.DueAt < d.StartAt {
|
||||
d.DueAt = d.StartAt
|
||||
@@ -140,6 +222,16 @@ func parseAITaskDrafts(kind, raw string) ([]AITaskDraft, error) {
|
||||
} else {
|
||||
d.Type, d.Description, d.StartAt = "", "", ""
|
||||
d.DueAt = normalizeTodoDue(d.DueAt)
|
||||
if d.DueAt == "" {
|
||||
// 待办也必须有截止日期,缺省按优先级估。
|
||||
days := 1
|
||||
if d.Priority == "high" {
|
||||
days = 0
|
||||
} else if d.Priority == "low" {
|
||||
days = 3
|
||||
}
|
||||
d.DueAt = time.Now().AddDate(0, 0, days).Format("2006-01-02") + "T18:00"
|
||||
}
|
||||
}
|
||||
out = append(out, d)
|
||||
if len(out) >= 20 {
|
||||
@@ -153,21 +245,45 @@ func parseAITaskDrafts(kind, raw string) ([]AITaskDraft, error) {
|
||||
}
|
||||
|
||||
func validDateStr(s string) bool {
|
||||
_, e := time.Parse("2006-01-02", strings.TrimSpace(s))
|
||||
return e == nil
|
||||
return dateOnly(s) != ""
|
||||
}
|
||||
|
||||
// normalizeTodoDue 接受 YYYY-MM-DDTHH:MM 或纯日期(补 18:00),其余返回空。
|
||||
// dateOnly 从 YYYY-MM-DD 或带时间的写法里抽出日期;解析失败返回空。
|
||||
func dateOnly(s string) string {
|
||||
s = strings.TrimSpace(s)
|
||||
if s == "" {
|
||||
return ""
|
||||
}
|
||||
if i := strings.IndexAny(s, "T "); i > 0 {
|
||||
s = s[:i]
|
||||
}
|
||||
if _, e := time.Parse("2006-01-02", s); e == nil {
|
||||
return s
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
// normalizeTodoDue 接受常见日期/时间写法,统一成 datetime-local 用的 YYYY-MM-DDTHH:MM。
|
||||
func normalizeTodoDue(s string) string {
|
||||
s = strings.TrimSpace(s)
|
||||
if s == "" {
|
||||
return ""
|
||||
}
|
||||
if _, e := time.Parse("2006-01-02T15:04", s); e == nil {
|
||||
return s
|
||||
s = strings.ReplaceAll(s, " ", "T")
|
||||
for _, layout := range []string{
|
||||
"2006-01-02T15:04",
|
||||
"2006-01-02T15:04:05",
|
||||
time.RFC3339,
|
||||
} {
|
||||
if t, e := time.Parse(layout, s); e == nil {
|
||||
return t.Format("2006-01-02T15:04")
|
||||
}
|
||||
}
|
||||
if validDateStr(s) {
|
||||
return s + "T18:00"
|
||||
}
|
||||
if i := strings.Index(s, "T"); i == 10 && validDateStr(s[:10]) {
|
||||
return s[:10] + "T18:00"
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user