新增工单工具
This commit is contained in:
@@ -26,9 +26,10 @@ type FunctionToolRuntime struct {
|
|||||||
// 这类工具的返回结果已经包含了用户问题所需的全部数据,调用之后无需再让模型
|
// 这类工具的返回结果已经包含了用户问题所需的全部数据,调用之后无需再让模型
|
||||||
// 决定是否继续调用其他工具,直接进入最终回答生成阶段,可以省掉一轮模型请求。
|
// 决定是否继续调用其他工具,直接进入最终回答生成阶段,可以省掉一轮模型请求。
|
||||||
var terminalFunctionTools = map[string]bool{
|
var terminalFunctionTools = map[string]bool{
|
||||||
"ops_ai_assistant": true,
|
"ops_ai_assistant": true,
|
||||||
"ops_ai_assistant_schedule_query": true,
|
"ops_ai_assistant_schedule_query": true,
|
||||||
"ops_ai_assistant_purchase_query": true,
|
"ops_ai_assistant_purchase_query": true,
|
||||||
|
"ops_ai_assistant_work_order_query": true,
|
||||||
}
|
}
|
||||||
|
|
||||||
// IsTerminalFunctionTool 判断给定工具名是否为终止工具。命名匹配采用与
|
// IsTerminalFunctionTool 判断给定工具名是否为终止工具。命名匹配采用与
|
||||||
@@ -60,6 +61,8 @@ func FunctionToolSchemas(configs []ToolConfig) []FunctionToolSchema {
|
|||||||
tools = append(tools, opsAIAssistantCurrentUserSchema())
|
tools = append(tools, opsAIAssistantCurrentUserSchema())
|
||||||
case "ops_ai_assistant_purchase_query":
|
case "ops_ai_assistant_purchase_query":
|
||||||
tools = append(tools, opsAIAssistantPurchaseQuerySchema())
|
tools = append(tools, opsAIAssistantPurchaseQuerySchema())
|
||||||
|
case "ops_ai_assistant_work_order_query":
|
||||||
|
tools = append(tools, opsAIAssistantWorkOrderQuerySchema())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return tools
|
return tools
|
||||||
@@ -99,6 +102,8 @@ func ExecuteFunctionTool(ctx context.Context, runtime FunctionToolRuntime, name
|
|||||||
return executeOpsAIAssistantCurrentUser(ctx, runtime, rawArgs)
|
return executeOpsAIAssistantCurrentUser(ctx, runtime, rawArgs)
|
||||||
case "ops_ai_assistant_purchase_query":
|
case "ops_ai_assistant_purchase_query":
|
||||||
return executeOpsAIAssistantPurchaseQuery(ctx, runtime, rawArgs)
|
return executeOpsAIAssistantPurchaseQuery(ctx, runtime, rawArgs)
|
||||||
|
case "ops_ai_assistant_work_order_query":
|
||||||
|
return executeOpsAIAssistantWorkOrderQuery(ctx, runtime, rawArgs)
|
||||||
default:
|
default:
|
||||||
return nil, fmt.Errorf("unknown tool: %s", name)
|
return nil, fmt.Errorf("unknown tool: %s", name)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -42,6 +42,85 @@ type PurchaseQueryArgs struct {
|
|||||||
Limit int `json:"limit,omitempty"`
|
Limit int `json:"limit,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type WorkOrderQueryArgs 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 WorkOrderQuery struct {
|
||||||
|
Action string
|
||||||
|
OrderID uint
|
||||||
|
Search string
|
||||||
|
Status string
|
||||||
|
StartDate string
|
||||||
|
EndDate string
|
||||||
|
Page int
|
||||||
|
Limit int
|
||||||
|
UserID uint
|
||||||
|
}
|
||||||
|
|
||||||
|
type WorkOrderCommit struct {
|
||||||
|
ID uint `json:"id"`
|
||||||
|
WorkOrderID uint `json:"work_order_id"`
|
||||||
|
UserID uint `json:"user_id"`
|
||||||
|
Action string `json:"action"`
|
||||||
|
Status string `json:"status,omitempty"`
|
||||||
|
StatusName string `json:"status_name,omitempty"`
|
||||||
|
OldStatus string `json:"old_status,omitempty"`
|
||||||
|
Comment string `json:"comment,omitempty"`
|
||||||
|
CreatedAt string `json:"created_at,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type WorkOrderCustomer struct {
|
||||||
|
ID uint `json:"id"`
|
||||||
|
FirstName string `json:"first_name,omitempty"`
|
||||||
|
LastName string `json:"last_name,omitempty"`
|
||||||
|
PrimaryPhone string `json:"primary_phone,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type WorkOrderItem struct {
|
||||||
|
ID uint `json:"id"`
|
||||||
|
Name string `json:"name,omitempty"`
|
||||||
|
SerialNumber string `json:"serial_number,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type WorkOrder struct {
|
||||||
|
ID uint `json:"id"`
|
||||||
|
UserID uint `json:"user_id"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
Description string `json:"description,omitempty"`
|
||||||
|
DetailURL string `json:"detail_url"`
|
||||||
|
CurrentStatus string `json:"current_status"`
|
||||||
|
CurrentStatusName string `json:"current_status_name"`
|
||||||
|
CreatedAt string `json:"created_at,omitempty"`
|
||||||
|
UpdatedAt string `json:"updated_at,omitempty"`
|
||||||
|
CanModify bool `json:"can_modify,omitempty"`
|
||||||
|
Customers []WorkOrderCustomer `json:"customers,omitempty"`
|
||||||
|
Items []WorkOrderItem `json:"items,omitempty"`
|
||||||
|
Commits []WorkOrderCommit `json:"commits,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type WorkOrderQueryResult 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 []WorkOrder `json:"orders,omitempty"`
|
||||||
|
Order *WorkOrder `json:"order,omitempty"`
|
||||||
|
Counts map[string]int64 `json:"counts,omitempty"`
|
||||||
|
Filters map[string]interface{} `json:"filters,omitempty"`
|
||||||
|
Message string `json:"message,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
type PurchaseQuery struct {
|
type PurchaseQuery struct {
|
||||||
Action string
|
Action string
|
||||||
OrderID uint
|
OrderID uint
|
||||||
@@ -177,8 +256,13 @@ type PurchaseProvider interface {
|
|||||||
QueryPurchases(ctx context.Context, query PurchaseQuery) (*PurchaseQueryResult, error)
|
QueryPurchases(ctx context.Context, query PurchaseQuery) (*PurchaseQueryResult, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type WorkOrderProvider interface {
|
||||||
|
QueryWorkOrders(ctx context.Context, query WorkOrderQuery) (*WorkOrderQueryResult, error)
|
||||||
|
}
|
||||||
|
|
||||||
var registeredScheduleProvider ScheduleProvider = nil
|
var registeredScheduleProvider ScheduleProvider = nil
|
||||||
var registeredPurchaseProvider PurchaseProvider = nil
|
var registeredPurchaseProvider PurchaseProvider = nil
|
||||||
|
var registeredWorkOrderProvider WorkOrderProvider = nil
|
||||||
|
|
||||||
func RegisterScheduleProvider(provider ScheduleProvider) {
|
func RegisterScheduleProvider(provider ScheduleProvider) {
|
||||||
registeredScheduleProvider = provider
|
registeredScheduleProvider = provider
|
||||||
@@ -188,6 +272,10 @@ func RegisterPurchaseProvider(provider PurchaseProvider) {
|
|||||||
registeredPurchaseProvider = provider
|
registeredPurchaseProvider = provider
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func RegisterWorkOrderProvider(provider WorkOrderProvider) {
|
||||||
|
registeredWorkOrderProvider = provider
|
||||||
|
}
|
||||||
|
|
||||||
func opsAIAssistantScheduleQuerySchema() FunctionToolSchema {
|
func opsAIAssistantScheduleQuerySchema() FunctionToolSchema {
|
||||||
return FunctionToolSchema{
|
return FunctionToolSchema{
|
||||||
Name: "ops_ai_assistant_schedule_query",
|
Name: "ops_ai_assistant_schedule_query",
|
||||||
@@ -254,6 +342,31 @@ func opsAIAssistantPurchaseQuerySchema() FunctionToolSchema {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func opsAIAssistantWorkOrderQuerySchema() FunctionToolSchema {
|
||||||
|
return FunctionToolSchema{
|
||||||
|
Name: "ops_ai_assistant_work_order_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", "checked", "parts_ordered", "repaired", "returned", "unrepairable"}, "description": "工单状态过滤:pending 待处理,checked 已检查,parts_ordered 已下单零件,repaired 已维修,returned 已送还,unrepairable 无法维修。"},
|
||||||
|
"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) {
|
func executeOpsAIAssistantCurrentUser(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) {
|
||||||
var args CurrentUserArgs
|
var args CurrentUserArgs
|
||||||
if len(rawArgs) > 0 {
|
if len(rawArgs) > 0 {
|
||||||
@@ -361,6 +474,78 @@ func executeOpsAIAssistantPurchaseQuery(ctx context.Context, runtime FunctionToo
|
|||||||
return json.Marshal(result)
|
return json.Marshal(result)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func executeOpsAIAssistantWorkOrderQuery(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) {
|
||||||
|
var args WorkOrderQueryArgs
|
||||||
|
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_work_order_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 registeredWorkOrderProvider == nil {
|
||||||
|
return json.Marshal(map[string]interface{}{
|
||||||
|
"ok": false,
|
||||||
|
"error": "工单查询服务未注册",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
result, err := registeredWorkOrderProvider.QueryWorkOrders(ctx, WorkOrderQuery{
|
||||||
|
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) {
|
func executeOpsAIAssistantScheduleQuery(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) {
|
||||||
var args ScheduleQueryArgs
|
var args ScheduleQueryArgs
|
||||||
if err := json.Unmarshal(rawArgs, &args); err != nil {
|
if err := json.Unmarshal(rawArgs, &args); err != nil {
|
||||||
|
|||||||
@@ -108,6 +108,7 @@ func ConfigAllInit() error {
|
|||||||
{Name: "time", Enabled: true, Description: "解析当前时间、相对日期和日期范围。"},
|
{Name: "time", Enabled: true, Description: "解析当前时间、相对日期和日期范围。"},
|
||||||
{Name: "ops_ai_assistant_current_user", 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_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_calendar_list", Enabled: true, Description: "查询当前用户可见的 OPS 日历列表。"},
|
||||||
{Name: "ops_ai_assistant_schedule_query", Enabled: true, Description: "按日期范围查询当前用户可见的 OPS 日历/日程。"},
|
{Name: "ops_ai_assistant_schedule_query", Enabled: true, Description: "按日期范围查询当前用户可见的 OPS 日历/日程。"},
|
||||||
{Name: "ops_ai_assistant_schedule_create", Enabled: true, Description: "创建 OPS 日历日程。"},
|
{Name: "ops_ai_assistant_schedule_create", Enabled: true, Description: "创建 OPS 日历日程。"},
|
||||||
|
|||||||
@@ -319,7 +319,7 @@ func handleChat(ctx *gin.Context) {
|
|||||||
toolNames = append(toolNames, tool.Function.Name)
|
toolNames = append(toolNames, tool.Function.Name)
|
||||||
}
|
}
|
||||||
emitTrace("function_tools", "prepare", "success", "已启用 Function Calling 工具", map[string]interface{}{"tools": toolNames})
|
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
|
var toolExecuted bool
|
||||||
openaiMsgs, toolExecuted, err = runOpenAIToolLoop(ctx.Request.Context(), profile, openaiMsgs, functionTools, currentUser, tracker, emitTrace)
|
openaiMsgs, toolExecuted, err = runOpenAIToolLoop(ctx.Request.Context(), profile, openaiMsgs, functionTools, currentUser, tracker, emitTrace)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -111,6 +111,7 @@ func ensureBuiltinAIChatTools() error {
|
|||||||
{Name: "time", Enabled: true, Description: "解析当前时间、相对日期和日期范围。", SortOrder: 0},
|
{Name: "time", Enabled: true, Description: "解析当前时间、相对日期和日期范围。", SortOrder: 0},
|
||||||
{Name: "ops_ai_assistant_current_user", Enabled: true, Description: "返回当前登录用户信息;未登录时提示需要登录才能获取信息。", SortOrder: 5},
|
{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_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_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_query", Enabled: true, Description: "按日期范围查询当前用户可见的 OPS 日历/日程。", SortOrder: 12},
|
||||||
{Name: "ops_ai_assistant_schedule_create", Enabled: true, Description: "创建 OPS 日历日程。", SortOrder: 14},
|
{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: "time", Enabled: true, Description: "解析当前时间、相对日期和日期范围。"},
|
||||||
{Name: "ops_ai_assistant_current_user", 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_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_calendar_list", Enabled: true, Description: "查询当前用户可见的 OPS 日历列表。"},
|
||||||
{Name: "ops_ai_assistant_schedule_query", Enabled: true, Description: "按日期范围查询当前用户可见的 OPS 日历/日程。"},
|
{Name: "ops_ai_assistant_schedule_query", Enabled: true, Description: "按日期范围查询当前用户可见的 OPS 日历/日程。"},
|
||||||
{Name: "ops_ai_assistant_schedule_create", Enabled: true, Description: "创建 OPS 日历日程。"},
|
{Name: "ops_ai_assistant_schedule_create", Enabled: true, Description: "创建 OPS 日历日程。"},
|
||||||
|
|||||||
@@ -1,8 +1,10 @@
|
|||||||
package routers
|
package routers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
parsefmt "fmt"
|
parsefmt "fmt"
|
||||||
|
"ops/agents"
|
||||||
"ops/models"
|
"ops/models"
|
||||||
"slices"
|
"slices"
|
||||||
"time"
|
"time"
|
||||||
@@ -81,6 +83,274 @@ type PurchaseOrderInfo struct {
|
|||||||
Status string `json:"status"`
|
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() {
|
func ApiWorkOrderInit() {
|
||||||
@@ -96,6 +366,8 @@ func ApiWorkOrderInit() {
|
|||||||
workOrderUserGroup.Type = "usergroup"
|
workOrderUserGroup.Type = "usergroup"
|
||||||
models.DB.Create(&workOrderUserGroup)
|
models.DB.Create(&workOrderUserGroup)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
agents.RegisterWorkOrderProvider(workOrderProvider{})
|
||||||
}
|
}
|
||||||
|
|
||||||
// ---------- 路由注册 ----------
|
// ---------- 路由注册 ----------
|
||||||
|
|||||||
Reference in New Issue
Block a user