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

54
backend/internal/mock/domain/agent/singleagent/model_mgr_mock.go
View File

@@ -1,9 +1,9 @@
// Code generated by MockGen. DO NOT EDIT.
// Source: model_manager.go
// Source: backend/infra/contract/modelmgr/modelmgr.go
//
// Generated by this command:
//
// mockgen -destination ../../../../internal/mock/domain/agent/singleagent/model_mgr_mock.go --package mock -source model_manager.go
// mockgen --destination backend/internal/mock/domain/agent/singleagent/model_mgr_mock.go --package mock -source backend/infra/contract/modelmgr/modelmgr.go
//
// Package mock is a generated GoMock package.
@@ -13,47 +13,59 @@ import (
context "context"
reflect "reflect"
modelmgr "github.com/coze-dev/coze-studio/backend/infra/contract/modelmgr"
gomock "go.uber.org/mock/gomock"
modelmgr "github.com/coze-dev/coze-studio/backend/domain/modelmgr"
entity "github.com/coze-dev/coze-studio/backend/domain/modelmgr/entity"
)
// MockModelMgr is a mock of ModelMgr interface.
type MockModelMgr struct {
// MockManager is a mock of Manager interface.
type MockManager struct {
ctrl *gomock.Controller
recorder *MockModelMgrMockRecorder
isgomock struct{}
recorder *MockManagerMockRecorder
}
// MockModelMgrMockRecorder is the mock recorder for MockModelMgr.
type MockModelMgrMockRecorder struct {
mock *MockModelMgr
// MockManagerMockRecorder is the mock recorder for MockManager.
type MockManagerMockRecorder struct {
mock *MockManager
}
// NewMockModelMgr creates a new mock instance.
func NewMockModelMgr(ctrl *gomock.Controller) *MockModelMgr {
mock := &MockModelMgr{ctrl: ctrl}
mock.recorder = &MockModelMgrMockRecorder{mock}
// NewMockManager creates a new mock instance.
func NewMockManager(ctrl *gomock.Controller) *MockManager {
mock := &MockManager{ctrl: ctrl}
mock.recorder = &MockManagerMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockModelMgr) EXPECT() *MockModelMgrMockRecorder {
func (m *MockManager) EXPECT() *MockManagerMockRecorder {
return m.recorder
}
// ListModel mocks base method.
func (m *MockManager) ListModel(ctx context.Context, req *modelmgr.ListModelRequest) (*modelmgr.ListModelResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListModel", ctx, req)
ret0, _ := ret[0].(*modelmgr.ListModelResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ListModel indicates an expected call of ListModel.
func (mr *MockManagerMockRecorder) ListModel(ctx, req any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListModel", reflect.TypeOf((*MockManager)(nil).ListModel), ctx, req)
}
// MGetModelByID mocks base method.
func (m *MockModelMgr) MGetModelByID(ctx context.Context, req *modelmgr.MGetModelRequest) ([]*entity.Model, error) {
func (m *MockManager) MGetModelByID(ctx context.Context, req *modelmgr.MGetModelRequest) ([]*modelmgr.Model, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "MGetModelByID", ctx, req)
ret0, _ := ret[0].([]*entity.Model)
ret0, _ := ret[0].([]*modelmgr.Model)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// MGetModelByID indicates an expected call of MGetModelByID.
func (mr *MockModelMgrMockRecorder) MGetModelByID(ctx, req any) *gomock.Call {
func (mr *MockManagerMockRecorder) MGetModelByID(ctx, req any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetModelByID", reflect.TypeOf((*MockModelMgr)(nil).MGetModelByID), ctx, req)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetModelByID", reflect.TypeOf((*MockManager)(nil).MGetModelByID), ctx, req)
}

2
backend/internal/testutil/chat_model.go
View File

@@ -27,7 +27,7 @@ import (
"github.com/cloudwego/eino/components/model"
"github.com/cloudwego/eino/schema"
"github.com/coze-dev/coze-studio/backend/api/model/crossdomain/modelmgr"
"github.com/coze-dev/coze-studio/backend/infra/contract/modelmgr"
"github.com/coze-dev/coze-studio/backend/pkg/logs"
)