geekai/types/config.go

package types

import (
	"net/http"
)

type Config struct {
	Listen     string
	Session    Session
	ProxyURL   []string
	EnableAuth bool   // 是否开启鉴权
	AccessKey  string // 管理员访问 AccessKey, 通过传入这个参数可以访问系统管理 API
	Chat       Chat
}

type User struct {
	Name           string `json:"name"`
	MaxCalls       int    `json:"max_calls"`         // 最多调用次数，如果为 0 则表示不限制
	RemainingCalls int    `json:"remaining_calls"`   // 剩余调用次数
	EnableHistory  bool   `json:"enable_history"`    // 是否启用聊天记录
	Status         bool   `json:"status"`            // 当前状态
	Term           int    `json:"term" default:"30"` // 会员有效期，单位：天
	ActiveTime     int64  `json:"active_time"`       // 激活时间
	ExpiredTime    int64  `json:"expired_time"`      // 到期时间
	ApiKey         string `json:"api_key"`           // OpenAI  API KEY
}

// Chat configs struct
type Chat struct {
	ApiURL                string
	ApiKeys               []string
	Model                 string
	Temperature           float32
	MaxTokens             int
	EnableContext         bool // 是否保持聊天上下文
	ChatContextExpireTime int  // 聊天上下文过期时间，单位：秒
}

// Session configs struct
type Session struct {
	SecretKey string // session encryption key
	Name      string
	Path      string
	Domain    string
	MaxAge    int
	Secure    bool
	HttpOnly  bool
	SameSite  http.SameSite
}