mirror of
https://github.com/EZ-Api/ez-api.git
synced 2026-01-13 17:47:51 +00:00
8645b22b83
Add admin and master authentication layers with JWT support. Replace direct key creation with hierarchical master/child key system. Update database schema to support master accounts with configurable limits and epoch-based key revocation. Add health check endpoint with system status monitoring. BREAKING CHANGE: Removed direct POST /keys endpoint in favor of master-based key issuance through /v1/tokens. Database migration requires dropping old User table and creating Master table with new relationships.
55 lines
2.1 KiB
Go
55 lines
2.1 KiB
Go
package model
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Admin is not a database model. It's configured via environment variables.
|
|
|
|
// Master represents a tenant account.
|
|
type Master struct {
|
|
gorm.Model
|
|
Name string `gorm:"size:255" json:"name"`
|
|
MasterKey string `gorm:"size:255;uniqueIndex" json:"-"` // Hashed master key
|
|
Group string `gorm:"size:100;default:'default'" json:"group"`
|
|
Epoch int64 `gorm:"default:1" json:"epoch"`
|
|
Status string `gorm:"size:50;default:'active'" json:"status"` // active, suspended
|
|
MaxChildKeys int `gorm:"default:5" json:"max_child_keys"`
|
|
GlobalQPS int `gorm:"default:3" json:"global_qps"`
|
|
}
|
|
|
|
// Key represents a child access token issued by a Master.
|
|
type Key struct {
|
|
gorm.Model
|
|
MasterID uint `gorm:"not null;index" json:"master_id"`
|
|
KeySecret string `gorm:"size:255;uniqueIndex" json:"key_secret"`
|
|
Group string `gorm:"size:100;default:'default'" json:"group"`
|
|
Scopes string `gorm:"size:1024" json:"scopes"` // Comma-separated scopes
|
|
IssuedAtEpoch int64 `gorm:"not null" json:"issued_at_epoch"`
|
|
Status string `gorm:"size:50;default:'active'" json:"status"` // active, suspended
|
|
}
|
|
|
|
// Provider remains the same.
|
|
type Provider struct {
|
|
gorm.Model
|
|
Name string `gorm:"not null" json:"name"`
|
|
Type string `gorm:"not null" json:"type"` // openai, anthropic, etc.
|
|
BaseURL string `json:"base_url"`
|
|
APIKey string `json:"api_key"`
|
|
Group string `gorm:"default:'default'" json:"group"` // routing group/tier
|
|
Models string `json:"models"` // comma-separated list of supported models (e.g. "gpt-4,gpt-3.5-turbo")
|
|
}
|
|
|
|
// Model remains the same.
|
|
type Model struct {
|
|
gorm.Model
|
|
Name string `gorm:"uniqueIndex;not null" json:"name"`
|
|
ContextWindow int `json:"context_window"`
|
|
CostPerToken float64 `json:"cost_per_token"`
|
|
SupportsVision bool `json:"supports_vision"`
|
|
SupportsFunctions bool `json:"supports_functions"`
|
|
SupportsToolChoice bool `json:"supports_tool_choice"`
|
|
SupportsFIM bool `json:"supports_fim"`
|
|
MaxOutputTokens int `json:"max_output_tokens"`
|
|
}
|