rpc call to add AI context

Author: sawka

frontend/app/aipanel/waveai-model.tsx

@@ -173,6 +173,18 @@ export class WaveAIModel {
         return input != null && input.trim().length > 0;
     }
 
+    appendText(text: string) {
+        const currentInput = globalStore.get(this.inputAtom);
+        let newInput = currentInput;
+
+        if (newInput.length > 0 && !newInput.endsWith(" ") && !newInput.endsWith("\n")) {
+            newInput += " ";
+        }
+
+        newInput += text;
+        globalStore.set(this.inputAtom, newInput);
+    }
+
     setModel(model: string) {
         const tabId = globalStore.get(atoms.staticTabId);
         RpcApi.SetMetaCommand(TabRpcClient, {

frontend/app/store/tabrpcclient.ts

@@ -1,6 +1,7 @@
 // Copyright 2025, Command Line Inc.
 // SPDX-License-Identifier: Apache-2.0
 
+import { WaveAIModel } from "@/app/aipanel/waveai-model";
 import { getApi } from "@/app/store/global";
 import { getLayoutModelForStaticTab } from "@/layout/index";
 import { RpcResponseHelper, WshClient } from "./wshclient";
@@ -56,4 +57,24 @@ export class TabClient extends WshClient {
 
         return await getApi().captureScreenshot(electronRect);
     }
+
+    async handle_waveaiaddcontext(rh: RpcResponseHelper, data: CommandWaveAIAddContextData): Promise<void> {
+        const model = WaveAIModel.getInstance();
+
+        if (data.files && data.files.length > 0) {
+            for (const fileData of data.files) {
+                const blob = new Blob([fileData.data], { type: fileData.type });
+                const file = new File([blob], fileData.name, { type: fileData.type });
+                await model.addFile(file);
+            }
+        }
+
+        if (data.text) {
+            model.appendText(data.text);
+        }
+
+        if (data.submit) {
+            model.focusInput();
+        }
+    }
 }

frontend/app/store/wshclientapi.ts

@@ -487,6 +487,11 @@ class RpcApiType {
         return client.wshRpcCall("waitforroute", data, opts);
     }
 
+    // command "waveaiaddcontext" [call]
+    WaveAIAddContextCommand(client: WshClient, data: CommandWaveAIAddContextData, opts?: RpcOpts): Promise<void> {
+        return client.wshRpcCall("waveaiaddcontext", data, opts);
+    }
+
     // command "waveaienabletelemetry" [call]
     WaveAIEnableTelemetryCommand(client: WshClient, opts?: RpcOpts): Promise<void> {
         return client.wshRpcCall("waveaienabletelemetry", null, opts);

frontend/types/gotypes.d.ts

@@ -5,6 +5,14 @@
 
 declare global {
 
+    // wshrpc.AIAttachedFile
+    type AIAttachedFile = {
+        name: string;
+        type: string;
+        size: number;
+        data: string;
+    };
+
     // wshrpc.ActivityDisplayType
     type ActivityDisplayType = {
         width: number;
@@ -320,6 +328,13 @@ declare global {
         waitms: number;
     };
 
+    // wshrpc.CommandWaveAIAddContextData
+    type CommandWaveAIAddContextData = {
+        files?: AIAttachedFile[];
+        text?: string;
+        submit?: boolean;
+    };
+
     // wshrpc.CommandWaveAIToolApproveData
     type CommandWaveAIToolApproveData = {
         toolcallid: string;

pkg/wshrpc/wshclient/wshclient.go

@@ -581,6 +581,12 @@ func WaitForRouteCommand(w *wshutil.WshRpc, data wshrpc.CommandWaitForRouteData,
 	return resp, err
 }
 
+// command "waveaiaddcontext", wshserver.WaveAIAddContextCommand
+func WaveAIAddContextCommand(w *wshutil.WshRpc, data wshrpc.CommandWaveAIAddContextData, opts *wshrpc.RpcOpts) error {
+	_, err := sendRpcRequestCallHelper[any](w, "waveaiaddcontext", data, opts)
+	return err
+}
+
 // command "waveaienabletelemetry", wshserver.WaveAIEnableTelemetryCommand
 func WaveAIEnableTelemetryCommand(w *wshutil.WshRpc, opts *wshrpc.RpcOpts) error {
 	_, err := sendRpcRequestCallHelper[any](w, "waveaienabletelemetry", nil, opts)

pkg/wshrpc/wshrpctypes.go

@@ -143,6 +143,7 @@ const (
 	Command_GetWaveAIChat         = "getwaveaichat"
 	Command_GetWaveAIRateLimit    = "getwaveairatelimit"
 	Command_WaveAIToolApprove     = "waveaitoolapprove"
+	Command_WaveAIAddContext      = "waveaiaddcontext"
 
 	Command_CaptureBlockScreenshot = "captureblockscreenshot"
 
@@ -273,6 +274,7 @@ type WshRpcInterface interface {
 	GetWaveAIChatCommand(ctx context.Context, data CommandGetWaveAIChatData) (*uctypes.UIChat, error)
 	GetWaveAIRateLimitCommand(ctx context.Context) (*uctypes.RateLimitInfo, error)
 	WaveAIToolApproveCommand(ctx context.Context, data CommandWaveAIToolApproveData) error
+	WaveAIAddContextCommand(ctx context.Context, data CommandWaveAIAddContextData) error
 
 	// screenshot
 	CaptureBlockScreenshotCommand(ctx context.Context, data CommandCaptureBlockScreenshotData) (string, error)
@@ -734,6 +736,19 @@ type CommandWaveAIToolApproveData struct {
 	Approval   string `json:"approval,omitempty"`
 }
 
+type AIAttachedFile struct {
+	Name string `json:"name"`
+	Type string `json:"type"`
+	Size int    `json:"size"`
+	Data []byte `json:"data"`
+}
+
+type CommandWaveAIAddContextData struct {
+	Files  []AIAttachedFile `json:"files,omitempty"`
+	Text   string           `json:"text,omitempty"`
+	Submit bool             `json:"submit,omitempty"`
+}
+
 type CommandCaptureBlockScreenshotData struct {
 	BlockId string `json:"blockid" wshcontext:"BlockId"`
 }