mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/yunionio/cloudpods.git
synced 2026-09-20 08:03:53 +08:00
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
package service
|
|
|
|
import (
|
|
"yunion.io/x/onecloud/pkg/appsrv"
|
|
"yunion.io/x/onecloud/pkg/appsrv/dispatcher"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db/taskman"
|
|
"yunion.io/x/onecloud/pkg/llm/models"
|
|
)
|
|
|
|
func InitHandlers(app *appsrv.Application, isSlave bool) {
|
|
db.InitAllManagers()
|
|
db.RegistUserCredCacheUpdater()
|
|
|
|
taskman.AddTaskHandler("", app, isSlave)
|
|
|
|
for _, manager := range []db.IModelManager{
|
|
taskman.TaskManager,
|
|
taskman.SubTaskManager,
|
|
taskman.TaskObjectManager,
|
|
taskman.ArchivedTaskManager,
|
|
|
|
db.SharedResourceManager,
|
|
db.UserCacheManager,
|
|
db.TenantCacheManager,
|
|
} {
|
|
db.RegisterModelManager(manager)
|
|
}
|
|
|
|
for _, manager := range []db.IModelManager{
|
|
db.OpsLog,
|
|
db.Metadata,
|
|
|
|
models.GetLLMImageManager(),
|
|
models.GetLLMSkuManager(),
|
|
models.GetDifySkuManager(),
|
|
models.GetVolumeManager(),
|
|
models.GetAccessInfoManager(),
|
|
models.GetLLMContainerManager(),
|
|
models.GetLLMManager(),
|
|
models.GetDifyManager(),
|
|
models.GetInstantModelManager(),
|
|
models.GetLLMInstantModelManager(),
|
|
} {
|
|
db.RegisterModelManager(manager)
|
|
handler := db.NewModelHandler(manager)
|
|
dispatcher.AddModelDispatcher("", app, handler, isSlave)
|
|
}
|
|
}
|