mirror of
https://github.com/coder/coder.git
synced 2026-06-03 13:08:25 +00:00
Hugo Dutka
84527390c6
Implement the backend for the desktop feature for agents. - Adds a new `/api/experimental/chats/$id/desktop` endpoint to coderd which exposes a VNC stream from a [portabledesktop](https://github.com/coder/portabledesktop) process running inside the workspace - Adds a new `spawn_computer_use_agent` tool to chatd, which spawns a subagent that has access to the `computer` tool which lets it interact with the `portabledesktop` process running inside the workspace - Adds the plumbing to make the above possible There's a follow up frontend PR here: https://github.com/coder/coder/pull/23006
82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
package chattool
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestComputeScaledScreenshotSize(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
tests := []struct {
|
|
name string
|
|
width, height int
|
|
wantW, wantH int
|
|
}{
|
|
{
|
|
name: "1920x1080_scales_down",
|
|
width: 1920,
|
|
height: 1080,
|
|
wantW: 1429,
|
|
wantH: 804,
|
|
},
|
|
{
|
|
name: "1280x800_no_scaling",
|
|
width: 1280,
|
|
height: 800,
|
|
wantW: 1280,
|
|
wantH: 800,
|
|
},
|
|
{
|
|
name: "3840x2160_large_display",
|
|
width: 3840,
|
|
height: 2160,
|
|
wantW: 1429,
|
|
wantH: 804,
|
|
},
|
|
{
|
|
name: "1568x1000_pixel_cap_applies",
|
|
width: 1568,
|
|
height: 1000,
|
|
wantW: 1342,
|
|
wantH: 856,
|
|
},
|
|
{
|
|
name: "100x100_small_display",
|
|
width: 100,
|
|
height: 100,
|
|
wantW: 100,
|
|
wantH: 100,
|
|
},
|
|
{
|
|
name: "4000x3000_stays_within_limits",
|
|
width: 4000,
|
|
// Both constraints apply. The function should keep
|
|
// the result within maxLongEdge=1568 and
|
|
// totalPixels<=1,150,000.
|
|
height: 3000,
|
|
wantW: 1238,
|
|
wantH: 928,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
gotW, gotH := computeScaledScreenshotSize(tt.width, tt.height)
|
|
assert.Equal(t, tt.wantW, gotW)
|
|
assert.Equal(t, tt.wantH, gotH)
|
|
|
|
// Invariant: results must respect Anthropic constraints.
|
|
const maxLongEdge = 1568
|
|
const maxTotalPixels = 1_150_000
|
|
longEdge := max(gotW, gotH)
|
|
assert.LessOrEqual(t, longEdge, maxLongEdge,
|
|
"long edge %d exceeds max %d", longEdge, maxLongEdge)
|
|
assert.LessOrEqual(t, gotW*gotH, maxTotalPixels,
|
|
"total pixels %d exceeds max %d", gotW*gotH, maxTotalPixels)
|
|
})
|
|
}
|
|
}
|