cmd

type APIFlags struct {
	CommandFlags
	Method   string   `flag:"method" short:"X" desc:"HTTP method, defaults to GET or POST if fields are provided"`
	Field    []string `flag:"field" short:"F" desc:"Add a string field (key=value), parsed as JSON value"`
	RawField []string `flag:"raw-field" short:"f" desc:"Add a raw string field (key=value)"`
	Header   []string `flag:"header" short:"H" desc:"Add a request header (key:value)"`
	Input    string   `flag:"input" desc:"Read request body from file (use - for stdin)"`
}

type APIInferenceFlags struct {
	APIFlags
	InferenceClientFlags
}

type APIManagementFlags struct {
	APIFlags
}

type AuditLogFlags struct {
	Start time.Time     `` /* 258-byte string literal not displayed */
	End   time.Time     `` /* 152-byte string literal not displayed */
	Since time.Duration `` /* 184-byte string literal not displayed */

	Limit int `` /* 135-byte string literal not displayed */

	// PageSize is the per-request fetch size while paging. Hidden; exists so
	// tests can force multiple pages without a full page of entries.
	PageSize int `flag:"page-size" hidden:"true" desc:"Entries fetched per backend request while paging." default:"200"`

	Direction string `` /* 145-byte string literal not displayed */
	Search    string `flag:"search" desc:"Case-insensitive substring matched against resource names and IDs in the entry."`

	EventTypeGroups []string `` /* 364-byte string literal not displayed */
	Sources         []string `flag:"source" desc:"Only return entries issued from one of these surfaces. May be repeated. One of: ui, api, mcp, other."`
	UserIDs         []string `flag:"user-id" desc:"Only return entries whose acting user is one of these IDs. May be repeated."`
	DeploymentIDs   []string `flag:"deployment-id" desc:"Only return entries referencing one of these model deployment IDs. May be repeated."`
	Environments    []string `flag:"environment" desc:"Only return entries for one of these environments. May be repeated."`
}

type AuthLoginFlags struct {
	CommandFlags
	Web             bool   `flag:"web" desc:"Use browser login without interactive prompts"`
	WithAPIKey      bool   `flag:"with-api-key" desc:"Read API key from stdin"`
	RemoteURL       string `flag:"remote-url" desc:"Baseten remote URL for this profile (default https://app.baseten.co)"`
	NoSwitch        bool   `flag:"no-switch" desc:"Store the profile without making it the current profile"`
	InsecureStorage bool   `flag:"insecure-storage" desc:"Store credentials in plain text instead of system keyring"`
}

type AuthLoginResult struct {
	Profile       string `json:"profile"`
	UserID        string `json:"user_id"`
	Email         string `json:"email"`
	Name          string `json:"name"`
	WorkspaceName string `json:"workspace_name"`
}

type AuthLogoutFlags struct {
	CommandFlags
}

type AuthLogoutResult struct {
	Profile string `json:"profile"`
}

type AuthStatusFlags struct {
	CommandFlags
}

type AuthStatusResult struct {
	Profile   string `json:"profile"`
	RemoteURL string `json:"remote_url"`
	AuthType  string `json:"auth_type"`
}

type AuthSwitchFlags struct {
	CommandFlags
}

type AuthSwitchResult struct {
	Profile string `json:"profile"`
}

type AutoscalingScheduleTimingFlags struct {
	Enabled OptionalFlag[bool] `flag:"enabled" desc:"Whether the schedule is enabled. Defaults to true on create."`

	Weekdays    []string          `` /* 157-byte string literal not displayed */
	StartHour   OptionalFlag[int] `` /* 151-byte string literal not displayed */
	StartMinute OptionalFlag[int] `flag:"start-minute" desc:"Minute the window opens. Recurring schedules only."`
	EndHour     OptionalFlag[int] `` /* 150-byte string literal not displayed */
	EndMinute   OptionalFlag[int] `flag:"end-minute" desc:"Minute the window closes. Recurring schedules only."`

	StartAt time.Time `` /* 211-byte string literal not displayed */
	EndAt   time.Time `flag:"end-at" desc:"Exclusive end of the window, ISO 8601. One-time schedules only."`
}

type AutoscalingSettingsFlags struct {
	MinReplica                  OptionalFlag[int] `flag:"min-replica" desc:"Minimum number of replicas."`
	MaxReplica                  OptionalFlag[int] `flag:"max-replica" desc:"Maximum number of replicas."`
	AutoscalingWindow           OptionalFlag[int] `flag:"autoscaling-window" desc:"Timeframe of traffic considered for autoscaling decisions, in seconds."`
	ScaleDownDelay              OptionalFlag[int] `flag:"scale-down-delay" desc:"Waiting period before scaling down any active replica, in seconds."`
	ConcurrencyTarget           OptionalFlag[int] `flag:"concurrency-target" desc:"Number of requests per replica before scaling up."`
	TargetUtilizationPercentage OptionalFlag[int] `flag:"target-utilization-percentage" desc:"Target utilization percentage for scaling up and down."`
	TargetInFlightTokens        OptionalFlag[int] `flag:"target-in-flight-tokens" desc:"Target number of in-flight tokens for autoscaling decisions. Early access only."`
	MaxScaleDownRate            OptionalFlag[int] `` /* 185-byte string literal not displayed */
}

type Command struct {
	Name        string
	Summary     string
	Description string
	Flags       any // nil, or struct value with flag tags
	Children    []Command
	// ArgsUsage is appended to the command name in help output (e.g. "[path]").
	ArgsUsage string
	// ExactArgs requires exactly this many positional arguments. Mutually
	// exclusive with MaxArgs.
	ExactArgs int
	// MaxArgs is the maximum number of positional arguments. 0 means no args
	// (the default), -1 means unlimited. Mutually exclusive with ExactArgs.
	MaxArgs int
	// DisableFlagParsing disables all flag parsing; everything after the command
	// name is passed as raw args. MaxArgs is ignored and assumed -1.
	DisableFlagParsing bool
	// Output declares the leaf command's stdout shape, text-mode behavior, and
	// examples. Required on every leaf (a command with no Children); must be
	// nil on commands with Children. Typically a *[CommandOutput][T].
	Output CommandOutputSpec
	// Errors lists command-declared typed errors. Each entry is built via
	// [ErrorDescOf] and documents one extra exit code surfaced by this leaf
	// beyond the standard set. Rendered in --help-output.
	Errors []ErrorDesc
	// Hidden keeps the command out of help listings and completion. The command
	// is still fully runnable.
	Hidden bool
}

type CommandError interface {
	error
	ExitCode() ExitCode
	Meaning() string
}

type CommandErrorMeta struct {
	// contains filtered or unexported fields
}

type CommandExample struct {
	// Description is a one-line "what this does" preceding the command.
	Description string
	// Command is a literal shell line beginning with "baseten ...".
	Command string
}

type CommandFlag struct {
	Name     string
	Short    string
	Desc     string
	Default  string
	Enum     []string
	Required bool
	Hidden   bool   // omitted from help output; still parsed and settable
	Oneof    string // group name: exactly one flag in the group must be set
	// Nullable allows an [OptionalFlag] to be passed as 'null'.
	Nullable  bool
	Type      reflect.Type
	FieldName string // Go struct field name
	// Group is the help-output flag-section bucket. Empty in raw metadata; the
	// loader fills in [DefaultFlagGroup] when no `group:` tag is set.
	Group string
	// GroupPri is the rendering priority declared via `group-pri:`. 0 means
	// "unset on this field". The whole group resolves to [DefaultFlagGroupPri]
	// when no field in the group declares one. Lower values render earlier.
	GroupPri int
}

type CommandFlags struct {
	Verbose bool   `flag:"verbose" short:"v" desc:"Enable verbose logging" group:"common" group-pri:"500"`
	Output  string `flag:"output" short:"o" desc:"Output format" default:"text" enum:"text,json,jsonl,none" group:"common"`
	JQ      string `` /* 137-byte string literal not displayed */
	Profile string `` /* 135-byte string literal not displayed */
}

type CommandOutput[JSONT any] struct {
	// TextDescription describes the --output text format. Free-form prose.
	TextDescription string
	// JSONDescription is optional prose describing the --output json shape
	// beyond what the JSON schema already conveys (e.g. how status fields
	// behave, what --dry-run emits). Free-form.
	JSONDescription string
	// Examples documents how to invoke the command. At least one required.
	Examples []CommandExample
	// JQExample is a required example that uses --jq. Surfaced separately so
	// every command guarantees a JQ-using example for agents to copy.
	// Optional on commands with DisableFlagParsing, since --jq is not honored
	// when the framework does not parse flags.
	JQExample CommandExample
	// JSONArrayStreamed indicates the command streams JSONT records: under
	// --output json the records are wrapped in a JSON array, under
	// --output jsonl one record per line, and --jq applies per record.
	JSONArrayStreamed bool
	// JSONOutputUnimportant marks a command that produces no meaningful JSON
	// stdout payload (e.g. a long-running watcher that streams status to
	// stderr). Such a command is exempt from the JQExample requirement, and
	// --help-output omits the JSON schema block for it.
	JSONOutputUnimportant bool
}

type CommandOutputSpec interface {
	// JSONOutputType returns the Go type of the stdout payload under
	// --output json, used for JSON schema generation and documentation.
	JSONOutputType() reflect.Type
	// Text describes the human-readable (--output text) output. Free-form
	// prose; surfaced via --help-output.
	Text() string
	// JSON describes the --output json shape beyond what the JSON schema
	// already conveys (special cases, status semantics, etc.). May be empty.
	JSON() string
	// ExampleList returns the declared usage examples. Every leaf must
	// declare at least one.
	ExampleList() []CommandExample
	// JQ returns the required example invoking --jq.
	JQ() CommandExample
	// JSONArrayStreamedBool reports whether the command streams JSONT records:
	// --output json wraps them in a JSON array, --output jsonl emits one
	// record per line, and --jq applies per record.
	JSONArrayStreamedBool() bool
	// JSONOutputUnimportantBool reports whether the command produces no
	// meaningful JSON stdout payload (so --jq is not useful and no JQ example
	// is required).
	JSONOutputUnimportantBool() bool
}

type ErrAuth struct{ CommandErrorMeta }

type ErrGeneric struct{ CommandErrorMeta }

type ErrNotFound struct{ CommandErrorMeta }

type ErrServer struct{ CommandErrorMeta }

type ErrUsage struct{ CommandErrorMeta }

type ErrValidation struct{ CommandErrorMeta }

type ErrorDesc struct {
	Name    string
	Code    ExitCode
	Meaning string
}

type ExitCode int

type InferenceClientFlags struct {
	ModelID     string `flag:"model-id" desc:"Model ID to target"`
	ChainID     string `flag:"chain-id" desc:"Chain ID to target"`
	Environment string `flag:"environment" desc:"Environment name (e.g. production)"`
}

type JSONAny = map[string]any

type JSONUndefined struct{}

type LogFlags struct {
	Tail bool `` /* 254-byte string literal not displayed */

	Start time.Time     `` /* 278-byte string literal not displayed */
	End   time.Time     `` /* 185-byte string literal not displayed */
	Since time.Duration `` /* 203-byte string literal not displayed */

	Limit int `` /* 196-byte string literal not displayed */

	// PageSize is the per-request fetch size while paging. Hidden; exists so
	// tests can force multiple pages without generating a full page of logs.
	PageSize int `flag:"page-size" hidden:"true" desc:"Log lines fetched per backend request while paging." default:"1000"`

	MinLevel      string   `flag:"min-level" desc:"Only return logs at or above this severity level." enum:"debug,info,warning,error"`
	Includes      []string `flag:"includes" desc:"Case-sensitive substring that must appear in the log message. May be repeated; all must match."`
	Excludes      []string `flag:"excludes" desc:"Case-sensitive substring; lines containing it are dropped. May be repeated."`
	SearchPattern string   `` /* 146-byte string literal not displayed */
	Replica       string   `flag:"replica" desc:"Only return logs emitted by this replica (5-char short ID)."`
	RequestID     string   `flag:"request-id" desc:"Only return logs tagged with this inference request ID."`
}

type LoopsCheckpointDeployFlags struct {
	CommandFlags
	TrussAuthFlags

	RunID        string   `flag:"run-id" desc:"Loops run whose checkpoints are deployed. Cannot be combined with --checkpoint-id."`
	Checkpoint   []string `` /* 196-byte string literal not displayed */
	CheckpointID []string `` /* 130-byte string literal not displayed */
	Config       string   `` /* 182-byte string literal not displayed */
	DryRun       bool     `flag:"dry-run" desc:"Print the generated model config without deploying anything."`
}

type LoopsCheckpointFilesFlags struct {
	CommandFlags

	CheckpointID string `flag:"checkpoint-id" desc:"ID of the Loops checkpoint." required:"true"`
}

type LoopsCheckpointListFlags struct {
	CommandFlags

	RunID     string `flag:"run-id" desc:"List checkpoints saved by this run." oneof:"loops-checkpoint-scope"`
	BaseModel string `flag:"base-model" desc:"List checkpoints across your runs of this base model." oneof:"loops-checkpoint-scope"`
	Direction string `` /* 130-byte string literal not displayed */
}

type LoopsLogFlags struct {
	Tail bool `` /* 244-byte string literal not displayed */

	Start time.Time     `` /* 278-byte string literal not displayed */
	End   time.Time     `` /* 185-byte string literal not displayed */
	Since time.Duration `` /* 203-byte string literal not displayed */

	Limit int `` /* 196-byte string literal not displayed */

	// PageSize is the per-request fetch size while paging. Hidden; exists so
	// tests can force multiple pages without generating a full page of logs.
	PageSize int `flag:"page-size" hidden:"true" desc:"Log lines fetched per backend request while paging." default:"1000"`

	MinLevel string `flag:"min-level" desc:"Only return logs at or above this severity level." enum:"debug,info,warning,error"`
}

type LoopsRunCreateFlags struct {
	CommandFlags

	BaseModel string `flag:"base-model" desc:"HuggingFace ID of the base model to fine-tune (for example 'Qwen/Qwen3-8B')." required:"true"`
	Name      string `flag:"name" desc:"Display name for the run. Defaults to the base model."`
	Replicas  int    `` /* 214-byte string literal not displayed */
	Team      string `flag:"team" desc:"Team name or ID that owns the run's infrastructure. Defaults to the organization's default team."`
}

type LoopsRunDeactivateFlags struct {
	CommandFlags
	LoopsRunRefFlags

	Yes bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type LoopsRunDescribeFlags struct {
	CommandFlags
	LoopsRunRefFlags
}

type LoopsRunListFlags struct {
	CommandFlags

	BaseModel string `flag:"base-model" desc:"Only list runs of this base model."`
	Org       bool   `flag:"org" desc:"List every run in the organization, with its owner, instead of only your own."`
	All       bool   `flag:"all" desc:"Include inactive runs."`
	Direction string `` /* 130-byte string literal not displayed */
}

type LoopsRunLogsFlags struct {
	CommandFlags
	LoopsRunRefFlags
	LoopsLogFlags

	Sampler bool `flag:"sampler" desc:"Fetch the paired sampler's logs instead of the run's trainer logs."`
}

type LoopsRunRefFlags struct {
	RunID string `flag:"run-id" desc:"ID of the Loops run." required:"true"`
}

type LoopsUsageFlags struct {
	CommandFlags

	Org  bool   `flag:"org" desc:"Report usage across the organization, with each allocation's owner, instead of only your own."`
	User string `flag:"user" desc:"Only report allocations owned by this email. Implies --org."`
	All  bool   `flag:"all" desc:"Include allocations holding no live GPUs (scaled to zero, or in a terminal state)."`
}

type LoopsUsageResult struct {
	Summary LoopsUsageSummary `json:"summary"`
	Rows    []LoopsUsageRow   `json:"rows"`
}

type LoopsUsageRow struct {
	RunID     string `json:"run_id,omitempty"`
	Owner     string `json:"owner,omitempty"`
	BaseModel string `json:"base_model"`
	CreatedAt string `json:"created_at"`

	TrainerStatus       string `json:"trainer_status,omitempty"`
	TrainerInstanceType string `json:"trainer_instance_type,omitempty"`
	TrainerNodeCount    int    `json:"trainer_node_count,omitempty"`
	TrainerGPUs         int    `json:"trainer_gpus"`

	SamplerStatus       string `json:"sampler_status,omitempty"`
	SamplerInstanceType string `json:"sampler_instance_type,omitempty"`
	SamplerNodeCount    int    `json:"sampler_node_count,omitempty"`
	SamplerGPUs         int    `json:"sampler_gpus"`
}

type LoopsUsageSummary struct {
	TrainerInUse        int `json:"trainer_in_use"`
	TrainerScaledToZero int `json:"trainer_scaled_to_zero"`
	SamplerInUse        int `json:"sampler_in_use"`
	SamplerScaledToZero int `json:"sampler_scaled_to_zero"`
}

type MetricsFlags struct {
	Mode string `` /* 324-byte string literal not displayed */

	Start time.Time     `` /* 314-byte string literal not displayed */
	End   time.Time     `` /* 221-byte string literal not displayed */
	Since time.Duration `` /* 198-byte string literal not displayed */

	Metric []string `` /* 206-byte string literal not displayed */

	NoChart bool `flag:"no-chart" desc:"For --mode series, emit a per-step table instead of sparklines."`
}

type ModelAPIDescribeFlags struct {
	CommandFlags

	Model string `flag:"model" desc:"Name of the Model API to describe." required:"true"`
}

type ModelAPIList struct {
	Items []managementapi.ModelAPI `json:"items"`
}

type ModelAPIListFlags struct {
	CommandFlags

	AddedOnly bool `flag:"added-only" desc:"Restrict to the Model APIs the workspace has added instead of the full visible catalog."`
}

type ModelAPIPredictFlags struct {
	CommandFlags

	URL   string `flag:"url" desc:"Endpoint to POST the request to. Defaults to https://inference.baseten.co/v1/chat/completions."`
	Model string `flag:"model" desc:"Name of the Model API. Required with --content, where it sets the request's model." `

	Content string `` /* 194-byte string literal not displayed */
	Data    string `flag:"data" desc:"Inline request body, sent verbatim." oneof:"predict-input"`
	File    string `flag:"file" desc:"Path to a file containing the request body, sent verbatim. Use '-' for stdin." oneof:"predict-input"`
}

type ModelAPIUsageFlags struct {
	CommandFlags

	Start time.Time     `` /* 127-byte string literal not displayed */
	End   time.Time     `flag:"end" desc:"End of the range, exclusive. ISO 8601, local when no timezone is given. Defaults to now."`
	Since time.Duration `flag:"since" desc:"Window from a relative time ago until now (e.g. '30m', '3d'). Mutually exclusive with --start and --end."`

	BucketWidth string   `` /* 147-byte string literal not displayed */
	GroupBy     []string `flag:"group-by" desc:"Dimension to break usage down by. May be repeated. One of: api-key, user, model. Defaults to model."`

	APIKeyPrefixes []string `flag:"api-key-prefix" desc:"Only return usage for these API key prefixes. May be repeated."`
	UserIDs        []string `flag:"user-id" desc:"Only return usage attributed to these user IDs. May be repeated."`
	Models         []string `flag:"model" desc:"Only return usage for these models. May be repeated."`

	Limit int `flag:"limit" desc:"Maximum number of time buckets, paging as needed. Rows per bucket depend on --group-by. 0 for no limit."`

	// PageSize is the per-request fetch size while paging. Hidden; exists so
	// tests can force multiple pages without a full page of buckets. Zero uses
	// the backend's maximum for the bucket width.
	PageSize int `flag:"page-size" hidden:"true" desc:"Time buckets fetched per backend request while paging."`
}

type ModelAuditLogsFlags struct {
	CommandFlags
	ModelRefFlags
	AuditLogFlags
}

type ModelDeleteFlags struct {
	CommandFlags
	ModelRefFlags

	Yes bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type ModelDeploymentActivateFlags struct {
	CommandFlags
	ModelDeploymentIDFlags
}

type ModelDeploymentConfigFlags struct {
	CommandFlags
	ModelDeploymentIDFlags
}

type ModelDeploymentDeactivateFlags struct {
	CommandFlags
	ModelDeploymentIDFlags

	Yes bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type ModelDeploymentDeleteFlags struct {
	CommandFlags
	ModelDeploymentIDFlags

	Yes bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type ModelDeploymentDescribeFlags struct {
	CommandFlags
	ModelDeploymentIDFlags
}

type ModelDeploymentDownloadFlags struct {
	CommandFlags
	ModelDeploymentIDFlags

	OutFile   string `flag:"out-file" desc:"Save the model source as an uncompressed tar file at this path." oneof:"download-out"`
	OutDir    string `flag:"out-dir" desc:"Extract the model source tar into this directory." oneof:"download-out"`
	Overwrite bool   `flag:"overwrite" desc:"Allow overwriting an existing file or non-empty directory."`
}

type ModelDeploymentDownloadResult struct {
	OutFile string `json:"out_file,omitempty"`
	OutDir  string `json:"out_dir,omitempty"`
}

type ModelDeploymentIDFlags struct {
	ModelRefFlags
	DeploymentID   string `flag:"deployment-id" desc:"ID of the deployment." oneof:"deployment-ref"`
	DeploymentName string `flag:"deployment-name" desc:"Name of the deployment." oneof:"deployment-ref"`
}

type ModelDeploymentListFlags struct {
	CommandFlags
	ModelRefFlags
}

type ModelDeploymentLogsFlags struct {
	CommandFlags
	ModelDeploymentIDFlags
	LogFlags
}

type ModelDeploymentMetricsFlags struct {
	CommandFlags
	ModelDeploymentIDFlags
	MetricsFlags
}

type ModelDeploymentPromoteFlags struct {
	CommandFlags
	ModelDeploymentIDFlags

	Environment             string `flag:"environment" desc:"Target environment name. Defaults to production." default:"production"`
	OverrideEnvInstanceType bool   `flag:"override-env-instance-type" desc:"Use this deployment's instance type instead of preserving the target environment's."`

	Yes bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type ModelDeploymentRenameFlags struct {
	CommandFlags
	ModelDeploymentIDFlags

	NewName string `` /* 182-byte string literal not displayed */
}

type ModelDeploymentReplicaIDFlags struct {
	ModelDeploymentIDFlags
	ReplicaID string `flag:"replica-id" desc:"ID of the replica." required:"true"`
}

type ModelDeploymentReplicaTerminateFlags struct {
	CommandFlags
	ModelDeploymentReplicaIDFlags

	Yes bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type ModelDeploymentUpdateAutoscalingFlags struct {
	CommandFlags
	ModelDeploymentIDFlags
	AutoscalingSettingsFlags
}

type ModelDeploymentUpdateRequestBackpressureFlags struct {
	CommandFlags
	ModelDeploymentIDFlags
	RequestBackpressureFlags
}

type ModelDescribeFlags struct {
	CommandFlags
	ModelRefFlags
}

type ModelEnvironmentActivateFlags struct {
	CommandFlags
	ModelEnvironmentFlags
}

type ModelEnvironmentAutoscalingScheduleCreateFlags struct {
	CommandFlags
	ModelEnvironmentFlags
	AutoscalingScheduleTimingFlags
	AutoscalingSettingsFlags

	Name     string `flag:"name" desc:"Name of the schedule, unique among the environment's schedules." required:"true"`
	Cadence  string `` /* 193-byte string literal not displayed */
	Timezone string `` /* 187-byte string literal not displayed */
}

type ModelEnvironmentAutoscalingScheduleDeleteFlags struct {
	CommandFlags
	ModelEnvironmentFlags
	ModelEnvironmentAutoscalingScheduleRefFlags
}

type ModelEnvironmentAutoscalingScheduleListFlags struct {
	CommandFlags
	ModelEnvironmentFlags
}

type ModelEnvironmentAutoscalingScheduleRefFlags struct {
	ScheduleID   string `flag:"schedule-id" desc:"Stable identifier of the schedule, from 'autoscaling-schedule list'." oneof:"schedule-ref"`
	ScheduleName string `flag:"schedule-name" desc:"Name of the schedule, unique among the environment's schedules." oneof:"schedule-ref"`
}

type ModelEnvironmentAutoscalingScheduleUpdateFlags struct {
	CommandFlags
	ModelEnvironmentFlags
	ModelEnvironmentAutoscalingScheduleRefFlags
	AutoscalingScheduleTimingFlags
	AutoscalingSettingsFlags

	Name    OptionalFlag[string] `flag:"name" desc:"New name for the schedule, unique among the environment's schedules."`
	Cadence OptionalFlag[string] `` /* 167-byte string literal not displayed */
}

type ModelEnvironmentAutoscalingScheduleUpdateSettingsFlags struct {
	CommandFlags
	ModelEnvironmentFlags

	Timezone string `` /* 129-byte string literal not displayed */
}

type ModelEnvironmentDeactivateFlags struct {
	CommandFlags
	ModelEnvironmentFlags

	Yes bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type ModelEnvironmentDescribeFlags struct {
	CommandFlags
	ModelEnvironmentFlags
}

type ModelEnvironmentFlags struct {
	ModelRefFlags
	Environment string `flag:"environment" desc:"Name of the environment (e.g. production)." required:"true"`
}

type ModelEnvironmentListFlags struct {
	CommandFlags
	ModelRefFlags
}

type ModelEnvironmentLogsFlags struct {
	CommandFlags
	ModelEnvironmentFlags
	LogFlags
}

type ModelEnvironmentMetricsFlags struct {
	CommandFlags
	ModelEnvironmentFlags
	MetricsFlags
}

type ModelEnvironmentUpdateAutoscalingFlags struct {
	CommandFlags
	ModelEnvironmentFlags
	AutoscalingSettingsFlags
}

type ModelEnvironmentUpdatePromotionFlags struct {
	CommandFlags
	ModelEnvironmentFlags

	RedeployOnPromotion      OptionalFlag[bool]   `` /* 234-byte string literal not displayed */
	RollingDeploy            OptionalFlag[bool]   `flag:"rolling-deploy" desc:"Whether the environment should rely on rolling deploy orchestration."`
	PromotionCleanupStrategy OptionalFlag[string] `` /* 134-byte string literal not displayed */
	RampUpWhilePromoting     OptionalFlag[bool]   `flag:"ramp-up-while-promoting" desc:"Whether to ramp up traffic while promoting."`
	RampUpDurationSeconds    OptionalFlag[int]    `flag:"ramp-up-duration-seconds" desc:"Duration of the ramp up, in seconds."`

	RollingDeployStrategy    OptionalFlag[string] `flag:"rolling-deploy-strategy" desc:"The rolling deploy strategy to use for promotions." enum:"replica"`
	MaxSurgePercent          OptionalFlag[int]    `flag:"max-surge-percent" desc:"The maximum surge percentage for rolling deploys."`
	MaxUnavailablePercent    OptionalFlag[int]    `flag:"max-unavailable-percent" desc:"The maximum unavailable percentage for rolling deploys."`
	StabilizationTimeSeconds OptionalFlag[int]    `flag:"stabilization-time-seconds" desc:"The stabilization time for rolling deploys, in seconds."`
	ReplicaOverheadPercent   OptionalFlag[int]    `flag:"replica-overhead-percent" desc:"The replica overhead percentage for rolling deploys."`
}

type ModelEnvironmentUpdateRequestBackpressureFlags struct {
	CommandFlags
	ModelEnvironmentFlags
	RequestBackpressureFlags
}

type ModelImageBuildFlags struct {
	ModelImageCommonFlags
	BuildDir string `flag:"build-dir" desc:"Directory for the Docker build context. Defaults to a temp dir removed after the build."`
	Tag      string `flag:"tag" desc:"Image tag. Defaults to '<model_name>:latest' from config.yaml."`
}

type ModelImageBuildResult struct {
	ImageID string `json:"image_id"`
	Tag     string `json:"tag"`
}

type ModelImageCommonFlags struct {
	CommandFlags
	// TrussFlags without credential forwarding: generating a build context runs
	// entirely locally and calls no API.
	TrussFlags

	Dir          string `flag:"dir" desc:"Model directory." default:"."`
	RootUser     bool   `flag:"root-user" desc:"Run the image's container as root instead of the default non-root user."`
	CacheMountID string `flag:"cache-mount-id" desc:"Enable a persistent apt/pip/uv build cache keyed by this id."`
}

type ModelImagePrepareFlags struct {
	ModelImageCommonFlags
	BuildDir string `flag:"build-dir" desc:"Directory for the Docker build context. Created if absent and kept." required:"true"`
}

type ModelImagePrepareResult struct {
	BuildDir   string `json:"build_dir"`
	Dockerfile string `json:"dockerfile"`
}

type ModelListFlags struct {
	CommandFlags

	Team string `flag:"team" desc:"Team name or ID to scope the listing to. Defaults to all teams the caller can see."`
}

type ModelPredictFlags struct {
	CommandFlags
	ModelRefFlags

	Environment    string `` /* 179-byte string literal not displayed */
	DeploymentID   string `` /* 132-byte string literal not displayed */
	DeploymentName string `` /* 135-byte string literal not displayed */
	Regional       string `` /* 162-byte string literal not displayed */

	Data string `flag:"data" desc:"Inline JSON request body." oneof:"predict-input"`
	File string `flag:"file" desc:"Path to a JSON file containing the request body. Use '-' for stdin." oneof:"predict-input"`

	Websocket bool `` /* 173-byte string literal not displayed */
}

type ModelPushFlags struct {
	CommandFlags

	Dir string `flag:"dir" desc:"Model directory to push. Defaults to the current directory." default:"."`

	Team string `flag:"team" desc:"Team the model belongs to. Only valid for new models."`

	DryRun bool `flag:"dry-run" desc:"Validate the push and request upload credentials without uploading or creating anything."`

	Environment    string `flag:"environment" desc:"Stable environment to push to."`
	DeploymentName string `flag:"deployment-name" desc:"Human-readable name for the new deployment."`
	Region         string `flag:"region" desc:"Slug of the region to deploy the model in. Defaults to a region Baseten selects."`

	NoBuildCache bool   `flag:"no-build-cache" desc:"Force a full rebuild without using cached layers."`
	Labels       string `flag:"labels" desc:"User-provided labels for the deployment as a JSON object, e.g. '{\"team\":\"ml\",\"priority\":1}'."`

	Tail bool `` /* 178-byte string literal not displayed */
	Wait bool `flag:"wait" desc:"Block until the deployment is active. Exits non-zero on a terminal-failure status."`

	Develop bool `` /* 201-byte string literal not displayed */

	Watch            bool `` /* 132-byte string literal not displayed */
	WatchHotReload   bool `` /* 154-byte string literal not displayed */
	WatchNoKeepalive bool `` /* 153-byte string literal not displayed */

	DeployTimeout string `flag:"deploy-timeout" desc:"Deployment timeout as a duration (e.g. 30m, 1h); allowed range 10m to 24h."`

	OverrideName            string `` /* 129-byte string literal not displayed */
	OverrideEnvInstanceType bool   `` /* 173-byte string literal not displayed */

	DisableArchiveDownload bool `flag:"disable-archive-download" desc:"Disable archive download for the new model. Only valid for new models."`
}

type ModelPushResult struct {
	Model      managementapi.Model      `json:"model"`
	Deployment managementapi.Deployment `json:"deployment"`
	PredictURL string                   `json:"predict_url"`
	LogsURL    string                   `json:"logs_url"`
}

type ModelRefFlags struct {
	ModelID   string `flag:"model-id" desc:"ID of the model." oneof:"model-ref"`
	ModelName string `` /* 133-byte string literal not displayed */
	Team      string `flag:"team" desc:"Team name or ID. Only valid with --model-name."`
}

type ModelWatchFlags struct {
	CommandFlags

	Dir string `flag:"dir" desc:"Model directory to watch. Defaults to the current directory." default:"."`

	Team string `flag:"team" desc:"Team the model belongs to. Use to disambiguate when the same model_name exists in multiple teams."`

	HotReload   bool `` /* 134-byte string literal not displayed */
	NoKeepalive bool `` /* 133-byte string literal not displayed */
}

type OptionalFlag[T any] struct {
	// contains filtered or unexported fields
}

type OrgAPIKeyCreateFlags struct {
	CommandFlags

	Type     string   `` /* 157-byte string literal not displayed */
	Name     string   `flag:"name" desc:"Optional human-readable name for the key."`
	ModelIDs []string `` /* 146-byte string literal not displayed */
	Team     string   `flag:"team" desc:"Team name or ID to create the key in. Defaults to the organization's default team."`
}

type OrgAPIKeyDeleteFlags struct {
	CommandFlags

	Name   string `flag:"name" desc:"Human-readable name of the API key to delete." oneof:"identifier"`
	Prefix string `flag:"prefix" desc:"Prefix of the API key to delete (as shown in list)." oneof:"identifier"`
}

type OrgAPIKeyListFlags struct {
	CommandFlags
}

type OrgAuditLogsFlags struct {
	CommandFlags
	AuditLogFlags
}

type OrgBillingUsageFlags struct {
	CommandFlags

	Since time.Duration `` /* 160-byte string literal not displayed */
	Start time.Time     `` /* 216-byte string literal not displayed */
	End   time.Time     `` /* 170-byte string literal not displayed */
}

type OrgSecretDeleteFlags struct {
	CommandFlags

	Name string `flag:"name" desc:"Name of the secret to delete." required:"true"`
	Team string `flag:"team" desc:"Team name or ID the secret belongs to. Defaults to the organization's default team."`
}

type OrgSecretListFlags struct {
	CommandFlags

	Team string `flag:"team" desc:"Filter to a specific team by name or ID. Defaults to all teams the caller belongs to."`
}

type OrgSecretSetFlags struct {
	CommandFlags

	Name  string `flag:"name" desc:"Name of the secret." required:"true"`
	Value string `flag:"value" desc:"Secret value. Discouraged: leaks into shell history and process list. Prefer stdin or prompt."`
	Team  string `flag:"team" desc:"Team name or ID the secret belongs to. Defaults to the organization's default team."`
}

type OrgTeamDescribeFlags struct {
	CommandFlags

	TeamID   string `flag:"team-id" desc:"Team ID to describe." oneof:"team-ref"`
	TeamName string `flag:"team-name" desc:"Team name to describe." oneof:"team-ref"`
}

type OrgTeamListFlags struct {
	CommandFlags
}

type OrgUserDescribeFlags struct {
	CommandFlags

	UserID    string `flag:"user-id" desc:"User ID to describe. Pass 'me' for the authenticated user." oneof:"user-ref"`
	UserEmail string `flag:"user-email" desc:"Email of the user to describe." oneof:"user-ref"`
}

type OrgUserList struct {
	Items []managementapi.UserInfo `json:"items"`
}

type OrgUserListFlags struct {
	CommandFlags
}

type RequestBackpressureFlags struct {
	Policy OptionalFlag[string] `` /* 157-byte string literal not displayed */
}

type SSHProxyFlags struct {
	CommandFlags

	DefaultProfile string `` /* 136-byte string literal not displayed */
}

type SSHSetupFlags struct {
	CommandFlags
}

type SSHSetupResult struct {
	KeyPath   string `json:"key_path"`
	KeyReused bool   `json:"key_reused"`
	Profile   string `json:"profile,omitempty"`
}

type SSHSignFlags struct {
	CommandFlags

	DefaultProfile string `` /* 136-byte string literal not displayed */
}

type TrainCapacityDescribeFlags struct {
	CommandFlags
}

type TrainCapacityUpdateFlags struct {
	CommandFlags

	Team    string `flag:"team" desc:"Team name or ID whose limit changes." required:"true"`
	GPUType string `flag:"gpu-type" desc:"GPU type the limit applies to (for example H100)." required:"true"`
	MaxGPUs int    `flag:"max-gpus" desc:"Maximum concurrent GPUs of this type the team may use." required:"true"`
}

type TrainCheckpointDeployFlags struct {
	CommandFlags
	TrussAuthFlags

	JobID        string `flag:"job-id" desc:"Training job whose checkpoints are deployed. Required unless --config names them."`
	Config       string `flag:"config" desc:"Python file defining a DeployCheckpointsConfig: which checkpoints deploy and the model that serves them."`
	ConfigOutDir string `` /* 127-byte string literal not displayed */
	DryRun       bool   `flag:"dry-run" desc:"Write the generated model config without deploying anything."`
}

type TrainCheckpointFilesFlags struct {
	CommandFlags
	TrainJobRefFlags
}

type TrainCheckpointListFlags struct {
	CommandFlags
	TrainJobRefFlags

	Direction string `` /* 130-byte string literal not displayed */
}

type TrainInitFlags struct {
	CommandFlags
	// Scaffolding writes files and downloads public examples, so no credential
	// is forwarded and TrussFlags is embedded rather than TrussAuthFlags.
	TrussFlags

	Dir          string   `` /* 141-byte string literal not displayed */
	Example      []string `flag:"example" desc:"Name of a published training example to download instead of the empty template. Repeatable."`
	ListExamples bool     `flag:"list-examples" desc:"List the available example names and exit."`
}

type TrainJobDescribeFlags struct {
	CommandFlags
	TrainJobRefFlags
}

type TrainJobDownloadFlags struct {
	CommandFlags
	TrainJobRefFlags

	OutFile   string `flag:"out-file" desc:"Save the artifact as an uncompressed tar file at this path." oneof:"download-out"`
	OutDir    string `flag:"out-dir" desc:"Extract the artifact tar into this directory." oneof:"download-out"`
	Overwrite bool   `flag:"overwrite" desc:"Allow overwriting an existing file or non-empty directory."`
}

type TrainJobDownloadResult struct {
	JobID   string `json:"job_id"`
	OutFile string `json:"out_file,omitempty"`
	OutDir  string `json:"out_dir,omitempty"`
}

type TrainJobListFlags struct {
	CommandFlags

	Project   string   `flag:"project" desc:"Only list jobs in this training project, by name or ID."`
	Status    []string `` /* 156-byte string literal not displayed */
	Direction string   `` /* 130-byte string literal not displayed */
}

type TrainJobLogsFlags struct {
	CommandFlags
	TrainJobRefFlags
	TrainLogFlags
}

type TrainJobMetricsFlags struct {
	CommandFlags
	TrainJobRefFlags

	Start time.Time     `` /* 141-byte string literal not displayed */
	End   time.Time     `` /* 153-byte string literal not displayed */
	Since time.Duration `` /* 184-byte string literal not displayed */
}

type TrainJobRecreateFlags struct {
	CommandFlags
	TrainJobRefFlags
}

type TrainJobRefFlags struct {
	JobID string `flag:"job-id" desc:"ID of the training job." required:"true"`
}

type TrainJobSessionDescribeFlags struct {
	CommandFlags
	TrainJobRefFlags
}

type TrainJobStopFlags struct {
	CommandFlags
	TrainJobRefFlags

	Yes bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type TrainJobUpdateFlags struct {
	CommandFlags
	TrainJobRefFlags

	Priority int `flag:"priority" desc:"New queue priority. Higher values are dequeued first." required:"true"`
}

type TrainLogFlags struct {
	Tail bool `` /* 244-byte string literal not displayed */

	Start time.Time     `` /* 278-byte string literal not displayed */
	End   time.Time     `` /* 185-byte string literal not displayed */
	Since time.Duration `` /* 203-byte string literal not displayed */

	Limit int `` /* 196-byte string literal not displayed */

	// PageSize is the per-request fetch size while paging. Hidden; exists so
	// tests can force multiple pages without generating a full page of logs.
	PageSize int `flag:"page-size" hidden:"true" desc:"Log lines fetched per backend request while paging." default:"1000"`

	MinLevel string `flag:"min-level" desc:"Only return logs at or above this severity level." enum:"debug,info,warning,error"`
}

type TrainProjectCacheDescribeFlags struct {
	CommandFlags

	Project string `flag:"project" desc:"Training project to describe, by name or ID." required:"true"`
	Sort    string `flag:"sort" desc:"Sort the listing by 'size' (largest first) or 'path' (alphabetical)." enum:"size,path" default:"size"`
}

type TrainProjectListFlags struct {
	CommandFlags
}

type TrainPushFlags struct {
	CommandFlags
	TrussAuthFlags

	Config string `flag:"config" desc:"Python file defining the training project and its job." required:"true"`

	JobName string `flag:"job-name" desc:"Name for the training job. Defaults to a generated name."`
	Team    string `flag:"team" desc:"Team name or ID that owns the training project."`

	Accelerator string `flag:"accelerator" desc:"Accelerator type and count, for example 'H200:8'. Overrides the config."`
	NodeCount   int    `flag:"node-count" desc:"Number of compute nodes. Overrides the config."`
	Entrypoint  string `flag:"entrypoint" desc:"Command the job runs. Overrides the config."`
	Priority    int    `flag:"priority" desc:"Queue priority. Higher values are dequeued first."`
	Spot        bool   `flag:"spot" desc:"Run on interruptible spot capacity, overriding the config. Checkpointing your own progress is up to you."`

	Interactive        string        `flag:"interactive" desc:"Start an interactive session on this trigger." enum:"on-startup,on-failure,on-demand"`
	InteractiveTimeout time.Duration `` /* 161-byte string literal not displayed */
}

type TrainWorkstationCreateFlags struct {
	CommandFlags
	TrussAuthFlags

	Accelerator string `flag:"accelerator" desc:"GPU accelerator type." default:"H100"`
	GPUCount    int    `flag:"gpu-count" desc:"Number of GPUs on a single node, 1 to 8. Mutually exclusive with --node-count."`
	NodeCount   int    `flag:"node-count" desc:"Number of eight-GPU nodes, 1 to 16. Mutually exclusive with --gpu-count."`
	Image       string `flag:"image" desc:"Docker base image the workstation runs."`

	Project string `flag:"project" desc:"Training project that owns the workstation. Defaults to 'workstation-<accelerator>'."`
	Team    string `flag:"team" desc:"Team name or ID that owns the training project."`

	Orchestrator string `flag:"orchestrator" desc:"Multi-node orchestrator set up across the nodes." enum:"slurm" default:"slurm"`

	EnableCheckpointing  bool   `flag:"enable-checkpointing" desc:"Attach checkpoint storage to the workstation."`
	CheckpointPath       string `flag:"checkpoint-path" desc:"Path inside the container where checkpoints are written."`
	CheckpointVolumeSize int    `flag:"checkpoint-volume-size" desc:"Size of the checkpoint volume in GiB."`
	CheckpointFromJob    string `flag:"checkpoint-from-job" desc:"Training job whose latest checkpoint the workstation starts from."`
}

type TrussAuthFlags struct {
	TrussFlags

	TrussNoForwardAuth bool `` /* 243-byte string literal not displayed */
}

type TrussDelegatedResult struct{}

type TrussFlags struct {
	TrussVersion    string `` /* 227-byte string literal not displayed */
	TrussExecutable string `` /* 301-byte string literal not displayed */
}

type TrussPassthroughFlags struct {
	TrussAuthFlags
}

type VersionFlags struct {
	CommandFlags
}

type VersionResult struct {
	Version string `json:"version"`
}

type WhoamiFlags struct {
	CommandFlags
}