更换搜索引擎并防止milvus使用系统代理

This commit is contained in:
2023-12-07 01:06:02 +08:00
parent 25b6d1ef8d
commit 04e1456eaa
4 changed files with 191 additions and 3 deletions

View File

@@ -267,7 +267,27 @@ func NewSchema(config Config) (graphql.Schema, error) {
var id_list []string
if args.Text != "" {
fmt.Println("args:", args)
id_list = models.ElasticsearchSearch(args.Text).GetIDList(args.First, args.Last, args.After, args.Before)
//id_list = models.ElasticsearchSearch(args.Text).GetIDList(args.First, args.Last, args.After, args.Before)
resp, err := models.ZincSearch(map[string]interface{}{
"query": map[string]interface{}{
"bool": map[string]interface{}{
"must": []map[string]interface{}{
{
"query_string": map[string]string{"query": "content:" + args.Text},
},
},
},
},
"sort": []string{},
"from": 0,
"size": 10,
})
if err != nil {
fmt.Println("获取图像列表失败", err)
return nil, err
}
id_list = resp.ToIDList(args.First, args.Last, args.After, args.Before)
id_list_str := strings.Trim(strings.Join(strings.Fields(fmt.Sprint(id_list)), ","), "[]")
if id_list_str == "" {
return map[string]interface{}{