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.
84 lines
2.0 KiB
84 lines
2.0 KiB
package tavilysearch
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
type (
|
|
// TavilyResponse Tavily Search API 最小结构
|
|
// 仅包含结果的核心字段
|
|
TavilyResponse struct {
|
|
Results []struct {
|
|
Title string `json:"title"`
|
|
URL string `json:"url"`
|
|
Content string `json:"content"`
|
|
} `json:"results"`
|
|
}
|
|
// ISys 搜索系统接口
|
|
// 仅实现网页搜索的最小能力
|
|
ISys interface {
|
|
// Search 执行网页搜索
|
|
// 参数:
|
|
// - ctx: 上下文
|
|
// - query: 搜索关键词
|
|
// - count: 返回条数
|
|
// - offset: 偏移量（保留，不参与）
|
|
//
|
|
// 返回值:
|
|
// - results: TavilyResponse 最小结构
|
|
// - err: 调用过程中的错误
|
|
//
|
|
// 异常:
|
|
// - 当请求失败或响应体解析失败时返回错误
|
|
Search(ctx context.Context, query string, count, offset int) (results *TavilyResponse, err error)
|
|
}
|
|
)
|
|
|
|
var defsys ISys
|
|
|
|
// OnInit 初始化全局系统对象
|
|
// 参数:
|
|
// - config: 配置映射
|
|
// - option: 可选项函数
|
|
//
|
|
// 返回值:
|
|
// - err: 初始化过程中的错误
|
|
//
|
|
// 异常:
|
|
// - 当内部对象创建失败时返回错误
|
|
func OnInit(config map[string]interface{}, option ...Option) (err error) {
|
|
defsys, err = newSys(newOptions(config, option...))
|
|
return
|
|
}
|
|
|
|
// NewSys 创建一个新的系统对象
|
|
// 参数:
|
|
// - option: 可选项函数
|
|
//
|
|
// 返回值:
|
|
// - sys: 系统对象
|
|
// - err: 创建过程中的错误
|
|
//
|
|
// 异常:
|
|
// - 当内部对象创建失败时返回错误
|
|
func NewSys(option ...Option) (sys ISys, err error) {
|
|
sys, err = newSys(newOptionsByOption(option...))
|
|
return
|
|
}
|
|
|
|
// Search 执行网页搜索
|
|
// 参数:
|
|
// - ctx: 上下文
|
|
// - query: 搜索关键词
|
|
// - count: 返回条数
|
|
// - offset: 偏移量（保留，不参与）
|
|
//
|
|
// 返回值:
|
|
// - results: TavilyResponse 最小结构
|
|
// - err: 调用过程中的错误
|
|
//
|
|
// 异常:
|
|
// - 当请求失败或响应体解析失败时返回错误
|
|
func Search(ctx context.Context, query string, count, offset int) (results *TavilyResponse, err error) {
|
|
return defsys.Search(ctx, query, count, offset)
|
|
}
|
|
|