youngkingdom/coze-studio
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(plugin): update params when tool added in agent

Browse Source
This commit is contained in:
lijunwen.gigoo
2025-10-16 16:00:08 +08:00
parent 8de8686bab
commit cd1dc490bf
4 changed files with 63 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/application/plugin/plugin.go
View File

@ -69,15 +69,8 @@ func (p *PluginApplicationService) CheckAndLockPluginEdit(ctx context.Context, r
}
func (p *PluginApplicationService) GetBotDefaultParams(ctx context.Context, req *pluginAPI.GetBotDefaultParamsRequest) (resp *pluginAPI.GetBotDefaultParamsResponse, err error) {
_, exist, err := p.pluginRepo.GetOnlinePlugin(ctx, req.PluginID, repository.WithPluginID())
if err != nil {
return nil, errorx.Wrapf(err, "GetOnlinePlugin failed, pluginID=%d", req.PluginID)
}
if !exist {
return nil, errorx.New(errno.ErrPluginRecordNotFound)
}
draftAgentTool, err := p.DomainSVC.GetDraftAgentToolByName(ctx, req.BotID, req.APIName)
draftAgentTool, err := p.DomainSVC.GetDraftAgentToolByName(ctx, req.BotID, req.PluginID, req.APIName)
if err != nil {
return nil, errorx.Wrapf(err, "GetDraftAgentToolByName failed, agentID=%d, toolName=%s", req.BotID, req.APIName)
}

1
backend/domain/plugin/dto/tool.go
View File

@ -69,6 +69,7 @@ type UpdateBotDefaultParamsRequest struct {
Parameters openapi3.Parameters
RequestBody *openapi3.RequestBodyRef
Responses openapi3.Responses
PluginFormat *common.PluginDataFormat
}
type UniqueToolAPI struct {
SubURL string

74
backend/domain/plugin/service/agent_tool.go
View File

@ -44,7 +44,7 @@ func (p *pluginServiceImpl) DuplicateDraftAgentTools(ctx context.Context, fromAg
return p.toolRepo.DuplicateDraftAgentTools(ctx, fromAgentID, toAgentID)
}
func (p *pluginServiceImpl) GetDraftAgentToolByName(ctx context.Context, agentID int64, toolName string) (tool *entity.ToolInfo, err error) {
func (p *pluginServiceImpl) GetDraftAgentToolByName(ctx context.Context, agentID int64, pluginID int64, toolName string) (tool *entity.ToolInfo, err error) {
draftAgentTool, exist, err := p.toolRepo.GetDraftAgentToolWithToolName(ctx, agentID, toolName)
if err != nil {
return nil, errorx.Wrapf(err, "GetDraftAgentToolWithToolName failed, agentID=%d, toolName=%s", agentID, toolName)
@ -53,9 +53,36 @@ func (p *pluginServiceImpl) GetDraftAgentToolByName(ctx context.Context, agentID
return nil, errorx.New(errno.ErrPluginRecordNotFound)
}
tool, exist, err = p.toolRepo.GetOnlineTool(ctx, draftAgentTool.ID)
if err != nil {
return nil, errorx.Wrapf(err, "GetOnlineTool failed, id=%d", draftAgentTool.ID)
if draftAgentTool.GetPluginFrom() == bot_common.PluginFrom_FromSaas {
tools, _, err := p.toolRepo.BatchGetSaasPluginToolsInfo(ctx, []int64{draftAgentTool.PluginID})
if err != nil {
return nil, errorx.Wrapf(err, "BatchGetSaasPluginToolsInfo failed, pluginIDs=%v", []int64{draftAgentTool.PluginID})
}
if len(tools) == 0 || len(tools[draftAgentTool.PluginID]) == 0 {
return nil, errorx.New(errno.ErrPluginRecordNotFound)
}
exist = false
for _, saasTool := range tools[draftAgentTool.PluginID] {
if saasTool.ID == draftAgentTool.ID {
tool = saasTool
exist = true
break
}
}
} else {
_, exist, err := p.pluginRepo.GetOnlinePlugin(ctx, pluginID, repository.WithPluginID())
if err != nil {
return nil, errorx.Wrapf(err, "GetOnlinePlugin failed, pluginID=%d", pluginID)
}
if !exist {
return nil, errorx.New(errno.ErrPluginRecordNotFound)
}
tool, exist, err = p.toolRepo.GetOnlineTool(ctx, draftAgentTool.ID)
if err != nil {
return nil, errorx.Wrapf(err, "GetOnlineTool failed, id=%d", draftAgentTool.ID)
}
}
if !exist {
return nil, errorx.New(errno.ErrPluginRecordNotFound)
@ -157,13 +184,6 @@ func (p *pluginServiceImpl) PublishAgentTools(ctx context.Context, agentID int64
}
func (p *pluginServiceImpl) UpdateBotDefaultParams(ctx context.Context, req *dto.UpdateBotDefaultParamsRequest) (err error) {
_, exist, err := p.pluginRepo.GetOnlinePlugin(ctx, req.PluginID, repository.WithPluginID())
if err != nil {
return errorx.Wrapf(err, "GetOnlinePlugin failed, pluginID=%d", req.PluginID)
}
if !exist {
return errorx.New(errno.ErrPluginRecordNotFound)
}
draftAgentTool, exist, err := p.toolRepo.GetDraftAgentToolWithToolName(ctx, req.AgentID, req.ToolName)
if err != nil {
@ -173,9 +193,35 @@ func (p *pluginServiceImpl) UpdateBotDefaultParams(ctx context.Context, req *dto
return errorx.New(errno.ErrPluginRecordNotFound)
}
onlineTool, exist, err := p.toolRepo.GetOnlineTool(ctx, draftAgentTool.ID)
if err != nil {
return errorx.Wrapf(err, "GetOnlineTool failed, id=%d", draftAgentTool.ID)
var onlineTool *entity.ToolInfo
if draftAgentTool.GetPluginFrom() == bot_common.PluginFrom_FromSaas {
tools, _, err := p.toolRepo.BatchGetSaasPluginToolsInfo(ctx, []int64{draftAgentTool.PluginID})
if err != nil {
return errorx.Wrapf(err, "BatchGetSaasPluginToolsInfo failed, pluginIDs=%v", []int64{draftAgentTool.PluginID})
}
if len(tools) == 0 || len(tools[draftAgentTool.PluginID]) == 0 {
return errorx.New(errno.ErrPluginRecordNotFound)
}
for _, saasTool := range tools[draftAgentTool.PluginID] {
if saasTool.ID == draftAgentTool.ID {
onlineTool = saasTool
exist = true
break
}
}
} else {
_, exist, err := p.pluginRepo.GetOnlinePlugin(ctx, req.PluginID, repository.WithPluginID())
if err != nil {
return errorx.Wrapf(err, "GetOnlinePlugin failed, pluginID=%d", req.PluginID)
}
if !exist {
return errorx.New(errno.ErrPluginRecordNotFound)
}
onlineTool, exist, err = p.toolRepo.GetOnlineTool(ctx, draftAgentTool.ID)
if err != nil {
return errorx.Wrapf(err, "GetOnlineTool failed, id=%d", draftAgentTool.ID)
}
}
if !exist {
return errorx.New(errno.ErrPluginRecordNotFound)

2
backend/domain/plugin/service/service.go
View File

@ -65,7 +65,7 @@ type PluginService interface {
// Agent Tool
BindAgentTools(ctx context.Context, agentID int64, bindTools []*model.BindToolInfo) (err error)
DuplicateDraftAgentTools(ctx context.Context, fromAgentID, toAgentID int64) (err error)
GetDraftAgentToolByName(ctx context.Context, agentID int64, toolName string) (tool *entity.ToolInfo, err error)
GetDraftAgentToolByName(ctx context.Context, agentID int64, pluginID int64, toolName string) (tool *entity.ToolInfo, err error)
MGetAgentTools(ctx context.Context, req *model.MGetAgentToolsRequest) (tools []*entity.ToolInfo, err error)
UpdateBotDefaultParams(ctx context.Context, req *dto.UpdateBotDefaultParamsRequest) (err error)