mirror of
https://github.com/wavetermdev/waveterm
synced 2026-05-22 08:18:29 +00:00
0e8eb83f62
Documentation Updates (removing AI Widget Information / deprecation) Hover effect on tool calls shows which widget is effected Remove AI Widget from sidebar unless there is customized presets Backend now provides blockid (if available) to frontend for tool calls
83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
// Copyright 2025, Command Line Inc.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package aiusechat
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/wavetermdev/waveterm/pkg/aiusechat/uctypes"
|
|
"github.com/wavetermdev/waveterm/pkg/wcore"
|
|
"github.com/wavetermdev/waveterm/pkg/wshrpc"
|
|
"github.com/wavetermdev/waveterm/pkg/wshrpc/wshclient"
|
|
"github.com/wavetermdev/waveterm/pkg/wshutil"
|
|
)
|
|
|
|
func makeTabCaptureBlockScreenshot(tabId string) func(any) (string, error) {
|
|
return func(input any) (string, error) {
|
|
inputMap, ok := input.(map[string]any)
|
|
if !ok {
|
|
return "", fmt.Errorf("invalid input format")
|
|
}
|
|
|
|
blockIdPrefix, ok := inputMap["widget_id"].(string)
|
|
if !ok {
|
|
return "", fmt.Errorf("missing or invalid widget_id parameter")
|
|
}
|
|
|
|
ctx, cancelFn := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancelFn()
|
|
|
|
fullBlockId, err := wcore.ResolveBlockIdFromPrefix(ctx, tabId, blockIdPrefix)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
rpcClient := wshclient.GetBareRpcClient()
|
|
screenshotData, err := wshclient.CaptureBlockScreenshotCommand(
|
|
rpcClient,
|
|
wshrpc.CommandCaptureBlockScreenshotData{BlockId: fullBlockId},
|
|
&wshrpc.RpcOpts{Route: wshutil.MakeTabRouteId(tabId)},
|
|
)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to capture screenshot: %w", err)
|
|
}
|
|
|
|
return screenshotData, nil
|
|
}
|
|
}
|
|
|
|
func GetCaptureScreenshotToolDefinition(tabId string) uctypes.ToolDefinition {
|
|
return uctypes.ToolDefinition{
|
|
Name: "capture_screenshot",
|
|
DisplayName: "Capture Screenshot",
|
|
Description: "Capture a screenshot of a widget and return it as an image",
|
|
ToolLogName: "gen:screenshot",
|
|
Strict: true,
|
|
InputSchema: map[string]any{
|
|
"type": "object",
|
|
"properties": map[string]any{
|
|
"widget_id": map[string]any{
|
|
"type": "string",
|
|
"description": "8-character widget ID of the widget to screenshot",
|
|
},
|
|
},
|
|
"required": []string{"widget_id"},
|
|
"additionalProperties": false,
|
|
},
|
|
ToolInputDesc: func(input any) string {
|
|
inputMap, ok := input.(map[string]any)
|
|
if !ok {
|
|
return "error parsing input: invalid format"
|
|
}
|
|
widgetId, ok := inputMap["widget_id"].(string)
|
|
if !ok {
|
|
return "error parsing input: missing widget_id"
|
|
}
|
|
return fmt.Sprintf("capturing screenshot of widget %s", widgetId)
|
|
},
|
|
ToolTextCallback: makeTabCaptureBlockScreenshot(tabId),
|
|
}
|
|
}
|