fix:代码优化

This commit is contained in:
fuzhongyun 2026-01-26 17:54:23 +08:00
parent 5560e879d0
commit 32cd8691b7
5 changed files with 12 additions and 46 deletions

View File

@ -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{

View File

@ -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隐藏]

View File

@ -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,
} }
} }

View File

@ -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
}

View File

@ -15,5 +15,4 @@ var ProviderSetServices = wire.NewSet(
NewHistoryService, NewHistoryService,
NewCapabilityService, NewCapabilityService,
NewCronService, NewCronService,
// NewDingtalkCardService,
) )