123 lines
4.5 KiB
Go
123 lines
4.5 KiB
Go
package biz
|
|
|
|
// import (
|
|
// "ai_scheduler/internal/config"
|
|
// "ai_scheduler/internal/data/impl"
|
|
// "ai_scheduler/internal/data/model"
|
|
// "ai_scheduler/internal/entitys"
|
|
// "ai_scheduler/internal/pkg"
|
|
// "ai_scheduler/internal/pkg/utils_ollama"
|
|
// "ai_scheduler/internal/tools"
|
|
// "ai_scheduler/utils"
|
|
// "encoding/json"
|
|
// "flag"
|
|
// "fmt"
|
|
// "os"
|
|
// "path/filepath"
|
|
// "testing"
|
|
|
|
// "github.com/gofiber/fiber/v2/log"
|
|
// )
|
|
|
|
// func Test_task(t *testing.T) {
|
|
// var c entitys.TaskConfig
|
|
// config := `{"param": {"type": "object", "required": ["number"], "properties": {"number": {"type": "string", "description": "订单编号/流水号"}}}, "request": {"url": "http://www.baidu.com/${number}", "headers": {"Authorization": "${authorization}"}, "method": "GET"}}`
|
|
// err := json.Unmarshal([]byte(config), &c)
|
|
// t.Log(err)
|
|
// }
|
|
|
|
// type configData struct {
|
|
// Param map[string]interface{} `json:"param"`
|
|
// Do map[string]interface{} `json:"do"`
|
|
// }
|
|
|
|
// func Test_Order(t *testing.T) {
|
|
// routerBiz := in()
|
|
// ch := make(chan entitys.Response, 5)
|
|
// defer close(ch)
|
|
// err := routerBiz.handleTask(ch, nil, &entitys.Match{Index: "order_diagnosis", Parameters: `{"order_number":"822895927188791297"}`}, &model.AiTask{Config: `{"tool": "zltxOrderDetail", "param": {"type": "object", "optional": [], "required": ["order_number"], "properties": {"order_number": {"type": "string", "description": "订单编号/流水号"}}}}`})
|
|
// select {
|
|
// case v := <-ch: // 尝试接收
|
|
// fmt.Println("接收到值:", v)
|
|
// default:
|
|
// fmt.Println("无数据可接收")
|
|
// }
|
|
// t.Log(err)
|
|
// }
|
|
|
|
// func Test_OrderLog(t *testing.T) {
|
|
// routerBiz := in()
|
|
// ch := make(chan entitys.Response, 5)
|
|
// defer close(ch)
|
|
// err := routerBiz.handleTask(ch, nil, &entitys.Match{Index: "order_diagnosis", Parameters: `{"order_number":"822979421673758721","serial_number":"822979421979938817"}`}, &model.AiTask{Config: `{"tool": "zltxOrderDirectLog", "param": {"type": "object", "optional": [], "required": ["order_number"], "properties": {"order_number": {"type": "string", "description": "订单编号/流水号"}}}}`})
|
|
// t.Log(err)
|
|
// }
|
|
|
|
// func Test_ProductLog(t *testing.T) {
|
|
// routerBiz := in()
|
|
// ch := make(chan entitys.Response, 5)
|
|
// defer close(ch)
|
|
// err := routerBiz.handleTask(ch, nil, &entitys.Match{Index: "order_diagnosis", Parameters: `{"name":"利楚测试"}`}, &model.AiTask{Config: `{"tool": "zltxProduct", "param": {"type": "object", "optional": [], "required": ["order_number"], "properties": {"order_number": {"type": "string", "description": "订单编号/流水号"}}}}`})
|
|
// t.Log(err)
|
|
// }
|
|
|
|
// func Test_ZltxStatistics(t *testing.T) {
|
|
// routerBiz := in()
|
|
// ch := make(chan entitys.Response, 5)
|
|
// defer close(ch)
|
|
// err := routerBiz.handleTask(ch, nil, &entitys.Match{Index: "order_diagnosis", Parameters: `{"number":"13737882067"}`}, &model.AiTask{Config: `{"tool": "zltxOrderStatistics", "param": {"type": "object", "optional": [], "required": ["number"], "properties": {"number": {"type": "string", "description": "充值账号/分销商ID"}}}}`})
|
|
// t.Log(err)
|
|
// }
|
|
|
|
// func in() *AiRouterBiz {
|
|
|
|
// modDir, err := getModuleDir()
|
|
// if err != nil {
|
|
// panic("1")
|
|
// }
|
|
// configPath := flag.String("config", fmt.Sprintf("%s/config/config.yaml", modDir), "Path to configuration file")
|
|
// flag.Parse()
|
|
|
|
// configConfig, err := config.LoadConfig(*configPath)
|
|
// if err != nil {
|
|
// panic("加载配置失败")
|
|
// }
|
|
// client, _, err := utils_ollama.NewClient(configConfig)
|
|
// allLogger := log.DefaultLogger()
|
|
// utilOllama := utils_ollama.NewUtilOllama(configConfig, allLogger)
|
|
// manager := tools.NewManager(configConfig, client)
|
|
|
|
// db, _ := utils.NewGormDb(configConfig)
|
|
// sessionImpl := impl.NewSessionImpl(db)
|
|
// sysImpl := impl.NewSysImpl(db)
|
|
// taskImpl := impl.NewTaskImpl(db)
|
|
// chatImpl := impl.NewChatImpl(db)
|
|
// safeChannelPool, _ := pkg.NewSafeChannelPool(configConfig)
|
|
// routerBiz := NewAiRouterBiz(manager, sessionImpl, sysImpl, taskImpl, chatImpl, configConfig, utilOllama, safeChannelPool, client)
|
|
|
|
// return routerBiz
|
|
// }
|
|
|
|
// func getModuleDir() (string, error) {
|
|
// dir, err := os.Getwd()
|
|
// if err != nil {
|
|
// return "", err
|
|
// }
|
|
|
|
// for {
|
|
// modPath := filepath.Join(dir, "go.mod")
|
|
// if _, err := os.Stat(modPath); err == nil {
|
|
// return dir, nil // 找到 go.mod
|
|
// }
|
|
|
|
// // 向上查找父目录
|
|
// parent := filepath.Dir(dir)
|
|
// if parent == dir {
|
|
// break // 到达根目录，未找到
|
|
// }
|
|
// dir = parent
|
|
// }
|
|
|
|
// return "", fmt.Errorf("go.mod not found in current directory or parents")
|
|
// }
|