54 lines
2.2 KiB
Go
54 lines
2.2 KiB
Go
package app
|
||
|
||
import (
|
||
"time"
|
||
|
||
"infinite-canvas/backend/internal/platform"
|
||
)
|
||
|
||
// 画布 Agent 单步的执行边界：一次模型调用最多输出多少 token、最多等多久。
|
||
// 两个值都来自运行时策略（管理端可改、环境变量可给部署级默认值），"0"是各自的关闭档位。
|
||
|
||
// cloudAgentStepOperation 是一次"模型调用"任务的操作名。它同时是任务中心里区分
|
||
// "这一条任务是 Agent 的某一步"的依据。
|
||
const cloudAgentStepOperation = "cloud_agent_step"
|
||
|
||
// cloudAgentStepBoostFallbackTokens 是"不限制输出"（策略值为 0）时放大重试用的预算：
|
||
// 不限制的本意是"让模型写完"，重试却必须有个上界，否则一次卡住的调用会一直占着单步墙钟。
|
||
const cloudAgentStepBoostFallbackTokens = 32_768
|
||
|
||
// cloudAgentStepLimits 是一次模型调用实际生效的执行边界。
|
||
type cloudAgentStepLimits struct {
|
||
// OutputTokens 是本次调用的输出上限；0 表示不限制（只有 Timeout 兜底）。
|
||
OutputTokens int
|
||
// Timeout 是单步墙钟：策略给了秒级值时用它，否则沿用文本任务超时。
|
||
Timeout time.Duration
|
||
}
|
||
|
||
// cloudAgentStepLimits 解析当前生效的单步边界；读取失败不得放宽管理员策略。
|
||
func (s *Service) cloudAgentStepLimits() (cloudAgentStepLimits, error) {
|
||
policy, err := s.runtimeConcurrencySetting()
|
||
if err != nil {
|
||
return cloudAgentStepLimits{}, err
|
||
}
|
||
limits := cloudAgentStepLimits{OutputTokens: policy.AgentStepMaxOutputTokens}
|
||
if policy.AgentStepTimeoutSeconds > 0 {
|
||
limits.Timeout = time.Duration(policy.AgentStepTimeoutSeconds) * time.Second
|
||
} else {
|
||
limits.Timeout = time.Duration(policy.TextTimeoutMinutes) * time.Minute
|
||
}
|
||
return limits, nil
|
||
}
|
||
|
||
// cloudAgentStepOutputBudget 把生效上限折算成本步请求要带的 maxOutputTokens：
|
||
// 空输出重试不能突破管理员配置的上限；只有原值为 0（不限制）时使用有界的重试预算。
|
||
func cloudAgentStepOutputBudget(limits cloudAgentStepLimits, boosted bool) int {
|
||
if !boosted {
|
||
return limits.OutputTokens
|
||
}
|
||
if limits.OutputTokens <= 0 {
|
||
return cloudAgentStepBoostFallbackTokens
|
||
}
|
||
return min(limits.OutputTokens, platform.MaxRuntimeAgentStepOutputTokens)
|
||
}
|