mirror of
https://github.com/mindoc-org/mindoc.git
synced 2025-10-07 15:48:33 +08:00
69 lines
2.2 KiB
Go
69 lines
2.2 KiB
Go
package mcp
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
|
||
"github.com/mark3labs/mcp-go/mcp"
|
||
"github.com/mindoc-org/mindoc/conf"
|
||
"github.com/mindoc-org/mindoc/models"
|
||
"github.com/mindoc-org/mindoc/utils/sqltil"
|
||
)
|
||
|
||
// GetGlobalSearchMcpTool 获取全局搜索的mcp工具
|
||
func GetGlobalSearchMcpTool() mcp.Tool {
|
||
return mcp.NewTool("MinDocGlobalSearch",
|
||
mcp.WithDescription("MinDoc全局文档内容搜索"),
|
||
mcp.WithString("keyword",
|
||
mcp.Required(),
|
||
mcp.Description("要执行全局搜索的关键词，多个搜索关键词请用空格分割，请使用最少的关键词来检索，结果中只会出现包含全部关键词的结果，过多的无关词会导致更少的检索结果"),
|
||
),
|
||
mcp.WithNumber("pageIndex",
|
||
mcp.Required(),
|
||
mcp.Description("全局搜索时指定分页的顺序下标，每页最多有10条结果，建议只查看1-10页文档内容的搜索结果"),
|
||
mcp.Enum("1", "2", "3", "4", "5", "6", "7", "8", "9", "10"),
|
||
),
|
||
)
|
||
}
|
||
|
||
// GlobalSearchMcpHandler 全局搜索的mcp处理函数
|
||
func GlobalSearchMcpHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||
paramMap := request.Params.Arguments.(map[string]any)
|
||
pageIndex := 1
|
||
if v, ok := paramMap["pageIndex"].(float64); ok {
|
||
pageIndex = int(v)
|
||
}
|
||
totalCount, result := globalSearchFunction(paramMap["keyword"].(string), pageIndex)
|
||
jsonContent, err := json.Marshal(result)
|
||
if err != nil {
|
||
return mcp.NewToolResultStructuredOnly(map[string]any{
|
||
"totalCount": 0,
|
||
"result": make([]map[string]any, 0),
|
||
}), err
|
||
}
|
||
|
||
structContent := make([]map[string]any, 0)
|
||
err = json.Unmarshal(jsonContent, &structContent)
|
||
if err != nil {
|
||
return mcp.NewToolResultStructuredOnly(map[string]any{
|
||
"totalCount": 0,
|
||
"result": make([]map[string]any, 0),
|
||
}), err
|
||
}
|
||
|
||
return mcp.NewToolResultStructuredOnly(map[string]any{
|
||
"totalCount": totalCount,
|
||
"result": structContent,
|
||
}), nil
|
||
}
|
||
|
||
func globalSearchFunction(keyword string, pageIndex int) (int, []*models.DocumentSearchResult) {
|
||
memberId := 0
|
||
searchResult, totalCount, err := models.NewDocumentSearchResult().FindToPager(sqltil.EscapeLike(keyword),
|
||
pageIndex, conf.PageSize, memberId)
|
||
if err != nil {
|
||
return 0, make([]*models.DocumentSearchResult, 0)
|
||
}
|
||
return totalCount, searchResult
|
||
}
|