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 APISandboxFlags struct {
	APIFlags
	SandboxURL string `flag:"sandbox-url" desc:"Sandbox URL to target, as shown by 'baseten sandbox describe'" required:"true"`
}

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
	// CommandLines is Command split over multiple lines for an invocation too
	// long to render on one. Help joins them with a trailing backslash and
	// indents the continuations, so write each line without either. Break at
	// argument boundaries the reader would choose, never inside a quoted
	// argument.
	CommandLines []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.
	//
	// A command declaring JSONAlternatives says here when JSONT itself
	// applies, since only the alternatives carry that with them.
	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
	// JSONAlternatives declares the stdout shapes this command produces
	// besides JSONT, for a command whose answer changes shape with its input.
	// See [CommandOutputAlternative].
	JSONAlternatives []CommandOutputAlternative
}

type CommandOutputAlternative struct {
	// When completes "when the ...", naming the input that produces this
	// shape: "ref names a namespace", "ref carries a path".
	When string
	// Type is the Go type of the payload, as [CommandOutput.JSONOutputType] is
	// for the primary shape.
	Type reflect.Type
}

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
	// JSONAlternativeList returns the stdout shapes the command produces
	// besides JSONOutputType, each with the input that selects it. Empty for a
	// command with one fixed shape, which is nearly all of them.
	JSONAlternativeList() []CommandOutputAlternative
}

type ErrAuth struct{ CommandErrorMeta }

type ErrGeneric struct{ CommandErrorMeta }

type ErrInterrupted 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 HarnessFlags struct {
	Harness   []string `flag:"harness" desc:"Harness to apply to. May be repeated." enum:"claude-code,codex,opencode,pi"`
	ConfigDir string   `` /* 129-byte string literal not displayed */
}

type HarnessPlan struct {
	Harness          string   `json:"harness"`
	Config           string   `json:"config"`
	Settings         []string `json:"settings"`
	ReplacedSettings []string `json:"replaced_settings,omitempty"`
	Managed          bool     `json:"managed"`
	Changed          bool     `json:"changed"`
}

type HarnessPlanList struct {
	Items []HarnessPlan `json:"items"`
	// DeletedAPIKeys lists the prefixes of routes API keys teardown deleted.
	DeletedAPIKeys []string `json:"deleted_api_keys,omitempty"`
}

type HarnessRoute struct {
	Name        string `json:"name"`
	DisplayName string `json:"display_name"`
}

type HarnessSetupFlags struct {
	CommandFlags
	HarnessFlags
	Team            string `` /* 142-byte string literal not displayed */
	KeyName         string `` /* 167-byte string literal not displayed */
	Route           string `flag:"route" desc:"Default route. Defaults to the first route the harness lists."`
	BackgroundRoute string `` /* 142-byte string literal not displayed */
	SubagentRoute   string `flag:"subagent-route" desc:"Route for subagents (Claude Code and OpenCode). Defaults to the harness's own setting."`
	FallbackRoute   string `` /* 128-byte string literal not displayed */
	DryRun          bool   `flag:"dry-run" desc:"Preview the configuration without changing files or creating an API key."`
	Yes             bool   `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type HarnessStatus struct {
	Harness         string         `json:"harness"`
	Config          string         `json:"config"`
	Installed       bool           `json:"installed"`
	Version         string         `json:"version"`
	State           string         `json:"state"`
	DefaultRoute    string         `json:"default_route,omitempty"`
	BackgroundRoute string         `json:"background_route,omitempty"`
	Routes          []HarnessRoute `json:"routes,omitempty"`
	ManagedSettings []string       `json:"managed_settings,omitempty"`
}

type HarnessStatusFlags struct {
	CommandFlags
	HarnessFlags
}

type HarnessStatusList struct {
	Items []HarnessStatus `json:"items"`
}

type HarnessTeardownFlags struct {
	CommandFlags
	HarnessFlags
	DryRun bool `flag:"dry-run" desc:"Preview removal without changing files."`
	Yes    bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type HarnessUsageFlags struct {
	CommandFlags

	Start time.Time     `` /* 170-byte string literal not displayed */
	End   time.Time     `` /* 144-byte string literal not displayed */
	Since time.Duration `flag:"since" desc:"Window from a relative time ago until now (e.g. '7d'). Mutually exclusive with --start and --end."`

	UserIDs []string `` /* 134-byte string literal not displayed */

	RouteUsageQueryFlags
}

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 JSONError struct {
	// Message is the human-readable failure message, identical to the line
	// written to stderr. For a failure originating in an API response this
	// includes the raw response body.
	Message string `json:"message"`
	// Type is the name of the typed error, matching the names listed in the
	// exit-code tables of `baseten --help-output` and each command's
	// --help-output.
	Type string `json:"type"`
	// ExitCode is the process exit code, the same value the shell sees.
	ExitCode ExitCode `json:"exit_code"`
	// APIStatusCode is the HTTP status of the failed API response. Absent when
	// the failure did not come from an API call.
	APIStatusCode int `json:"api_status_code,omitempty"`
	// APIErrorCode is the machine-readable error code from the API response
	// body. Absent when the failure did not come from an API call, or when the
	// response carried no code (some failures are reported by an upstream
	// proxy whose body has only a message).
	APIErrorCode string `json:"api_error_code,omitempty"`
	// APIDetails is the structured, per-failure-kind payload from the API
	// response body. Absent when the failure did not come from an API call, or
	// when the response carried no details.
	APIDetails map[string]any `json:"api_details,omitempty"`
}

type JSONErrorEnvelope struct {
	Error JSONError `json:"error"`
}

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 `` /* 186-byte string literal not displayed */
	Config       string   `` /* 280-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 `` /* 127-byte string literal not displayed */
}

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 LoopsExecFlags struct {
	CommandFlags
	TrussAuthFlags

	Accelerator string `` /* 158-byte string literal not displayed */
	GPUCount    int    `flag:"gpu-count" desc:"Number of GPUs, from 1 to 8. Requires --accelerator."`
	CPUCount    int    `flag:"cpu-count" desc:"Number of CPUs to request." default:"16"`
	Memory      string `flag:"memory" desc:"Memory to request, for example 8Gi." default:"64Gi"`

	ProjectName string   `flag:"project-name" desc:"Training project name. Defaults to the selected directory's name."`
	Image       string   `flag:"image" desc:"Custom Docker base image."`
	Dir         string   `` /* 130-byte string literal not displayed */
	ExcludeDir  []string `flag:"exclude-dir" desc:"Top-level directory of the workspace root to leave out of the upload. Repeatable."`
	ExternalDir []string `` /* 148-byte string literal not displayed */

	Env      []string `flag:"env" desc:"Environment variable for the job as KEY=VALUE. Repeatable."`
	Secret   []string `flag:"secret" desc:"Environment variable sourced from a Baseten workspace secret as KEY=SECRET_NAME. Repeatable."`
	NoAPIKey bool     `flag:"no-api-key" desc:"Do not provision BASETEN_API_KEY in the job from a per-team workspace secret."`
	NoSSH    bool     `` /* 126-byte string literal not displayed */
	WithUV   bool     `flag:"with-uv" desc:"Make uv available in the job image. The start command must invoke uv itself."`
	Team     string   `flag:"team" desc:"Team name that owns the training project."`
	Tail     bool     `` /* 134-byte string literal not displayed */
}

type LoopsExecResult struct {
	JobID   string `json:"job_id"`
	Project struct {
		ID   *string `json:"id"`
		Name *string `json:"name"`
	} `json:"project"`
	SSHHostname          *string  `json:"ssh_hostname"`
	StartCommand         string   `json:"start_command"`
	EnvironmentVariables []string `json:"environment_variables"`
	Compute              struct {
		CPUCount    int     `json:"cpu_count"`
		Memory      string  `json:"memory"`
		Accelerator *string `json:"accelerator"`
		GPUCount    *int    `json:"gpu_count"`
	} `json:"compute"`
	Job map[string]any `json:"job"`
}

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 `` /* 196-byte string literal not displayed */
	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 `` /* 157-byte string literal not displayed */
}

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 `` /* 169-byte string literal not displayed */
	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"`
	PreserveEnvInstanceType bool   `` /* 139-byte string literal not displayed */

	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 `` /* 152-byte string literal not displayed */
}

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 `` /* 247-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 `` /* 176-byte string literal not displayed */
	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. Run 'baseten model environment list' to see a model's environments."`
	CreateEnvironmentIfMissing bool   `` /* 274-byte string literal not displayed */
	DeploymentName             string `flag:"deployment-name" desc:"Human-readable name for the new deployment."`
	Region                     string `` /* 155-byte string literal not displayed */

	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 */
	PreserveEnvInstanceType bool   `` /* 187-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 ModelRenameFlags struct {
	CommandFlags
	ModelRefFlags

	NewName string `flag:"new-name" desc:"New name for the model, unique within its team." required:"true"`
}

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   `` /* 143-byte string literal not displayed */
}

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 string `` /* 163-byte string literal not displayed */
}

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 OrgDescribeFlags struct {
	CommandFlags
}

type OrgRegionsFlags struct {
	CommandFlags

	Team string `` /* 146-byte string literal not displayed */
}

type OrgSecretDeleteFlags struct {
	CommandFlags

	Name string `flag:"name" desc:"Name of the secret to delete." required:"true"`
	Team string `` /* 144-byte string literal not displayed */
}

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 `` /* 144-byte string literal not displayed */
}

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 RouteAPIKeyDeleteFlags struct {
	CommandFlags
	Prefix string `flag:"prefix" desc:"Prefix of the key to delete, as shown by route api-key list." oneof:"key"`
	All    bool   `flag:"all" desc:"Delete all your routes API keys, on every machine and team." oneof:"key"`
	Yes    bool   `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type RouteAPIKeyDeleteResult struct {
	Deleted []string `json:"deleted"`
	Failed  []string `json:"failed"`
}

type RouteAPIKeyListFlags struct{ CommandFlags }

type RouteCreateFlags struct {
	CommandFlags
	RouteTargetFlags
	Team        string               `` /* 143-byte string literal not displayed */
	DisplayName OptionalFlag[string] `flag:"display-name" desc:"Display name (1 to 255 characters). Defaults to the route name."`
	Description OptionalFlag[string] `flag:"description" desc:"Optional route description (up to 1000 characters)."`
}

type RouteDeleteFlags struct {
	CommandFlags
	RouteRefFlags
	Yes bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type RouteDescribeFlags struct {
	CommandFlags
	RouteRefFlags
}

type RouteList struct {
	Items []managementapi.Route `json:"items"`
}

type RouteListFlags struct {
	CommandFlags
	Team string `flag:"team" desc:"Team name or ID to scope the listing to. Defaults to all routes you can invoke."`
}

type RouteRefFlags struct {
	ID   string `flag:"id" desc:"Stable route ID." oneof:"route-ref"`
	Name string `flag:"name" desc:"Exact route name, including its organization-owned prefix." oneof:"route-ref"`
}

type RouteTargetFlags struct {
	TargetType   string `flag:"target-type" desc:"Target type." enum:"baseten-model-api,anthropic,openai,xai" required:"true"`
	TargetModel  string `flag:"target-model" desc:"Model API name or external provider model name." required:"true"`
	TargetSecret string `flag:"target-secret" desc:"Name of an existing secret in the route's team. Required for external targets."`
}

type RouteUpdateFlags struct {
	CommandFlags
	RouteRefFlags
	DisplayName OptionalFlag[string] `flag:"display-name" desc:"New display name (1 to 255 characters). Omit to keep it unchanged."`
	Description OptionalFlag[string] `flag:"description" desc:"New description (up to 1000 characters). Pass an empty string to clear it."`
}

type RouteUsageFlags struct {
	CommandFlags

	Start time.Time     `` /* 156-byte string literal not displayed */
	End   time.Time     `` /* 144-byte string literal not displayed */
	Since time.Duration `flag:"since" desc:"Window from a relative time ago until now (e.g. '7d'). Mutually exclusive with --start and --end."`

	UserIDs []string `` /* 137-byte string literal not displayed */

	RouteUsageQueryFlags
}

type RouteUsageQueryFlags struct {
	GroupBy []string `flag:"group-by" desc:"Dimension to break usage down by. May be repeated." enum:"user,model,provider" default:"model"`

	Models    []string `flag:"model" desc:"Only return usage for these models. May be repeated."`
	Providers []string `` /* 149-byte string literal not displayed */

	Limit int `flag:"limit" desc:"Maximum number of daily buckets, paging as needed. 0 for no limit."`

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

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 SandboxConnectFlags struct {
	CommandFlags
	SandboxRefFlags
}

type SandboxCreateFlags struct {
	CommandFlags
	SandboxEnvFlags

	Name        string   `flag:"name" desc:"Unique name of the sandbox. The server generates one when omitted."`
	Team        string   `` /* 171-byte string literal not displayed */
	IfNotExists bool     `` /* 152-byte string literal not displayed */
	Region      string   `flag:"region" desc:"Region to create the sandbox in."`
	Memory      int      `flag:"memory" desc:"Memory in MB, which also sets the CPU allocation."`
	Image       string   `flag:"image" desc:"Image to create the sandbox from, including its tag, such as my-image:latest."`
	Label       []string `flag:"label" desc:"Label as KEY=VALUE. Repeatable."`
}

type SandboxDeleteFlags struct {
	CommandFlags
	SandboxRefFlags

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

type SandboxDescribeFlags struct {
	CommandFlags
	SandboxRefFlags

	ShowSecrets bool `` /* 151-byte string literal not displayed */
}

type SandboxEnvFlags struct {
	Env      []string `flag:"env" desc:"Secret environment variable as KEY=VALUE, masked when read back. Repeatable."`
	PlainEnv []string `flag:"plain-env" desc:"Environment variable that is not secret, as KEY=VALUE, readable when read back. Repeatable."`
}

type SandboxExecFlags struct {
	CommandFlags
	SandboxRefFlags
	SandboxProcessOptionFlags

	Stdin bool `` /* 200-byte string literal not displayed */
}

type SandboxImageDeleteFlags struct {
	CommandFlags
	SandboxImageRefFlags

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

type SandboxImageLibraryList struct {
	Items []managementapi.SandboxLibraryImage `json:"items"`
}

type SandboxImageList struct {
	Items []managementapi.SandboxImageSummary `json:"items"`
}

type SandboxImageListFlags struct {
	CommandFlags

	Team  string `` /* 164-byte string literal not displayed */
	Limit int    `flag:"limit" desc:"Most images to list. 0 lists all." default:"1000"`
}

type SandboxImageListLibraryFlags struct {
	CommandFlags
}

type SandboxImageLogsFlags struct {
	CommandFlags
	SandboxImageRefFlags

	Start time.Time `flag:"start" desc:"Earliest log time to print, such as 2026-10-05T15:04:05Z."`
	End   time.Time `flag:"end" desc:"Latest log time to print, such as 2026-10-05T15:04:05Z."`
}

type SandboxImagePushFlags struct {
	CommandFlags
	SandboxImageRefFlags

	Dir           string `flag:"dir" desc:"Directory to build the image from, with a Dockerfile at its root." oneof:"image-source"`
	RegistryImage string `flag:"registry-image" desc:"Registry image reference to import, such as registry.example.com/app:v1." oneof:"image-source"`
	DockerConfig  string `` /* 127-byte string literal not displayed */
	Wait          bool   `flag:"wait" desc:"Wait until the image is built. Exits non-zero if the build fails. Does not stop the build if interrupted."`
}

type SandboxImageRefCommandFlags struct {
	CommandFlags
	SandboxImageRefFlags
}

type SandboxImageRefFlags struct {
	Name string `flag:"name" desc:"Name of the image." required:"true"`
	Team string `` /* 167-byte string literal not displayed */
}

type SandboxImageTagList struct {
	Items []managementapi.SandboxImageTag `json:"items"`
}

type SandboxList struct {
	Items []managementapi.Sandbox `json:"items"`
}

type SandboxListFlags struct {
	CommandFlags

	Team   string   `` /* 167-byte string literal not displayed */
	Query  string   `flag:"query" desc:"Free-text search over sandbox names and labels."`
	Status []string `` /* 203-byte string literal not displayed */
	Limit  int      `flag:"limit" desc:"Most sandboxes to list. 0 lists all." default:"1000"`
}

type SandboxProcessDescribeFlags struct {
	CommandFlags
	SandboxRefFlags
	SandboxProcessRefFlags
}

type SandboxProcessList struct {
	Items []sandboxapi.ProcessResponse `json:"items"`
}

type SandboxProcessListFlags struct {
	CommandFlags
	SandboxRefFlags
}

type SandboxProcessLogLine struct {
	Stream string `json:"stream,omitempty"`
	Text   string `json:"text"`
}

type SandboxProcessLogsFlags struct {
	CommandFlags
	SandboxRefFlags
	SandboxProcessRefFlags

	Tail bool `` /* 204-byte string literal not displayed */
}

type SandboxProcessOptionFlags struct {
	Env         []string      `flag:"env" desc:"Environment variable for the command as KEY=VALUE, on top of the sandbox's own. Repeatable."`
	WorkingDir  string        `flag:"working-dir" desc:"Directory to run the command in."`
	Timeout     time.Duration `flag:"timeout" desc:"Stop the command if it runs longer than this, such as 10m."`
	ProcessName string        `flag:"process-name" desc:"Name for the process, to find it later with --process-name."`
}

type SandboxProcessRefCommandFlags struct {
	CommandFlags
	SandboxRefFlags
	SandboxProcessRefFlags
}

type SandboxProcessRefFlags struct {
	PID         string `flag:"pid" desc:"PID of the process." oneof:"process-ref"`
	ProcessName string `flag:"process-name" desc:"Name the process was started with." oneof:"process-ref"`
}

type SandboxProcessStartFlags struct {
	CommandFlags
	SandboxRefFlags
	SandboxProcessOptionFlags

	KeepAlive        bool     `flag:"keep-alive" desc:"Keep the sandbox from going to sleep while the process runs."`
	RestartOnFailure bool     `flag:"restart-on-failure" desc:"Restart the process when it exits with a failure."`
	MaxRestarts      int      `flag:"max-restarts" desc:"Most restarts with --restart-on-failure."`
	WaitForPort      []string `flag:"wait-for-port" desc:"Return only once the process listens on this port. Repeatable."`
}

type SandboxProcessWaitFlags struct {
	CommandFlags
	SandboxRefFlags
	SandboxProcessRefFlags

	Timeout time.Duration `flag:"timeout" desc:"Fail if the process has not exited after this long, such as 30m. The process keeps running."`
}

type SandboxRefFlags struct {
	Name string `flag:"name" desc:"Name of the sandbox." required:"true"`
	Team string `` /* 169-byte string literal not displayed */
}

type SandboxUpdateFlags struct {
	CommandFlags
	SandboxRefFlags
	SandboxEnvFlags

	Label []string `flag:"label" desc:"Label as KEY=VALUE, replacing all of the sandbox's labels. Repeatable."`
}

type TrainCapacityDescribeFlags struct {
	CommandFlags
}

type TrainCapacityUpdateFlags struct {
	CommandFlags

	Team    string `flag:"team" desc:"Team name or ID whose limit changes. Run 'baseten org team list' to see teams." 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 `` /* 223-byte string literal not displayed */
	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."`
	AvailabilityModel string `` /* 251-byte string literal not displayed */
}

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. Run 'baseten org team list' to see teams."`

	Accelerator string `` /* 183-byte string literal not displayed */
	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."`
	Spot        bool   `flag:"spot" desc:"Run on interruptible spot capacity."`

	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. Run 'baseten org team list' to see teams."`

	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 VolumeCatFlags struct {
	CommandFlags
}

type VolumeEntry struct {
	// Path is the entry's path within the version, slash-prefixed and with no
	// trailing slash, which is the form a ref takes: appending it to the ref
	// this listing was asked for names the entry.
	Path string `json:"path"`
	// Kind is "file", "directory", or "symlink".
	Kind string `json:"kind"`
	// SizeBytes is a file's length, and zero for the other kinds.
	SizeBytes int64 `json:"size_bytes"`
	// Mode is the recorded permission bits in octal, as "0644". Absent for a
	// directory the version describes only by the paths beneath it.
	Mode string `json:"mode,omitempty"`
	// Modified is when the entry was last modified, as recorded at push.
	// Absent when the version records none.
	Modified *time.Time `json:"modified,omitempty"`
	// LinkTarget is a symlink's target exactly as recorded, and absent for
	// every other kind. It may point outside the volume.
	LinkTarget string `json:"link_target,omitempty"`
}

type VolumeEntryDetail struct {
	// VersionRef is the version the entry was read from, pinned to its digest.
	VersionRef string `json:"version_ref"`
	VolumeEntry
}

type VolumeEntryList struct {
	// VersionRef is the version the entries were read from, pinned to its
	// digest, so the same listing can be read again.
	VersionRef string        `json:"version_ref"`
	Items      []VolumeEntry `json:"items"`
}

type VolumeList struct {
	Items []managementapi.Volume `json:"items"`
}

type VolumeLsFlags struct {
	CommandFlags

	Recursive bool `flag:"recursive" short:"R" desc:"List every entry beneath the ref instead of only its immediate children."`
}

type VolumeNamespaceList struct {
	Items []string `json:"items"`
}

type VolumePullFlags struct {
	CommandFlags
	VolumeRefFlags
	VolumeTransferFlags

	Overwrite bool `` /* 136-byte string literal not displayed */

	Include []string `` /* 287-byte string literal not displayed */
	Restart bool     `flag:"restart" desc:"Discard a partly downloaded tree from an earlier attempt instead of continuing it."`

	StripPrefix bool `` /* 167-byte string literal not displayed */
}

type VolumePullResult struct {
	// DurationSeconds is total elapsed SDK transfer time, including setup and finalization.
	DurationSeconds float64 `json:"duration_seconds"`
	// VersionRef is the version that was downloaded, pinned to its digest.
	VersionRef string `json:"version_ref"`
	DestDir    string `json:"dest_dir"`
	Files      int64  `json:"files"`
	Bytes      int64  `json:"bytes"`
	// SelectedFiles and TotalFiles report what a path or --include narrowed
	// to, and are equal when the whole version was downloaded.
	SelectedFiles int64 `json:"selected_files"`
	TotalFiles    int64 `json:"total_files"`
	// ChunksFetched counts the chunks this download transferred; ChunksReused
	// counts those already on disk from an earlier attempt.
	ChunksFetched int64 `json:"chunks_fetched"`
	ChunksReused  int64 `json:"chunks_reused"`
	// Warnings are containment findings that did not stop the download, which
	// only volumes published before the containment rule carry.
	Warnings []string `json:"warnings"`
}

type VolumePushFlags struct {
	CommandFlags
	VolumeRefFlags
	VolumeTransferFlags

	Tags []string `` /* 177-byte string literal not displayed */

	SourceURI string `` /* 228-byte string literal not displayed */

	FileJobs int `flag:"file-jobs" desc:"Number of files processed concurrently. Defaults to 16." group:"transfer"`
}

type VolumePushResult struct {
	// DurationSeconds is total elapsed SDK transfer time, including setup and finalization.
	DurationSeconds float64 `json:"duration_seconds"`
	// VersionRef is the published version, pinned to its digest, which is what
	// config.yaml takes to mount exactly this tree.
	VersionRef string `json:"version_ref"`
	Sequence   int64  `json:"sequence"`
	// HeadUpdated is false when head already pointed at this exact version,
	// which is what re-pushing an unchanged tree does. HeadMoveDenied means
	// the version was published but head was left where it was, because the
	// credential's grants did not cover moving it.
	HeadUpdated    bool     `json:"head_updated"`
	HeadMoveDenied bool     `json:"head_move_denied"`
	TagsApplied    []string `json:"tags_applied"`
	Files          int64    `json:"files"`
	Bytes          int64    `json:"bytes"`
	// Chunks counts every chunk the push accounted for, and the three below
	// partition it: ChunksUnique were uploaded, ChunksReused never reached the
	// network because a previous version or an earlier file in this push
	// already had those bytes, and ChunksExisting were offered and the volume
	// already had them.
	Chunks         int64 `json:"chunks"`
	ChunksUnique   int64 `json:"chunks_unique"`
	ChunksReused   int64 `json:"chunks_reused"`
	ChunksExisting int64 `json:"chunks_existing"`
}

type VolumeRefFlags struct {
	FullRef bool `flag:"full-ref" desc:"Write refs with the whole digest instead of shortening it to 12 hex characters."`
}

type VolumeRestoreFlags struct {
	CommandFlags
	VolumeRefFlags
}

type VolumeRmFlags struct {
	CommandFlags
	VolumeRefFlags

	Recursive bool `` /* 138-byte string literal not displayed */
	Yes       bool `flag:"yes" desc:"Skip the interactive confirmation prompt. Required when stdin is not a terminal."`
}

type VolumeStatFlags struct {
	CommandFlags
	VolumeRefFlags
}

type VolumeSyncIDFlags struct {
	CommandFlags

	SyncID string `flag:"sync-id" desc:"ID of the volume sync operation." required:"true"`
}

type VolumeSyncList struct {
	Items []managementapi.VolumeSync `json:"items"`
}

type VolumeSyncListFlags struct {
	CommandFlags

	Destination string `flag:"dest" desc:"Only return syncs whose destination exactly matches this ref."`
}

type VolumeSyncStartFlags struct {
	CommandFlags

	Source      string   `` /* 130-byte string literal not displayed */
	Destination string   `flag:"dest" desc:"Destination ref as bdn:<namespace>/<volume>, with an optional :<tag>." required:"true"`
	Include     []string `flag:"include" desc:"Glob selecting source-relative files to include. May be repeated."`
	Exclude     []string `flag:"exclude" desc:"Glob selecting source-relative files to exclude. May be repeated."`

	AuthSecretName                      string `` /* 169-byte string literal not displayed */
	AuthAWSAssumeRoleARN                string `` /* 139-byte string literal not displayed */
	AuthAWSAssumeRoleRegion             string `` /* 140-byte string literal not displayed */
	AuthAWSOIDCRoleARN                  string `` /* 153-byte string literal not displayed */
	AuthAWSOIDCRegion                   string `` /* 130-byte string literal not displayed */
	AuthGCPOIDCServiceAccount           string `` /* 187-byte string literal not displayed */
	AuthGCPOIDCWorkloadIdentityProvider string `` /* 170-byte string literal not displayed */

	Wait bool `` /* 126-byte string literal not displayed */
}

type VolumeTransferFlags struct {
	ChunkOperations int `` /* 188-byte string literal not displayed */
	MaxInFlightMiB  int `flag:"max-in-flight-mib" desc:"Cap on the chunk data held in memory, in MiB. Defaults to 2048." group:"transfer"`
}

type VolumeVersionsFlags struct {
	CommandFlags

	IncludeTombstoned bool `flag:"include-tombstoned" desc:"Include deleted versions that are still inside their recovery window."`
}

type WhoamiFlags struct {
	CommandFlags
}