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

refactor(workflow): Move the knowledge component in the Workflow package into the common crossdomain package (#708)

Browse Source
This commit is contained in:
Ryo
2025-08-12 17:10:36 +08:00
committed by GitHub
parent 9ff065cebd
commit b38ab95623
27 changed files with 586 additions and 710 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
backend/domain/workflow/internal/compose/test/llm_test.go
View File

@@ -34,8 +34,9 @@ import (
"github.com/stretchr/testify/assert"
"go.uber.org/mock/gomock"
"github.com/coze-dev/coze-studio/backend/domain/workflow/crossdomain/model"
mockmodel "github.com/coze-dev/coze-studio/backend/domain/workflow/crossdomain/model/modelmock"
model "github.com/coze-dev/coze-studio/backend/api/model/crossdomain/modelmgr"
crossmodelmgr "github.com/coze-dev/coze-studio/backend/crossdomain/contract/modelmgr"
mockmodel "github.com/coze-dev/coze-studio/backend/crossdomain/contract/modelmgr/modelmock"
"github.com/coze-dev/coze-studio/backend/domain/workflow/entity"
"github.com/coze-dev/coze-studio/backend/domain/workflow/entity/vo"
compose2 "github.com/coze-dev/coze-studio/backend/domain/workflow/internal/compose"
@@ -63,7 +64,7 @@ func TestLLM(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockModelManager := mockmodel.NewMockManager(ctrl)
mockey.Mock(model.GetManager).Return(mockModelManager).Build()
mockey.Mock(crossmodelmgr.DefaultSVC).Return(mockModelManager).Build()
if len(accessKey) > 0 && len(baseURL) > 0 && len(modelName) > 0 {
openaiModel, err = openai.NewChatModel(context.Background(), &openai.ChatModelConfig{

7
backend/domain/workflow/internal/compose/test/question_answer_test.go
View File

@@ -36,9 +36,10 @@ import (
"gorm.io/driver/mysql"
"gorm.io/gorm"
model "github.com/coze-dev/coze-studio/backend/api/model/crossdomain/modelmgr"
crossmodelmgr "github.com/coze-dev/coze-studio/backend/crossdomain/contract/modelmgr"
mockmodel "github.com/coze-dev/coze-studio/backend/crossdomain/contract/modelmgr/modelmock"
"github.com/coze-dev/coze-studio/backend/domain/workflow"
"github.com/coze-dev/coze-studio/backend/domain/workflow/crossdomain/model"
mockmodel "github.com/coze-dev/coze-studio/backend/domain/workflow/crossdomain/model/modelmock"
"github.com/coze-dev/coze-studio/backend/domain/workflow/entity"
"github.com/coze-dev/coze-studio/backend/domain/workflow/entity/vo"
compose2 "github.com/coze-dev/coze-studio/backend/domain/workflow/internal/compose"
@@ -61,7 +62,7 @@ func TestQuestionAnswer(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockModelManager := mockmodel.NewMockManager(ctrl)
mockey.Mock(model.GetManager).Return(mockModelManager).Build()
mockey.Mock(crossmodelmgr.DefaultSVC).Return(mockModelManager).Build()
accessKey := os.Getenv("OPENAI_API_KEY")
baseURL := os.Getenv("OPENAI_BASE_URL")