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

feat: Support for Chat Flow & Agent Support for binding a single chat flow (#765)

Browse Source 
Co-authored-by: Yu Yang <72337138+tomasyu985@users.noreply.github.com>
Co-authored-by: zengxiaohui <csu.zengxiaohui@gmail.com>
Co-authored-by: lijunwen.gigoo <lijunwen.gigoo@bytedance.com>
Co-authored-by: lvxinyu.1117 <lvxinyu.1117@bytedance.com>
Co-authored-by: liuyunchao.0510 <liuyunchao.0510@bytedance.com>
Co-authored-by: haozhenfei <37089575+haozhenfei@users.noreply.github.com>
Co-authored-by: July <jiangxujin@bytedance.com>
Co-authored-by: tecvan-fe <fanwenjie.fe@bytedance.com>
This commit is contained in:
Zhj
2025-08-28 21:53:32 +08:00
committed by GitHub
parent bbc615a18e
commit d70101c979
503 changed files with 48036 additions and 3427 deletions
Download Patch File Download Diff File Expand all files Collapse all files

810
backend/internal/mock/domain/workflow/interface.go
View File

@@ -38,6 +38,7 @@ import (
config "github.com/coze-dev/coze-studio/backend/domain/workflow/config"
entity "github.com/coze-dev/coze-studio/backend/domain/workflow/entity"
vo "github.com/coze-dev/coze-studio/backend/domain/workflow/entity/vo"
storage "github.com/coze-dev/coze-studio/backend/infra/contract/storage"
gomock "go.uber.org/mock/gomock"
)
@@ -109,6 +110,20 @@ func (mr *MockServiceMockRecorder) AsyncResume(ctx, req, arg2 any) *gomock.Call
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AsyncResume", reflect.TypeOf((*MockService)(nil).AsyncResume), ctx, req, arg2)
}
// BindConvRelatedInfo mocks base method.
func (m *MockService) BindConvRelatedInfo(ctx context.Context, convID int64, info entity.ConvRelatedInfo) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "BindConvRelatedInfo", ctx, convID, info)
ret0, _ := ret[0].(error)
return ret0
}
// BindConvRelatedInfo indicates an expected call of BindConvRelatedInfo.
func (mr *MockServiceMockRecorder) BindConvRelatedInfo(ctx, convID, info any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BindConvRelatedInfo", reflect.TypeOf((*MockService)(nil).BindConvRelatedInfo), ctx, convID, info)
}
// Cancel mocks base method.
func (m *MockService) Cancel(ctx context.Context, wfExeID, wfID, spaceID int64) error {
m.ctrl.T.Helper()
@@ -123,6 +138,21 @@ func (mr *MockServiceMockRecorder) Cancel(ctx, wfExeID, wfID, spaceID any) *gomo
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Cancel", reflect.TypeOf((*MockService)(nil).Cancel), ctx, wfExeID, wfID, spaceID)
}
// CheckWorkflowsToReplace mocks base method.
func (m *MockService) CheckWorkflowsToReplace(ctx context.Context, appID, templateID int64) ([]*entity.Workflow, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CheckWorkflowsToReplace", ctx, appID, templateID)
ret0, _ := ret[0].([]*entity.Workflow)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// CheckWorkflowsToReplace indicates an expected call of CheckWorkflowsToReplace.
func (mr *MockServiceMockRecorder) CheckWorkflowsToReplace(ctx, appID, templateID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckWorkflowsToReplace", reflect.TypeOf((*MockService)(nil).CheckWorkflowsToReplace), ctx, appID, templateID)
}
// CopyWorkflow mocks base method.
func (m *MockService) CopyWorkflow(ctx context.Context, workflowID int64, policy vo.CopyWorkflowPolicy) (*entity.Workflow, error) {
m.ctrl.T.Helper()
@@ -168,6 +198,36 @@ func (mr *MockServiceMockRecorder) Create(ctx, meta any) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockService)(nil).Create), ctx, meta)
}
// CreateChatFlowRole mocks base method.
func (m *MockService) CreateChatFlowRole(ctx context.Context, role *vo.ChatFlowRoleCreate) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CreateChatFlowRole", ctx, role)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// CreateChatFlowRole indicates an expected call of CreateChatFlowRole.
func (mr *MockServiceMockRecorder) CreateChatFlowRole(ctx, role any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateChatFlowRole", reflect.TypeOf((*MockService)(nil).CreateChatFlowRole), ctx, role)
}
// CreateDraftConversationTemplate mocks base method.
func (m *MockService) CreateDraftConversationTemplate(ctx context.Context, template *vo.CreateConversationTemplateMeta) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CreateDraftConversationTemplate", ctx, template)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// CreateDraftConversationTemplate indicates an expected call of CreateDraftConversationTemplate.
func (mr *MockServiceMockRecorder) CreateDraftConversationTemplate(ctx, template any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDraftConversationTemplate", reflect.TypeOf((*MockService)(nil).CreateDraftConversationTemplate), ctx, template)
}
// Delete mocks base method.
func (m *MockService) Delete(ctx context.Context, policy *vo.DeletePolicy) ([]int64, error) {
m.ctrl.T.Helper()
@@ -183,6 +243,50 @@ func (mr *MockServiceMockRecorder) Delete(ctx, policy any) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockService)(nil).Delete), ctx, policy)
}
// DeleteChatFlowRole mocks base method.
func (m *MockService) DeleteChatFlowRole(ctx context.Context, id, workflowID int64) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DeleteChatFlowRole", ctx, id, workflowID)
ret0, _ := ret[0].(error)
return ret0
}
// DeleteChatFlowRole indicates an expected call of DeleteChatFlowRole.
func (mr *MockServiceMockRecorder) DeleteChatFlowRole(ctx, id, workflowID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteChatFlowRole", reflect.TypeOf((*MockService)(nil).DeleteChatFlowRole), ctx, id, workflowID)
}
// DeleteDraftConversationTemplate mocks base method.
func (m *MockService) DeleteDraftConversationTemplate(ctx context.Context, templateID int64, wfID2ConversationName map[int64]string) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DeleteDraftConversationTemplate", ctx, templateID, wfID2ConversationName)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// DeleteDraftConversationTemplate indicates an expected call of DeleteDraftConversationTemplate.
func (mr *MockServiceMockRecorder) DeleteDraftConversationTemplate(ctx, templateID, wfID2ConversationName any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDraftConversationTemplate", reflect.TypeOf((*MockService)(nil).DeleteDraftConversationTemplate), ctx, templateID, wfID2ConversationName)
}
// DeleteDynamicConversation mocks base method.
func (m *MockService) DeleteDynamicConversation(ctx context.Context, env vo.Env, templateID int64) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DeleteDynamicConversation", ctx, env, templateID)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// DeleteDynamicConversation indicates an expected call of DeleteDynamicConversation.
func (mr *MockServiceMockRecorder) DeleteDynamicConversation(ctx, env, templateID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDynamicConversation", reflect.TypeOf((*MockService)(nil).DeleteDynamicConversation), ctx, env, templateID)
}
// DuplicateWorkflowsByAppID mocks base method.
func (m *MockService) DuplicateWorkflowsByAppID(ctx context.Context, sourceAPPID, targetAppID int64, related vo.ExternalResourceRelated) ([]*entity.Workflow, error) {
m.ctrl.T.Helper()
@@ -213,6 +317,70 @@ func (mr *MockServiceMockRecorder) Get(ctx, policy any) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockService)(nil).Get), ctx, policy)
}
// GetChatFlowRole mocks base method.
func (m *MockService) GetChatFlowRole(ctx context.Context, workflowID int64, version string) (*entity.ChatFlowRole, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetChatFlowRole", ctx, workflowID, version)
ret0, _ := ret[0].(*entity.ChatFlowRole)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetChatFlowRole indicates an expected call of GetChatFlowRole.
func (mr *MockServiceMockRecorder) GetChatFlowRole(ctx, workflowID, version any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetChatFlowRole", reflect.TypeOf((*MockService)(nil).GetChatFlowRole), ctx, workflowID, version)
}
// GetConvRelatedInfo mocks base method.
func (m *MockService) GetConvRelatedInfo(ctx context.Context, convID int64) (*entity.ConvRelatedInfo, bool, func() error, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetConvRelatedInfo", ctx, convID)
ret0, _ := ret[0].(*entity.ConvRelatedInfo)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(func() error)
ret3, _ := ret[3].(error)
return ret0, ret1, ret2, ret3
}
// GetConvRelatedInfo indicates an expected call of GetConvRelatedInfo.
func (mr *MockServiceMockRecorder) GetConvRelatedInfo(ctx, convID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetConvRelatedInfo", reflect.TypeOf((*MockService)(nil).GetConvRelatedInfo), ctx, convID)
}
// GetConversationNameByID mocks base method.
func (m *MockService) GetConversationNameByID(ctx context.Context, env vo.Env, appID, connectorID, conversationID int64) (string, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetConversationNameByID", ctx, env, appID, connectorID, conversationID)
ret0, _ := ret[0].(string)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetConversationNameByID indicates an expected call of GetConversationNameByID.
func (mr *MockServiceMockRecorder) GetConversationNameByID(ctx, env, appID, connectorID, conversationID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetConversationNameByID", reflect.TypeOf((*MockService)(nil).GetConversationNameByID), ctx, env, appID, connectorID, conversationID)
}
// GetDynamicConversationByName mocks base method.
func (m *MockService) GetDynamicConversationByName(ctx context.Context, env vo.Env, appID, connectorID, userID int64, name string) (*entity.DynamicConversation, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetDynamicConversationByName", ctx, env, appID, connectorID, userID, name)
ret0, _ := ret[0].(*entity.DynamicConversation)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetDynamicConversationByName indicates an expected call of GetDynamicConversationByName.
func (mr *MockServiceMockRecorder) GetDynamicConversationByName(ctx, env, appID, connectorID, userID, name any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDynamicConversationByName", reflect.TypeOf((*MockService)(nil).GetDynamicConversationByName), ctx, env, appID, connectorID, userID, name)
}
// GetExecution mocks base method.
func (m *MockService) GetExecution(ctx context.Context, wfExe *entity.WorkflowExecution, includeNodes bool) (*entity.WorkflowExecution, error) {
m.ctrl.T.Helper()
@@ -277,6 +445,38 @@ func (mr *MockServiceMockRecorder) GetNodeExecution(ctx, exeID, nodeID any) *gom
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeExecution", reflect.TypeOf((*MockService)(nil).GetNodeExecution), ctx, exeID, nodeID)
}
// GetOrCreateConversation mocks base method.
func (m *MockService) GetOrCreateConversation(ctx context.Context, env vo.Env, appID, connectorID, userID int64, conversationName string) (int64, int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetOrCreateConversation", ctx, env, appID, connectorID, userID, conversationName)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(int64)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetOrCreateConversation indicates an expected call of GetOrCreateConversation.
func (mr *MockServiceMockRecorder) GetOrCreateConversation(ctx, env, appID, connectorID, userID, conversationName any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrCreateConversation", reflect.TypeOf((*MockService)(nil).GetOrCreateConversation), ctx, env, appID, connectorID, userID, conversationName)
}
// GetTemplateByName mocks base method.
func (m *MockService) GetTemplateByName(ctx context.Context, env vo.Env, appID int64, templateName string) (*entity.ConversationTemplate, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetTemplateByName", ctx, env, appID, templateName)
ret0, _ := ret[0].(*entity.ConversationTemplate)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetTemplateByName indicates an expected call of GetTemplateByName.
func (mr *MockServiceMockRecorder) GetTemplateByName(ctx, env, appID, templateName any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateByName", reflect.TypeOf((*MockService)(nil).GetTemplateByName), ctx, env, appID, templateName)
}
// GetWorkflowDependenceResource mocks base method.
func (m *MockService) GetWorkflowDependenceResource(ctx context.Context, workflowID int64) (*vo.DependenceResource, error) {
m.ctrl.T.Helper()
@@ -307,6 +507,65 @@ func (mr *MockServiceMockRecorder) GetWorkflowReference(ctx, id any) *gomock.Cal
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkflowReference", reflect.TypeOf((*MockService)(nil).GetWorkflowReference), ctx, id)
}
// GetWorkflowVersionsByConnector mocks base method.
func (m *MockService) GetWorkflowVersionsByConnector(ctx context.Context, connectorID, workflowID int64, limit int) ([]string, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetWorkflowVersionsByConnector", ctx, connectorID, workflowID, limit)
ret0, _ := ret[0].([]string)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetWorkflowVersionsByConnector indicates an expected call of GetWorkflowVersionsByConnector.
func (mr *MockServiceMockRecorder) GetWorkflowVersionsByConnector(ctx, connectorID, workflowID, limit any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkflowVersionsByConnector", reflect.TypeOf((*MockService)(nil).GetWorkflowVersionsByConnector), ctx, connectorID, workflowID, limit)
}
// InitApplicationDefaultConversationTemplate mocks base method.
func (m *MockService) InitApplicationDefaultConversationTemplate(ctx context.Context, spaceID, appID, userID int64) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "InitApplicationDefaultConversationTemplate", ctx, spaceID, appID, userID)
ret0, _ := ret[0].(error)
return ret0
}
// InitApplicationDefaultConversationTemplate indicates an expected call of InitApplicationDefaultConversationTemplate.
func (mr *MockServiceMockRecorder) InitApplicationDefaultConversationTemplate(ctx, spaceID, appID, userID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InitApplicationDefaultConversationTemplate", reflect.TypeOf((*MockService)(nil).InitApplicationDefaultConversationTemplate), ctx, spaceID, appID, userID)
}
// ListConversationTemplate mocks base method.
func (m *MockService) ListConversationTemplate(ctx context.Context, env vo.Env, policy *vo.ListConversationTemplatePolicy) ([]*entity.ConversationTemplate, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListConversationTemplate", ctx, env, policy)
ret0, _ := ret[0].([]*entity.ConversationTemplate)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ListConversationTemplate indicates an expected call of ListConversationTemplate.
func (mr *MockServiceMockRecorder) ListConversationTemplate(ctx, env, policy any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListConversationTemplate", reflect.TypeOf((*MockService)(nil).ListConversationTemplate), ctx, env, policy)
}
// ListDynamicConversation mocks base method.
func (m *MockService) ListDynamicConversation(ctx context.Context, env vo.Env, policy *vo.ListConversationPolicy) ([]*entity.DynamicConversation, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListDynamicConversation", ctx, env, policy)
ret0, _ := ret[0].([]*entity.DynamicConversation)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ListDynamicConversation indicates an expected call of ListDynamicConversation.
func (mr *MockServiceMockRecorder) ListDynamicConversation(ctx, env, policy any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDynamicConversation", reflect.TypeOf((*MockService)(nil).ListDynamicConversation), ctx, env, policy)
}
// ListNodeMeta mocks base method.
func (m *MockService) ListNodeMeta(ctx context.Context, nodeTypes map[entity.NodeType]bool) (map[string][]*entity.NodeTypeMeta, []entity.Category, error) {
m.ctrl.T.Helper()
@@ -339,6 +598,21 @@ func (mr *MockServiceMockRecorder) MGet(ctx, policy any) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGet", reflect.TypeOf((*MockService)(nil).MGet), ctx, policy)
}
// MGetStaticConversation mocks base method.
func (m *MockService) MGetStaticConversation(ctx context.Context, env vo.Env, userID, connectorID int64, templateIDs []int64) ([]*entity.StaticConversation, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "MGetStaticConversation", ctx, env, userID, connectorID, templateIDs)
ret0, _ := ret[0].([]*entity.StaticConversation)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// MGetStaticConversation indicates an expected call of MGetStaticConversation.
func (mr *MockServiceMockRecorder) MGetStaticConversation(ctx, env, userID, connectorID, templateIDs any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetStaticConversation", reflect.TypeOf((*MockService)(nil).MGetStaticConversation), ctx, env, userID, connectorID, templateIDs)
}
// Publish mocks base method.
func (m *MockService) Publish(ctx context.Context, policy *vo.PublishPolicy) error {
m.ctrl.T.Helper()
@@ -353,6 +627,20 @@ func (mr *MockServiceMockRecorder) Publish(ctx, policy any) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Publish", reflect.TypeOf((*MockService)(nil).Publish), ctx, policy)
}
// PublishChatFlowRole mocks base method.
func (m *MockService) PublishChatFlowRole(ctx context.Context, policy *vo.PublishRolePolicy) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "PublishChatFlowRole", ctx, policy)
ret0, _ := ret[0].(error)
return ret0
}
// PublishChatFlowRole indicates an expected call of PublishChatFlowRole.
func (mr *MockServiceMockRecorder) PublishChatFlowRole(ctx, policy any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishChatFlowRole", reflect.TypeOf((*MockService)(nil).PublishChatFlowRole), ctx, policy)
}
// QueryNodeProperties mocks base method.
func (m *MockService) QueryNodeProperties(ctx context.Context, id int64) (map[string]*vo.NodeProperty, error) {
m.ctrl.T.Helper()
@@ -383,6 +671,20 @@ func (mr *MockServiceMockRecorder) ReleaseApplicationWorkflows(ctx, appID, arg2
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReleaseApplicationWorkflows", reflect.TypeOf((*MockService)(nil).ReleaseApplicationWorkflows), ctx, appID, arg2)
}
// ReleaseConversationTemplate mocks base method.
func (m *MockService) ReleaseConversationTemplate(ctx context.Context, appID int64, version string) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ReleaseConversationTemplate", ctx, appID, version)
ret0, _ := ret[0].(error)
return ret0
}
// ReleaseConversationTemplate indicates an expected call of ReleaseConversationTemplate.
func (mr *MockServiceMockRecorder) ReleaseConversationTemplate(ctx, appID, version any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReleaseConversationTemplate", reflect.TypeOf((*MockService)(nil).ReleaseConversationTemplate), ctx, appID, version)
}
// Save mocks base method.
func (m *MockService) Save(ctx context.Context, id int64, arg2 string) error {
m.ctrl.T.Helper()
@@ -427,6 +729,21 @@ func (mr *MockServiceMockRecorder) StreamResume(ctx, req, arg2 any) *gomock.Call
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamResume", reflect.TypeOf((*MockService)(nil).StreamResume), ctx, req, arg2)
}
// Suggest mocks base method.
func (m *MockService) Suggest(ctx context.Context, input *vo.SuggestInfo) ([]string, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Suggest", ctx, input)
ret0, _ := ret[0].([]string)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Suggest indicates an expected call of Suggest.
func (mr *MockServiceMockRecorder) Suggest(ctx, input any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Suggest", reflect.TypeOf((*MockService)(nil).Suggest), ctx, input)
}
// SyncExecute mocks base method.
func (m *MockService) SyncExecute(ctx context.Context, arg1 workflow.ExecuteConfig, input map[string]any) (*entity.WorkflowExecution, vo.TerminatePlan, error) {
m.ctrl.T.Helper()
@@ -457,6 +774,49 @@ func (mr *MockServiceMockRecorder) SyncRelatedWorkflowResources(ctx, appID, rela
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncRelatedWorkflowResources", reflect.TypeOf((*MockService)(nil).SyncRelatedWorkflowResources), ctx, appID, relatedWorkflows, related)
}
// UpdateChatFlowRole mocks base method.
func (m *MockService) UpdateChatFlowRole(ctx context.Context, workflowID int64, role *vo.ChatFlowRoleUpdate) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateChatFlowRole", ctx, workflowID, role)
ret0, _ := ret[0].(error)
return ret0
}
// UpdateChatFlowRole indicates an expected call of UpdateChatFlowRole.
func (mr *MockServiceMockRecorder) UpdateChatFlowRole(ctx, workflowID, role any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateChatFlowRole", reflect.TypeOf((*MockService)(nil).UpdateChatFlowRole), ctx, workflowID, role)
}
// UpdateConversation mocks base method.
func (m *MockService) UpdateConversation(ctx context.Context, env vo.Env, appID, connectorID, userID int64, conversationName string) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateConversation", ctx, env, appID, connectorID, userID, conversationName)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// UpdateConversation indicates an expected call of UpdateConversation.
func (mr *MockServiceMockRecorder) UpdateConversation(ctx, env, appID, connectorID, userID, conversationName any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateConversation", reflect.TypeOf((*MockService)(nil).UpdateConversation), ctx, env, appID, connectorID, userID, conversationName)
}
// UpdateDraftConversationTemplateName mocks base method.
func (m *MockService) UpdateDraftConversationTemplateName(ctx context.Context, appID, userID, templateID int64, name string) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateDraftConversationTemplateName", ctx, appID, userID, templateID, name)
ret0, _ := ret[0].(error)
return ret0
}
// UpdateDraftConversationTemplateName indicates an expected call of UpdateDraftConversationTemplateName.
func (mr *MockServiceMockRecorder) UpdateDraftConversationTemplateName(ctx, appID, userID, templateID, name any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDraftConversationTemplateName", reflect.TypeOf((*MockService)(nil).UpdateDraftConversationTemplateName), ctx, appID, userID, templateID, name)
}
// UpdateMeta mocks base method.
func (m *MockService) UpdateMeta(ctx context.Context, id int64, metaUpdate *vo.MetaUpdate) error {
m.ctrl.T.Helper()
@@ -545,6 +905,21 @@ func (mr *MockServiceMockRecorder) WorkflowAsModelTool(ctx, policies any) *gomoc
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WorkflowAsModelTool", reflect.TypeOf((*MockService)(nil).WorkflowAsModelTool), ctx, policies)
}
// WorkflowSchemaCheck mocks base method.
func (m *MockService) WorkflowSchemaCheck(ctx context.Context, wf *entity.Workflow, checks []workflow0.CheckType) ([]*workflow0.CheckResult, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "WorkflowSchemaCheck", ctx, wf, checks)
ret0, _ := ret[0].([]*workflow0.CheckResult)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// WorkflowSchemaCheck indicates an expected call of WorkflowSchemaCheck.
func (mr *MockServiceMockRecorder) WorkflowSchemaCheck(ctx, wf, checks any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WorkflowSchemaCheck", reflect.TypeOf((*MockService)(nil).WorkflowSchemaCheck), ctx, wf, checks)
}
// MockRepository is a mock of Repository interface.
type MockRepository struct {
ctrl *gomock.Controller
@@ -583,6 +958,34 @@ func (mr *MockRepositoryMockRecorder) BatchCreateConnectorWorkflowVersion(ctx, a
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreateConnectorWorkflowVersion", reflect.TypeOf((*MockRepository)(nil).BatchCreateConnectorWorkflowVersion), ctx, appID, connectorID, workflowIDs, version)
}
// BatchCreateOnlineConversationTemplate mocks base method.
func (m *MockRepository) BatchCreateOnlineConversationTemplate(ctx context.Context, templates []*entity.ConversationTemplate, version string) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "BatchCreateOnlineConversationTemplate", ctx, templates, version)
ret0, _ := ret[0].(error)
return ret0
}
// BatchCreateOnlineConversationTemplate indicates an expected call of BatchCreateOnlineConversationTemplate.
func (mr *MockRepositoryMockRecorder) BatchCreateOnlineConversationTemplate(ctx, templates, version any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreateOnlineConversationTemplate", reflect.TypeOf((*MockRepository)(nil).BatchCreateOnlineConversationTemplate), ctx, templates, version)
}
// BindConvRelatedInfo mocks base method.
func (m *MockRepository) BindConvRelatedInfo(ctx context.Context, convID int64, info entity.ConvRelatedInfo) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "BindConvRelatedInfo", ctx, convID, info)
ret0, _ := ret[0].(error)
return ret0
}
// BindConvRelatedInfo indicates an expected call of BindConvRelatedInfo.
func (mr *MockRepositoryMockRecorder) BindConvRelatedInfo(ctx, convID, info any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BindConvRelatedInfo", reflect.TypeOf((*MockRepository)(nil).BindConvRelatedInfo), ctx, convID, info)
}
// CancelAllRunningNodes mocks base method.
func (m *MockRepository) CancelAllRunningNodes(ctx context.Context, wfExeID int64) error {
m.ctrl.T.Helper()
@@ -597,6 +1000,20 @@ func (mr *MockRepositoryMockRecorder) CancelAllRunningNodes(ctx, wfExeID any) *g
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CancelAllRunningNodes", reflect.TypeOf((*MockRepository)(nil).CancelAllRunningNodes), ctx, wfExeID)
}
// CopyTemplateConversationByAppID mocks base method.
func (m *MockRepository) CopyTemplateConversationByAppID(ctx context.Context, appID, toAppID int64) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CopyTemplateConversationByAppID", ctx, appID, toAppID)
ret0, _ := ret[0].(error)
return ret0
}
// CopyTemplateConversationByAppID indicates an expected call of CopyTemplateConversationByAppID.
func (mr *MockRepositoryMockRecorder) CopyTemplateConversationByAppID(ctx, appID, toAppID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CopyTemplateConversationByAppID", reflect.TypeOf((*MockRepository)(nil).CopyTemplateConversationByAppID), ctx, appID, toAppID)
}
// CopyWorkflow mocks base method.
func (m *MockRepository) CopyWorkflow(ctx context.Context, workflowID int64, policy vo.CopyWorkflowPolicy) (*entity.Workflow, error) {
m.ctrl.T.Helper()
@@ -612,6 +1029,36 @@ func (mr *MockRepositoryMockRecorder) CopyWorkflow(ctx, workflowID, policy any)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CopyWorkflow", reflect.TypeOf((*MockRepository)(nil).CopyWorkflow), ctx, workflowID, policy)
}
// CreateChatFlowRoleConfig mocks base method.
func (m *MockRepository) CreateChatFlowRoleConfig(ctx context.Context, chatFlowRole *entity.ChatFlowRole) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CreateChatFlowRoleConfig", ctx, chatFlowRole)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// CreateChatFlowRoleConfig indicates an expected call of CreateChatFlowRoleConfig.
func (mr *MockRepositoryMockRecorder) CreateChatFlowRoleConfig(ctx, chatFlowRole any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateChatFlowRoleConfig", reflect.TypeOf((*MockRepository)(nil).CreateChatFlowRoleConfig), ctx, chatFlowRole)
}
// CreateDraftConversationTemplate mocks base method.
func (m *MockRepository) CreateDraftConversationTemplate(ctx context.Context, template *vo.CreateConversationTemplateMeta) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CreateDraftConversationTemplate", ctx, template)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// CreateDraftConversationTemplate indicates an expected call of CreateDraftConversationTemplate.
func (mr *MockRepositoryMockRecorder) CreateDraftConversationTemplate(ctx, template any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDraftConversationTemplate", reflect.TypeOf((*MockRepository)(nil).CreateDraftConversationTemplate), ctx, template)
}
// CreateMeta mocks base method.
func (m *MockRepository) CreateMeta(ctx context.Context, meta *vo.Meta) (int64, error) {
m.ctrl.T.Helper()
@@ -711,6 +1158,50 @@ func (mr *MockRepositoryMockRecorder) Delete(ctx, id any) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockRepository)(nil).Delete), ctx, id)
}
// DeleteChatFlowRoleConfig mocks base method.
func (m *MockRepository) DeleteChatFlowRoleConfig(ctx context.Context, id, workflowID int64) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DeleteChatFlowRoleConfig", ctx, id, workflowID)
ret0, _ := ret[0].(error)
return ret0
}
// DeleteChatFlowRoleConfig indicates an expected call of DeleteChatFlowRoleConfig.
func (mr *MockRepositoryMockRecorder) DeleteChatFlowRoleConfig(ctx, id, workflowID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteChatFlowRoleConfig", reflect.TypeOf((*MockRepository)(nil).DeleteChatFlowRoleConfig), ctx, id, workflowID)
}
// DeleteDraftConversationTemplate mocks base method.
func (m *MockRepository) DeleteDraftConversationTemplate(ctx context.Context, templateID int64) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DeleteDraftConversationTemplate", ctx, templateID)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// DeleteDraftConversationTemplate indicates an expected call of DeleteDraftConversationTemplate.
func (mr *MockRepositoryMockRecorder) DeleteDraftConversationTemplate(ctx, templateID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDraftConversationTemplate", reflect.TypeOf((*MockRepository)(nil).DeleteDraftConversationTemplate), ctx, templateID)
}
// DeleteDynamicConversation mocks base method.
func (m *MockRepository) DeleteDynamicConversation(ctx context.Context, env vo.Env, id int64) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DeleteDynamicConversation", ctx, env, id)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// DeleteDynamicConversation indicates an expected call of DeleteDynamicConversation.
func (mr *MockRepositoryMockRecorder) DeleteDynamicConversation(ctx, env, id any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDynamicConversation", reflect.TypeOf((*MockRepository)(nil).DeleteDynamicConversation), ctx, env, id)
}
// DraftV2 mocks base method.
func (m *MockRepository) DraftV2(ctx context.Context, id int64, commitID string) (*vo.DraftInfo, error) {
m.ctrl.T.Helper()
@@ -772,6 +1263,55 @@ func (mr *MockRepositoryMockRecorder) Get(ctx, checkPointID any) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockRepository)(nil).Get), ctx, checkPointID)
}
// GetChatFlowRoleConfig mocks base method.
func (m *MockRepository) GetChatFlowRoleConfig(ctx context.Context, workflowID int64, version string) (*entity.ChatFlowRole, error, bool) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetChatFlowRoleConfig", ctx, workflowID, version)
ret0, _ := ret[0].(*entity.ChatFlowRole)
ret1, _ := ret[1].(error)
ret2, _ := ret[2].(bool)
return ret0, ret1, ret2
}
// GetChatFlowRoleConfig indicates an expected call of GetChatFlowRoleConfig.
func (mr *MockRepositoryMockRecorder) GetChatFlowRoleConfig(ctx, workflowID, version any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetChatFlowRoleConfig", reflect.TypeOf((*MockRepository)(nil).GetChatFlowRoleConfig), ctx, workflowID, version)
}
// GetConvRelatedInfo mocks base method.
func (m *MockRepository) GetConvRelatedInfo(ctx context.Context, convID int64) (*entity.ConvRelatedInfo, bool, func() error, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetConvRelatedInfo", ctx, convID)
ret0, _ := ret[0].(*entity.ConvRelatedInfo)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(func() error)
ret3, _ := ret[3].(error)
return ret0, ret1, ret2, ret3
}
// GetConvRelatedInfo indicates an expected call of GetConvRelatedInfo.
func (mr *MockRepositoryMockRecorder) GetConvRelatedInfo(ctx, convID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetConvRelatedInfo", reflect.TypeOf((*MockRepository)(nil).GetConvRelatedInfo), ctx, convID)
}
// GetConversationTemplate mocks base method.
func (m *MockRepository) GetConversationTemplate(ctx context.Context, env vo.Env, policy vo.GetConversationTemplatePolicy) (*entity.ConversationTemplate, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetConversationTemplate", ctx, env, policy)
ret0, _ := ret[0].(*entity.ConversationTemplate)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetConversationTemplate indicates an expected call of GetConversationTemplate.
func (mr *MockRepositoryMockRecorder) GetConversationTemplate(ctx, env, policy any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetConversationTemplate", reflect.TypeOf((*MockRepository)(nil).GetConversationTemplate), ctx, env, policy)
}
// GetDraftWorkflowsByAppID mocks base method.
func (m *MockRepository) GetDraftWorkflowsByAppID(ctx context.Context, AppID int64) (map[int64]*vo.DraftInfo, map[int64]string, error) {
m.ctrl.T.Helper()
@@ -788,6 +1328,38 @@ func (mr *MockRepositoryMockRecorder) GetDraftWorkflowsByAppID(ctx, AppID any) *
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDraftWorkflowsByAppID", reflect.TypeOf((*MockRepository)(nil).GetDraftWorkflowsByAppID), ctx, AppID)
}
// GetDynamicConversationByID mocks base method.
func (m *MockRepository) GetDynamicConversationByID(ctx context.Context, env vo.Env, appID, connectorID, conversationID int64) (*entity.DynamicConversation, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetDynamicConversationByID", ctx, env, appID, connectorID, conversationID)
ret0, _ := ret[0].(*entity.DynamicConversation)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetDynamicConversationByID indicates an expected call of GetDynamicConversationByID.
func (mr *MockRepositoryMockRecorder) GetDynamicConversationByID(ctx, env, appID, connectorID, conversationID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDynamicConversationByID", reflect.TypeOf((*MockRepository)(nil).GetDynamicConversationByID), ctx, env, appID, connectorID, conversationID)
}
// GetDynamicConversationByName mocks base method.
func (m *MockRepository) GetDynamicConversationByName(ctx context.Context, env vo.Env, appID, connectorID, userID int64, name string) (*entity.DynamicConversation, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetDynamicConversationByName", ctx, env, appID, connectorID, userID, name)
ret0, _ := ret[0].(*entity.DynamicConversation)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetDynamicConversationByName indicates an expected call of GetDynamicConversationByName.
func (mr *MockRepositoryMockRecorder) GetDynamicConversationByName(ctx, env, appID, connectorID, userID, name any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDynamicConversationByName", reflect.TypeOf((*MockRepository)(nil).GetDynamicConversationByName), ctx, env, appID, connectorID, userID, name)
}
// GetEntity mocks base method.
func (m *MockRepository) GetEntity(ctx context.Context, policy *vo.GetPolicy) (*entity.Workflow, error) {
m.ctrl.T.Helper()
@@ -938,6 +1510,92 @@ func (mr *MockRepositoryMockRecorder) GetNodeOfCodeConfig() *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeOfCodeConfig", reflect.TypeOf((*MockRepository)(nil).GetNodeOfCodeConfig))
}
// GetObjectUrl mocks base method.
func (m *MockRepository) GetObjectUrl(ctx context.Context, objectKey string, opts ...storage.GetOptFn) (string, error) {
m.ctrl.T.Helper()
varargs := []any{ctx, objectKey}
for _, a := range opts {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "GetObjectUrl", varargs...)
ret0, _ := ret[0].(string)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetObjectUrl indicates an expected call of GetObjectUrl.
func (mr *MockRepositoryMockRecorder) GetObjectUrl(ctx, objectKey any, opts ...any) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]any{ctx, objectKey}, opts...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetObjectUrl", reflect.TypeOf((*MockRepository)(nil).GetObjectUrl), varargs...)
}
// GetOrCreateDynamicConversation mocks base method.
func (m *MockRepository) GetOrCreateDynamicConversation(ctx context.Context, env vo.Env, idGen workflow1.ConversationIDGenerator, meta *vo.CreateDynamicConversation) (int64, int64, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetOrCreateDynamicConversation", ctx, env, idGen, meta)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(int64)
ret2, _ := ret[2].(bool)
ret3, _ := ret[3].(error)
return ret0, ret1, ret2, ret3
}
// GetOrCreateDynamicConversation indicates an expected call of GetOrCreateDynamicConversation.
func (mr *MockRepositoryMockRecorder) GetOrCreateDynamicConversation(ctx, env, idGen, meta any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrCreateDynamicConversation", reflect.TypeOf((*MockRepository)(nil).GetOrCreateDynamicConversation), ctx, env, idGen, meta)
}
// GetOrCreateStaticConversation mocks base method.
func (m *MockRepository) GetOrCreateStaticConversation(ctx context.Context, env vo.Env, idGen workflow1.ConversationIDGenerator, meta *vo.CreateStaticConversation) (int64, int64, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetOrCreateStaticConversation", ctx, env, idGen, meta)
ret0, _ := ret[0].(int64)
ret1, _ := ret[1].(int64)
ret2, _ := ret[2].(bool)
ret3, _ := ret[3].(error)
return ret0, ret1, ret2, ret3
}
// GetOrCreateStaticConversation indicates an expected call of GetOrCreateStaticConversation.
func (mr *MockRepositoryMockRecorder) GetOrCreateStaticConversation(ctx, env, idGen, meta any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrCreateStaticConversation", reflect.TypeOf((*MockRepository)(nil).GetOrCreateStaticConversation), ctx, env, idGen, meta)
}
// GetStaticConversationByID mocks base method.
func (m *MockRepository) GetStaticConversationByID(ctx context.Context, env vo.Env, appID, connectorID, conversationID int64) (string, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetStaticConversationByID", ctx, env, appID, connectorID, conversationID)
ret0, _ := ret[0].(string)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetStaticConversationByID indicates an expected call of GetStaticConversationByID.
func (mr *MockRepositoryMockRecorder) GetStaticConversationByID(ctx, env, appID, connectorID, conversationID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStaticConversationByID", reflect.TypeOf((*MockRepository)(nil).GetStaticConversationByID), ctx, env, appID, connectorID, conversationID)
}
// GetStaticConversationByTemplateID mocks base method.
func (m *MockRepository) GetStaticConversationByTemplateID(ctx context.Context, env vo.Env, userID, connectorID, templateID int64) (*entity.StaticConversation, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetStaticConversationByTemplateID", ctx, env, userID, connectorID, templateID)
ret0, _ := ret[0].(*entity.StaticConversation)
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetStaticConversationByTemplateID indicates an expected call of GetStaticConversationByTemplateID.
func (mr *MockRepositoryMockRecorder) GetStaticConversationByTemplateID(ctx, env, userID, connectorID, templateID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStaticConversationByTemplateID", reflect.TypeOf((*MockRepository)(nil).GetStaticConversationByTemplateID), ctx, env, userID, connectorID, templateID)
}
// GetTestRunLatestExeID mocks base method.
func (m *MockRepository) GetTestRunLatestExeID(ctx context.Context, wfID, uID int64) (int64, error) {
m.ctrl.T.Helper()
@@ -954,12 +1612,13 @@ func (mr *MockRepositoryMockRecorder) GetTestRunLatestExeID(ctx, wfID, uID any)
}
// GetVersion mocks base method.
func (m *MockRepository) GetVersion(ctx context.Context, id int64, version string) (*vo.VersionInfo, error) {
func (m *MockRepository) GetVersion(ctx context.Context, id int64, version string) (*vo.VersionInfo, bool, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetVersion", ctx, id, version)
ret0, _ := ret[0].(*vo.VersionInfo)
ret1, _ := ret[1].(error)
return ret0, ret1
ret1, _ := ret[1].(bool)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// GetVersion indicates an expected call of GetVersion.
@@ -968,6 +1627,21 @@ func (mr *MockRepositoryMockRecorder) GetVersion(ctx, id, version any) *gomock.C
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockRepository)(nil).GetVersion), ctx, id, version)
}
// GetVersionListByConnectorAndWorkflowID mocks base method.
func (m *MockRepository) GetVersionListByConnectorAndWorkflowID(ctx context.Context, connectorID, workflowID int64, limit int) ([]string, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetVersionListByConnectorAndWorkflowID", ctx, connectorID, workflowID, limit)
ret0, _ := ret[0].([]string)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetVersionListByConnectorAndWorkflowID indicates an expected call of GetVersionListByConnectorAndWorkflowID.
func (mr *MockRepositoryMockRecorder) GetVersionListByConnectorAndWorkflowID(ctx, connectorID, workflowID, limit any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersionListByConnectorAndWorkflowID", reflect.TypeOf((*MockRepository)(nil).GetVersionListByConnectorAndWorkflowID), ctx, connectorID, workflowID, limit)
}
// GetWorkflowCancelFlag mocks base method.
func (m *MockRepository) GetWorkflowCancelFlag(ctx context.Context, wfExeID int64) (bool, error) {
m.ctrl.T.Helper()
@@ -1014,6 +1688,36 @@ func (mr *MockRepositoryMockRecorder) IsApplicationConnectorWorkflowVersion(ctx,
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsApplicationConnectorWorkflowVersion", reflect.TypeOf((*MockRepository)(nil).IsApplicationConnectorWorkflowVersion), ctx, connectorID, workflowID, version)
}
// ListConversationTemplate mocks base method.
func (m *MockRepository) ListConversationTemplate(ctx context.Context, env vo.Env, policy *vo.ListConversationTemplatePolicy) ([]*entity.ConversationTemplate, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListConversationTemplate", ctx, env, policy)
ret0, _ := ret[0].([]*entity.ConversationTemplate)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ListConversationTemplate indicates an expected call of ListConversationTemplate.
func (mr *MockRepositoryMockRecorder) ListConversationTemplate(ctx, env, policy any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListConversationTemplate", reflect.TypeOf((*MockRepository)(nil).ListConversationTemplate), ctx, env, policy)
}
// ListDynamicConversation mocks base method.
func (m *MockRepository) ListDynamicConversation(ctx context.Context, env vo.Env, policy *vo.ListConversationPolicy) ([]*entity.DynamicConversation, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListDynamicConversation", ctx, env, policy)
ret0, _ := ret[0].([]*entity.DynamicConversation)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ListDynamicConversation indicates an expected call of ListDynamicConversation.
func (mr *MockRepositoryMockRecorder) ListDynamicConversation(ctx, env, policy any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDynamicConversation", reflect.TypeOf((*MockRepository)(nil).ListDynamicConversation), ctx, env, policy)
}
// ListInterruptEvents mocks base method.
func (m *MockRepository) ListInterruptEvents(ctx context.Context, wfExeID int64) ([]*entity.InterruptEvent, error) {
m.ctrl.T.Helper()
@@ -1106,6 +1810,21 @@ func (mr *MockRepositoryMockRecorder) MGetReferences(ctx, policy any) *gomock.Ca
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetReferences", reflect.TypeOf((*MockRepository)(nil).MGetReferences), ctx, policy)
}
// MGetStaticConversation mocks base method.
func (m *MockRepository) MGetStaticConversation(ctx context.Context, env vo.Env, userID, connectorID int64, templateIDs []int64) ([]*entity.StaticConversation, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "MGetStaticConversation", ctx, env, userID, connectorID, templateIDs)
ret0, _ := ret[0].([]*entity.StaticConversation)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// MGetStaticConversation indicates an expected call of MGetStaticConversation.
func (mr *MockRepositoryMockRecorder) MGetStaticConversation(ctx, env, userID, connectorID, templateIDs any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetStaticConversation", reflect.TypeOf((*MockRepository)(nil).MGetStaticConversation), ctx, env, userID, connectorID, templateIDs)
}
// PopFirstInterruptEvent mocks base method.
func (m *MockRepository) PopFirstInterruptEvent(ctx context.Context, wfExeID int64) (*entity.InterruptEvent, bool, error) {
m.ctrl.T.Helper()
@@ -1192,6 +1911,21 @@ func (mr *MockRepositoryMockRecorder) SetWorkflowCancelFlag(ctx, wfExeID any) *g
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetWorkflowCancelFlag", reflect.TypeOf((*MockRepository)(nil).SetWorkflowCancelFlag), ctx, wfExeID)
}
// Suggest mocks base method.
func (m *MockRepository) Suggest(ctx context.Context, input *vo.SuggestInfo) ([]string, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Suggest", ctx, input)
ret0, _ := ret[0].([]string)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Suggest indicates an expected call of Suggest.
func (mr *MockRepositoryMockRecorder) Suggest(ctx, input any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Suggest", reflect.TypeOf((*MockRepository)(nil).Suggest), ctx, input)
}
// TryLockWorkflowExecution mocks base method.
func (m *MockRepository) TryLockWorkflowExecution(ctx context.Context, wfExeID, resumingEventID int64) (bool, entity.WorkflowExecuteStatus, error) {
m.ctrl.T.Helper()
@@ -1208,6 +1942,62 @@ func (mr *MockRepositoryMockRecorder) TryLockWorkflowExecution(ctx, wfExeID, res
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TryLockWorkflowExecution", reflect.TypeOf((*MockRepository)(nil).TryLockWorkflowExecution), ctx, wfExeID, resumingEventID)
}
// UpdateChatFlowRoleConfig mocks base method.
func (m *MockRepository) UpdateChatFlowRoleConfig(ctx context.Context, workflowID int64, chatFlowRole *vo.ChatFlowRoleUpdate) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateChatFlowRoleConfig", ctx, workflowID, chatFlowRole)
ret0, _ := ret[0].(error)
return ret0
}
// UpdateChatFlowRoleConfig indicates an expected call of UpdateChatFlowRoleConfig.
func (mr *MockRepositoryMockRecorder) UpdateChatFlowRoleConfig(ctx, workflowID, chatFlowRole any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateChatFlowRoleConfig", reflect.TypeOf((*MockRepository)(nil).UpdateChatFlowRoleConfig), ctx, workflowID, chatFlowRole)
}
// UpdateDraftConversationTemplateName mocks base method.
func (m *MockRepository) UpdateDraftConversationTemplateName(ctx context.Context, templateID int64, name string) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateDraftConversationTemplateName", ctx, templateID, name)
ret0, _ := ret[0].(error)
return ret0
}
// UpdateDraftConversationTemplateName indicates an expected call of UpdateDraftConversationTemplateName.
func (mr *MockRepositoryMockRecorder) UpdateDraftConversationTemplateName(ctx, templateID, name any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDraftConversationTemplateName", reflect.TypeOf((*MockRepository)(nil).UpdateDraftConversationTemplateName), ctx, templateID, name)
}
// UpdateDynamicConversation mocks base method.
func (m *MockRepository) UpdateDynamicConversation(ctx context.Context, env vo.Env, conversationID, newConversationID int64) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateDynamicConversation", ctx, env, conversationID, newConversationID)
ret0, _ := ret[0].(error)
return ret0
}
// UpdateDynamicConversation indicates an expected call of UpdateDynamicConversation.
func (mr *MockRepositoryMockRecorder) UpdateDynamicConversation(ctx, env, conversationID, newConversationID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDynamicConversation", reflect.TypeOf((*MockRepository)(nil).UpdateDynamicConversation), ctx, env, conversationID, newConversationID)
}
// UpdateDynamicConversationNameByID mocks base method.
func (m *MockRepository) UpdateDynamicConversationNameByID(ctx context.Context, env vo.Env, templateID int64, name string) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateDynamicConversationNameByID", ctx, env, templateID, name)
ret0, _ := ret[0].(error)
return ret0
}
// UpdateDynamicConversationNameByID indicates an expected call of UpdateDynamicConversationNameByID.
func (mr *MockRepositoryMockRecorder) UpdateDynamicConversationNameByID(ctx, env, templateID, name any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDynamicConversationNameByID", reflect.TypeOf((*MockRepository)(nil).UpdateDynamicConversationNameByID), ctx, env, templateID, name)
}
// UpdateFirstInterruptEvent mocks base method.
func (m *MockRepository) UpdateFirstInterruptEvent(ctx context.Context, wfExeID int64, event *entity.InterruptEvent) error {
m.ctrl.T.Helper()
@@ -1264,6 +2054,20 @@ func (mr *MockRepositoryMockRecorder) UpdateNodeExecutionStreaming(ctx, executio
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNodeExecutionStreaming", reflect.TypeOf((*MockRepository)(nil).UpdateNodeExecutionStreaming), ctx, execution)
}
// UpdateStaticConversation mocks base method.
func (m *MockRepository) UpdateStaticConversation(ctx context.Context, env vo.Env, templateID, connectorID, userID, newConversationID int64) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateStaticConversation", ctx, env, templateID, connectorID, userID, newConversationID)
ret0, _ := ret[0].(error)
return ret0
}
// UpdateStaticConversation indicates an expected call of UpdateStaticConversation.
func (mr *MockRepositoryMockRecorder) UpdateStaticConversation(ctx, env, templateID, connectorID, userID, newConversationID any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateStaticConversation", reflect.TypeOf((*MockRepository)(nil).UpdateStaticConversation), ctx, env, templateID, connectorID, userID, newConversationID)
}
// UpdateWorkflowDraftTestRunSuccess mocks base method.
func (m *MockRepository) UpdateWorkflowDraftTestRunSuccess(ctx context.Context, id int64) error {
m.ctrl.T.Helper()