mod_ai_token_auth

type Action ActionFile

type ActionFile struct {
	Cmd string
}

type ConfModAITokenAuth struct {
	Basic struct {
		ProductRulePath string
	}

	// redis conf
	Redis struct {
		Bns            string // bns name for redis proxy
		ConnectTimeout int    // connect timeout (ms)
		ReadTimeout    int    // read timeout (ms)
		WriteTimeout   int    // write timeout(ms)

		// max idle connections in pool
		MaxIdle int

		// redis password，ignore if not set
		Password string

		// max active connections in pool,
		// when set 0, there is no connection num limit
		MaxActive int
	}

	Log struct {
		OpenDebug bool
	}
}

type ModuleAITokenAuth struct {
	// contains filtered or unexported fields
}

type ModuleAITokenAuthState struct {
	ReqTotal    *metrics.Counter
	ReqAuth     *metrics.Counter
	ReqAuthFail *metrics.Counter
}

type ProductRules map[string]*tokenRuleList

type ProductRulesFile map[string]*tokenRuleFileList

type ProductTokenFiles map[string]*tokenFileMap

type ProductTokens map[string]*tokenMap

type Token struct {
	Key            string
	Status         int
	Name           string
	UpdateTime     int64
	ExpiredTime    int64
	RemainQuota    int64
	UsedQuota      *atomic.Uint64
	UnlimitedQuota bool
	Models         []string
	Subnet         []*net.IPNet
}

type TokenAuthContext struct {
	Token            *Token
	PromptTokens     int64 // number of tokens in the prompt
	CompletionTokens int64 // number of tokens in the completion
	UsedQuota        int64 // used quota for this request
}

type TokenFile struct {
	Key            string  `json:"key"`
	Status         int     `json:"status"`
	Name           string  `json:"name"`
	UpdateTime     int64   `json:"update_time"`
	ExpiredTime    int64   `json:"expired_time"` // -1 means never expired
	RemainQuota    int64   `json:"remain_quota"`
	UnlimitedQuota bool    `json:"unlimited_quota"`
	Models         *string `json:"models"` // allowed models
	Subnet         *string `json:"subnet"` // allowed subnet
	// contains filtered or unexported fields
}

type TokenRuleTable struct {
	// contains filtered or unexported fields
}