conf

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

type CONTRACT struct {
	UpgradeName       string
	SwanToken         string `toml:"SWAN_CONTRACT"`
	CpAccountRegister string `toml:"REGISTER_CP_CONTRACT"`

	JobCollateral     string `toml:"SWAN_COLLATERAL_CONTRACT"`
	JobManager        string `toml:"SWAN_JOB_CONTRACT"`
	JobManagerCreated uint64 `toml:"JobManagerCreated"`

	TaskRegister           string `toml:"REGISTER_TASK_CONTRACT"`
	ZkCollateral           string `toml:"ZK_COLLATERAL_CONTRACT"`
	Sequencer              string `toml:"SEQUENCER_CONTRACT"`
	EdgeTaskPayment        string `toml:"EDGE_TASK_PAYMENT"`
	EdgeTaskPaymentCreated uint64 `toml:"EdgeTaskPaymentCreated"`

	JobCollateralUbiZero string `toml:"SWAN_COLLATERAL_UBI_ZERO_CONTRACT"`
	ZkCollateralUbiZero  string `toml:"ZK_COLLATERAL_UBI_ZERO_CONTRACT"`
}

type ComputeNode struct {
	API       API
	UBI       UBI
	LOG       LOG
	Registry  Registry
	RPC       RPC
	CONTRACT  CONTRACT  `toml:"CONTRACT,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
	ChainRPC           string   `toml:"ChainRPC"`           // Base Sepolia RPC for Swan Inference
	CollateralContract string   `toml:"CollateralContract"` // Collateral contract address
	TaskContract       string   `toml:"TaskContract"`       // Task contract address
}

type LOG struct {
	CrtFile string
	KeyFile string
}

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"` // Actual model name to use in requests (e.g., Ollama model name)
}

type Pricing bool

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

type Registry struct {
	ServerAddress string
	UserName      string
	Password      string
}

type UBI struct {
	UbiEnginePk     string
	EnableSequencer bool
	AutoChainProof  bool
	SequencerUrl    string
	EdgeUrl         string
	VerifySign      bool
}