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.
92 lines
1.8 KiB
92 lines
1.8 KiB
package tavilysearch
|
|
|
|
import (
|
|
"yunyan/lego/sys/log"
|
|
"yunyan/lego/utils/mapstructure"
|
|
)
|
|
|
|
type Option func(*Options)
|
|
|
|
type Options struct {
|
|
Debug bool // 日志是否开启
|
|
Log log.ILogger
|
|
ApiKey string // Tavily API Key
|
|
SearchDepth string // 搜索深度: basic | advanced
|
|
Include []string // 可选：只包含的域名
|
|
Exclude []string // 可选：排除的域名
|
|
}
|
|
|
|
// SetApiKey 设置 Tavily API 密钥
|
|
// 参数:
|
|
// - v: API 密钥字符串
|
|
//
|
|
// 返回值:
|
|
// - 无
|
|
//
|
|
// 异常:
|
|
// - 无
|
|
func SetApiKey(v string) Option {
|
|
return func(o *Options) {
|
|
o.ApiKey = v
|
|
}
|
|
}
|
|
|
|
// SetSearchDepth 设置搜索深度
|
|
// 参数:
|
|
// - v: 搜索深度 basic/advanced
|
|
//
|
|
// 返回值:
|
|
// - 无
|
|
//
|
|
// 异常:
|
|
// - 无
|
|
func SetSearchDepth(v string) Option {
|
|
return func(o *Options) {
|
|
o.SearchDepth = v
|
|
}
|
|
}
|
|
|
|
// newOptions 初始化配置（从 map 与可选项合并）
|
|
// 参数:
|
|
// - config: 配置映射
|
|
// - opts: 可选项函数
|
|
//
|
|
// 返回值:
|
|
// - Options: 合并后的选项
|
|
//
|
|
// 异常:
|
|
// - 无
|
|
func newOptions(config map[string]interface{}, opts ...Option) Options {
|
|
options := Options{}
|
|
if config != nil {
|
|
mapstructure.Decode(config, &options)
|
|
}
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
if options.Log == nil {
|
|
options.Log = log.NewTurnlog(options.Debug, log.Clone("sys.tavilysearch", 3))
|
|
}
|
|
return options
|
|
}
|
|
|
|
// newOptionsByOption 仅基于可选项初始化配置
|
|
// 参数:
|
|
// - opts: 可选项函数
|
|
//
|
|
// 返回值:
|
|
// - Options: 合并后的选项
|
|
//
|
|
// 异常:
|
|
// - 无
|
|
func newOptionsByOption(opts ...Option) Options {
|
|
options := Options{}
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
if options.Log == nil {
|
|
options.Log = log.NewTurnlog(options.Debug, log.Clone("sys.tavilysearch", 3))
|
|
}
|
|
return options
|
|
}
|
|
|
|
|