62 lines
2.4 KiB
Go

package models
import (
"time"
)
type Price struct {
ID uint `json:"id"`
Model string `json:"model"`
ModelType string `json:"model_type"` // text2text, text2image, etc.
BillingType string `json:"billing_type"` // tokens or times
ChannelType string `json:"channel_type"`
Currency string `json:"currency"` // USD or CNY
InputPrice float64 `json:"input_price"`
OutputPrice float64 `json:"output_price"`
PriceSource string `json:"price_source"`
Status string `json:"status"` // pending, approved, rejected
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
CreatedBy string `json:"created_by"`
// 临时字段,用于存储待审核的更新
TempModel *string `json:"temp_model,omitempty"`
TempModelType *string `json:"temp_model_type,omitempty"`
TempBillingType *string `json:"temp_billing_type,omitempty"`
TempChannelType *string `json:"temp_channel_type,omitempty"`
TempCurrency *string `json:"temp_currency,omitempty"`
TempInputPrice *float64 `json:"temp_input_price,omitempty"`
TempOutputPrice *float64 `json:"temp_output_price,omitempty"`
TempPriceSource *string `json:"temp_price_source,omitempty"`
UpdatedBy *string `json:"updated_by,omitempty"`
}
// CreatePriceTableSQL 返回创建价格表的 SQL
func CreatePriceTableSQL() string {
return `
CREATE TABLE IF NOT EXISTS price (
id BIGINT UNSIGNED AUTO_INCREMENT PRIMARY KEY,
model VARCHAR(255) NOT NULL,
model_type VARCHAR(50) NOT NULL,
billing_type VARCHAR(50) NOT NULL,
channel_type VARCHAR(50) NOT NULL,
currency VARCHAR(10) NOT NULL,
input_price DECIMAL(10,6) NOT NULL,
output_price DECIMAL(10,6) NOT NULL,
price_source VARCHAR(255) NOT NULL,
status VARCHAR(50) NOT NULL DEFAULT 'pending',
created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
created_by VARCHAR(255) NOT NULL,
temp_model VARCHAR(255),
temp_model_type VARCHAR(50),
temp_billing_type VARCHAR(50),
temp_channel_type VARCHAR(50),
temp_currency VARCHAR(10),
temp_input_price DECIMAL(10,6),
temp_output_price DECIMAL(10,6),
temp_price_source VARCHAR(255),
updated_by VARCHAR(255),
FOREIGN KEY (channel_type) REFERENCES provider(id)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci`
}