runtime

type ApprovedArgs struct {
	ContextSize int
	BatchSize   int
	GPULayers   string
	Device      string
	Parallel    int
	Host        string
	Port        int
}

type ChatMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type CompletionRequest struct {
	Model       string        `json:"model"`
	Messages    []ChatMessage `json:"messages"`
	Temperature float64       `json:"temperature"`
	MaxTokens   int           `json:"max_tokens"`
	Stream      bool          `json:"stream"`
}

type CompletionResponse struct {
	Choices []struct {
		Message      ChatMessage `json:"message"`
		FinishReason string      `json:"finish_reason"`
	} `json:"choices"`
	Usage CompletionUsage `json:"usage"`
}

type CompletionUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type InstalledRuntime struct {
	Engine         string `json:"engine"`
	Version        string `json:"version"`
	BuildID        string `json:"build_id"`
	PlatformKey    string `json:"platform_key"`
	Path           string `json:"path"`
	ExecutablePath string `json:"executable_path"`
	SHA256         string `json:"sha256"`
	InstalledAt    string `json:"installed_at"`
}

type LaunchConfig struct {
	ExecutablePath  string
	ModelPath       string
	WorkDir         string
	LogDir          string
	Args            ApprovedArgs
	HealthInterval  time.Duration
	StartupTimeout  time.Duration
	ShutdownTimeout time.Duration
	MaxLogBytes     int64
	Redactions      []string
	Env             []string
}

type Manager struct {
	Root        string
	PublicKey   ed25519.PublicKey
	PlatformKey string
	// GPUVendor picks a vendor-specific runtime build when the manifest has
	// one. Empty or "unknown" falls back to the bare platform artifact.
	GPUVendor  string
	HTTPClient *http.Client
}

type ManifestSignature struct {
	KeyID string `json:"key_id"`
	Value string `json:"value"`
}

type ReleaseManifest struct {
	SchemaVersion int                        `json:"schema_version"`
	Engine        string                     `json:"engine"`
	Version       string                     `json:"version"`
	BuildID       string                     `json:"build_id"`
	Artifacts     map[string]RuntimeArtifact `json:"artifacts"`
	Signature     ManifestSignature          `json:"signature"`
}

type RunningRuntime struct {
	BaseURL string
	Port    int
	// contains filtered or unexported fields
}

type RuntimeArtifact struct {
	URL              string            `json:"url"`
	SHA256           string            `json:"sha256"`
	SizeBytes        int64             `json:"size_bytes"`
	ArchiveType      string            `json:"archive_type"`
	ExecutablePath   string            `json:"executable_path"`
	SupportArtifacts []SupportArtifact `json:"support_artifacts,omitempty"`
}

type SupportArtifact struct {
	Name        string `json:"name"`
	URL         string `json:"url"`
	SHA256      string `json:"sha256"`
	SizeBytes   int64  `json:"size_bytes"`
	ArchiveType string `json:"archive_type"`
}