fix:代码优化
This commit is contained in:
parent
5560e879d0
commit
32cd8691b7
|
|
@ -269,8 +269,8 @@ func (g *GroupConfigBiz) handleMatch(ctx context.Context, rec *entitys.Recognize
|
||||||
return g.handleReport(ctx, rec, pointTask, groupConfig)
|
return g.handleReport(ctx, rec, pointTask, groupConfig)
|
||||||
case constants.TaskTypeCozeWorkflow:
|
case constants.TaskTypeCozeWorkflow:
|
||||||
return g.handleCozeWorkflow(ctx, rec, pointTask)
|
return g.handleCozeWorkflow(ctx, rec, pointTask)
|
||||||
case constants.TaskTypeKnowle: // 知识库V2版本
|
case constants.TaskTypeKnowle: // 知识库lightRAG版本
|
||||||
return g.handleKnowledgeV2(ctx, rec, groupConfig, callback)
|
return g.handleKnowledge(ctx, rec, groupConfig, callback)
|
||||||
default:
|
default:
|
||||||
return g.otherTask(ctx, rec)
|
return g.otherTask(ctx, rec)
|
||||||
}
|
}
|
||||||
|
|
@ -477,7 +477,7 @@ func (g *GroupConfigBiz) GetReportCache(ctx context.Context, day time.Time, tota
|
||||||
}
|
}
|
||||||
|
|
||||||
// handleKnowledgeV2 处理知识库V2版本
|
// handleKnowledgeV2 处理知识库V2版本
|
||||||
func (g *GroupConfigBiz) handleKnowledgeV2(ctx context.Context, rec *entitys.Recognize, groupConfig *model.AiBotGroupConfig, callback *chatbot.BotCallbackDataModel) (err error) {
|
func (g *GroupConfigBiz) handleKnowledge(ctx context.Context, rec *entitys.Recognize, groupConfig *model.AiBotGroupConfig, callback *chatbot.BotCallbackDataModel) (err error) {
|
||||||
// 请求知识库工具
|
// 请求知识库工具
|
||||||
knowledgeBase := knowledge_base.New(g.conf.KnowledgeConfig)
|
knowledgeBase := knowledge_base.New(g.conf.KnowledgeConfig)
|
||||||
knowledgeResp, err := knowledgeBase.Query(&knowledge_base.QueryRequest{
|
knowledgeResp, err := knowledgeBase.Query(&knowledge_base.QueryRequest{
|
||||||
|
|
@ -591,13 +591,14 @@ func (g *GroupConfigBiz) shouldCreateIssueHandlingGroup(ctx context.Context, rec
|
||||||
}
|
}
|
||||||
issueOwnerStr := strings.Join(userNames, "、")
|
issueOwnerStr := strings.Join(userNames, "、")
|
||||||
|
|
||||||
|
// 获取机器人应用配置
|
||||||
botConfig, err := g.botConfigImpl.GetRobotConfig(callback.RobotCode)
|
botConfig, err := g.botConfigImpl.GetRobotConfig(callback.RobotCode)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("获取机器人配置失败,err: %v", err)
|
return fmt.Errorf("获取机器人配置失败,err: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 构建卡片 OutTrackId
|
// 构建卡片 OutTrackId
|
||||||
outTrackId := constants.BuildCardOutTrackId(callback.ConversationId, botConfig.ClientId)
|
outTrackId := constants.BuildCardOutTrackId(callback.ConversationId, callback.RobotCode)
|
||||||
|
|
||||||
// 发送钉钉卡片
|
// 发送钉钉卡片
|
||||||
_, err = g.dingtalkCardClient.CreateAndDeliver(dingtalk.AppKey{
|
_, err = g.dingtalkCardClient.CreateAndDeliver(dingtalk.AppKey{
|
||||||
|
|
|
||||||
|
|
@ -137,4 +137,4 @@ var GroupTemplateRobotIdMap = map[string]string{
|
||||||
GroupTemplateRobotIdIssueHandling: "ding5wwvnf9hxeyjau7t",
|
GroupTemplateRobotIdIssueHandling: "ding5wwvnf9hxeyjau7t",
|
||||||
}
|
}
|
||||||
|
|
||||||
const CardDebugToolEntryShow string = "show" // 卡片调试工具 [show:展示 hide:隐藏]
|
const CardDebugToolEntryShow string = "hide" // 卡片调试工具 [show:展示 hide:隐藏]
|
||||||
|
|
|
||||||
|
|
@ -35,12 +35,17 @@ func NewDingTalkBotServer(
|
||||||
for _, service := range services {
|
for _, service := range services {
|
||||||
serviceConfigs, err := service.GetServiceCfg()
|
serviceConfigs, err := service.GetServiceCfg()
|
||||||
for _, serviceConf := range serviceConfigs {
|
for _, serviceConf := range serviceConfigs {
|
||||||
|
// 配置不全
|
||||||
if serviceConf.ClientId == "" || serviceConf.ClientSecret == "" {
|
if serviceConf.ClientId == "" || serviceConf.ClientSecret == "" {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
// 非应用主机器人 pass ,防止重复
|
||||||
|
if serviceConf.ClientId != serviceConf.BotIndex {
|
||||||
|
continue
|
||||||
|
}
|
||||||
cli := DingBotServerInit(serviceConf.ClientId, serviceConf.ClientSecret, service)
|
cli := DingBotServerInit(serviceConf.ClientId, serviceConf.ClientSecret, service)
|
||||||
if cli == nil {
|
if cli == nil {
|
||||||
log.Info("%s客户端初始失败:%s", serviceConf.BotIndex, err.Error())
|
log.Infof("%s客户端初始失败:%s", serviceConf.BotIndex, err.Error())
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
clients[serviceConf.BotIndex] = cli
|
clients[serviceConf.BotIndex] = cli
|
||||||
|
|
@ -53,11 +58,9 @@ func NewDingTalkBotServer(
|
||||||
|
|
||||||
func ProvideAllDingBotServices(
|
func ProvideAllDingBotServices(
|
||||||
dingBotSvc *services.DingBotService,
|
dingBotSvc *services.DingBotService,
|
||||||
// dingCardSvc *services.DingtalkCardService,
|
|
||||||
) []DingBotServiceInterface {
|
) []DingBotServiceInterface {
|
||||||
return []DingBotServiceInterface{
|
return []DingBotServiceInterface{
|
||||||
dingBotSvc,
|
dingBotSvc,
|
||||||
// dingCardSvc,
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,37 +0,0 @@
|
||||||
package services
|
|
||||||
|
|
||||||
// 当前不用,后续切钉钉新sdk考虑使用
|
|
||||||
|
|
||||||
import (
|
|
||||||
"ai_scheduler/internal/biz"
|
|
||||||
"ai_scheduler/internal/config"
|
|
||||||
"ai_scheduler/internal/entitys"
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"gitea.cdlsxd.cn/self-tools/l-dingtalk-stream-sdk-go/card"
|
|
||||||
"gitea.cdlsxd.cn/self-tools/l-dingtalk-stream-sdk-go/chatbot"
|
|
||||||
)
|
|
||||||
|
|
||||||
type DingtalkCardService struct {
|
|
||||||
config *config.Config
|
|
||||||
dingtalkCardBiz *biz.DingTalkBotBiz
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewDingtalkCardService(config *config.Config, dingtalkCardBiz *biz.DingTalkBotBiz) *DingtalkCardService {
|
|
||||||
return &DingtalkCardService{
|
|
||||||
config: config,
|
|
||||||
dingtalkCardBiz: dingtalkCardBiz,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *DingtalkCardService) GetServiceCfg() (config []entitys.DingTalkBot, err error) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *DingtalkCardService) OnChatBotMessageReceived(ctx context.Context, data *chatbot.BotCallbackDataModel) (content []byte, err error) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *DingtalkCardService) OnCardMessageReceived(ctx context.Context, data *card.CardRequest) (resp *card.CardResponse, err error) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
@ -15,5 +15,4 @@ var ProviderSetServices = wire.NewSet(
|
||||||
NewHistoryService,
|
NewHistoryService,
|
||||||
NewCapabilityService,
|
NewCapabilityService,
|
||||||
NewCronService,
|
NewCronService,
|
||||||
// NewDingtalkCardService,
|
|
||||||
)
|
)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue