refactor(*): 平台架构 - 拆分后端领域模块并同步插件协议与模型能力
This commit is contained in:
ddCat committed
1 parent
916043fb5a
commit
6e5cbc8b05
644 files changed
+62731
-2320
No files matched your search
@@ -0,0 +1,518 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"infinite-canvas/backend/internal/model"
|
||||
"infinite-canvas/backend/internal/repository"
|
||||
)
|
||||
|
||||
// CreateTask 收敛任务进入系统前的 admission 流程：输入标准化、逻辑模型路由、
|
||||
// 能力/额度校验和持久化。执行阶段由 worker 与 provider 相关模块负责。
|
||||
// CreateTask 校验并创建一条生成任务。
|
||||
// 这是常规模型生成任务的写入口：客户端只提交创作意图，模型、渠道、协议和计价信息必须由服务端目录重新解析，
|
||||
// 以保证“可展示的模型”与“实际执行及扣费的模型”来自同一份有效配置。
|
||||
func (s *Service) CreateTask(userID string, req CreateTaskRequest) (*model.Task, error) {
|
||||
if s.IsDraining() {
|
||||
return nil, &AppError{Status: 503, Code: 503, Message: "服务正在维护，暂不接受新的生成任务", Retryable: true}
|
||||
}
|
||||
prompt := strings.TrimSpace(req.Prompt)
|
||||
if prompt == "" {
|
||||
return nil, errors.New("prompt is required")
|
||||
}
|
||||
taskType := strings.TrimSpace(req.Type)
|
||||
if err := validateTaskType(taskType); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
normalizedInput, err := normalizeTaskInput(req.Input)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var routed *RoutedModel
|
||||
logicalModelID := strings.TrimSpace(req.LogicalModelID)
|
||||
workflowProviderTask := taskInputUsesWorkflowProvider(normalizedInput)
|
||||
frontendEnabled := false
|
||||
if workflowProviderTask {
|
||||
config, _ := normalizedInput["config"].(map[string]any)
|
||||
if err := s.RequireWorkflowPluginForUser(userID, strings.TrimSpace(fmt.Sprint(config["interfaceType"]))); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
} else {
|
||||
// 工作流是独立执行器；普通模型仍严格使用主线的目录和路由校验。
|
||||
frontendEnabled, err = s.FeatureEnabled(FeatureFrontendModels)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
if !workflowProviderTask {
|
||||
routed, normalizedInput, err = s.resolveTaskModelSelection(normalizedInput, logicalModelID, taskType, req.Operation, frontendEnabled)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
if strings.HasPrefix(taskType, "video_") && !hasExecutableProviderVideoConfig(normalizedInput) {
|
||||
if mode, _ := normalizedInput["mode"].(string); mode != "video" {
|
||||
return nil, errors.New("视频任务必须使用 video 模式")
|
||||
}
|
||||
return nil, errors.New("视频任务缺少可执行的模型配置")
|
||||
}
|
||||
// 前端自管的文本持久化任务：直连模型生成、增量上报 text-deltas，不排入 worker 队列生成。
|
||||
if isTextReplayTaskRequest(normalizedInput) {
|
||||
return s.createTextReplayTask(userID, req, normalizedInput)
|
||||
}
|
||||
if err := s.requireCustomChannelsForTaskInput(normalizedInput); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := s.ValidateTaskCapability(normalizedInput); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if containsInlineMediaDataURL(normalizedInput) {
|
||||
return nil, BadAuthRequest("任务输入不能包含内嵌媒体，请先上传到资源存储")
|
||||
}
|
||||
policy, err := s.RuntimePolicy()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
activeTasks, err := s.repo.ActiveTaskCountForUser(userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if activeTasks >= int64(policy.Task.ActiveTaskLimit) {
|
||||
return nil, BadAuthRequest(fmt.Sprintf("同时排队或运行的任务最多 %d 个，请等待已有任务完成", policy.Task.ActiveTaskLimit))
|
||||
}
|
||||
task := model.Task{ID: newID(), UserID: userID, TraceID: req.TraceID, RequestID: req.RequestID, SessionID: req.SessionID, ProjectID: req.ProjectID, Type: taskType, Status: model.TaskStatusQueued, Stage: "等待队列调度", Progress: 5, Prompt: prompt, Operation: req.Operation, Provider: req.Provider, Model: req.Model}
|
||||
if routed != nil {
|
||||
task.LogicalModelID = routed.LogicalModel.ID
|
||||
task.LogicalModelRevisionID = routed.Revision.ID
|
||||
task.RouteID = routed.Route.ID
|
||||
task.ChannelModelID = routed.ChannelModel.ID
|
||||
task.RouteRun = 1
|
||||
task.Model = routed.LogicalModel.Code
|
||||
task.Provider = "managed"
|
||||
}
|
||||
if err := s.ensureTaskProjectActive(userID, req.ProjectID); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
billingOrder, err := s.taskBillingOrder(userID, &task, normalizedInput)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if req.creationPrepare != nil {
|
||||
encoded, encodeErr := json.Marshal(normalizedInput)
|
||||
if encodeErr != nil {
|
||||
return nil, encodeErr
|
||||
}
|
||||
task.InputJSON = string(encoded)
|
||||
req.creationPrepare.Order = billingOrder
|
||||
return &task, nil
|
||||
}
|
||||
if err := s.protectTaskSecrets(normalizedInput); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
inputJSON, err := json.Marshal(normalizedInput)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("序列化任务输入失败：%w", err)
|
||||
}
|
||||
task.InputJSON = string(inputJSON)
|
||||
if billingOrder != nil {
|
||||
task.BillingOrderID = billingOrder.ID
|
||||
}
|
||||
err = s.createTaskWithinStorageQuota(&task, billingOrder, policy)
|
||||
if errors.Is(err, repository.ErrActiveTaskLimit) {
|
||||
return nil, BadAuthRequest(fmt.Sprintf("同时排队或运行的任务最多 %d 个，请等待已有任务完成", policy.Task.ActiveTaskLimit))
|
||||
}
|
||||
if errors.Is(err, repository.ErrInsufficientCredits) {
|
||||
return nil, BadAuthRequest("积分不足，请先使用兑换码充值")
|
||||
}
|
||||
if errors.Is(err, repository.ErrLogicalModelUnavailable) {
|
||||
return nil, BadAuthRequest("所选模型已停用、归档或配置已更新，请重新选择")
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
s.recordActivity(userID, "task", 1)
|
||||
_ = s.log(userID, task.ID, "info", "任务已进入队列", "")
|
||||
return taskForOutput(task), nil
|
||||
}
|
||||
|
||||
// resolveTaskModelSelection 根据请求实际携带的模型选择决定路由方式。
|
||||
// 显式系统渠道和用户自定义渠道请求不能被全局前台模型开关误判；
|
||||
// 它们仍分别进入系统目录校验或自定义渠道的功能、能力与安全校验。
|
||||
func (s *Service) resolveTaskModelSelection(input map[string]any, logicalModelID string, taskType string, operation string, frontendEnabled bool) (*RoutedModel, map[string]any, error) {
|
||||
customChannelTask := taskInputUsesCustomChannel(input)
|
||||
if frontendEnabled && !taskInputUsesSystemChannel(input) && !customChannelTask {
|
||||
if logicalModelID == "" {
|
||||
return nil, input, InvalidModelSelection("前台模型模式下必须指定 logicalModelId")
|
||||
}
|
||||
intent := ModelRequestIntentFromTaskInput(input, taskType, operation)
|
||||
routed, err := s.ResolveLogicalModel(logicalModelID, intent)
|
||||
if err != nil {
|
||||
return nil, input, err
|
||||
}
|
||||
return routed, applyRoutedProviderSelection(input, routed), nil
|
||||
}
|
||||
|
||||
if logicalModelID != "" {
|
||||
return nil, input, ModelCatalogMismatch("模型目录已更新，请重新选择")
|
||||
}
|
||||
// 自定义渠道没有系统 channelId；它会在后续由自定义渠道功能开关、
|
||||
// 能力校验和 provider 配置校验共同处理，不能误报为“缺少系统渠道”。
|
||||
if !customChannelTask {
|
||||
resolvedInput, err := s.resolveSystemChannelModelSelection(input, taskType, operation)
|
||||
if err != nil {
|
||||
return nil, input, err
|
||||
}
|
||||
return nil, resolvedInput, nil
|
||||
}
|
||||
return nil, input, nil
|
||||
}
|
||||
|
||||
func applyRoutedProviderSelection(input map[string]any, routed *RoutedModel) map[string]any {
|
||||
config, _ := input["config"].(map[string]any)
|
||||
nextConfig := make(map[string]any, len(config)+2)
|
||||
for key, value := range config {
|
||||
switch key {
|
||||
case "channelId", "channelModelKey", "priceTierId", "providerModelKey", "apiFormat", "interfaceType", "baseUrl", "allowLocalChannel", "apiKey", "secretKey", "headers", "model", "capabilityConfig":
|
||||
continue
|
||||
default:
|
||||
nextConfig[key] = value
|
||||
}
|
||||
}
|
||||
for key, value := range routed.Defaults {
|
||||
canonical := canonicalCapabilityOptionName(key)
|
||||
if existing, exists := nextConfig[canonical]; !exists || existing == nil || strings.TrimSpace(fmt.Sprint(existing)) == "" {
|
||||
nextConfig[canonical] = providerConfigOptionValue(value)
|
||||
}
|
||||
}
|
||||
// 路由匹配和真实请求必须使用同一组参数；逻辑能力参数覆盖空的页面配置，但不携带供应链字段。
|
||||
if options, ok := input["capabilityOptions"].(map[string]any); ok {
|
||||
for key, value := range options {
|
||||
canonical := canonicalCapabilityOptionName(key)
|
||||
if isProviderCapabilityOption(canonical) {
|
||||
nextConfig[canonical] = providerConfigOptionValue(value)
|
||||
}
|
||||
}
|
||||
}
|
||||
nextConfig["channelId"] = routed.ChannelModel.ChannelID
|
||||
nextConfig["model"] = routed.ChannelModel.ModelKey
|
||||
nextConfig["channelModelKey"] = routed.ChannelModel.ModelKey
|
||||
if routed.PriceTier != nil {
|
||||
nextConfig["priceTierId"] = routed.PriceTier.ID
|
||||
nextConfig["providerModelKey"] = routed.PriceTier.ProviderModelKey
|
||||
}
|
||||
input["config"] = nextConfig
|
||||
return input
|
||||
}
|
||||
|
||||
func providerConfigOptionValue(value any) string {
|
||||
switch typed := value.(type) {
|
||||
case string:
|
||||
return typed
|
||||
case json.Number:
|
||||
return typed.String()
|
||||
default:
|
||||
return fmt.Sprint(value)
|
||||
}
|
||||
}
|
||||
|
||||
// 所有任务输入先收敛为 JSON 对象，确保计费与密钥保护不会因 Go 结构体类型不同而被绕过。
|
||||
func normalizeTaskInput(input map[string]any) (map[string]any, error) {
|
||||
if input == nil {
|
||||
return map[string]any{}, nil
|
||||
}
|
||||
encoded, err := json.Marshal(input)
|
||||
if err != nil {
|
||||
return nil, BadAuthRequest("任务输入格式无效")
|
||||
}
|
||||
var normalized map[string]any
|
||||
if err := json.Unmarshal(encoded, &normalized); err != nil {
|
||||
return nil, BadAuthRequest("任务输入格式无效")
|
||||
}
|
||||
if snapshot, ok := normalized["canvasSnapshot"]; ok {
|
||||
normalized["canvasSnapshot"] = compactPersistedValue(snapshot)
|
||||
}
|
||||
return normalized, nil
|
||||
}
|
||||
|
||||
// createTextReplayTask 创建前端自管的文本持久化任务：状态为 text_replay，
|
||||
// 不排队执行、不计 active 队列、不产生计费，仅作为正文增量（text-deltas）的存储容器。
|
||||
func (s *Service) createTextReplayTask(userID string, req CreateTaskRequest, normalizedInput map[string]any) (*model.Task, error) {
|
||||
prompt := strings.TrimSpace(req.Prompt)
|
||||
if prompt == "" {
|
||||
prompt = strings.TrimSpace(fmt.Sprint(normalizedInput["prompt"]))
|
||||
}
|
||||
if prompt == "" {
|
||||
return nil, errors.New("prompt is required")
|
||||
}
|
||||
taskType := strings.TrimSpace(req.Type)
|
||||
if err := validateTaskType(taskType); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
task := model.Task{
|
||||
ID: newID(), UserID: userID, TraceID: req.TraceID, RequestID: req.RequestID, SessionID: req.SessionID, ProjectID: req.ProjectID,
|
||||
Type: taskType, Status: model.TaskStatusTextReplay, Stage: "文本持久化（前端自管）", Progress: 5,
|
||||
Prompt: prompt, Operation: req.Operation, Provider: req.Provider, Model: strings.TrimSpace(req.Model),
|
||||
}
|
||||
if err := s.protectTaskSecrets(normalizedInput); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
inputJSON, _ := json.Marshal(normalizedInput)
|
||||
task.InputJSON = string(inputJSON)
|
||||
policy, err := s.RuntimePolicy()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := s.createTaskWithinStorageQuota(&task, nil, policy); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
_ = s.log(userID, task.ID, "info", "文本持久化任务已创建（前端自管）", "")
|
||||
return taskForOutput(task), nil
|
||||
}
|
||||
|
||||
// validateTaskType 是任务进入队列前的边界校验。视频任务允许携带具体操作后缀，
|
||||
// 其他任务类型必须是已实现的执行分支，避免未知类型落入假成功工作流。
|
||||
func validateTaskType(taskType string) error {
|
||||
switch taskType {
|
||||
case "text", "agent_storyboard", "agent_storyboard_rows", "canvas_text", "canvas_image", "canvas_video", "canvas_audio":
|
||||
return nil
|
||||
}
|
||||
if strings.HasPrefix(taskType, "video_") && strings.TrimPrefix(taskType, "video_") != "" {
|
||||
return nil
|
||||
}
|
||||
if taskType == "" {
|
||||
return errors.New("task type is required")
|
||||
}
|
||||
return fmt.Errorf("不支持的任务类型：%s", taskType)
|
||||
}
|
||||
|
||||
func (s *Service) requireCustomChannelsForTaskInput(input map[string]any) error {
|
||||
if !taskInputUsesCustomChannel(input) {
|
||||
return nil
|
||||
}
|
||||
return s.RequireFeature(FeatureCustomChannels)
|
||||
}
|
||||
|
||||
// resolveSystemChannelModelSelection 是系统渠道任务的 admission 边界。
|
||||
// 客户端只负责表达创作参数；模型协议、能力合同、价格档和上游模型标识必须从服务端记录重建，
|
||||
// 避免出现“按一个规格校验/计费，却按另一个规格执行”的跨阶段漂移。
|
||||
func (s *Service) resolveSystemChannelModelSelection(input map[string]any, taskType string, operation string) (map[string]any, error) {
|
||||
config, ok := input["config"].(map[string]any)
|
||||
if !ok {
|
||||
return input, InvalidModelSelection("缺少模型配置")
|
||||
}
|
||||
|
||||
channelID := strings.TrimSpace(stringValue(config["channelId"]))
|
||||
modelKey := strings.TrimPrefix(strings.TrimSpace(stringValue(config["model"])), "models/")
|
||||
|
||||
if channelID == "" || modelKey == "" {
|
||||
return input, InvalidModelSelection("必须指定系统渠道和模型")
|
||||
}
|
||||
|
||||
channel, err := s.repo.SystemChannel(channelID)
|
||||
if err != nil {
|
||||
return input, InvalidModelSelection("指定的渠道不存在")
|
||||
}
|
||||
if !channel.Enabled || channel.Scope != model.ChannelScopeSystem {
|
||||
return input, InvalidModelSelection("指定的渠道不可用")
|
||||
}
|
||||
|
||||
channelModel, err := s.repo.ChannelModelByKey(channelID, modelKey)
|
||||
if err != nil {
|
||||
return input, InvalidModelSelection("指定的模型不存在")
|
||||
}
|
||||
if !channelModel.Enabled {
|
||||
return input, InvalidModelSelection("指定的模型已停用")
|
||||
}
|
||||
if channelModel.Protocol == "" {
|
||||
return input, InvalidModelSelection("指定的模型未配置请求协议")
|
||||
}
|
||||
|
||||
nextConfig := make(map[string]any, len(config)+6)
|
||||
for key, value := range config {
|
||||
switch key {
|
||||
case "channelId", "channelModelKey", "priceTierId", "providerModelKey", "apiFormat", "interfaceType", "baseUrl", "allowLocalChannel", "apiKey", "secretKey", "headers", "model", "capabilityConfig":
|
||||
continue
|
||||
default:
|
||||
nextConfig[key] = value
|
||||
}
|
||||
}
|
||||
// capabilityOptions 是路由、校验和计价共同使用的请求规格；存在时必须覆盖 config 中的同名字段，
|
||||
// 不能让两份客户端数据分别驱动计费与真实上游请求。
|
||||
if options, ok := input["capabilityOptions"].(map[string]any); ok {
|
||||
for key, value := range options {
|
||||
canonical := canonicalCapabilityOptionName(key)
|
||||
if isCapabilityOptionFor(channelModel.Capability, canonical) {
|
||||
nextConfig[canonical] = value
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
capabilityConfig, err := normalizedChannelModelCapability(channelModel)
|
||||
if err != nil {
|
||||
return input, InvalidModelSelection("指定的模型能力配置无效，请联系管理员")
|
||||
}
|
||||
// 只有真实能力配置声明过的参数才能进入路由意图。客户端 config 可能保留
|
||||
// 旧模型的质量/分辨率值；若直接重新汇总，会把已关闭的参数误报为“不支持”。
|
||||
var capabilitySpec *CapabilitySpec
|
||||
if normalizeCapability(channelModel.Capability) != "audio" {
|
||||
spec, specErr := CapabilitySpecFromModelCapabilityConfig(capabilityConfig, channelModel.Capability)
|
||||
if specErr != nil {
|
||||
return input, InvalidModelSelection("指定的模型能力配置无效，请联系管理员")
|
||||
}
|
||||
capabilitySpec = &spec
|
||||
}
|
||||
applyChannelCapabilityDefaults(nextConfig, channelModel.Capability, capabilityConfig)
|
||||
input["config"] = nextConfig
|
||||
var declaredOptions map[string]OptionConstraint
|
||||
if capabilitySpec != nil {
|
||||
declaredOptions = capabilitySpec.Options
|
||||
}
|
||||
input["capabilityOptions"] = capabilityOptionsFromConfig(channelModel.Capability, nextConfig, declaredOptions)
|
||||
|
||||
intent := ModelRequestIntentFromTaskInput(input, taskType, operation)
|
||||
if normalizeCapability(intent.Capability) != normalizeCapability(channelModel.Capability) {
|
||||
return input, ModelCapabilityNotSupported("所选模型与任务能力不匹配")
|
||||
}
|
||||
if normalizeCapability(channelModel.Capability) != "audio" {
|
||||
if capabilitySpec == nil {
|
||||
return input, InvalidModelSelection("指定的模型能力配置无效，请联系管理员")
|
||||
}
|
||||
if match := MatchCapability(*capabilitySpec, intent); !match.Matched {
|
||||
return input, ModelCapabilityNotSupported("所选模型不支持当前请求：" + strings.Join(match.Reasons, "；"))
|
||||
}
|
||||
}
|
||||
|
||||
priceTier := channelModelPriceTierForIntent(*channelModel, intent)
|
||||
if priceTier == nil || !ValidatePriceTierPrice(priceTier, channelModel.Capability, channelModel.Protocol) {
|
||||
return input, ModelPriceNotConfigured("指定的模型未配置当前规格的有效价格")
|
||||
}
|
||||
|
||||
nextConfig["channelId"] = channel.ID
|
||||
nextConfig["model"] = channelModel.ModelKey
|
||||
nextConfig["channelModelKey"] = channelModel.ModelKey
|
||||
nextConfig["priceTierId"] = priceTier.ID
|
||||
nextConfig["providerModelKey"] = firstNonEmpty(priceTier.ProviderModelKey, channelModel.ProviderModelKey, channelModel.ModelKey)
|
||||
nextConfig["interfaceType"] = string(channelModel.Protocol)
|
||||
nextConfig["apiFormat"] = channelAPIFormatForProtocol(channel.APIFormat, channelModel.Protocol)
|
||||
return input, nil
|
||||
}
|
||||
|
||||
// applyChannelCapabilityDefaults 只采用管理员保存的能力默认值，且仅填补客户端未表达的参数。
|
||||
// 这些值随后会写回 capabilityOptions，使能力校验、SKU 选择和 provider 执行看到同一份规格。
|
||||
func applyChannelCapabilityDefaults(config map[string]any, capability string, profile *ModelCapabilityConfig) {
|
||||
setDefault := func(key string, value any) {
|
||||
if existing, exists := config[key]; !exists || existing == nil || strings.TrimSpace(fmt.Sprint(existing)) == "" {
|
||||
config[key] = value
|
||||
}
|
||||
}
|
||||
switch normalizeCapability(capability) {
|
||||
case "image":
|
||||
if profile == nil || profile.Image == nil {
|
||||
return
|
||||
}
|
||||
if profile.Image.Size.Parameter != "none" {
|
||||
setDefault("size", profile.Image.Size.Default)
|
||||
}
|
||||
if profile.Image.Quality.Supported {
|
||||
setDefault("quality", profile.Image.Quality.Default)
|
||||
}
|
||||
setDefault("transparentBackground", profile.Image.TransparentBackground.Default)
|
||||
setDefault("count", 1)
|
||||
case "video":
|
||||
if profile == nil || profile.Video == nil {
|
||||
return
|
||||
}
|
||||
if videoDurationSupported(profile.Video) {
|
||||
setDefault("videoSeconds", profile.Video.Duration.Default)
|
||||
}
|
||||
setDefault("size", profile.Video.DefaultRatio)
|
||||
setDefault("vquality", profile.Video.DefaultResolution)
|
||||
setDefault("videoGenerateAudio", profile.Video.GenerateAudio.Default)
|
||||
setDefault("videoWatermark", profile.Video.Watermark.Default)
|
||||
}
|
||||
}
|
||||
|
||||
func capabilityOptionsFromConfig(capability string, config map[string]any, declared map[string]OptionConstraint) map[string]any {
|
||||
options := map[string]any{}
|
||||
for key, value := range config {
|
||||
canonical := canonicalCapabilityOptionName(key)
|
||||
if !isCapabilityOptionFor(capability, canonical) || value == nil || strings.TrimSpace(fmt.Sprint(value)) == "" {
|
||||
continue
|
||||
}
|
||||
if declared != nil {
|
||||
if _, ok := declared[canonical]; !ok {
|
||||
continue
|
||||
}
|
||||
}
|
||||
options[canonical] = value
|
||||
}
|
||||
return options
|
||||
}
|
||||
|
||||
func taskInputUsesCustomChannel(input map[string]any) bool {
|
||||
if taskInputUsesWorkflowProvider(input) {
|
||||
return false
|
||||
}
|
||||
config, ok := input["config"].(map[string]any)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
channelID, _ := config["channelId"].(string)
|
||||
baseURL, _ := config["baseUrl"].(string)
|
||||
apiKey, _ := config["apiKey"].(string)
|
||||
if strings.TrimSpace(channelID) != "" || systemChannelIDFromBaseURL(baseURL) != "" {
|
||||
return false
|
||||
}
|
||||
return strings.TrimSpace(baseURL) != "" && strings.TrimSpace(apiKey) != ""
|
||||
}
|
||||
|
||||
func taskInputUsesSystemChannel(input map[string]any) bool {
|
||||
config, ok := input["config"].(map[string]any)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
channelID, _ := config["channelId"].(string)
|
||||
return strings.TrimSpace(channelID) != ""
|
||||
}
|
||||
|
||||
func taskInputUsesWorkflowProvider(input map[string]any) bool {
|
||||
config, ok := input["config"].(map[string]any)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
// 系统渠道的 interfaceType 是客户端缓存，不是授权事实；必须先走系统模型 admission，
|
||||
// 不能通过伪造工作流协议绕开渠道模型、能力和价格校验。
|
||||
if strings.TrimSpace(stringValue(config["channelId"])) != "" {
|
||||
return false
|
||||
}
|
||||
return isWorkflowProviderInterface(strings.TrimSpace(fmt.Sprint(config["interfaceType"])))
|
||||
}
|
||||
|
||||
func compactPersistedValue(value interface{}) interface{} {
|
||||
switch item := value.(type) {
|
||||
case map[string]interface{}:
|
||||
result := make(map[string]interface{}, len(item))
|
||||
for key, child := range item {
|
||||
if text, ok := child.(string); ok && strings.HasPrefix(text, "data:") {
|
||||
result[key] = ""
|
||||
continue
|
||||
}
|
||||
result[key] = compactPersistedValue(child)
|
||||
}
|
||||
return result
|
||||
case []interface{}:
|
||||
result := make([]interface{}, len(item))
|
||||
for index, child := range item {
|
||||
result[index] = compactPersistedValue(child)
|
||||
}
|
||||
return result
|
||||
default:
|
||||
return value
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user