conf

type API struct {
	Port                          int
	MultiAddress                  string
	Domain                        string
	NodeName                      string
	Pricing                       Pricing  `toml:"pricing"`
	AutoDeleteImage               bool     `toml:"AutoDeleteImage"`
	ClearLogDuration              int      `toml:"ClearLogDuration"`
	PortRange                     []string `toml:"PortRange"`
	GpuUtilizationRejectThreshold float64  `toml:"GpuUtilizationRejectThreshold"`
}

type ComputeNode struct {
	API       API
	RPC       RPC       `toml:"RPC,omitempty"`
	Inference Inference `toml:"Inference,omitempty"`
}

type Inference struct {
	Enable       bool     `toml:"Enable"`
	ServiceURL   string   `toml:"ServiceURL"`   // HTTP API URL (e.g., http://localhost:8080)
	WebSocketURL string   `toml:"WebSocketURL"` // WebSocket URL (e.g., wss://inference-ws.swanchain.io)
	ApiKey       string   `toml:"ApiKey"`       // Provider API key for authentication (sk-prov-*)
	Models       []string `toml:"Models"`       // Models this provider serves
}

type ModelConfig struct {
	Container  string `json:"container,omitempty"`
	Endpoint   string `json:"endpoint"`
	GPUMemory  int    `json:"gpu_memory"`
	Category   string `json:"category"`
	LocalModel string `json:"local_model,omitempty"`
}

type Pricing bool

type RPC struct {
	SwanChainRpc string `toml:"SWAN_CHAIN_RPC"`
}