fuzhongyun
/
ai_scheduler
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: fuzhongyun:a8cfb118febc0d004624ba91a138b8c0b00b0b3f
Branches Tags
fuzhongyun:master
fuzhongyun:test
fuzhongyun:v3
fuzhongyun:v2
fuzhongyun:feature/wc/v2
fuzhongyun:feature/toolSelect
fuzhongyun:dev
fuzhongyun:feature/session
fuzhongyun:feature/fiber
..
compare: fuzhongyun:5067932e9246299b92f42523e6b66ad276e6d1bc
Branches Tags
fuzhongyun:master
fuzhongyun:test
fuzhongyun:v3
fuzhongyun:v2
fuzhongyun:feature/wc/v2
fuzhongyun:feature/toolSelect
fuzhongyun:dev
fuzhongyun:feature/session
fuzhongyun:feature/fiber

No commits in common. "a8cfb118febc0d004624ba91a138b8c0b00b0b3f" and "5067932e9246299b92f42523e6b66ad276e6d1bc" have entirely different histories.
a8cfb118fe ... 5067932e92

25 changed files with 235 additions and 415 deletions
Show Stats Expand all files Collapse all files

1
README2.md Normal file
View File

File diff suppressed because one or more lines are too long

2
cmd/server/main.go
View File

@ -10,7 +10,7 @@ import (
)
func main() {
configPath := flag.String("config", "./config/config_test.yaml", "Path to configuration file")
configPath := flag.String("config", "./config/config.yaml", "Path to configuration file")
flag.Parse()
bc, err := config.LoadConfig(*configPath)
if err != nil {

5
config/config_test.yaml
View File

@ -7,7 +7,7 @@ ollama:
base_url: "http://127.0.0.1:11434"
model: "qwen3-coder:480b-cloud"
generate_model: "qwen3-coder:480b-cloud"
vl_model: "qwen2.5vl:3b"
vl_model: "qwen2.5vl:7b"
timeout: "120s"
level: "info"
format: "json"
@ -55,9 +55,6 @@ tools:
enabled: true
api_key: "dingsbbntrkeiyazcfdg"
api_secret: "ObqxwyR20r9rVNhju0sCPQyQA98_FZSc32W4vgxnGFH_b02HZr1BPCJsOAF816nu"
zltxOrderAfterSaleSupplier:
enabled: true
base_url: "https://revcl.1688sup.com/api/admin/afterSales/reseller_supplier"
default_prompt:
img_recognize:

2
gen.sh
View File

@ -16,4 +16,4 @@
gentool --dsn "root:SD###sdf323r343@tcp(121.199.38.107:3306)/sys_ai_test?charset=utf8mb4&parseTime=true&loc=Asia%2FShanghai" -outPath ${modeldir} -onlyModel -modelPkgName "model" -tables ${prefix}${tables}
gentool --dsn "root:SD###sdf323r343@tcp(121.199.38.107:3306)/sys_ai?charset=utf8mb4&parseTime=true&loc=Asia%2FShanghai" -outPath ${modeldir} -onlyModel -modelPkgName "model" -tables ${prefix}${tables}

BIN
img.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 3.8 MiB

56
internal/biz/chat_history.go
View File

@ -5,8 +5,6 @@ import (
"ai_scheduler/internal/data/model"
"ai_scheduler/internal/entitys"
"context"
"xorm.io/builder"
)
type ChatHistoryBiz struct {
@ -17,41 +15,35 @@ func NewChatHistoryBiz(chatRepo *impl.ChatImpl) *ChatHistoryBiz {
s := &ChatHistoryBiz{
chatRepo: chatRepo,
}
//go s.AsyncProcess(context.Background())
go s.AsyncProcess(context.Background())
return s
}
//func (s *ChatHistoryBiz) create(ctx context.Context, sessionID, role, content string) error {
// chat := model.AiChatHi{
// SessionID: sessionID,
// Role: role,
// Content: content,
// }
//
// return s.chatRepo.Create(&chat)
//}
//
//// 添加会话历史
//func (s *ChatHistoryBiz) Create(ctx context.Context, chat entitys.ChatHistory) error {
// return s.create(ctx, chat.SessionID, chat.Role.String(), chat.Content)
//}
func (s *ChatHistoryBiz) create(ctx context.Context, sessionID, role, content string) error {
chat := model.AiChatHi{
SessionID: sessionID,
Role: role,
Content: content,
}
return s.chatRepo.Create(&chat)
}
// 添加会话历史
func (s *ChatHistoryBiz) Create(ctx context.Context, chat entitys.ChatHistory) error {
return s.create(ctx, chat.SessionID, chat.Role.String(), chat.Content)
}
// 异步添加会话历史
//func (s *ChatHistoryBiz) AsyncCreate(ctx context.Context, chat entitys.ChatHistory) {
// s.chatRepo.AsyncCreate(ctx, model.AiChatHi{
// SessionID: chat.SessionID,
// Role: chat.Role.String(),
// Content: chat.Content,
// })
//}
func (s *ChatHistoryBiz) AsyncCreate(ctx context.Context, chat entitys.ChatHistory) {
s.chatRepo.AsyncCreate(ctx, model.AiChatHi{
SessionID: chat.SessionID,
Role: chat.Role.String(),
Content: chat.Content,
})
}
// 异步处理会话历史
//func (s *ChatHistoryBiz) AsyncProcess(ctx context.Context) {
// s.chatRepo.AsyncProcess(ctx)
//}
func (s *ChatHistoryBiz) Update(ctx context.Context, chat *entitys.UseFulRequest) error {
cond := builder.NewCond()
cond = cond.And(builder.Eq{"his_id": chat.HisId})
return s.chatRepo.UpdateByCond(&cond, &model.AiChatHi{HisID: chat.HisId, Useful: chat.Useful})
func (s *ChatHistoryBiz) AsyncProcess(ctx context.Context) {
s.chatRepo.AsyncProcess(ctx)
}

34
internal/biz/do/ctx.go
View File

@ -94,7 +94,7 @@ func (d *Do) DataAuth(c *websocket.Conn) (err error) {
func (d *Do) MakeCh(c *websocket.Conn) (ctx context.Context, deferFunc func()) {
d.Ctx.Ch = make(chan entitys.Response)
ctx, cancel := context.WithCancel(context.Background())
done := d.startMessageHandler(ctx, c)
done := d.startMessageHandler(ctx, c, d.hisImpl)
return ctx, func() {
close(d.Ctx.Ch) //关闭主通道
<-done // 等待消息处理完成
@ -186,7 +186,7 @@ func (d *Do) getTasks(sysId int32) (tasks []model.AiTask, err error) {
func (d *Do) startMessageHandler(
ctx context.Context,
c *websocket.Conn,
hisImpl *impl.ChatImpl,
) <-chan struct{} {
done := make(chan struct{})
var chat []string
@ -195,25 +195,23 @@ func (d *Do) startMessageHandler(
defer func() {
close(done)
// 保存历史记录
var (
hisLog = &entitys.ChatHisLog{}
)
if len(chat) > 0 {
AiRes := &model.AiChatHi{
var his = []*model.AiChatHi{
{
SessionID: d.Ctx.Session,
Ques: d.Ctx.Req.Text,
Ans: strings.Join(chat, ""),
Files: d.Ctx.Req.Img,
Role: "user",
Content: d.Ctx.Req.Text, // 用户输入在外部处理
},
}
d.hisImpl.AddWithData(AiRes)
hisLog.HisId = AiRes.HisID
}
_ = entitys.MsgSend(c, entitys.Response{
Content: pkg.JsonStringIgonErr(hisLog),
Type: entitys.ResponseEnd,
if len(chat) > 0 {
his = append(his, &model.AiChatHi{
SessionID: d.Ctx.Session,
Role: "assistant",
Content: strings.Join(chat, ""),
})
}
for _, hi := range his {
hisImpl.Add(hi)
}
}()
for v := range d.Ctx.Ch { // 自动检测通道关闭

2
internal/biz/do/handle.go
View File

@ -90,7 +90,7 @@ func (r *Handle) HandleMatch(ctx context.Context, requireData *entitys.RequireDa
if pointTask == nil || pointTask.Index == "other" {
return r.OtherTask(ctx, requireData)
}
switch constants.TaskType(pointTask.Type) {
switch pointTask.Type {
case constants.TaskTypeApi:
return r.handleApiTask(ctx, requireData, pointTask)
case constants.TaskTypeFunc:

39
internal/biz/llm_service/common.go
View File

@ -1,11 +1,9 @@
package llm_service
import (
"ai_scheduler/internal/data/constants"
"ai_scheduler/internal/data/model"
"ai_scheduler/internal/entitys"
"context"
"time"
)
type LlmService interface {
@ -26,18 +24,11 @@ func buildAssistant(his []model.AiChatHi) (chatHis entitys.ChatHis) {
if len(chatHis.SessionId) == 0 {
chatHis.SessionId = item.SessionID
}
chatHis.Messages = append(chatHis.Messages, []entitys.HisMessage{
{
Role: constants.RoleUser,
Content: item.Ques,
Timestamp: item.CreateAt.Format(time.DateTime),
},
{
Role: constants.RoleAssistant,
Content: item.Ans,
Timestamp: item.CreateAt.Format(time.DateTime),
},
}...)
chatHis.Messages = append(chatHis.Messages, entitys.HisMessage{
Role: item.Role,
Content: item.Content,
Timestamp: item.CreateAt.Format("2006-01-02 15:04:05"),
})
}
chatHis.Context = entitys.HisContext{
UserLanguage: "zh-CN",
@ -45,23 +36,3 @@ func buildAssistant(his []model.AiChatHi) (chatHis entitys.ChatHis) {
}
return
}
func BuildChatHisMessage(his []model.AiChatHi) (chatHis []entitys.HisMessage) {
for _, item := range his {
chatHis = append(chatHis, []entitys.HisMessage{
{
Role: constants.RoleUser,
Content: item.Ques,
Timestamp: item.CreateAt.Format(time.DateTime),
},
{
Role: constants.RoleAssistant,
Content: item.Ans,
Timestamp: item.CreateAt.Format(time.DateTime),
},
}...)
}
return
}

52
internal/biz/llm_service/ollama.go
View File

@ -2,7 +2,6 @@ package llm_service
import (
"ai_scheduler/internal/config"
"ai_scheduler/internal/data/impl"
"ai_scheduler/internal/data/model"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
@ -14,24 +13,20 @@ import (
"time"
"github.com/ollama/ollama/api"
"xorm.io/builder"
)
type OllamaService struct {
client *utils_ollama.Client
config *config.Config
chatHis *impl.ChatImpl
}
func NewOllamaGenerate(
client *utils_ollama.Client,
config *config.Config,
chatHis *impl.ChatImpl,
) *OllamaService {
return &OllamaService{
client: client,
config: config,
chatHis: chatHis,
}
}
@ -71,10 +66,6 @@ func (r *OllamaService) getPrompt(ctx context.Context, requireData *entitys.Requ
var (
prompt = make([]api.Message, 0)
)
content, err := r.getUserContent(ctx, requireData)
if err != nil {
return nil, err
}
prompt = append(prompt, api.Message{
Role: "system",
Content: buildSystemPrompt(requireData.Sys.SysPrompt),
@ -83,17 +74,32 @@ func (r *OllamaService) getPrompt(ctx context.Context, requireData *entitys.Requ
Content: "### 聊天记录:" + pkg.JsonStringIgonErr(buildAssistant(requireData.Histories)),
}, api.Message{
Role: "user",
Content: content,
Content: r.getUserContent(requireData),
//Images: requireData.ImgByte,
})
if len(requireData.ImgByte) > 0 {
desc, err := r.RecognizeWithImg(ctx, requireData)
if err != nil {
return nil, err
}
var imgs strings.Builder
imgs.WriteString("### 用户上传图片解析内容:\n")
prompt = append(prompt, api.Message{
Role: "image_desc",
Content: "" + desc.Response,
})
}
return prompt, nil
}
func (r *OllamaService) getUserContent(ctx context.Context, requireData *entitys.RequireData) (string, error) {
func (r *OllamaService) getUserContent(requireData *entitys.RequireData) string {
var content strings.Builder
content.WriteString(requireData.Req.Text)
if len(requireData.ImgByte) > 0 {
content.WriteString("\n")
content.WriteString("### 图片内容已经解析到image_desc里")
}
if len(requireData.Req.Tags) > 0 {
@ -101,29 +107,7 @@ func (r *OllamaService) getUserContent(ctx context.Context, requireData *entitys
content.WriteString("### 工具必须使用：")
content.WriteString(requireData.Req.Tags)
}
if len(requireData.ImgByte) > 0 {
desc, err := r.RecognizeWithImg(ctx, requireData)
if err != nil {
return "", err
}
content.WriteString("### 上传图片解析内容:\n")
content.WriteString(requireData.Req.Tags)
content.WriteString(desc.Response)
}
if requireData.Req.MarkHis > 0 {
var his model.AiChatHi
cond := builder.NewCond()
cond = cond.And(builder.Eq{"his_id": requireData.Req.MarkHis})
err := r.chatHis.GetOneBySearchToStrut(&cond, &his)
if err != nil {
return "", err
}
content.WriteString("### 引用历史聊天记录:\n")
content.WriteString(pkg.JsonStringIgonErr(BuildChatHisMessage([]model.AiChatHi{his})))
}
return content.String(), nil
return content.String()
}
func (r *OllamaService) RecognizeWithImg(ctx context.Context, requireData *entitys.RequireData) (desc api.GenerateResponse, err error) {

26
internal/biz/session.go
View File

@ -80,11 +80,11 @@ func (s *SessionBiz) SessionInit(ctx context.Context, req *entitys.SessionInitRe
result.Prologue = sysConfig.Prologue
// 开场白写入会话历史
//s.chatRepo.AsyncCreate(ctx, model.AiChatHi{
// SessionID: chat.SessionID,
// Role: chat.Role.String(),
// Content: chat.Content,
//})
s.chatRepo.AsyncCreate(ctx, model.AiChatHi{
SessionID: chat.SessionID,
Role: chat.Role.String(),
Content: chat.Content,
})
} else {
result.SessionId = session.SessionID
@ -102,20 +102,12 @@ func (s *SessionBiz) SessionInit(ctx context.Context, req *entitys.SessionInitRe
// 转换为 entitys.ChatHistory 类型
for _, chat := range chatList {
result.Chat = append(result.Chat, []entitys.ChatHistory{
{
result.Chat = append(result.Chat, entitys.ChatHistory{
SessionID: chat.SessionID,
Role: constants.RoleUser,
Content: chat.Ques,
Role: constants.Caller(chat.Role),
Content: chat.Content,
Prologue: sysConfig.Prologue,
},
{
SessionID: chat.SessionID,
Role: constants.RoleAssistant,
Content: chat.Ans,
Prologue: sysConfig.Prologue,
},
}...)
})
}
}

2
internal/config/config.go
View File

@ -93,8 +93,6 @@ type ToolsConfig struct {
ZltxOrderAfterSaleDetail ToolConfig `mapstructure:"zltxOrderAfterSaleDetail"`
//下游订单预检
ZltxOrderAfterSalePreCheck ToolConfig `mapstructure:"zltxOrderAfterSalePreCheck"`
// 上游订单售后
ZltxOrderAfterSaleSupplier ToolConfig `mapstructure:"zltxOrderAfterSaleSupplier"`
}
// ToolConfig 单个工具配置

3
internal/data/constants/bot.go
View File

@ -4,7 +4,4 @@ type BotTools string
const (
BotToolsBugOptimizationSubmit = "bug_optimization_submit" // 系统的bug/优化建议
BotToolsAfterSalesSupplier = "after_sales_supplier" // 供应商售后
BotToolsAfterSalesResellerSingle = "after_sales_reseller_single" // 分销商单条售后
BotToolsAfterSalesResellerBatch = "after_sales_reseller_batch" // 分销商批量售后
)

22
internal/data/constants/const.go
View File

@ -11,22 +11,8 @@ const (
type TaskType int32
const (
TaskTypeApi TaskType = 1
TaskTypeKnowle TaskType = 2
TaskTypeFunc TaskType = 3
TaskTypeBot TaskType = 4
TaskTypeApi = 1
TaskTypeKnowle = 2
TaskTypeFunc = 3
TaskTypeBot = 4
)
type UseFul int32
const (
UseFulNotSolve UseFul = 2
UseFulNotUnclear UseFul = 3
UseFulNotError UseFul = 4
)
var UseFulMap = map[UseFul]string{
UseFulNotSolve: "未解决问题",
UseFulNotUnclear: "回答不明确",
UseFulNotError: "理解错误",
}

11
internal/data/model/ai_chat_his.gen.go
View File

@ -12,14 +12,11 @@ const TableNameAiChatHi = "ai_chat_his"
// AiChatHi mapped from table <ai_chat_his>
type AiChatHi struct {
HisID int64 `gorm:"column:his_id;primaryKey;autoIncrement:true" json:"his_id"`
HisID int64 `gorm:"column:his_id;primaryKey" json:"his_id"`
SessionID string `gorm:"column:session_id;not null" json:"session_id"`
Ques string `gorm:"column:ques;not null" json:"ques"`
Ans string `gorm:"column:ans;not null" json:"ans"`
Files string `gorm:"column:files;not null" json:"files"`
Useful int32 `gorm:"column:useful;not null;comment:0不评价,1有用，其他为无用" json:"useful"` // 0不评价,1有用，其他为无用
CreateAt time.Time `gorm:"column:create_at;default:CURRENT_TIMESTAMP" json:"create_at"`
UpdatedAt time.Time `gorm:"column:updated_at;default:CURRENT_TIMESTAMP" json:"updated_at"`
Role string `gorm:"column:role;not null;comment:system系统输出，assistant助手输出,user用户输入" json:"role"` // system系统输出，assistant助手输出,user用户输入
Content string `gorm:"column:content;not null" json:"content"`
CreateAt *time.Time `gorm:"column:create_at;default:CURRENT_TIMESTAMP" json:"create_at"`
}
// TableName AiChatHi's table name

4
internal/entitys/chat_history.go
View File

@ -10,7 +10,3 @@ type ChatHistory struct {
Content string `json:"content"`
Prologue string `json:"prologue"`
}
type ChatHisLog struct {
HisId int64 `json:"his_id"`
}

5
internal/entitys/session.go
View File

@ -23,8 +23,3 @@ type SessionListRequest struct {
type TaskRequest struct {
SysId int32 `json:"sys_id"`
}
type UseFulRequest struct {
HisId int64 `json:"his_id"`
Useful int32 `json:"useful"`
}

5
internal/entitys/types.go
View File

@ -1,7 +1,6 @@
package entitys
import (
"ai_scheduler/internal/data/constants"
"ai_scheduler/internal/data/model"
"context"
@ -32,9 +31,7 @@ type FirstSockRequest struct {
type ChatSockRequest struct {
Text string `json:"text" binding:"required"`
Img string `json:"img" binding:"required"`
File string `json:"file" binding:"required"`
Tags string `json:"tags" binding:"required"`
MarkHis int64 `json:"mark_his" `
Caller string `json:"caller" binding:"required"`
SessionID string `json:"session_id"`
}
@ -133,7 +130,7 @@ type ChatHis struct {
Context HisContext `json:"context"`
}
type HisMessage struct {
Role constants.Caller `json:"role"`
Role string `json:"role"`
Content string `json:"content"`
Timestamp string `json:"timestamp"`
}

4
internal/server/router/router.go
View File

@ -50,11 +50,7 @@ func SetupRoutes(app *fiber.App, ChatService *services.ChatService, sessionServi
r.Post("/session/init", sessionService.SessionInit) // 会话初始化,不存在则创建,存在则返回会话ID和默认条数会话历史
r.Post("/session/list", sessionService.SessionList)
r.Post("/sys/tasks", task.Tasks)
// 评价
r.Post("/chat/useful/list", ChatService.UsefulList)
r.Post("/chat/useful", ChatService.Useful)
// 回调
r.Post("/callback", callbackService.Callback)
//广播

34
internal/services/chat.go
View File

@ -13,7 +13,6 @@ import (
"sync"
"time"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/websocket/v2"
)
@ -22,19 +21,13 @@ type ChatService struct {
routerBiz *biz.AiRouterBiz
Gw *gateway.Gateway
mu sync.Mutex
ChatHis *biz.ChatHistoryBiz
}
// NewChatHandler 创建聊天处理器
func NewChatService(
routerService *biz.AiRouterBiz,
chatHis *biz.ChatHistoryBiz,
gw *gateway.Gateway,
) *ChatService {
func NewChatService(routerService *biz.AiRouterBiz, gw *gateway.Gateway) *ChatService {
return &ChatService{
routerBiz: routerService,
Gw: gw,
ChatHis: chatHis,
}
}
@ -125,7 +118,10 @@ func (h *ChatService) Chat(c *websocket.Conn) {
Type: entitys.ResponseText,
})
}
_ = entitys.MsgSend(c, entitys.Response{
Content: "",
Type: entitys.ResponseEnd,
})
}
h.Gw.RemoveClient(clientID)
_ = c.Close()
@ -152,23 +148,3 @@ func (h *ChatService) handleMessageToString(c *websocket.Conn, msgType int, msg
}
return msg.([]byte), constants.ConnStatusIgnore
}
func (s *ChatService) Useful(c *fiber.Ctx) error {
req := &entitys.UseFulRequest{}
if err := c.BodyParser(req); err != nil {
return err
}
err := s.ChatHis.Update(c.Context(), req)
if err != nil {
return err
}
return nil
}
func (s *ChatService) UsefulList(c *fiber.Ctx) error {
return c.JSON(constants.UseFulMap)
}

6
internal/tools/manager.go
View File

@ -81,12 +81,6 @@ func NewManager(config *config.Config, llm *utils_ollama.Client) *Manager {
m.tools[zltxOrderAfterSalePreCheckTool.Name()] = zltxOrderAfterSalePreCheckTool
}
// 注册直连天下上游售后订单工具
if config.Tools.ZltxOrderAfterSaleSupplier.Enabled {
zltxOrderAfterSaleSupplierTool := NewZltxOrderAfterSaleSupplierTool(config.Tools.ZltxOrderAfterSaleSupplier)
m.tools[zltxOrderAfterSaleSupplierTool.Name()] = zltxOrderAfterSaleSupplierTool
}
// 普通对话
chat := NewNormalChatTool(m.llm, config)
m.tools[chat.Name()] = chat

131
internal/tools/zltx_after_sale.go Normal file
View File

@ -0,0 +1,131 @@
package tools
import (
"ai_scheduler/internal/config"
"ai_scheduler/internal/entitys"
"context"
"encoding/json"
"fmt"
"sort"
"gitea.cdlsxd.cn/self-tools/l_request"
)
type ZltxOrderAfterSaleTool struct {
config config.ToolConfig
}
func (z ZltxOrderAfterSaleTool) Name() string {
return "zltxOrderAfterSale"
}
func (z ZltxOrderAfterSaleTool) Description() string {
return "查询直连天下订单售后信息"
}
func (z ZltxOrderAfterSaleTool) Definition() entitys.ToolDefinition {
return entitys.ToolDefinition{
Type: "function",
Function: entitys.FunctionDef{
Name: z.Name(),
Description: z.Description(),
Parameters: map[string]interface{}{
"type": "object",
"properties": map[string]interface{}{
"number": map[string]interface{}{
"type": "string",
"description": "账号或分销商号",
},
},
"required": []string{"number"},
},
},
}
}
type ZltxOrderAfterSaleRequest struct {
Number string `json:"number"`
}
func (z ZltxOrderAfterSaleTool) Execute(ctx context.Context, requireData *entitys.RequireData) error {
var req ZltxOrderAfterSaleRequest
if err := json.Unmarshal([]byte(requireData.Match.Parameters), &req); err != nil {
return err
}
if req.Number == "" {
return fmt.Errorf("number is required")
}
return z.getZltxOrderAfterSale(req.Number, requireData)
}
type ZltxOrderAfterSaleResponse struct {
Code int `json:"code"`
Data struct {
List []*ZltxOrderAfterSaleData `json:"list"`
} `json:"data"`
Error string `json:"error"`
}
type ZltxOrderAfterSaleData struct {
// 处理方式 价款/扣款
HandleMethod string `json:"handle_method"`
// 售后金额
AfterSaleAmount float64 `json:"after_sale_amount"`
// 售后原因
AfterSaleReason string `json:"after_sale_reason"`
// 原单信息
OriginalOrderInfo *OrderInfo
}
type OrderInfo struct {
SerialNumber string `json:"serial_number"` // 订单流水号
SupplierName string `json:"supplier_name"` // 供应商名称
SigningBody string `json:"signing_body"` // 签约主体
ProductName string `json:"product_name"` // 商品名称
UpstreamPrice float64 `json:"upstream_price"` // 上游价格
RechargeAccount string `json:"recharge_account"` // 充值账号
RechargeStatus string `json:"recharge_status"` // 充值状态
}
func (z ZltxOrderAfterSaleTool) getZltxOrderAfterSale(number string, requireData *entitys.RequireData) error {
//查询订单详情
url := fmt.Sprintf("%s%s", z.config.BaseURL, number)
req := l_request.Request{
Url: url,
Headers: map[string]string{
"Authorization": fmt.Sprintf("Bearer %s", requireData.Auth),
},
Method: "GET",
}
res, err := req.Send()
var resData ZltxOrderAfterSaleResponse
if err != nil {
return err
}
if err := json.Unmarshal(res.Content, &resData); err != nil {
return err
}
if resData.Code != 200 {
return fmt.Errorf("为获取到数据，请检查权限: %s", string(res.Content))
}
//按照日期排序
sort.Slice(resData.Data.RecentThreeDays, func(i, j int) bool {
return resData.Data.RecentThreeDays[i].Date < resData.Data.RecentThreeDays[j].Date
})
sort.Slice(resData.Data.RecentOneMonth, func(i, j int) bool {
return resData.Data.RecentOneMonth[i].Date < resData.Data.RecentOneMonth[j].Date
})
jsonByte, err := json.Marshal(resData)
if err != nil {
return err
}
entitys.ResJson(requireData.Ch, z.Name(), string(jsonByte))
return nil
}
func NewZltxOrderAfterSaleTool(config config.ToolConfig) *ZltxOrderAfterSaleTool {
return &ZltxOrderAfterSaleTool{
config: config,
}
}

169
internal/tools/zltx_order_after_supplier.go
View File

@ -1,169 +0,0 @@
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
}

1
internal/tools_bot/dtalk_bot.go
View File

@ -17,7 +17,6 @@ type BotTool struct {
llm *utils_ollama.Client
sessionImpl *impl.SessionImpl
taskMap map[string]string // task_id -> session_id
// zltxOrderAfterSaleTool tools.ZltxOrderAfterSaleTool
}
// NewBotTool 创建直连天下订单详情工具

8
tmpl/dataTemp/queryTempl.go
View File

@ -68,14 +68,6 @@ func (k DataTemp) Add(data interface{}) (id int, err error) {
return primary.Id, add.Error
}
func (k DataTemp) AddWithData(data interface{}) (interface{}, error) {
result := k.Db.Model(k.Model).Create(data)
if result.Error != nil {
return data, result.Error
}
return data, nil
}
func (k DataTemp) GetList(cond *builder.Cond, pageBoIn *ReqPageBo) (list []map[string]interface{}, pageBoOut *RespPageBo, err error) {
var (
query, _ = builder.ToBoundSQL(*cond)