You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
1.5 KiB
77 lines
1.5 KiB
package gemini
|
|
|
|
import (
|
|
"yunyan/lego/sys/log"
|
|
"yunyan/lego/utils/mapstructure"
|
|
)
|
|
|
|
type Option func(*Options)
|
|
|
|
type Options struct {
|
|
Debug bool
|
|
Log log.ILogger
|
|
Apikey string // Gemini API Key（AIzaSy... 形式）
|
|
Model string // 默认模型 ID，如 gemini-2.5-flash / gemini-2.5-pro
|
|
Endpoint string // 自定义 endpoint（可选，默认 generativelanguage.googleapis.com）
|
|
Timeout int // HTTP 超时（秒），默认 120
|
|
}
|
|
|
|
func SetApikey(v string) Option {
|
|
return func(o *Options) {
|
|
o.Apikey = v
|
|
}
|
|
}
|
|
|
|
func SetModel(v string) Option {
|
|
return func(o *Options) {
|
|
o.Model = v
|
|
}
|
|
}
|
|
|
|
func SetEndpoint(v string) Option {
|
|
return func(o *Options) {
|
|
o.Endpoint = v
|
|
}
|
|
}
|
|
|
|
func SetTimeout(v int) Option {
|
|
return func(o *Options) {
|
|
o.Timeout = v
|
|
}
|
|
}
|
|
|
|
func newOptions(config map[string]interface{}, opts ...Option) Options {
|
|
options := Options{}
|
|
if config != nil {
|
|
mapstructure.Decode(config, &options)
|
|
}
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
applyDefaults(&options)
|
|
return options
|
|
}
|
|
|
|
func newOptionsByOption(opts ...Option) Options {
|
|
options := Options{}
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
applyDefaults(&options)
|
|
return options
|
|
}
|
|
|
|
func applyDefaults(o *Options) {
|
|
if o.Endpoint == "" {
|
|
o.Endpoint = "https://generativelanguage.googleapis.com"
|
|
}
|
|
if o.Model == "" {
|
|
o.Model = "gemini-2.5-flash"
|
|
}
|
|
if o.Timeout <= 0 {
|
|
o.Timeout = 120
|
|
}
|
|
if o.Log == nil {
|
|
o.Log = log.NewTurnlog(o.Debug, log.Clone("sys.google.gemini", 3))
|
|
}
|
|
}
|
|
|