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.
51 lines
1.4 KiB
51 lines
1.4 KiB
package bochasearch
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
type (
|
|
// 根响应结构体
|
|
BochaResponse struct {
|
|
Code int `json:"code"`
|
|
LogID string `json:"log_id"`
|
|
ConversationID string `json:"conversation_id"`
|
|
Messages []Message `json:"messages"`
|
|
}
|
|
|
|
// Message 消息结构体
|
|
Message struct {
|
|
Role string `json:"role"`
|
|
Type string `json:"type"`
|
|
ContentType string `json:"content_type"`
|
|
Content string `json:"content"` // 注意：根据实际内容可能需要进一步解析为具体结构体
|
|
}
|
|
ISys interface {
|
|
Search(ctx context.Context, query, freshness string, count int) (results *BochaResponse, err error)
|
|
}
|
|
)
|
|
|
|
var defsys ISys
|
|
|
|
func OnInit(config map[string]interface{}, option ...Option) (err error) {
|
|
defsys, err = newSys(newOptions(config, option...))
|
|
return
|
|
}
|
|
|
|
func NewSys(option ...Option) (sys ISys, err error) {
|
|
sys, err = newSys(newOptionsByOption(option...))
|
|
return
|
|
}
|
|
|
|
/*
|
|
- oneDay，一天内
|
|
- oneWeek，一周内
|
|
- oneMonth，一个月内
|
|
- oneYear，一年内
|
|
- noLimit，不限（默认）
|
|
- YYYY-MM-DD..YYYY-MM-DD，搜索日期范围，例如："2025-01-01..2025-04-06"
|
|
- YYYY-MM-DD，搜索指定日期，例如："2025-04-06"
|
|
*/
|
|
func Search(ctx context.Context, query, freshness string, count int) (results *BochaResponse, err error) {
|
|
return defsys.Search(ctx, query, freshness, count)
|
|
}
|
|
|