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

feat: refactor model manager

Browse Source 
* chore: mv model icon
* fix: model icon
* fix: model icon
* feat: refactor model manager
* fix: model icon
* fix: model icon
* feat: refactor model manager

See merge request: !905
This commit is contained in:
徐兆楠
2025-07-24 13:12:44 +00:00
parent 12f7762797
commit 9b3814e2c5
114 changed files with 2888 additions and 4982 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
backend/domain/workflow/internal/nodes/llm/model_with_info.go
View File

@@ -24,28 +24,27 @@ import (
"github.com/cloudwego/eino/components"
"github.com/cloudwego/eino/components/model"
"github.com/cloudwego/eino/schema"
crossmodelmgr "github.com/coze-dev/coze-studio/backend/api/model/crossdomain/modelmgr"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/execute"
"github.com/coze-dev/coze-studio/backend/infra/contract/modelmgr"
)
type ModelWithInfo interface {
model.BaseChatModel
Info(ctx context.Context) *crossmodelmgr.Model
Info(ctx context.Context) *modelmgr.Model
}
type ModelForLLM struct {
Model model.BaseChatModel
MInfo *crossmodelmgr.Model
MInfo *modelmgr.Model
FallbackModel model.BaseChatModel
FallbackInfo *crossmodelmgr.Model
FallbackInfo *modelmgr.Model
UseFallback func(ctx context.Context) bool
modelEnableCallback bool
fallbackEnableCallback bool
}
func NewModel(m model.BaseChatModel, info *crossmodelmgr.Model) *ModelForLLM {
func NewModel(m model.BaseChatModel, info *modelmgr.Model) *ModelForLLM {
return &ModelForLLM{
Model: m,
MInfo: info,
@@ -57,7 +56,7 @@ func NewModel(m model.BaseChatModel, info *crossmodelmgr.Model) *ModelForLLM {
}
}
func NewModelWithFallback(m, f model.BaseChatModel, info, fInfo *crossmodelmgr.Model) *ModelForLLM {
func NewModelWithFallback(m, f model.BaseChatModel, info, fInfo *modelmgr.Model) *ModelForLLM {
return &ModelForLLM{
Model: m,
MInfo: info,
@@ -175,7 +174,7 @@ func (m *ModelForLLM) IsCallbacksEnabled() bool {
return true
}
func (m *ModelForLLM) Info(ctx context.Context) *crossmodelmgr.Model {
func (m *ModelForLLM) Info(ctx context.Context) *modelmgr.Model {
if m.UseFallback(ctx) {
return m.FallbackInfo
}

2
backend/domain/workflow/internal/nodes/llm/prompt.go
View File

@@ -22,8 +22,8 @@ import (
"github.com/cloudwego/eino/components/prompt"
"github.com/cloudwego/eino/schema"
"github.com/coze-dev/coze-studio/backend/infra/contract/modelmgr"
"github.com/coze-dev/coze-studio/backend/api/model/crossdomain/modelmgr"
"github.com/coze-dev/coze-studio/backend/domain/workflow/entity/vo"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/execute"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/nodes"