fix: replace workflow hardcoding chat model for knowledge recall (#233)

backend/domain/workflow/interface.go

@@ -19,6 +19,7 @@ package workflow
 import (
 	"context"
 
+	"github.com/cloudwego/eino/components/model"
 	"github.com/cloudwego/eino/compose"
 
 	"github.com/coze-dev/coze-studio/backend/api/model/ocean/cloud/workflow"
@@ -96,6 +97,8 @@ type Repository interface {
 
 	compose.CheckPointStore
 	idgen.IDGenerator
+
+	GetKnowledgeRecallChatModel() model.BaseChatModel
 }
 
 var repositorySingleton Repository

backend/domain/workflow/internal/compose/test/question_answer_test.go

@@ -102,7 +102,7 @@ func TestQuestionAnswer(t *testing.T) {
 		mockTos := storageMock.NewMockStorage(ctrl)
 		mockTos.EXPECT().GetObjectUrl(gomock.Any(), gomock.Any(), gomock.Any()).Return("", nil).AnyTimes()
 		repo := repo2.NewRepository(mockIDGen, db, redisClient, mockTos,
-			checkpoint.NewRedisStore(redisClient))
+			checkpoint.NewRedisStore(redisClient), nil)
 		mockey.Mock(workflow.GetRepository).Return(repo).Build()
 
 		t.Run("answer directly, no structured output", func(t *testing.T) {

backend/domain/workflow/internal/compose/to_node.go

@@ -223,9 +223,9 @@ func (s *NodeSchema) ToLLMConfig(ctx context.Context) (*llm.Config, error) {
 		}
 
 		if fcParams.KnowledgeFCParam != nil && len(fcParams.KnowledgeFCParam.KnowledgeList) > 0 {
-			kwChatModel, err := knowledgeRecallChatModel(ctx)
-			if err != nil {
-				return nil, err
+			kwChatModel := workflow2.GetRepository().GetKnowledgeRecallChatModel()
+			if kwChatModel == nil {
+				return nil, fmt.Errorf("workflow builtin chat model for knowledge recall not configured")
 			}
 			knowledgeOperator := crossknowledge.GetKnowledgeOperator()
 			setting := fcParams.KnowledgeFCParam.GlobalSetting
@@ -650,15 +650,3 @@ func totRetrievalSearchType(s int64) (crossknowledge.SearchType, error) {
 		return "", fmt.Errorf("invalid retrieval search type %v", s)
 	}
 }
-
-// knowledgeRecallChatModel the chat model used by the knowledge base recall in the LLM node, not the user-configured model
-func knowledgeRecallChatModel(ctx context.Context) (einomodel.BaseChatModel, error) {
-	defaultChatModelParma := &model.LLMParams{
-		ModelName:   "豆包·1.5·Pro·32k",
-		ModelType:   1,
-		Temperature: ptr.Of(0.5),
-		MaxTokens:   4096,
-	}
-	m, _, err := model.GetManager().GetModel(ctx, defaultChatModelParma)
-	return m, err
-}

backend/domain/workflow/internal/repo/repository.go

@@ -41,6 +41,7 @@ import (
 	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/execute"
 	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/model"
 	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/query"
+	cm "github.com/coze-dev/coze-studio/backend/infra/contract/chatmodel"
 	"github.com/coze-dev/coze-studio/backend/infra/contract/idgen"
 	"github.com/coze-dev/coze-studio/backend/infra/contract/storage"
 	"github.com/coze-dev/coze-studio/backend/pkg/errorx"
@@ -66,10 +67,11 @@ type RepositoryImpl struct {
 	workflow.InterruptEventStore
 	workflow.CancelSignalStore
 	workflow.ExecuteHistoryStore
+	builtinModel cm.BaseChatModel
 }
 
 func NewRepository(idgen idgen.IDGenerator, db *gorm.DB, redis *redis.Client, tos storage.Storage,
-	cpStore einoCompose.CheckPointStore) workflow.Repository {
+	cpStore einoCompose.CheckPointStore, chatModel cm.BaseChatModel) workflow.Repository {
 	return &RepositoryImpl{
 		IDGenerator:     idgen,
 		query:           query.Use(db),
@@ -86,6 +88,7 @@ func NewRepository(idgen idgen.IDGenerator, db *gorm.DB, redis *redis.Client, to
 			query: query.Use(db),
 			redis: redis,
 		},
+		builtinModel: chatModel,
 	}
 }
 
@@ -1583,6 +1586,10 @@ func (r *RepositoryImpl) BatchCreateConnectorWorkflowVersion(ctx context.Context
 	return nil
 }
 
+func (r *RepositoryImpl) GetKnowledgeRecallChatModel() cm.BaseChatModel {
+	return r.builtinModel
+}
+
 func filterDisabledAPIParameters(parametersCfg []*workflow3.APIParameter, m map[string]any) map[string]any {
 	result := make(map[string]any, len(m))
 	responseParameterMap := slices.ToMap(parametersCfg, func(p *workflow3.APIParameter) (string, *workflow3.APIParameter) {

backend/domain/workflow/service/service_impl.go

@@ -39,6 +39,7 @@ import (
 	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/canvas/adaptor"
 	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/compose"
 	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo"
+	"github.com/coze-dev/coze-studio/backend/infra/contract/chatmodel"
 	"github.com/coze-dev/coze-studio/backend/infra/contract/idgen"
 	"github.com/coze-dev/coze-studio/backend/infra/contract/storage"
 	"github.com/coze-dev/coze-studio/backend/pkg/lang/ptr"
@@ -67,8 +68,8 @@ func NewWorkflowService(repo workflow.Repository) workflow.Service {
 }
 
 func NewWorkflowRepository(idgen idgen.IDGenerator, db *gorm.DB, redis *redis.Client, tos storage.Storage,
-	cpStore einoCompose.CheckPointStore) workflow.Repository {
-	return repo.NewRepository(idgen, db, redis, tos, cpStore)
+	cpStore einoCompose.CheckPointStore, chatModel chatmodel.BaseChatModel) workflow.Repository {
+	return repo.NewRepository(idgen, db, redis, tos, cpStore, chatModel)
 }
 
 func (i *impl) ListNodeMeta(ctx context.Context, nodeTypes map[entity.NodeType]bool) (map[string][]*entity.NodeTypeMeta, []entity.Category, error) {