新增工单工具

2026-06-16 16:15:34 +08:00
parent 4d154ebdd5
commit 9d3c489302
6 changed files with 469 additions and 4 deletions
@@ -319,7 +319,7 @@ func handleChat(ctx *gin.Context) {
 			toolNames = append(toolNames, tool.Function.Name)
 		}
 		emitTrace("function_tools", "prepare", "success", "已启用 Function Calling 工具", map[string]interface{}{"tools": toolNames})
-		openaiMsgs = append([]openaiMessage{{Role: "system", Content: "可用工具使用规则：当用户询问“我是谁”“当前登录用户是谁”“我的用户信息”等当前身份问题时，调用 ops_ai_assistant_current_user；工具返回 loggedIn=true 时按工具结果回答当前用户信息，返回 loggedIn=false 时说明不知道并提示需要登录才能获取信息。当用户询问采购订单列表、采购订单详情、采购状态或数量统计时，调用 ops_ai_assistant_purchase_query；该工具只允许查询，禁止新增、修改、删除采购数据。当用户询问本月、今天、本周、下周等相对日期的日程时，先调用 time 获取明确 start_date/end_date，再调用 ops_ai_assistant_schedule_query 查询日程。用户询问有哪些日历或创建日程但未提供 calendar_id 时，调用 ops_ai_assistant_calendar_list 获取可用日历；用户明确要求新增/修改/删除日程时，分别调用 ops_ai_assistant_schedule_create、ops_ai_assistant_schedule_update、ops_ai_assistant_schedule_delete，写入工具必须基于用户明确指令和明确日期，不要自行猜测日历或事件 ID。不要臆造工具结果中不存在的信息。"}}, openaiMsgs...)
+		openaiMsgs = append([]openaiMessage{{Role: "system", Content: "可用工具使用规则：当用户询问“我是谁”“当前登录用户是谁”“我的用户信息”等当前身份问题时，调用 ops_ai_assistant_current_user；工具返回 loggedIn=true 时按工具结果回答当前用户信息，返回 loggedIn=false 时说明不知道并提示需要登录才能获取信息。当用户询问采购订单列表、采购订单详情、采购状态或数量统计时，调用 ops_ai_assistant_purchase_query；该工具只允许查询，禁止新增、修改、删除采购数据。当用户询问工单、维修单、服务单、工单状态（待处理/已检查/已下单零件/已维修/已送还/无法维修）或数量统计时，调用 ops_ai_assistant_work_order_query；该工具只允许查询，禁止新增、修改、删除工单数据。当用户询问本月、今天、本周、下周等相对日期的日程时，先调用 time 获取明确 start_date/end_date，再调用 ops_ai_assistant_schedule_query 查询日程。用户询问有哪些日历或创建日程但未提供 calendar_id 时，调用 ops_ai_assistant_calendar_list 获取可用日历；用户明确要求新增/修改/删除日程时，分别调用 ops_ai_assistant_schedule_create、ops_ai_assistant_schedule_update、ops_ai_assistant_schedule_delete，写入工具必须基于用户明确指令和明确日期，不要自行猜测日历或事件 ID。不要臆造工具结果中不存在的信息。"}}, openaiMsgs...)
 		var toolExecuted bool
 		openaiMsgs, toolExecuted, err = runOpenAIToolLoop(ctx.Request.Context(), profile, openaiMsgs, functionTools, currentUser, tracker, emitTrace)
 		if err != nil {
@@ -111,6 +111,7 @@ func ensureBuiltinAIChatTools() error {
 		{Name: "time", Enabled: true, Description: "解析当前时间、相对日期和日期范围。", SortOrder: 0},
 		{Name: "ops_ai_assistant_current_user", Enabled: true, Description: "返回当前登录用户信息；未登录时提示需要登录才能获取信息。", SortOrder: 5},
 		{Name: "ops_ai_assistant_purchase_query", Enabled: true, Description: "查询采购订单列表、详情和状态数量统计。", SortOrder: 8},
+		{Name: "ops_ai_assistant_work_order_query", Enabled: true, Description: "查询工单（维修/服务）列表、详情和状态数量统计。", SortOrder: 9},
 		{Name: "ops_ai_assistant_calendar_list", Enabled: true, Description: "查询当前用户可见的 OPS 日历列表。", SortOrder: 10},
 		{Name: "ops_ai_assistant_schedule_query", Enabled: true, Description: "按日期范围查询当前用户可见的 OPS 日历/日程。", SortOrder: 12},
 		{Name: "ops_ai_assistant_schedule_create", Enabled: true, Description: "创建 OPS 日历日程。", SortOrder: 14},
@@ -212,6 +213,7 @@ func seedAIChatConfigFromYAMLIfEmpty() error {
 				{Name: "time", Enabled: true, Description: "解析当前时间、相对日期和日期范围。"},
 				{Name: "ops_ai_assistant_current_user", Enabled: true, Description: "返回当前登录用户信息；未登录时提示需要登录才能获取信息。"},
 				{Name: "ops_ai_assistant_purchase_query", Enabled: true, Description: "查询采购订单列表、详情和状态数量统计。"},
+				{Name: "ops_ai_assistant_work_order_query", Enabled: true, Description: "查询工单（维修/服务）列表、详情和状态数量统计。"},
 				{Name: "ops_ai_assistant_calendar_list", Enabled: true, Description: "查询当前用户可见的 OPS 日历列表。"},
 				{Name: "ops_ai_assistant_schedule_query", Enabled: true, Description: "按日期范围查询当前用户可见的 OPS 日历/日程。"},
 				{Name: "ops_ai_assistant_schedule_create", Enabled: true, Description: "创建 OPS 日历日程。"},
@@ -1,8 +1,10 @@
 package routers

 import (
+	"context"
 	"encoding/json"
 	parsefmt "fmt"
+	"ops/agents"
 	"ops/models"
 	"slices"
 	"time"
@@ -81,6 +83,274 @@ type PurchaseOrderInfo struct {
 	Status string `json:"status"`
 }

+// ---------- AI 工单查询 Provider ----------
+
+type workOrderProvider struct{}
+
+func (workOrderProvider) QueryWorkOrders(ctx context.Context, query agents.WorkOrderQuery) (*agents.WorkOrderQueryResult, error) {
+	select {
+	case <-ctx.Done():
+		return nil, ctx.Err()
+	default:
+	}
+
+	result := &agents.WorkOrderQueryResult{
+		Ok:       true,
+		Action:   query.Action,
+		LoggedIn: query.UserID > 0,
+	}
+	if !result.LoggedIn {
+		result.Message = "需要登录才能查询工单模块信息。"
+		return result, nil
+	}
+
+	switch query.Action {
+	case "count":
+		counts, err := queryWorkOrderCounts(query)
+		if err != nil {
+			return nil, err
+		}
+		result.Counts = counts
+		return result, nil
+	case "get":
+		order, err := queryWorkOrderDetail(query.OrderID, query.UserID)
+		if err != nil {
+			return nil, err
+		}
+		result.Order = order
+		if order != nil {
+			result.Count = 1
+		}
+		return result, nil
+	default:
+		orders, total, err := queryWorkOrderList(query)
+		if err != nil {
+			return nil, err
+		}
+		result.Orders = orders
+		result.Count = len(orders)
+		result.Total = total
+		result.Page = query.Page
+		result.Limit = query.Limit
+		result.Filters = map[string]interface{}{
+			"search":     query.Search,
+			"status":     query.Status,
+			"start_date": query.StartDate,
+			"end_date":   query.EndDate,
+		}
+		return result, nil
+	}
+}
+
+func applyWorkOrderQueryFilters(db *gorm.DB, query agents.WorkOrderQuery) (*gorm.DB, error) {
+	if query.Search != "" {
+		var id uint
+		if _, err := parsefmt.Sscanf(query.Search, "%d", &id); err == nil && id > 0 {
+			db = db.Where("id = ?", id)
+		} else {
+			db = db.Where("title LIKE ? OR description LIKE ?", "%"+query.Search+"%", "%"+query.Search+"%")
+		}
+	}
+	if query.Status != "" {
+		db = db.Where("current_status = ?", query.Status)
+	}
+	if query.StartDate != "" {
+		startDate, err := time.Parse("2006-01-02", query.StartDate)
+		if err != nil {
+			return nil, err
+		}
+		db = db.Where("created_at >= ?", startDate)
+	}
+	if query.EndDate != "" {
+		endDate, err := time.Parse("2006-01-02", query.EndDate)
+		if err != nil {
+			return nil, err
+		}
+		db = db.Where("created_at < ?", endDate.AddDate(0, 0, 1))
+	}
+	return db, nil
+}
+
+func queryWorkOrderList(query agents.WorkOrderQuery) ([]agents.WorkOrder, int64, error) {
+	db, err := applyWorkOrderQueryFilters(models.DB.Model(&TabWorkOrder{}), query)
+	if err != nil {
+		return nil, 0, err
+	}
+	var total int64
+	if err := db.Count(&total).Error; err != nil {
+		return nil, 0, err
+	}
+
+	var rows []TabWorkOrder
+	if err := db.Order("id DESC").Offset(query.Limit * (query.Page - 1)).Limit(query.Limit).Find(&rows).Error; err != nil {
+		return nil, 0, err
+	}
+
+	orders := make([]agents.WorkOrder, 0, len(rows))
+	for _, row := range rows {
+		order := buildWorkOrder(row, query.UserID)
+		order.Customers = loadWorkOrderCustomers(row.ID)
+		order.Items = loadWorkOrderItems(row.ID)
+		orders = append(orders, order)
+	}
+	return orders, total, nil
+}
+
+func queryWorkOrderDetail(orderID uint, userID uint) (*agents.WorkOrder, error) {
+	var row TabWorkOrder
+	if err := models.DB.Where("id = ?", orderID).First(&row).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return nil, nil
+		}
+		return nil, err
+	}
+	order := buildWorkOrder(row, userID)
+	order.Customers = loadWorkOrderCustomers(orderID)
+	order.Items = loadWorkOrderItems(orderID)
+
+	var commits []TabWorkOrderCommit
+	if err := models.DB.Where("work_order_id = ?", orderID).Order("created_at DESC").Find(&commits).Error; err != nil {
+		return nil, err
+	}
+	order.Commits = make([]agents.WorkOrderCommit, 0, len(commits))
+	for _, commit := range commits {
+		order.Commits = append(order.Commits, agents.WorkOrderCommit{
+			ID:          commit.ID,
+			WorkOrderID: commit.WorkOrderID,
+			UserID:      commit.UserID,
+			Action:      commit.Action,
+			Status:      commit.Status,
+			StatusName:  workOrderStatusName(commit.Status),
+			OldStatus:   commit.OldStatus,
+			Comment:     commit.Comment,
+			CreatedAt:   formatTimePtr(commit.CreatedAt),
+		})
+	}
+	return &order, nil
+}
+
+func queryWorkOrderCounts(query agents.WorkOrderQuery) (map[string]int64, error) {
+	counts := map[string]int64{}
+	statuses := []string{"pending", "checked", "parts_ordered", "repaired", "returned", "unrepairable"}
+	var total int64
+	base, err := applyWorkOrderQueryFilters(models.DB.Model(&TabWorkOrder{}), agents.WorkOrderQuery{Search: query.Search, StartDate: query.StartDate, EndDate: query.EndDate})
+	if err != nil {
+		return nil, err
+	}
+	if err := base.Count(&total).Error; err != nil {
+		return nil, err
+	}
+	counts["total"] = total
+	for _, status := range statuses {
+		statusQuery := agents.WorkOrderQuery{Search: query.Search, Status: status, StartDate: query.StartDate, EndDate: query.EndDate}
+		db, err := applyWorkOrderQueryFilters(models.DB.Model(&TabWorkOrder{}), statusQuery)
+		if err != nil {
+			return nil, err
+		}
+		var count int64
+		if err := db.Count(&count).Error; err != nil {
+			return nil, err
+		}
+		counts[status] = count
+	}
+	return counts, nil
+}
+
+func buildWorkOrder(row TabWorkOrder, currentUserID uint) agents.WorkOrder {
+	return agents.WorkOrder{
+		ID:                row.ID,
+		UserID:            row.UserID,
+		Title:             row.Title,
+		Description:       row.Description,
+		DetailURL:         workOrderDetailURL(row.ID),
+		CurrentStatus:     row.CurrentStatus,
+		CurrentStatusName: workOrderStatusName(row.CurrentStatus),
+		CreatedAt:         formatTimePtr(row.CreatedAt),
+		UpdatedAt:         formatTimePtr(row.UpdatedAt),
+		CanModify:         canModifyWorkOrder(currentUserID, row.UserID),
+	}
+}
+
+func loadWorkOrderCustomers(workOrderID uint) []agents.WorkOrderCustomer {
+	var customerBinds []TabWorkOrderCustomerBind
+	models.DB.Where("work_order_id = ?", workOrderID).Find(&customerBinds)
+	if len(customerBinds) == 0 {
+		return nil
+	}
+	customerIDs := make([]uint, 0, len(customerBinds))
+	for _, b := range customerBinds {
+		customerIDs = append(customerIDs, b.CustomerID)
+	}
+	var customers []TabCustomer
+	if err := models.DB.Where("id IN ?", customerIDs).Find(&customers).Error; err != nil {
+		return nil
+	}
+	result := make([]agents.WorkOrderCustomer, 0, len(customers))
+	for _, c := range customers {
+		item := agents.WorkOrderCustomer{
+			ID:        c.ID,
+			FirstName: c.FirstName,
+			LastName:  c.LastName,
+		}
+		var phone TabCustomerPhone
+		if err := models.DB.Where("customer_id = ? AND is_primary = ?", c.ID, true).First(&phone).Error; err == nil {
+			item.PrimaryPhone = phone.Phone
+		} else if err := models.DB.Where("customer_id = ?", c.ID).First(&phone).Error; err == nil {
+			item.PrimaryPhone = phone.Phone
+		}
+		result = append(result, item)
+	}
+	return result
+}
+
+func loadWorkOrderItems(workOrderID uint) []agents.WorkOrderItem {
+	var itemBinds []TabWarehouseItemWorkOrderBind
+	models.DB.Where("work_order_id = ?", workOrderID).Find(&itemBinds)
+	if len(itemBinds) == 0 {
+		return nil
+	}
+	itemIDs := make([]uint, 0, len(itemBinds))
+	for _, b := range itemBinds {
+		itemIDs = append(itemIDs, b.ItemID)
+	}
+	var items []TabWarehouseItem
+	if err := models.DB.Where("id IN ?", itemIDs).Find(&items).Error; err != nil {
+		return nil
+	}
+	result := make([]agents.WorkOrderItem, 0, len(items))
+	for _, it := range items {
+		result = append(result, agents.WorkOrderItem{
+			ID:           it.ID,
+			Name:         it.Name,
+			SerialNumber: it.SerialNumber,
+		})
+	}
+	return result
+}
+
+func workOrderDetailURL(orderID uint) string {
+	return parsefmt.Sprintf("/workorder/showorder/%d", orderID)
+}
+
+func workOrderStatusName(status string) string {
+	switch status {
+	case "pending":
+		return "待处理"
+	case "checked":
+		return "已检查"
+	case "parts_ordered":
+		return "已下单零件"
+	case "repaired":
+		return "已维修"
+	case "returned":
+		return "已送还"
+	case "unrepairable":
+		return "无法维修"
+	default:
+		return status
+	}
+}
+
 // ---------- 初始化 ----------

 func ApiWorkOrderInit() {
@@ -96,6 +366,8 @@ func ApiWorkOrderInit() {
 		workOrderUserGroup.Type = "usergroup"
 		models.DB.Create(&workOrderUserGroup)
 	}
+
+	agents.RegisterWorkOrderProvider(workOrderProvider{})
 }

 // ---------- 路由注册 ----------