ai

type AiModel struct {
	// Name is the model name, e.g. "gpt-4o".
	Name string
	// Format is the model format, e.g. "OpenAI".
	Format string
	// LifecycleStatus is the model lifecycle status, e.g. "preview", "stable".
	LifecycleStatus string
	// Capabilities lists the model's capabilities, e.g. ["chat", "embeddings"].
	Capabilities []string
	// Versions lists the available versions of this model.
	Versions []AiModelVersion
	// Locations lists the Azure locations where this model is available.
	Locations []string
}

type AiModelDeployment struct {
	// ModelName is the model name, e.g. "gpt-4o".
	ModelName string
	// Format is the model format, e.g. "OpenAI".
	Format string
	// Version is the model version, e.g. "2024-05-13".
	Version string
	// Location is the Azure location for this deployment.
	Location string
	// Sku is the selected SKU for this deployment.
	Sku AiModelSku
	// Capacity is the resolved deployment capacity in units.
	// Resolved from: DeploymentOptions.Capacity → Sku.DefaultCapacity → 0 (caller must handle).
	Capacity int32
	// RemainingQuota is the subscription quota remaining at this location for this SKU.
	// Only populated when a quota check is performed. nil means no quota check was done.
	RemainingQuota *float64
}

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

type AiModelSku struct {
	// Name is the SKU name, e.g. "GlobalStandard", "Standard".
	Name string
	// UsageName is the quota usage name used to join with usage/quota data,
	// e.g. "OpenAI.Standard.gpt-4o".
	UsageName string
	// DefaultCapacity is the suggested deployment capacity (0 if unavailable).
	DefaultCapacity int32
	// MinCapacity is the minimum allowed deployment capacity.
	MinCapacity int32
	// MaxCapacity is the maximum allowed deployment capacity.
	MaxCapacity int32
	// CapacityStep is the capacity increment granularity.
	CapacityStep int32
}

type AiModelUsage struct {
	// Name is the quota usage name, e.g. "OpenAI.Standard.gpt-4o".
	Name string
	// CurrentValue is the amount of quota currently consumed.
	CurrentValue float64
	// Limit is the total quota limit for this usage name.
	Limit float64
}

type AiModelVersion struct {
	// Version is the version string, e.g. "2024-05-13".
	Version string
	// IsDefault indicates whether this is the default version.
	IsDefault bool
	// Skus lists the available SKUs for this version.
	Skus []AiModelSku
}

type ComponentConfig struct {
	Name      osutil.ExpandableString            `yaml:"name,omitempty"`
	Path      string                             `yaml:"path,omitempty"`
	Overrides map[string]osutil.ExpandableString `yaml:"overrides,omitempty"`
}

type DeploymentConfig struct {
	ComponentConfig `yaml:",inline"`
	// A map of environment variables to set for the deployment
	Environment osutil.ExpandableMap `yaml:"environment,omitempty"`
}

type DeploymentOptions struct {
	// Locations lists preferred locations. If empty, location is left unset on results.
	Locations []string
	// Versions lists preferred versions. If empty, all versions are included.
	Versions []string
	// Skus lists preferred SKU names, e.g. ["GlobalStandard", "Standard"]. If empty, all SKUs are included.
	Skus []string
	// Capacity is the preferred deployment capacity. If set and valid
	// (within min/max, aligned to step), used directly. If nil, uses SKU default.
	Capacity *int32
	// IncludeFinetuneSkus controls whether fine-tune SKUs (usage names ending with
	// "-finetune") are included. Defaults to false (excluded).
	IncludeFinetuneSkus bool
}

type EndpointDeploymentConfig struct {
	Workspace   osutil.ExpandableString `yaml:"workspace,omitempty"`
	Environment *ComponentConfig        `yaml:"environment,omitempty"`
	Model       *ComponentConfig        `yaml:"model,omitempty"`
	Flow        *ComponentConfig        `yaml:"flow,omitempty"`
	Deployment  *DeploymentConfig       `yaml:"deployment,omitempty"`
}

type FilterOptions struct {
	// Locations filters to models available at these locations.
	Locations []string
	// Capabilities filters by model capabilities, e.g. ["chat", "embeddings"].
	Capabilities []string
	// Formats filters by model format, e.g. ["OpenAI"].
	Formats []string
	// Statuses filters by lifecycle status, e.g. ["preview", "stable"].
	Statuses []string
	// ExcludeModelNames excludes models by name (for multi-model selection flows).
	ExcludeModelNames []string
}

type Flow struct {
	Name        string            `json:"name"`
	Description string            `json:"description"`
	Type        string            `json:"type"`
	Path        string            `json:"path"`
	DisplayName string            `json:"display_name"`
	Tags        map[string]string `json:"tags"`
}

type ModelLocationQuota struct {
	// Location is the Azure location name.
	Location string
	// MaxRemainingQuota is the maximum remaining quota across model SKUs with usage entries.
	MaxRemainingQuota float64
}

type PythonBridge interface {
	Initialize(ctx context.Context) error
	RequiredExternalTools(ctx context.Context) []tools.ExternalTool
	Run(ctx context.Context, scriptName ScriptPath, args ...string) (*exec.RunResult, error)
}

type QuotaCheckOptions struct {
	// MinRemainingCapacity is the minimum remaining quota required per SKU.
	// Models/deployments where no SKU meets this threshold are excluded.
	// 0 means "any remaining > 0" (i.e. not fully exhausted).
	MinRemainingCapacity float64
}

type QuotaRequirement struct {
	// UsageName is the quota usage name to check, e.g. "OpenAI.Standard.gpt-4o".
	UsageName string
	// MinCapacity is the minimum remaining capacity needed. If 0, defaults to 1.
	MinCapacity float64
}

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

type ScriptPath string