修改工具调用机制

2026-06-10 18:54:17 +08:00
parent a838a812a0
commit 04485b6b0e
6 changed files with 589 additions and 78 deletions
@@ -9,6 +9,25 @@ import (

 type TimeTool struct{}

+type TimeRangeArgs struct {
+	Range     string `json:"range"`
+	Timezone  string `json:"timezone,omitempty"`
+	StartDate string `json:"start_date,omitempty"`
+	EndDate   string `json:"end_date,omitempty"`
+}
+
+type TimeRangeResult struct {
+	Ok           bool   `json:"ok"`
+	Now          string `json:"now"`
+	NowWeekday   string `json:"now_weekday"`
+	StartDate    string `json:"start_date"`
+	StartWeekday string `json:"start_weekday"`
+	EndDate      string `json:"end_date"`
+	EndWeekday   string `json:"end_weekday"`
+	Label        string `json:"label"`
+	Timezone     string `json:"timezone"`
+}
+
 func init() {
 	Register(TimeTool{})
 }
@@ -56,6 +75,101 @@ func (TimeTool) Enrich(ctx context.Context, messages []ChatMessage, config ToolC
 	return enriched, nil
 }

+func ResolveTimeRange(args TimeRangeArgs, now time.Time) (TimeRangeResult, error) {
+	loc := now.Location()
+	if args.Timezone != "" {
+		if loaded, err := time.LoadLocation(args.Timezone); err == nil {
+			loc = loaded
+			now = now.In(loc)
+		}
+	}
+
+	todayStart := dateStart(now)
+	weekStart := todayStart.AddDate(0, 0, -int((int(todayStart.Weekday())+6)%7))
+	monthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, loc)
+	yearStart := time.Date(now.Year(), 1, 1, 0, 0, 0, 0, loc)
+
+	rangeName := strings.TrimSpace(strings.ToLower(args.Range))
+	start := todayStart
+	end := todayStart
+	label := "今天"
+
+	switch rangeName {
+	case "", "today":
+		label = "今天"
+		start = todayStart
+		end = todayStart
+	case "yesterday":
+		label = "昨天"
+		start = todayStart.AddDate(0, 0, -1)
+		end = start
+	case "tomorrow":
+		label = "明天"
+		start = todayStart.AddDate(0, 0, 1)
+		end = start
+	case "this_week":
+		label = "本周"
+		start = weekStart
+		end = weekStart.AddDate(0, 0, 6)
+	case "last_week":
+		label = "上周"
+		start = weekStart.AddDate(0, 0, -7)
+		end = weekStart.AddDate(0, 0, -1)
+	case "next_week":
+		label = "下周"
+		start = weekStart.AddDate(0, 0, 7)
+		end = weekStart.AddDate(0, 0, 13)
+	case "this_month":
+		label = "本月"
+		start = monthStart
+		end = monthStart.AddDate(0, 1, -1)
+	case "last_month":
+		label = "上月"
+		start = monthStart.AddDate(0, -1, 0)
+		end = monthStart.AddDate(0, 0, -1)
+	case "next_month":
+		label = "下月"
+		start = monthStart.AddDate(0, 1, 0)
+		end = monthStart.AddDate(0, 2, -1)
+	case "this_year":
+		label = "今年"
+		start = yearStart
+		end = yearStart.AddDate(1, 0, -1)
+	case "custom":
+		if args.StartDate == "" || args.EndDate == "" {
+			return TimeRangeResult{Ok: false, Now: now.Format("2006-01-02 15:04:05"), Timezone: loc.String()}, fmt.Errorf("custom range requires start_date and end_date")
+		}
+		parsedStart, err := time.ParseInLocation("2006-01-02", args.StartDate, loc)
+		if err != nil {
+			return TimeRangeResult{Ok: false, Now: now.Format("2006-01-02 15:04:05"), Timezone: loc.String()}, fmt.Errorf("invalid start_date: %w", err)
+		}
+		parsedEnd, err := time.ParseInLocation("2006-01-02", args.EndDate, loc)
+		if err != nil {
+			return TimeRangeResult{Ok: false, Now: now.Format("2006-01-02 15:04:05"), Timezone: loc.String()}, fmt.Errorf("invalid end_date: %w", err)
+		}
+		if parsedEnd.Before(parsedStart) {
+			return TimeRangeResult{Ok: false, Now: now.Format("2006-01-02 15:04:05"), Timezone: loc.String()}, fmt.Errorf("end_date must be after start_date")
+		}
+		label = "自定义"
+		start = parsedStart
+		end = parsedEnd
+	default:
+		return TimeRangeResult{Ok: false, Now: now.Format("2006-01-02 15:04:05"), Timezone: loc.String()}, fmt.Errorf("unsupported range: %s", args.Range)
+	}
+
+	return TimeRangeResult{
+		Ok:           true,
+		Now:          now.Format("2006-01-02 15:04:05"),
+		NowWeekday:   weekdayName(now.Weekday()),
+		StartDate:    start.Format("2006-01-02"),
+		StartWeekday: weekdayName(start.Weekday()),
+		EndDate:      end.Format("2006-01-02"),
+		EndWeekday:   weekdayName(end.Weekday()),
+		Label:        label,
+		Timezone:     loc.String(),
+	}, nil
+}
+
 func buildTimeContext(now time.Time) string {
 	todayStart := dateStart(now)
 	weekStart := todayStart.AddDate(0, 0, -int((int(todayStart.Weekday())+6)%7))