package tools import ( "ai_scheduler/internal/config" "ai_scheduler/internal/entitys" "context" "encoding/json" "fmt" ) type ZltxOrderAfterSaleSupplierTool struct { config config.ToolConfig } // NewZltxOrderAfterSaleSupplierTool 创建售后订单预检工具 func NewZltxOrderAfterSaleSupplierTool(config config.ToolConfig) *ZltxOrderAfterSaleSupplierTool { return &ZltxOrderAfterSaleSupplierTool{config: config} } // Name 返回工具名称 func (t *ZltxOrderAfterSaleSupplierTool) Name() string { return "zltxOrderAfterSaleSupplier" } func (t *ZltxOrderAfterSaleSupplierTool) Description() string { return "直连天下售后订单供应商工具" } func (t *ZltxOrderAfterSaleSupplierTool) Definition() entitys.ToolDefinition { return entitys.ToolDefinition{ Type: "function", Function: entitys.FunctionDef{ Name: t.Name(), Description: t.Description(), Parameters: map[string]interface{}{ "type": "object", "properties": map[string]interface{}{ "orderType": map[string]interface{}{ "type": "integer", "description": "售后订单类型", }, "orderNumber": map[string]interface{}{ "type": "string", "description": "售后订单号", }, }, "required": []string{"orderType", "orderNumber"}, }, }, } } type ZltxOrderAfterSaleSupplierRequest struct { OrderNumber string `json:"orderNumber"` // 订单号 SerialNumber string `json:"serialNumber"` // 流水号 Account string `json:"account"` // 充值账号 OrderTimeStart string `json:"orderTimeStart"` // 订单执行开始时间 OrderTimeEnd string `json:"orderTimeEnd"` // 订单执行结束时间 AfterSalesReason string `json:"afterSalesReason"` // 售后原因 AfterSalesPrice string `json:"afterSalesPrice"` // 售后金额 } type ZltxOrderAfterSaleSupplierResponse struct { Code int `json:"code"` Msg string `json:"msg"` Data []CheckResult2 `json:"data"` } type CheckResult2 struct { OrderType int `json:"orderType"` OrderNumber string `json:"orderNumber"` OrderAmount float64 `json:"orderAmount"` OrderPrice float64 `json:"orderPrice"` SignCompany int `json:"signCompany"` OrderQuantity int `json:"orderQuantity"` ResellerID int `json:"resellerId"` ResellerName string `json:"resellerName"` OurProductID int `json:"ourProductId"` OurProductTitle string `json:"ourProductTitle"` Account []string `json:"account"` Platforms struct { Num4 string `json:"4"` } `json:"platforms"` } func (t *ZltxOrderAfterSaleSupplierTool) Execute(ctx context.Context, requireData *entitys.RequireData) error { var req ZltxOrderAfterSaleSupplierRequest if err := json.Unmarshal([]byte(requireData.Match.Parameters), &req); err != nil { return err } if req.OrderNumber == "" { return fmt.Errorf("orderType and orderNumber are required") } return t.checkZltxOrderAfterSaleSupplier(req.OrderNumber, requireData) } func (t *ZltxOrderAfterSaleSupplierTool) checkZltxOrderAfterSaleSupplier(orderNumber string, requireData *entitys.RequireData) error { // req := l_request.Request{ // Url: t.config.BaseURL, // Headers: map[string]string{ // "Authorization": fmt.Sprintf("Bearer %s", requireData.Auth), // }, // Method: "POST", // Data: map[string]string{ // // "orderType": fmt.Sprintf("%d", orderType), // "orderNumber": orderNumber, // }, // } // res, err := req.Send() // if err != nil { // return err // } // // 解析响应 // var resp ZltxOrderAfterSaleSupplierResponse // if err := json.Unmarshal(res.Content, &resp); err != nil { // return err // } // if resp.Code != 0 { // return fmt.Errorf("check failed: %s", resp.Msg) // } resp := ZltxOrderAfterSaleSupplierResponse{ Code: 0, Msg: "success", Data: []CheckResult2{ { OrderType: 1, OrderNumber: orderNumber, OrderAmount: 100, OrderPrice: 100, SignCompany: 1, OrderQuantity: 1, ResellerID: 1, ResellerName: "测试", OurProductID: 1, OurProductTitle: "测试", Account: []string{"123456"}, Platforms: struct { Num4 string `json:"4"` }{ Num4: "123456", }, }, { OrderType: 2, OrderNumber: orderNumber, OrderAmount: 100, OrderPrice: 100, SignCompany: 1, OrderQuantity: 1, ResellerID: 1, ResellerName: "测试", OurProductID: 1, OurProductTitle: "测试", Account: []string{"123456"}, Platforms: struct { Num4 string `json:"4"` }{ Num4: "123456", }, }, }, } jsonByte, err := json.Marshal(resp) if err != nil { return err } entitys.ResJson(requireData.Ch, t.Name(), string(jsonByte)) return nil }