Skip to content

Commit e310007

Browse files
author
piexlMax(奇淼
committed
refactor(mcp): 统一使用WithBoolean替换WithBool并优化错误处理
1 parent a45a038 commit e310007

File tree

1 file changed

+10
-11
lines changed

1 file changed

+10
-11
lines changed

server/mcp/dictionary_query.go

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -56,10 +56,10 @@ func (d *DictionaryQuery) New() mcp.Tool {
5656
mcp.WithString("dictType",
5757
mcp.Description("可选:指定字典类型进行精确查询,如果不提供则返回所有字典"),
5858
),
59-
mcp.WithBool("includeDisabled",
59+
mcp.WithBoolean("includeDisabled",
6060
mcp.Description("是否包含已禁用的字典和字典项,默认为false(只返回启用的)"),
6161
),
62-
mcp.WithBool("detailsOnly",
62+
mcp.WithBoolean("detailsOnly",
6363
mcp.Description("是否只返回字典详情信息(不包含字典基本信息),默认为false"),
6464
),
6565
)
@@ -87,7 +87,6 @@ func (d *DictionaryQuery) Handle(ctx context.Context, request mcp.CallToolReques
8787

8888
// 获取字典服务
8989
dictionaryService := service.ServiceGroupApp.SystemServiceGroup.DictionaryService
90-
dictionaryDetailService := service.ServiceGroupApp.SystemServiceGroup.DictionaryDetailService
9190

9291
var dictionaries []DictionaryInfo
9392
var err error
@@ -103,8 +102,8 @@ func (d *DictionaryQuery) Handle(ctx context.Context, request mcp.CallToolReques
103102
if err != nil {
104103
global.GVA_LOG.Error("查询字典失败", zap.Error(err))
105104
return &mcp.CallToolResult{
106-
Content: []interface{}{
107-
mcp.NewTextContent(fmt.Sprintf(`{"success": false, "message": "查询字典失败: %v", "total": 0, "dictionaries": []}`), err.Error())),
105+
Content: []mcp.Content{
106+
mcp.NewTextContent(fmt.Sprintf(`{"success": false, "message": "查询字典失败: %v", "total": 0, "dictionaries": []}`, err.Error())),
108107
},
109108
}, nil
110109
}
@@ -153,8 +152,8 @@ func (d *DictionaryQuery) Handle(ctx context.Context, request mcp.CallToolReques
153152
if err != nil {
154153
global.GVA_LOG.Error("查询字典列表失败", zap.Error(err))
155154
return &mcp.CallToolResult{
156-
Content: []interface{}{
157-
mcp.NewTextContent(fmt.Sprintf(`{"success": false, "message": "查询字典列表失败: %v", "total": 0, "dictionaries": []}`), err.Error())),
155+
Content: []mcp.Content{
156+
mcp.NewTextContent(fmt.Sprintf(`{"success": false, "message": "查询字典列表失败: %v", "total": 0, "dictionaries": []}`, err.Error())),
158157
},
159158
}, nil
160159
}
@@ -203,7 +202,7 @@ func (d *DictionaryQuery) Handle(ctx context.Context, request mcp.CallToolReques
203202

204203
responseJSON, _ := json.Marshal(response)
205204
return &mcp.CallToolResult{
206-
Content: []interface{}{
205+
Content: []mcp.Content{
207206
mcp.NewTextContent(string(responseJSON)),
208207
},
209208
}, nil
@@ -221,14 +220,14 @@ func (d *DictionaryQuery) Handle(ctx context.Context, request mcp.CallToolReques
221220
if err != nil {
222221
global.GVA_LOG.Error("序列化响应失败", zap.Error(err))
223222
return &mcp.CallToolResult{
224-
Content: []interface{}{
225-
mcp.NewTextContent(fmt.Sprintf(`{"success": false, "message": "序列化响应失败: %v", "total": 0, "dictionaries": []}`), err.Error())),
223+
Content: []mcp.Content{
224+
mcp.NewTextContent(fmt.Sprintf(`{"success": false, "message": "序列化响应失败: %v", "total": 0, "dictionaries": []}`, err.Error())),
226225
},
227226
}, nil
228227
}
229228

230229
return &mcp.CallToolResult{
231-
Content: []interface{}{
230+
Content: []mcp.Content{
232231
mcp.NewTextContent(string(responseJSON)),
233232
},
234233
}, nil

0 commit comments

Comments
 (0)