refactor: GetScreenshotBase64WithSize

This commit is contained in:
lilong.129
2025-06-13 12:01:21 +08:00
parent f6e7e970f8
commit 409cd693f0
4 changed files with 26 additions and 37 deletions

View File

@@ -34,7 +34,7 @@ func (t *ToolScreenShot) Implement() server.ToolHandlerFunc {
if err != nil {
return nil, err
}
bufferBase64, err := GetScreenShotBufferBase64(driverExt.IDriver)
bufferBase64, _, err := driverExt.GetScreenshotBase64WithSize()
if err != nil {
log.Error().Err(err).Msg("ScreenShot failed")
return mcp.NewToolResultError(fmt.Sprintf("Failed to take screenshot: %v", err)), nil