File size: 1,741 Bytes
5710dd0 | 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 | import type {
AgentRuntimeBindingSnapshot,
SessionWorkspaceAssociationSnapshot,
WorkspaceInstanceSnapshot,
WorkspaceInstancesSnapshot,
} from '@moonshot-ai/agent-core-v2';
import { DEBUG_RPC_BASE, type InspectClient } from '../channel';
import { RPCError } from '../channel/errors';
export function fetchWorkspaceSnapshots(client: InspectClient): Promise<WorkspaceInstancesSnapshot> {
return fetchSnapshot(client, '/workspaces');
}
export function fetchWorkspaceSnapshot(
client: InspectClient,
workspaceId: string,
): Promise<WorkspaceInstanceSnapshot> {
return fetchSnapshot(client, `/workspace/${encodeURIComponent(workspaceId)}/snapshot`);
}
export function fetchSessionWorkspaceAssociation(
client: InspectClient,
sessionId: string,
): Promise<SessionWorkspaceAssociationSnapshot> {
return fetchSnapshot(client, `/session/${encodeURIComponent(sessionId)}/association`);
}
export function fetchAgentRuntimeBinding(
client: InspectClient,
sessionId: string,
agentId: string,
): Promise<AgentRuntimeBindingSnapshot> {
return fetchSnapshot(
client,
`/session/${encodeURIComponent(sessionId)}/agent/${encodeURIComponent(agentId)}/runtime-binding`,
);
}
async function fetchSnapshot<T>(client: InspectClient, path: string): Promise<T> {
const headers: Record<string, string> = {};
if (client.token !== undefined && client.token !== '') {
headers['authorization'] = `Bearer ${client.token}`;
}
const response = await fetch(`${client.baseUrl}${DEBUG_RPC_BASE}${path}`, { headers });
const envelope = (await response.json()) as {
code: number;
msg: string;
data: T;
};
if (envelope.code !== 0) throw new RPCError(envelope.code, envelope.msg);
return envelope.data;
}
|