package agents import ( "context" "encoding/json" "fmt" "time" ) type ScheduleQueryArgs struct { Action string `json:"action,omitempty"` StartDate string `json:"start_date"` EndDate string `json:"end_date"` CalendarID uint `json:"calendar_id,omitempty"` Limit int `json:"limit,omitempty"` } type CurrentUserArgs struct { Action string `json:"action,omitempty"` } type CurrentUserInfo struct { ID uint `json:"id,omitempty"` UserID uint `json:"user_id,omitempty"` FirstName string `json:"first_name,omitempty"` Username string `json:"username,omitempty"` Birthdate string `json:"birthdate,omitempty"` Gender string `json:"gender,omitempty"` AvatarPath string `json:"avatar_path,omitempty"` Region string `json:"region,omitempty"` Language string `json:"language,omitempty"` } type PurchaseQueryArgs struct { Action string `json:"action"` OrderID uint `json:"order_id,omitempty"` Search string `json:"search,omitempty"` Status string `json:"status,omitempty"` StartDate string `json:"start_date,omitempty"` EndDate string `json:"end_date,omitempty"` Page int `json:"page,omitempty"` Limit int `json:"limit,omitempty"` } type PurchaseQuery struct { Action string OrderID uint Search string Status string StartDate string EndDate string Page int Limit int UserID uint } type PurchaseCost struct { ID uint `json:"id"` OrderID uint `json:"order_id"` UserID uint `json:"user_id"` Price int `json:"price"` Quantity int `json:"quantity"` CurrencyType int `json:"currency_type"` CurrencyName string `json:"currency_name"` CostType int `json:"cost_type"` CostTypeName string `json:"cost_type_name"` } type PurchaseCommit struct { ID uint `json:"id"` OrderID uint `json:"order_id"` UserID uint `json:"user_id"` Action string `json:"action"` Status string `json:"status,omitempty"` OldStatus string `json:"old_status,omitempty"` Comment string `json:"comment,omitempty"` CreatedAt string `json:"created_at,omitempty"` } type PurchaseOrder struct { ID uint `json:"id"` UserID uint `json:"user_id"` Title string `json:"title"` Remark string `json:"remark,omitempty"` Link string `json:"link,omitempty"` DetailURL string `json:"detail_url"` Styles string `json:"styles,omitempty"` OrderStatus string `json:"order_status"` OrderStatusName string `json:"order_status_name"` CreatedAt string `json:"created_at,omitempty"` UpdatedAt string `json:"updated_at,omitempty"` CanModify bool `json:"can_modify,omitempty"` Costs []PurchaseCost `json:"costs,omitempty"` Commits []PurchaseCommit `json:"commits,omitempty"` } type PurchaseQueryResult struct { Ok bool `json:"ok"` Action string `json:"action"` LoggedIn bool `json:"loggedIn"` Count int `json:"count,omitempty"` Total int64 `json:"total,omitempty"` Page int `json:"page,omitempty"` Limit int `json:"limit,omitempty"` Orders []PurchaseOrder `json:"orders,omitempty"` Order *PurchaseOrder `json:"order,omitempty"` Counts map[string]int64 `json:"counts,omitempty"` Filters map[string]interface{} `json:"filters,omitempty"` Message string `json:"message,omitempty"` } type ScheduleCalendar struct { ID uint `json:"id"` Name string `json:"name"` } type CalendarListOutput struct { ID uint `json:"id"` Name string `json:"name"` Description string `json:"description,omitempty"` Color string `json:"color,omitempty"` IsPublic bool `json:"is_public"` CanEdit bool `json:"canEdit"` } type CreateScheduleInput struct { CalendarID uint Title string StartDate string EndDate string ScheduleType string IsPublic bool Remark string } type UpdateScheduleInput struct { EventID uint Title string StartDate string EndDate string ScheduleType string IsPublic bool Remark string } type ScheduleEvent struct { ID uint `json:"id"` CalendarID uint `json:"calendar_id"` UserID uint `json:"user_id"` Title string `json:"title"` StartDate string `json:"start_date"` EndDate string `json:"end_date"` ScheduleType string `json:"schedule_type"` IsPublic bool `json:"is_public"` Remark string `json:"remark,omitempty"` AccessNote string `json:"access_note,omitempty"` CanEdit bool `json:"canEdit"` } type ScheduleQuery struct { StartDate string EndDate string CalendarID uint UserID uint Limit int } type ScheduleProvider interface { QuerySchedules(ctx context.Context, query ScheduleQuery) ([]ScheduleEvent, error) ListCalendars(ctx context.Context, userID uint) ([]CalendarListOutput, error) CreateScheduleEvent(ctx context.Context, userID uint, input CreateScheduleInput) (*ScheduleEvent, error) UpdateScheduleEvent(ctx context.Context, userID uint, input UpdateScheduleInput) (*ScheduleEvent, error) DeleteScheduleEvent(ctx context.Context, userID uint, eventID uint) error } type PurchaseProvider interface { QueryPurchases(ctx context.Context, query PurchaseQuery) (*PurchaseQueryResult, error) } var registeredScheduleProvider ScheduleProvider = nil var registeredPurchaseProvider PurchaseProvider = nil func RegisterScheduleProvider(provider ScheduleProvider) { registeredScheduleProvider = provider } func RegisterPurchaseProvider(provider PurchaseProvider) { registeredPurchaseProvider = provider } func opsAIAssistantScheduleQuerySchema() FunctionToolSchema { return FunctionToolSchema{ Name: "ops_ai_assistant_schedule_query", Description: "只读工具:按明确日期范围查询当前用户可见的 OPS 日历/日程。禁止新增、修改、删除数据,禁止批量添加、批量导入或执行任何写操作。相对日期需先调用 time 获取 start_date/end_date。", Parameters: map[string]interface{}{ "type": "object", "properties": map[string]interface{}{ "action": map[string]interface{}{ "type": "string", "enum": []string{"query"}, "description": "固定为 query。本工具只允许查询,禁止 create/update/delete/batch_add/import 等写操作。", }, "start_date": map[string]interface{}{"type": "string", "description": "开始日期,格式 YYYY-MM-DD。"}, "end_date": map[string]interface{}{"type": "string", "description": "结束日期,格式 YYYY-MM-DD。"}, "calendar_id": map[string]interface{}{"type": "integer", "description": "可选日历 ID;不传则查询全部可见日程。"}, "limit": map[string]interface{}{"type": "integer", "description": "可选返回上限,默认 100,最大 200。"}, }, "required": []string{"action", "start_date", "end_date"}, }, // ScheduleQueryResult is the return type packed as JSON } } func opsAIAssistantCurrentUserSchema() FunctionToolSchema { return FunctionToolSchema{ Name: "ops_ai_assistant_current_user", Description: "只读工具:当用户询问“我是谁”“当前登录用户是谁”“我的用户信息”等当前身份问题时调用。先判断是否已登录;已登录则返回当前登录用户信息,未登录则返回 unknown 并提示需要登录才能获取信息。", Parameters: map[string]interface{}{ "type": "object", "properties": map[string]interface{}{ "action": map[string]interface{}{ "type": "string", "enum": []string{"get"}, "description": "固定为 get。本工具只读取当前登录用户信息。", }, }, "required": []string{"action"}, }, } } func opsAIAssistantPurchaseQuerySchema() FunctionToolSchema { return FunctionToolSchema{ Name: "ops_ai_assistant_purchase_query", Description: "只读工具:查询采购模块订单。用户询问采购订单、采购状态、待处理/已下单/已到达/已收件/丢件/退件数量或列表、指定采购订单详情时调用。禁止新增、修改、删除订单或状态。", Parameters: map[string]interface{}{ "type": "object", "properties": map[string]interface{}{ "action": map[string]interface{}{ "type": "string", "enum": []string{"list", "get", "count"}, "description": "list 查询订单列表;get 查询单个订单详情;count 统计各状态数量。", }, "order_id": map[string]interface{}{"type": "integer", "description": "action=get 时的采购订单 ID。"}, "search": map[string]interface{}{"type": "string", "description": "按订单 ID、标题或备注搜索。"}, "status": map[string]interface{}{"type": "string", "enum": []string{"", "pending", "ordered", "arrived", "received", "lost", "returned"}, "description": "订单状态过滤:pending 待处理,ordered 已下单,arrived 已到达,received 已收件,lost 丢件,returned 退件。"}, "start_date": map[string]interface{}{"type": "string", "description": "可选创建日期开始,格式 YYYY-MM-DD。"}, "end_date": map[string]interface{}{"type": "string", "description": "可选创建日期结束,格式 YYYY-MM-DD。"}, "page": map[string]interface{}{"type": "integer", "description": "分页页码,默认 1。"}, "limit": map[string]interface{}{"type": "integer", "description": "返回上限,默认 20,最大 100。"}, }, "required": []string{"action"}, }, } } func executeOpsAIAssistantCurrentUser(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) { var args CurrentUserArgs if len(rawArgs) > 0 { if err := json.Unmarshal(rawArgs, &args); err != nil { return nil, err } } if args.Action != "" && args.Action != "get" { return json.Marshal(map[string]interface{}{ "ok": false, "error": "ops_ai_assistant_current_user 是只读工具,仅允许 get 读取当前用户信息", }) } if runtime.UserID <= 0 { return json.Marshal(map[string]interface{}{ "ok": true, "loggedIn": false, "unknown": true, "message": "不知道你是谁,需要登录才能获取当前用户信息。", }) } return json.Marshal(map[string]interface{}{ "ok": true, "loggedIn": true, "user": map[string]interface{}{ "id": runtime.UserID, "name": runtime.UserName, "email": runtime.UserEmail, "type": runtime.UserType, "info": runtime.UserInfo, }, }) } func executeOpsAIAssistantPurchaseQuery(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) { var args PurchaseQueryArgs if err := json.Unmarshal(rawArgs, &args); err != nil { return nil, err } if args.Action != "list" && args.Action != "get" && args.Action != "count" { return json.Marshal(map[string]interface{}{ "ok": false, "error": "ops_ai_assistant_purchase_query 是只读工具,仅允许 list/get/count 查询操作", }) } if runtime.UserID <= 0 { return json.Marshal(map[string]interface{}{ "ok": true, "action": args.Action, "loggedIn": false, "message": "需要登录才能查询采购模块信息。", }) } if args.Action == "get" && args.OrderID <= 0 { return nil, fmt.Errorf("order_id is required when action=get") } if args.StartDate != "" { if _, err := time.Parse("2006-01-02", args.StartDate); err != nil { return nil, fmt.Errorf("invalid start_date: %w", err) } } if args.EndDate != "" { if _, err := time.Parse("2006-01-02", args.EndDate); err != nil { return nil, fmt.Errorf("invalid end_date: %w", err) } } if args.StartDate != "" && args.EndDate != "" { startDate, _ := time.Parse("2006-01-02", args.StartDate) endDate, _ := time.Parse("2006-01-02", args.EndDate) if endDate.Before(startDate) { return nil, fmt.Errorf("end_date must be after start_date") } } if args.Page <= 0 { args.Page = 1 } if args.Limit <= 0 { args.Limit = 20 } if args.Limit > 100 { args.Limit = 100 } if registeredPurchaseProvider == nil { return json.Marshal(map[string]interface{}{ "ok": false, "error": "采购查询服务未注册", }) } result, err := registeredPurchaseProvider.QueryPurchases(ctx, PurchaseQuery{ Action: args.Action, OrderID: args.OrderID, Search: args.Search, Status: args.Status, StartDate: args.StartDate, EndDate: args.EndDate, Page: args.Page, Limit: args.Limit, UserID: runtime.UserID, }) if err != nil { return nil, err } return json.Marshal(result) } func executeOpsAIAssistantScheduleQuery(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) { var args ScheduleQueryArgs if err := json.Unmarshal(rawArgs, &args); err != nil { return nil, err } if args.Action != "" && args.Action != "query" { return json.Marshal(map[string]interface{}{ "ok": false, "error": "ops_ai_assistant 是只读工具,仅允许 query 查询操作,禁止新增、修改、删除、批量添加或导入数据", }) } startDate, err := time.Parse("2006-01-02", args.StartDate) if err != nil { return nil, fmt.Errorf("invalid start_date: %w", err) } endDate, err := time.Parse("2006-01-02", args.EndDate) if err != nil { return nil, fmt.Errorf("invalid end_date: %w", err) } if endDate.Before(startDate) { return nil, fmt.Errorf("end_date must be after start_date") } limit := args.Limit if limit <= 0 { limit = 100 } if limit > 200 { limit = 200 } if registeredScheduleProvider == nil { return json.Marshal(map[string]interface{}{ "ok": false, "error": "日程查询服务未注册", }) } events, err := registeredScheduleProvider.QuerySchedules(ctx, ScheduleQuery{ StartDate: args.StartDate, EndDate: args.EndDate, CalendarID: args.CalendarID, UserID: runtime.UserID, Limit: limit, }) if err != nil { return nil, err } if events == nil { events = []ScheduleEvent{} } return json.Marshal(map[string]interface{}{ "ok": true, "start_date": args.StartDate, "end_date": args.EndDate, "count": len(events), "limit": limit, "events": events, }) } type CalendarListArgs struct { Action string `json:"action,omitempty"` } type ScheduleEventCreateArgs struct { Action string `json:"action,omitempty"` CalendarID uint `json:"calendar_id"` Title string `json:"title"` StartDate string `json:"start_date"` EndDate string `json:"end_date"` ScheduleType string `json:"schedule_type,omitempty"` IsPublic bool `json:"is_public,omitempty"` Remark string `json:"remark,omitempty"` } type ScheduleEventUpdateArgs struct { Action string `json:"action,omitempty"` EventID uint `json:"event_id"` Title string `json:"title"` StartDate string `json:"start_date"` EndDate string `json:"end_date"` ScheduleType string `json:"schedule_type,omitempty"` IsPublic bool `json:"is_public,omitempty"` Remark string `json:"remark,omitempty"` } type ScheduleEventDeleteArgs struct { Action string `json:"action,omitempty"` EventID uint `json:"event_id"` } func opsAIAssistantCalendarListSchema() FunctionToolSchema { return FunctionToolSchema{ Name: "ops_ai_assistant_calendar_list", Description: "只读工具:查询当前用户可见的 OPS 日历列表。创建/修改日程前如用户未提供 calendar_id,应先调用本工具让用户选择日历。", Parameters: map[string]interface{}{ "type": "object", "properties": map[string]interface{}{ "action": map[string]interface{}{ "type": "string", "enum": []string{"list"}, "description": "固定为 list。", }, }, "required": []string{"action"}, }, } } func opsAIAssistantScheduleCreateSchema() FunctionToolSchema { return FunctionToolSchema{ Name: "ops_ai_assistant_schedule_create", Description: "写入工具:在指定 OPS 日历中创建日程。仅当用户明确要求新增/创建/添加日程且已提供日历 ID、标题和明确日期范围时调用;相对日期需先调用 time 获取 start_date/end_date。", Parameters: map[string]interface{}{ "type": "object", "properties": scheduleEventWriteProperties(), "required": []string{"action", "calendar_id", "title", "start_date", "end_date"}, }, } } func opsAIAssistantScheduleUpdateSchema() FunctionToolSchema { props := scheduleEventWriteProperties() props["event_id"] = map[string]interface{}{"type": "integer", "description": "要修改的日程事件 ID。"} delete(props, "calendar_id") props["action"] = map[string]interface{}{ "type": "string", "enum": []string{"update"}, "description": "固定为 update。", } return FunctionToolSchema{ Name: "ops_ai_assistant_schedule_update", Description: "写入工具:修改指定 OPS 日程事件。仅当用户明确要求修改/更新已有日程且已提供事件 ID、标题和明确日期范围时调用;相对日期需先调用 time 获取 start_date/end_date。", Parameters: map[string]interface{}{ "type": "object", "properties": props, "required": []string{"action", "event_id", "title", "start_date", "end_date"}, }, } } func opsAIAssistantScheduleDeleteSchema() FunctionToolSchema { return FunctionToolSchema{ Name: "ops_ai_assistant_schedule_delete", Description: "写入工具:删除指定 OPS 日程事件。仅当用户明确要求删除/移除日程且已提供事件 ID 时调用。", Parameters: map[string]interface{}{ "type": "object", "properties": map[string]interface{}{ "action": map[string]interface{}{ "type": "string", "enum": []string{"delete"}, "description": "固定为 delete。", }, "event_id": map[string]interface{}{"type": "integer", "description": "要删除的日程事件 ID。"}, }, "required": []string{"action", "event_id"}, }, } } func scheduleEventWriteProperties() map[string]interface{} { return map[string]interface{}{ "action": map[string]interface{}{ "type": "string", "enum": []string{"create"}, "description": "固定为 create。", }, "calendar_id": map[string]interface{}{"type": "integer", "description": "目标日历 ID。"}, "title": map[string]interface{}{"type": "string", "description": "日程标题。"}, "start_date": map[string]interface{}{"type": "string", "description": "开始日期,格式 YYYY-MM-DD。"}, "end_date": map[string]interface{}{"type": "string", "description": "结束日期,格式 YYYY-MM-DD。"}, "schedule_type": map[string]interface{}{"type": "string", "enum": []string{"", "work", "duty", "exam", "standby", "personal_holiday", "public_holiday"}, "description": "日程类型,默认 work。"}, "is_public": map[string]interface{}{"type": "boolean", "description": "是否公开,默认 false。"}, "remark": map[string]interface{}{"type": "string", "description": "备注。"}, } } func executeOpsAIAssistantCalendarList(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) { var args CalendarListArgs if len(rawArgs) > 0 { if err := json.Unmarshal(rawArgs, &args); err != nil { return nil, err } } if args.Action != "" && args.Action != "list" { return json.Marshal(map[string]interface{}{ "ok": false, "error": "ops_ai_assistant_calendar_list 仅允许 list 查询操作", }) } if runtime.UserID <= 0 { return json.Marshal(map[string]interface{}{ "ok": true, "loggedIn": false, "message": "需要登录才能查询日历列表。", }) } if registeredScheduleProvider == nil { return json.Marshal(map[string]interface{}{ "ok": false, "error": "日历服务未注册", }) } calendars, err := registeredScheduleProvider.ListCalendars(ctx, runtime.UserID) if err != nil { return nil, err } if calendars == nil { calendars = []CalendarListOutput{} } return json.Marshal(map[string]interface{}{ "ok": true, "loggedIn": true, "count": len(calendars), "calendars": calendars, }) } func executeOpsAIAssistantScheduleCreate(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) { var args ScheduleEventCreateArgs if err := json.Unmarshal(rawArgs, &args); err != nil { return nil, err } if args.Action != "" && args.Action != "create" { return json.Marshal(map[string]interface{}{ "ok": false, "error": "ops_ai_assistant_schedule_create 仅允许 create 操作", }) } if err := validateScheduleWrite(runtime.UserID, args.Title, args.StartDate, args.EndDate); err != nil { return nil, err } if args.CalendarID <= 0 { return nil, fmt.Errorf("calendar_id is required") } if registeredScheduleProvider == nil { return json.Marshal(map[string]interface{}{ "ok": false, "error": "日历服务未注册", }) } event, err := registeredScheduleProvider.CreateScheduleEvent(ctx, runtime.UserID, CreateScheduleInput{ CalendarID: args.CalendarID, Title: args.Title, StartDate: args.StartDate, EndDate: args.EndDate, ScheduleType: args.ScheduleType, IsPublic: args.IsPublic, Remark: args.Remark, }) if err != nil { return nil, err } return json.Marshal(map[string]interface{}{ "ok": true, "loggedIn": true, "event": event, "message": "日程已创建。", }) } func executeOpsAIAssistantScheduleUpdate(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) { var args ScheduleEventUpdateArgs if err := json.Unmarshal(rawArgs, &args); err != nil { return nil, err } if args.Action != "" && args.Action != "update" { return json.Marshal(map[string]interface{}{ "ok": false, "error": "ops_ai_assistant_schedule_update 仅允许 update 操作", }) } if args.EventID <= 0 { return nil, fmt.Errorf("event_id is required") } if err := validateScheduleWrite(runtime.UserID, args.Title, args.StartDate, args.EndDate); err != nil { return nil, err } if registeredScheduleProvider == nil { return json.Marshal(map[string]interface{}{ "ok": false, "error": "日历服务未注册", }) } event, err := registeredScheduleProvider.UpdateScheduleEvent(ctx, runtime.UserID, UpdateScheduleInput{ EventID: args.EventID, Title: args.Title, StartDate: args.StartDate, EndDate: args.EndDate, ScheduleType: args.ScheduleType, IsPublic: args.IsPublic, Remark: args.Remark, }) if err != nil { return nil, err } return json.Marshal(map[string]interface{}{ "ok": true, "loggedIn": true, "event": event, "message": "日程已更新。", }) } func executeOpsAIAssistantScheduleDelete(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) { var args ScheduleEventDeleteArgs if err := json.Unmarshal(rawArgs, &args); err != nil { return nil, err } if args.Action != "" && args.Action != "delete" { return json.Marshal(map[string]interface{}{ "ok": false, "error": "ops_ai_assistant_schedule_delete 仅允许 delete 操作", }) } if runtime.UserID <= 0 { return json.Marshal(map[string]interface{}{ "ok": true, "loggedIn": false, "message": "需要登录才能删除日程。", }) } if args.EventID <= 0 { return nil, fmt.Errorf("event_id is required") } if registeredScheduleProvider == nil { return json.Marshal(map[string]interface{}{ "ok": false, "error": "日历服务未注册", }) } if err := registeredScheduleProvider.DeleteScheduleEvent(ctx, runtime.UserID, args.EventID); err != nil { return nil, err } return json.Marshal(map[string]interface{}{ "ok": true, "loggedIn": true, "event_id": args.EventID, "message": "日程已删除。", }) } func validateScheduleWrite(userID uint, title string, startDate string, endDate string) error { if userID <= 0 { return fmt.Errorf("需要登录才能操作日程") } if title == "" { return fmt.Errorf("title is required") } start, err := time.Parse("2006-01-02", startDate) if err != nil { return fmt.Errorf("invalid start_date: %w", err) } end, err := time.Parse("2006-01-02", endDate) if err != nil { return fmt.Errorf("invalid end_date: %w", err) } if end.Before(start) { return fmt.Errorf("end_date must be after start_date") } return nil }