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)
case constants.TaskTypeCozeWorkflow:
return g.handleCozeWorkflow(ctx, rec, pointTask)
case constants.TaskTypeKnowle: // 知识库V2版本
return g.handleKnowledgeV2(ctx, rec, groupConfig, callback)
case constants.TaskTypeKnowle: // 知识库lightRAG版本
return g.handleKnowledge(ctx, rec, groupConfig, callback)
default:
return g.otherTask(ctx, rec)
}
@ -477,7 +477,7 @@ func (g *GroupConfigBiz) GetReportCache(ctx context.Context, day time.Time, tota
}
// 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)
knowledgeResp, err := knowledgeBase.Query(&knowledge_base.QueryRequest{
@ -591,13 +591,14 @@ func (g *GroupConfigBiz) shouldCreateIssueHandlingGroup(ctx context.Context, rec
}
issueOwnerStr := strings.Join(userNames, "、")
// 获取机器人应用配置
botConfig, err := g.botConfigImpl.GetRobotConfig(callback.RobotCode)
if err != nil {
return fmt.Errorf("获取机器人配置失败err: %v", err)
}
// 构建卡片 OutTrackId
outTrackId := constants.BuildCardOutTrackId(callback.ConversationId, botConfig.ClientId)
outTrackId := constants.BuildCardOutTrackId(callback.ConversationId, callback.RobotCode)
// 发送钉钉卡片
_, err = g.dingtalkCardClient.CreateAndDeliver(dingtalk.AppKey{

View File

@ -137,4 +137,4 @@ var GroupTemplateRobotIdMap = map[string]string{
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 {
serviceConfigs, err := service.GetServiceCfg()
for _, serviceConf := range serviceConfigs {
// 配置不全
if serviceConf.ClientId == "" || serviceConf.ClientSecret == "" {
continue
}
// 非应用主机器人 pass ,防止重复
if serviceConf.ClientId != serviceConf.BotIndex {
continue
}
cli := DingBotServerInit(serviceConf.ClientId, serviceConf.ClientSecret, service)
if cli == nil {
log.Info("%s客户端初始失败:%s", serviceConf.BotIndex, err.Error())
log.Infof("%s客户端初始失败:%s", serviceConf.BotIndex, err.Error())
continue
}
clients[serviceConf.BotIndex] = cli
@ -53,11 +58,9 @@ func NewDingTalkBotServer(
func ProvideAllDingBotServices(
dingBotSvc *services.DingBotService,
// dingCardSvc *services.DingtalkCardService,
) []DingBotServiceInterface {
return []DingBotServiceInterface{
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,
NewCapabilityService,
NewCronService,
// NewDingtalkCardService,
)