File size: 3,113 Bytes
3d700dd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | import { FileClient } from "@openhands/typescript-client/clients";
import { RemoteWorkspace } from "@openhands/typescript-client/workspace/remote-workspace";
import { getAgentServerClientOptions } from "#/api/agent-server-client-options";
import { getActiveBackend } from "#/api/backend-registry/active-store";
import { callCloudProxy } from "#/api/cloud/proxy";
import { buildHttpBaseUrl } from "#/utils/websocket-url";
export interface CommandResult {
exit_code: number;
stdout: string;
stderr: string;
}
/**
* Cloud-aware runtime operations for agent-server conversations.
*
* In **local** mode the runtime is reachable directly from the browser
* (e.g. `127.0.0.1:18000`) so the SDK's typed clients work fine.
* In **cloud** mode the runtime lives at `*.prod-runtime.all-hands.dev`,
* which doesn't allow CORS from `localhost`, so all calls go through
* `callCloudProxy` with the runtime URL as `hostOverride` and the
* conversation's `session_api_key` as auth — server-side hop, no CORS.
*/
class AgentServerRuntimeService {
static async executeCommand(
conversationUrl: string | null | undefined,
sessionApiKey: string | null | undefined,
command: string,
cwd?: string,
timeout = 30,
): Promise<CommandResult> {
const active = getActiveBackend().backend;
if (active.kind === "cloud" && conversationUrl) {
const output = await callCloudProxy<{
exit_code?: number;
stdout?: string;
stderr?: string;
}>({
backend: active,
method: "POST",
hostOverride: buildHttpBaseUrl(conversationUrl),
path: "/api/bash/execute_bash_command",
body: {
command,
...(cwd ? { cwd } : {}),
timeout: Math.floor(timeout),
},
authMode: "session-api-key",
sessionApiKey,
timeoutSeconds: timeout + 10,
});
return {
exit_code: output.exit_code ?? -1,
stdout: output.stdout ?? "",
stderr: output.stderr ?? "",
};
}
const result = await new RemoteWorkspace(
getAgentServerClientOptions({ conversationUrl, sessionApiKey }),
).executeCommand(command, cwd, timeout);
return {
exit_code: result.exit_code,
stdout: result.stdout,
stderr: result.stderr,
};
}
static async downloadFile(
conversationUrl: string | null | undefined,
sessionApiKey: string | null | undefined,
path: string,
): Promise<ArrayBuffer> {
const active = getActiveBackend().backend;
if (active.kind === "cloud" && conversationUrl) {
const blob = await callCloudProxy<Blob>({
backend: active,
method: "GET",
hostOverride: buildHttpBaseUrl(conversationUrl),
path: `/api/file/download?path=${encodeURIComponent(path)}`,
authMode: "session-api-key",
sessionApiKey,
responseType: "blob",
});
return blob.arrayBuffer();
}
return new FileClient(
getAgentServerClientOptions({ conversationUrl, sessionApiKey }),
).downloadFile(path);
}
}
export default AgentServerRuntimeService;
|