GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-07-10 23:12:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: optimize MCP tools response format with automatic schema generation

Browse Source 
- Remove all manual ReturnSchema() methods from tools
- Implement automatic schema generation using reflection
- Unify response format to flat structure with action/success/message fields
- Simplify tool implementation by removing MCPResponse embedding
- Update documentation to reflect new architecture
- Achieve ~70% code reduction while maintaining type safety
This commit is contained in:
lilong.129
2025-06-05 23:17:06 +08:00
parent 56831845ca
commit 6e1bd5bbe2
15 changed files with 990 additions and 797 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
uixt/mcp_tools_device.go
View File

@@ -5,7 +5,6 @@ import (
"fmt"
"github.com/danielpaulus/go-ios/ios"
"github.com/httprunner/httprunner/v5/internal/json"
"github.com/httprunner/httprunner/v5/pkg/gadb"
"github.com/httprunner/httprunner/v5/uixt/option"
"github.com/mark3labs/mcp-go/mcp"
@@ -14,7 +13,14 @@ import (
)
// ToolListAvailableDevices implements the list_available_devices tool call.
type ToolListAvailableDevices struct{}
type ToolListAvailableDevices struct {
// Return data fields - these define the structure of data returned by this tool
AndroidDevices []string `json:"androidDevices" desc:"List of Android device serial numbers"`
IosDevices []string `json:"iosDevices" desc:"List of iOS device UDIDs"`
TotalCount int `json:"totalCount" desc:"Total number of available devices"`
AndroidCount int `json:"androidCount" desc:"Number of Android devices"`
IosCount int `json:"iosCount" desc:"Number of iOS devices"`
}
func (t *ToolListAvailableDevices) Name() option.ActionName {
return option.ACTION_ListAvailableDevices
@@ -59,8 +65,19 @@ func (t *ToolListAvailableDevices) Implement() server.ToolHandlerFunc {
deviceList["iosDevices"] = serialList
}
jsonResult, _ := json.Marshal(deviceList)
return mcp.NewToolResultText(string(jsonResult)), nil
// Create structured response
totalDevices := len(deviceList["androidDevices"]) + len(deviceList["iosDevices"])
message := fmt.Sprintf("Found %d available devices (%d Android, %d iOS)",
totalDevices, len(deviceList["androidDevices"]), len(deviceList["iosDevices"]))
returnData := ToolListAvailableDevices{
AndroidDevices: deviceList["androidDevices"],
IosDevices: deviceList["iosDevices"],
TotalCount: totalDevices,
AndroidCount: len(deviceList["androidDevices"]),
IosCount: len(deviceList["iosDevices"]),
}
return NewMCPSuccessResponse(message, &returnData), nil
}
}
@@ -68,15 +85,11 @@ func (t *ToolListAvailableDevices) ConvertActionToCallToolRequest(action option.
return buildMCPCallToolRequest(t.Name(), map[string]any{}), nil
}
func (t *ToolListAvailableDevices) ReturnSchema() map[string]string {
return map[string]string{
"androidDevices": "[]string: List of Android device serial numbers",
"iosDevices": "[]string: List of iOS device UDIDs",
}
}
// ToolSelectDevice implements the select_device tool call.
type ToolSelectDevice struct{}
type ToolSelectDevice struct {
// Return data fields - these define the structure of data returned by this tool
DeviceUUID string `json:"deviceUUID" desc:"UUID of the selected device"`
}
func (t *ToolSelectDevice) Name() option.ActionName {
return option.ACTION_SelectDevice
@@ -101,16 +114,13 @@ func (t *ToolSelectDevice) Implement() server.ToolHandlerFunc {
}
uuid := driverExt.IDriver.GetDevice().UUID()
return mcp.NewToolResultText(fmt.Sprintf("Selected device: %s", uuid)), nil
message := fmt.Sprintf("Selected device: %s", uuid)
returnData := ToolSelectDevice{DeviceUUID: uuid}
return NewMCPSuccessResponse(message, &returnData), nil
}
}
func (t *ToolSelectDevice) ConvertActionToCallToolRequest(action option.MobileAction) (mcp.CallToolRequest, error) {
return buildMCPCallToolRequest(t.Name(), map[string]any{}), nil
}
func (t *ToolSelectDevice) ReturnSchema() map[string]string {
return map[string]string{
"message": "string: Success message with selected device UUID",
}
}