edgegate-mcp 0.9.1 → 0.11.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/client.d.ts +44 -2
- package/dist/client.js +28 -0
- package/dist/client.js.map +1 -1
- package/dist/server.js +60 -7
- package/dist/server.js.map +1 -1
- package/dist/tools/attach_byo_role.d.ts +28 -0
- package/dist/tools/attach_byo_role.js +96 -0
- package/dist/tools/attach_byo_role.js.map +1 -0
- package/dist/tools/check_llm_compile_status.d.ts +15 -0
- package/dist/tools/check_llm_compile_status.js +49 -0
- package/dist/tools/check_llm_compile_status.js.map +1 -0
- package/dist/tools/create_pipeline.d.ts +139 -17
- package/dist/tools/create_pipeline.js +155 -29
- package/dist/tools/create_pipeline.js.map +1 -1
- package/dist/tools/import_huggingface_model.js +23 -15
- package/dist/tools/import_huggingface_model.js.map +1 -1
- package/dist/tools/llm_compile.d.ts +47 -0
- package/dist/tools/llm_compile.js +111 -0
- package/dist/tools/llm_compile.js.map +1 -0
- package/dist/tools/register_byo_bucket.js +6 -1
- package/dist/tools/register_byo_bucket.js.map +1 -1
- package/dist/tools/setup_byo_storage.d.ts +44 -0
- package/dist/tools/setup_byo_storage.js +247 -0
- package/dist/tools/setup_byo_storage.js.map +1 -0
- package/dist/types.d.ts +66 -2
- package/dist/version.d.ts +2 -2
- package/dist/version.js +1 -1
- package/dist/version.js.map +1 -1
- package/package.json +1 -1
package/dist/client.d.ts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import type { APIKeyCreateResponse, APIKeyCreatedResponse, APIKeyListItem, ArtifactResponse, ByoArtifactRegisterRequest, ByoAuditPage, ByoGrant, DeviceListResponse, HFImportJob, HuggingFaceConnectResponse, HuggingFaceIntegrationStatus, InputSpec, Member, Pipeline, PromptPackCreateBody, PromptPackSummary, QaihubIntegration, RunBundle, RunComparison, RunDetail, RunSummary, Workspace, WorkflowTemplate, WorkspaceRole } from "./types.js";
|
|
1
|
+
import type { APIKeyCreateResponse, APIKeyCreatedResponse, APIKeyListItem, ArtifactResponse, ByoArtifactRegisterRequest, ByoAuditPage, ByoGrant, ByoSetupInfo, DeviceListResponse, HFImportJob, HuggingFaceConnectResponse, HuggingFaceIntegrationStatus, InputSpec, LLMCompileJob, Member, Pipeline, PromptPackCreateBody, PromptPackSummary, QaihubIntegration, RunBundle, RunComparison, RunDetail, RunSummary, Workspace, WorkflowTemplate, WorkspaceRole } from "./types.js";
|
|
2
2
|
export interface EdgeGateClientOptions {
|
|
3
3
|
apiUrl: string;
|
|
4
4
|
apiKey: string;
|
|
@@ -41,6 +41,13 @@ export declare class EdgeGateClient {
|
|
|
41
41
|
operator: string;
|
|
42
42
|
threshold: number;
|
|
43
43
|
description?: string;
|
|
44
|
+
} | {
|
|
45
|
+
type: "composition";
|
|
46
|
+
op_type: string;
|
|
47
|
+
compute_unit: "NPU" | "GPU" | "CPU";
|
|
48
|
+
operator: string;
|
|
49
|
+
threshold: number;
|
|
50
|
+
description?: string;
|
|
44
51
|
}>;
|
|
45
52
|
run_policy?: {
|
|
46
53
|
warmup_runs?: number;
|
|
@@ -49,13 +56,29 @@ export declare class EdgeGateClient {
|
|
|
49
56
|
timeout_minutes?: number;
|
|
50
57
|
};
|
|
51
58
|
model_matrix?: Array<{
|
|
52
|
-
artifact_id
|
|
59
|
+
artifact_id?: string;
|
|
53
60
|
label?: string;
|
|
61
|
+
llm_compile_source?: {
|
|
62
|
+
source_artifact_ids: string[];
|
|
63
|
+
sequence_lengths?: number[];
|
|
64
|
+
context_lengths?: number[];
|
|
65
|
+
roles?: string[];
|
|
66
|
+
target_runtime?: string;
|
|
67
|
+
};
|
|
54
68
|
}>;
|
|
55
69
|
/** Optional explicit AI Hub compile input shapes. Omit to let EdgeGate
|
|
56
70
|
* auto-detect from the ONNX file (works for most models). */
|
|
57
71
|
input_specs?: Record<string, InputSpec>;
|
|
58
72
|
}): Promise<Pipeline>;
|
|
73
|
+
submitLLMCompile(workspaceId: string, body: {
|
|
74
|
+
source_artifact_ids: string[];
|
|
75
|
+
device_id: string;
|
|
76
|
+
target_runtime?: string;
|
|
77
|
+
sequence_lengths?: number[];
|
|
78
|
+
context_lengths?: number[];
|
|
79
|
+
roles?: string[];
|
|
80
|
+
}): Promise<LLMCompileJob>;
|
|
81
|
+
getLLMCompileJob(workspaceId: string, compileJobId: string): Promise<LLMCompileJob>;
|
|
59
82
|
listPipelines(workspaceId: string): Promise<Pipeline[]>;
|
|
60
83
|
triggerRun(workspaceId: string, body: {
|
|
61
84
|
pipeline_id: string;
|
|
@@ -105,6 +128,25 @@ export declare class EdgeGateClient {
|
|
|
105
128
|
region: string;
|
|
106
129
|
kms_key_id?: string;
|
|
107
130
|
}): Promise<ByoGrant>;
|
|
131
|
+
/**
|
|
132
|
+
* Phase-1 of the two-phase wizard: register the grant WITHOUT a role_arn.
|
|
133
|
+
* Backend persists with status='pending_role' and returns the External ID
|
|
134
|
+
* the customer needs to write into their trust policy. The orchestrator
|
|
135
|
+
* tool calls this, then asks the agent to run AWS CLI to create the role,
|
|
136
|
+
* then calls attachByoRole below.
|
|
137
|
+
*/
|
|
138
|
+
registerByoPendingGrant(workspaceId: string, body: {
|
|
139
|
+
bucket: string;
|
|
140
|
+
region: string;
|
|
141
|
+
kms_key_id?: string;
|
|
142
|
+
}): Promise<ByoGrant>;
|
|
143
|
+
/**
|
|
144
|
+
* Phase-2 of the two-phase wizard: hand a Role ARN to a pending_role grant.
|
|
145
|
+
* Backend runs the readiness probe and either flips status to 'active' or
|
|
146
|
+
* 'failed' with last_verify_error populated.
|
|
147
|
+
*/
|
|
148
|
+
attachByoRole(workspaceId: string, roleArn: string): Promise<ByoGrant>;
|
|
149
|
+
getByoSetupInfo(workspaceId: string): Promise<ByoSetupInfo>;
|
|
108
150
|
getByoGrant(workspaceId: string): Promise<ByoGrant>;
|
|
109
151
|
verifyByoGrant(workspaceId: string): Promise<ByoGrant>;
|
|
110
152
|
deleteByoGrant(workspaceId: string): Promise<void>;
|
package/dist/client.js
CHANGED
|
@@ -39,6 +39,13 @@ export class EdgeGateClient {
|
|
|
39
39
|
async createPipeline(workspaceId, body) {
|
|
40
40
|
return this.request("POST", `/v1/workspaces/${workspaceId}/pipelines`, body);
|
|
41
41
|
}
|
|
42
|
+
// LLM compile (Plan B) ───────────────────────────────────────────────
|
|
43
|
+
async submitLLMCompile(workspaceId, body) {
|
|
44
|
+
return this.request("POST", `/v1/workspaces/${workspaceId}/llm-compile`, body);
|
|
45
|
+
}
|
|
46
|
+
async getLLMCompileJob(workspaceId, compileJobId) {
|
|
47
|
+
return this.request("GET", `/v1/workspaces/${workspaceId}/llm-compile/${compileJobId}`);
|
|
48
|
+
}
|
|
42
49
|
async listPipelines(workspaceId) {
|
|
43
50
|
return this.request("GET", `/v1/workspaces/${workspaceId}/pipelines`);
|
|
44
51
|
}
|
|
@@ -134,6 +141,27 @@ export class EdgeGateClient {
|
|
|
134
141
|
async registerByoGrant(workspaceId, body) {
|
|
135
142
|
return this.request("POST", `/v1/workspaces/${workspaceId}/byo-storage/grants`, body);
|
|
136
143
|
}
|
|
144
|
+
/**
|
|
145
|
+
* Phase-1 of the two-phase wizard: register the grant WITHOUT a role_arn.
|
|
146
|
+
* Backend persists with status='pending_role' and returns the External ID
|
|
147
|
+
* the customer needs to write into their trust policy. The orchestrator
|
|
148
|
+
* tool calls this, then asks the agent to run AWS CLI to create the role,
|
|
149
|
+
* then calls attachByoRole below.
|
|
150
|
+
*/
|
|
151
|
+
async registerByoPendingGrant(workspaceId, body) {
|
|
152
|
+
return this.request("POST", `/v1/workspaces/${workspaceId}/byo-storage/grants`, body);
|
|
153
|
+
}
|
|
154
|
+
/**
|
|
155
|
+
* Phase-2 of the two-phase wizard: hand a Role ARN to a pending_role grant.
|
|
156
|
+
* Backend runs the readiness probe and either flips status to 'active' or
|
|
157
|
+
* 'failed' with last_verify_error populated.
|
|
158
|
+
*/
|
|
159
|
+
async attachByoRole(workspaceId, roleArn) {
|
|
160
|
+
return this.request("POST", `/v1/workspaces/${workspaceId}/byo-storage/grants/attach-role`, { role_arn: roleArn });
|
|
161
|
+
}
|
|
162
|
+
async getByoSetupInfo(workspaceId) {
|
|
163
|
+
return this.request("GET", `/v1/workspaces/${workspaceId}/byo-storage/setup-info`);
|
|
164
|
+
}
|
|
137
165
|
async getByoGrant(workspaceId) {
|
|
138
166
|
return this.request("GET", `/v1/workspaces/${workspaceId}/byo-storage/grants`);
|
|
139
167
|
}
|
package/dist/client.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"client.js","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;
|
|
1
|
+
{"version":3,"file":"client.js","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAsC1C,MAAM,OAAO,aAAc,SAAQ,KAAK;IAEpB;IACA;IACA;IAHlB,YACkB,MAAc,EACd,MAAc,EACd,GAAW;QAE3B,KAAK,CAAC,YAAY,MAAM,OAAO,GAAG,KAAK,MAAM,EAAE,CAAC,CAAC;QAJjC,WAAM,GAAN,MAAM,CAAQ;QACd,WAAM,GAAN,MAAM,CAAQ;QACd,QAAG,GAAH,GAAG,CAAQ;QAG3B,IAAI,CAAC,IAAI,GAAG,eAAe,CAAC;IAC9B,CAAC;CACF;AAED,MAAM,OAAO,cAAc;IACR,MAAM,CAAS;IACf,MAAM,CAAS;IACf,YAAY,CAAS;IACrB,UAAU,CAAS;IACnB,SAAS,CAAS;IAEnC,YAAY,IAA2B;QACrC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,IAAI,GAAG,CAAC;QAC7C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,MAAM,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,WAAmB;QACpC,OAAO,IAAI,CAAC,OAAO,CAAY,KAAK,EAAE,kBAAkB,WAAW,EAAE,CAAC,CAAC;IACzE,CAAC;IACD,KAAK,CAAC,cAAc;QAClB,OAAO,IAAI,CAAC,OAAO,CAAc,KAAK,EAAE,gBAAgB,CAAC,CAAC;IAC5D,CAAC;IACD,KAAK,CAAC,WAAW;QACf,OAAO,IAAI,CAAC,OAAO,CAAqB,KAAK,EAAE,mBAAmB,CAAC,CAAC;IACtE,CAAC;IACD,KAAK,CAAC,YAAY,CAChB,WAAmB,EACnB,IAA2C;QAE3C,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,WAAW,EACxC,IAAI,CACL,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,cAAc,CAClB,WAAmB,EACnB,IAuCC;QAED,OAAO,IAAI,CAAC,OAAO,CAAW,MAAM,EAAE,kBAAkB,WAAW,YAAY,EAAE,IAAI,CAAC,CAAC;IACzF,CAAC;IAED,uEAAuE;IACvE,KAAK,CAAC,gBAAgB,CACpB,WAAmB,EACnB,IAOC;QAED,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,cAAc,EAC3C,IAAI,CACL,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,gBAAgB,CACpB,WAAmB,EACnB,YAAoB;QAEpB,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,gBAAgB,YAAY,EAAE,CAC5D,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,aAAa,CAAC,WAAmB;QACrC,OAAO,IAAI,CAAC,OAAO,CAAa,KAAK,EAAE,kBAAkB,WAAW,YAAY,CAAC,CAAC;IACpF,CAAC;IACD,KAAK,CAAC,UAAU,CACd,WAAmB,EACnB,IAA2E;QAE3E,OAAO,IAAI,CAAC,OAAO,CAAa,MAAM,EAAE,kBAAkB,WAAW,OAAO,EAAE,IAAI,CAAC,CAAC;IACtF,CAAC;IACD,KAAK,CAAC,MAAM,CAAC,WAAmB,EAAE,KAAa;QAC7C,OAAO,IAAI,CAAC,OAAO,CAAY,KAAK,EAAE,kBAAkB,WAAW,SAAS,KAAK,EAAE,CAAC,CAAC;IACvF,CAAC;IACD,KAAK,CAAC,QAAQ,CAAC,WAAmB,EAAE,KAAK,GAAG,EAAE;QAC5C,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,eAAe,KAAK,EAAE,CACpD,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,kBAAkB,CACtB,WAAmB,EACnB,UAAkB,EAClB,KAAK,GAAG,EAAE;QAEV,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,qBAAqB,UAAU,UAAU,KAAK,EAAE,CAC9E,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,UAAU,CAAC,WAAmB,EAAE,KAAa;QACjD,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,SAAS,KAAK,OAAO,CACnD,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,YAAY,CAAC,WAAmB,EAAE,KAAa;QACnD,OAAO,IAAI,CAAC,OAAO,CAAY,KAAK,EAAE,kBAAkB,WAAW,SAAS,KAAK,SAAS,CAAC,CAAC;IAC9F,CAAC;IACD,KAAK,CAAC,mBAAmB,CAAC,WAAmB;QAC3C,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,yBAAyB,CACvD,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,sBAAsB,CAC1B,WAAmB,EACnB,IAAkE;QAElE,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,+BAA+B,EAC5D,IAAI,CACL,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,uBAAuB,CAAC,WAAmB,EAAE,KAAa;QAC9D,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,iCAAiC,KAAK,EAAE,CACtE,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,6BAA6B,CACjC,WAAmB,EACnB,KAAa;QAEb,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,2BAA2B,EACxD,EAAE,KAAK,EAAE,CACV,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,4BAA4B,CAChC,WAAmB,EACnB,KAAa;QAEb,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,kCAAkC,EAC/D,EAAE,KAAK,EAAE,CACV,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,yBAAyB,CAC7B,WAAmB;QAEnB,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,2BAA2B,CACzD,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,4BAA4B,CAAC,WAAmB;QACpD,MAAM,IAAI,CAAC,OAAO,CAChB,QAAQ,EACR,kBAAkB,WAAW,2BAA2B,CACzD,CAAC;IACJ,CAAC;IAED,wEAAwE;IACxE,KAAK,CAAC,wBAAwB,CAC5B,WAAmB,EACnB,KAAa;QAEb,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,sBAAsB,EACnD,EAAE,KAAK,EAAE,CACV,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,uBAAuB,CAC3B,WAAmB,EACnB,KAAa;QAEb,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,6BAA6B,EAC1D,EAAE,KAAK,EAAE,CACV,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,oBAAoB,CAAC,WAAmB;QAC5C,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,sBAAsB,CACpD,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,uBAAuB,CAAC,WAAmB;QAC/C,MAAM,IAAI,CAAC,OAAO,CAChB,QAAQ,EACR,kBAAkB,WAAW,sBAAsB,CACpD,CAAC;IACJ,CAAC;IAED,uEAAuE;IACvE,KAAK,CAAC,eAAe,CAAC,IAAY;QAChC,OAAO,IAAI,CAAC,OAAO,CAAY,MAAM,EAAE,gBAAgB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,uEAAuE;IACvE,KAAK,CAAC,WAAW,CAAC,WAAmB;QACnC,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,WAAW,CACzC,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,YAAY,CAChB,WAAmB,EACnB,IAAkD;QAElD,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,WAAW,EACxC,IAAI,CACL,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,YAAY,CAAC,WAAmB,EAAE,KAAa;QACnD,MAAM,IAAI,CAAC,OAAO,CAChB,QAAQ,EACR,kBAAkB,WAAW,aAAa,KAAK,EAAE,CAClD,CAAC;IACJ,CAAC;IAED,uEAAuE;IACvE,KAAK,CAAC,WAAW,CAAC,WAAmB;QACnC,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,UAAU,CACxC,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,SAAS,CACb,WAAmB,EACnB,IAAiD;QAEjD,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,UAAU,EACvC,IAAI,CACL,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,gBAAgB,CACpB,WAAmB,EACnB,MAAc,EACd,IAAmB;QAEnB,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,YAAY,MAAM,EAAE,EACjD,EAAE,IAAI,EAAE,CACT,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,YAAY,CAAC,WAAmB,EAAE,MAAc;QACpD,MAAM,IAAI,CAAC,OAAO,CAChB,QAAQ,EACR,kBAAkB,WAAW,YAAY,MAAM,EAAE,CAClD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,WAAmB;QACvC,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,cAAc,CAC5C,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,gBAAgB,CACpB,WAAmB,EACnB,IAA0B;QAE1B,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,cAAc,EAC3C,IAAI,CACL,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,iBAAiB,CACrB,WAAmB,EACnB,YAAoB,EACpB,OAAe;QAEf,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,gBAAgB,YAAY,IAAI,OAAO,UAAU,CAC/E,CAAC;IACJ,CAAC;IAED,wEAAwE;IACxE,KAAK,CAAC,gBAAgB,CACpB,WAAmB,EACnB,IAKC;QAED,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,qBAAqB,EAClD,IAAI,CACL,CAAC;IACJ,CAAC;IACD;;;;;;OAMG;IACH,KAAK,CAAC,uBAAuB,CAC3B,WAAmB,EACnB,IAA6D;QAE7D,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,qBAAqB,EAClD,IAAI,CACL,CAAC;IACJ,CAAC;IACD;;;;OAIG;IACH,KAAK,CAAC,aAAa,CACjB,WAAmB,EACnB,OAAe;QAEf,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,iCAAiC,EAC9D,EAAE,QAAQ,EAAE,OAAO,EAAE,CACtB,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,eAAe,CAAC,WAAmB;QACvC,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,yBAAyB,CACvD,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,WAAW,CAAC,WAAmB;QACnC,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,qBAAqB,CACnD,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,cAAc,CAAC,WAAmB;QACtC,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,4BAA4B,CAC1D,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,cAAc,CAAC,WAAmB;QACtC,MAAM,IAAI,CAAC,OAAO,CAChB,QAAQ,EACR,kBAAkB,WAAW,qBAAqB,CACnD,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,mBAAmB,CACvB,WAAmB,EACnB,IAAgC;QAEhC,OAAO,IAAI,CAAC,OAAO,CACjB,MAAM,EACN,kBAAkB,WAAW,gBAAgB,EAC7C,IAAI,CACL,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,WAAW,CACf,WAAmB,EACnB,SAMI,EAAE;QAEN,MAAM,EAAE,GAAG,IAAI,eAAe,EAAE,CAAC;QACjC,IAAI,MAAM,CAAC,WAAW;YAAE,EAAE,CAAC,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QAClE,IAAI,MAAM,CAAC,MAAM;YAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QACnD,IAAI,MAAM,CAAC,KAAK;YAAE,EAAE,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAChD,IAAI,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ;YAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAC/E,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;YAAE,EAAE,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACxD,OAAO,IAAI,CAAC,OAAO,CACjB,KAAK,EACL,kBAAkB,WAAW,qBAAqB,MAAM,EAAE,CAC3D,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,OAAO,CACnB,MAAyC,EACzC,IAAY,EACZ,IAAc;QAEd,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC;QACpC,MAAM,YAAY,GAAG,MAAM,KAAK,KAAK,CAAC;QACtC,MAAM,QAAQ,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,QAAQ,EAAE,OAAO,EAAE,EAAE,CAAC;YACrD,MAAM,OAAO,GAA2B;gBACtC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;gBACtC,YAAY,EAAE,UAAU;aACzB,CAAC;YACF,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACvB,OAAO,CAAC,cAAc,CAAC,GAAG,kBAAkB,CAAC;YAC/C,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAC5B,MAAM;gBACN,OAAO;gBACP,IAAI,EAAE,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;gBAC3D,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC;aAC5C,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;gBACZ,OAAO,IAAS,CAAC;YACnB,CAAC;YACD,MAAM,MAAM,GACV,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,QAAQ,IAAI,IAAI;gBAClD,CAAC,CAAC,MAAM,CAAE,IAA4B,CAAC,MAAM,CAAC;gBAC9C,CAAC,CAAC,IAAI,IAAI,eAAe,CAAC;YAC9B,IAAI,IAAI,CAAC,MAAM,IAAI,GAAG,IAAI,YAAY,IAAI,OAAO,GAAG,QAAQ,EAAE,CAAC;gBAC7D,MAAM,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAC/B,SAAS;YACX,CAAC;YACD,MAAM,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;QACpD,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;CACF;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,IAAI,CAAC;QAAC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAAC,CAAC;IAAC,MAAM,CAAC;QAAC,OAAO,IAAI,CAAC;IAAC,CAAC;AACzD,CAAC;AACD,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;AAC/C,CAAC"}
|
package/dist/server.js
CHANGED
|
@@ -33,10 +33,14 @@ import { inviteMemberHandler, inviteMemberInputSchema, } from "./tools/invite_me
|
|
|
33
33
|
import { changeMemberRoleHandler, changeMemberRoleInputSchema, } from "./tools/change_member_role.js";
|
|
34
34
|
import { removeMemberHandler, removeMemberInputSchema, } from "./tools/remove_member.js";
|
|
35
35
|
import { registerByoBucketHandler, registerByoBucketInputSchema, } from "./tools/register_byo_bucket.js";
|
|
36
|
+
import { setupByoStorageHandler, setupByoStorageInputSchema, } from "./tools/setup_byo_storage.js";
|
|
37
|
+
import { attachByoRoleHandler, attachByoRoleInputSchema, } from "./tools/attach_byo_role.js";
|
|
36
38
|
import { checkByoBucketHandler, checkByoBucketInputSchema, } from "./tools/check_byo_bucket.js";
|
|
37
39
|
import { registerByoArtifactHandler, registerByoArtifactInputSchema, } from "./tools/register_byo_artifact.js";
|
|
38
40
|
import { disconnectByoBucketHandler, disconnectByoBucketInputSchema, } from "./tools/disconnect_byo_bucket.js";
|
|
39
41
|
import { getByoAuditHandler, getByoAuditInputSchema, } from "./tools/get_byo_audit.js";
|
|
42
|
+
import { llmCompileHandler, llmCompileInputSchema, } from "./tools/llm_compile.js";
|
|
43
|
+
import { checkLLMCompileStatusHandler, checkLLMCompileStatusInputSchema, } from "./tools/check_llm_compile_status.js";
|
|
40
44
|
const TOOLS = [
|
|
41
45
|
{
|
|
42
46
|
name: "edgegate_setup_workspace",
|
|
@@ -48,7 +52,12 @@ const TOOLS = [
|
|
|
48
52
|
{
|
|
49
53
|
name: "edgegate_create_pipeline",
|
|
50
54
|
description: "Create a new EdgeGate regression pipeline. Define which model(s), which device(s), " +
|
|
51
|
-
"and which gates (e.g. inference_time_ms ≤ 10) the pipeline will enforce."
|
|
55
|
+
"and which gates (e.g. inference_time_ms ≤ 10) the pipeline will enforce. " +
|
|
56
|
+
"For LLMs: set llm_compile_source on a model instead of artifact_id. EdgeGate will " +
|
|
57
|
+
"compile + link via AI Hub on first run; subsequent runs reuse the cached composite. " +
|
|
58
|
+
"ttft_ms + tps gates work; both are derived from per-component profile (prompt-role " +
|
|
59
|
+
"inference_time → TTFT; 1000/token-role inference_time → TPS). Each LLM gate run = " +
|
|
60
|
+
"3 AI Hub profile jobs (one per component) ≈ 3× CV cost.",
|
|
52
61
|
schema: createPipelineInputSchema,
|
|
53
62
|
handler: createPipelineHandler,
|
|
54
63
|
},
|
|
@@ -259,14 +268,39 @@ const TOOLS = [
|
|
|
259
268
|
schema: removeMemberInputSchema,
|
|
260
269
|
handler: removeMemberHandler,
|
|
261
270
|
},
|
|
271
|
+
{
|
|
272
|
+
name: "edgegate_setup_byo_storage",
|
|
273
|
+
description: "Enterprise only. Zero-friction BYO storage setup — creates a pending grant " +
|
|
274
|
+
"with EdgeGate (returns the External ID) and returns the exact AWS CLI " +
|
|
275
|
+
"commands the agent should run to create the IAM role in the customer's " +
|
|
276
|
+
"AWS account. Pair with edgegate_attach_byo_role to finalize. Prefer this " +
|
|
277
|
+
"over edgegate_register_byo_bucket for new setups — the agent doesn't have " +
|
|
278
|
+
"to figure out the trust policy or guess at role names. Requires owner role. " +
|
|
279
|
+
"402 from non-Enterprise workspaces; if a non-pending grant exists, instructs " +
|
|
280
|
+
"the agent to disconnect first.",
|
|
281
|
+
schema: setupByoStorageInputSchema,
|
|
282
|
+
handler: setupByoStorageHandler,
|
|
283
|
+
},
|
|
284
|
+
{
|
|
285
|
+
name: "edgegate_attach_byo_role",
|
|
286
|
+
description: "Phase 2 of edgegate_setup_byo_storage. Hands the freshly-created Role ARN " +
|
|
287
|
+
"back to EdgeGate, which runs sts:AssumeRole + a deny-by-default HEAD " +
|
|
288
|
+
"probe to verify the trust + permission policies are correct. Flips the " +
|
|
289
|
+
"grant to 'active' on success, or returns a typed BYO_* error with a " +
|
|
290
|
+
"checklist of common misconfigurations on failure. Re-callable with the " +
|
|
291
|
+
"same role_arn after fixing IAM.",
|
|
292
|
+
schema: attachByoRoleInputSchema,
|
|
293
|
+
handler: attachByoRoleHandler,
|
|
294
|
+
},
|
|
262
295
|
{
|
|
263
296
|
name: "edgegate_register_byo_bucket",
|
|
264
|
-
description: "Enterprise only.
|
|
265
|
-
"
|
|
266
|
-
"
|
|
267
|
-
"
|
|
268
|
-
"
|
|
269
|
-
"
|
|
297
|
+
description: "Enterprise only. **Use edgegate_setup_byo_storage instead for new setups** — " +
|
|
298
|
+
"this tool only works if you already have an IAM role and just want to register " +
|
|
299
|
+
"its ARN. Registers the workspace's customer-owned S3 bucket + IAM role as a BYO " +
|
|
300
|
+
"storage grant. EdgeGate's workers will AssumeRole into your AWS account to read " +
|
|
301
|
+
"model bytes — they never leave your account. Returns the External ID you must " +
|
|
302
|
+
"add to your role's trust policy. Requires owner role. 402 from non-Enterprise " +
|
|
303
|
+
"workspaces; 409 if a grant already exists.",
|
|
270
304
|
schema: registerByoBucketInputSchema,
|
|
271
305
|
handler: registerByoBucketHandler,
|
|
272
306
|
},
|
|
@@ -298,6 +332,25 @@ const TOOLS = [
|
|
|
298
332
|
schema: disconnectByoBucketInputSchema,
|
|
299
333
|
handler: disconnectByoBucketHandler,
|
|
300
334
|
},
|
|
335
|
+
{
|
|
336
|
+
name: "edgegate_llm_compile",
|
|
337
|
+
description: "Submit a multi-component LLM compile + link job via Qualcomm AI Hub. Returns a " +
|
|
338
|
+
"compile_job_id; poll with edgegate_check_llm_compile_status. Spend is gated by " +
|
|
339
|
+
"the workspace's monthly LLM compile cap (default 100/mo Pro tier). Each compile " +
|
|
340
|
+
"produces a composite QNN_DLC linked model + 3 component artifacts (prompt / token / " +
|
|
341
|
+
"kv_cache). Compile target_runtime is QNN_DLC under the hood despite the genie " +
|
|
342
|
+
"label — the label is for downstream profile dispatch hints only.",
|
|
343
|
+
schema: llmCompileInputSchema,
|
|
344
|
+
handler: llmCompileHandler,
|
|
345
|
+
},
|
|
346
|
+
{
|
|
347
|
+
name: "edgegate_check_llm_compile_status",
|
|
348
|
+
description: "Poll an LLM compile job. Returns status (queued|running|completed|failed), " +
|
|
349
|
+
"composite_artifact_id when complete, error_detail when failed, and " +
|
|
350
|
+
"progress.compile_jobs_done / total.",
|
|
351
|
+
schema: checkLLMCompileStatusInputSchema,
|
|
352
|
+
handler: checkLLMCompileStatusHandler,
|
|
353
|
+
},
|
|
301
354
|
{
|
|
302
355
|
name: "edgegate_get_byo_audit",
|
|
303
356
|
description: "Fetch the workspace's append-only BYO storage audit log (every AssumeRole, " +
|
package/dist/server.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GACvB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,OAAO,EAAE,qBAAqB,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AAC9F,OAAO,EAAE,qBAAqB,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AAC9F,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACrF,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EAAE,qBAAqB,EAAE,yBAAyB,EAAE,MAAM,6BAA6B,CAAC;AAC/F,OAAO,EACL,wBAAwB,EACxB,4BAA4B,GAC7B,MAAM,gCAAgC,CAAC;AACxC,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,sBAAsB,EACtB,0BAA0B,GAC3B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,6BAA6B,EAC7B,iCAAiC,GAClC,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EACL,sBAAsB,EACtB,0BAA0B,GAC3B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,uBAAuB,EACvB,2BAA2B,GAC5B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,wBAAwB,EACxB,4BAA4B,GAC7B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,yBAAyB,EACzB,6BAA6B,GAC9B,MAAM,gCAAgC,CAAC;AACxC,OAAO,EACL,4BAA4B,EAC5B,gCAAgC,GACjC,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EACL,gCAAgC,EAChC,oCAAoC,GACrC,MAAM,wCAAwC,CAAC;AAChD,OAAO,EACL,oBAAoB,EACpB,wBAAwB,GACzB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,2BAA2B,EAC3B,+BAA+B,GAChC,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EACL,uBAAuB,EACvB,2BAA2B,GAC5B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,sBAAsB,EACtB,0BAA0B,GAC3B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,uBAAuB,EACvB,2BAA2B,GAC5B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,wBAAwB,EACxB,4BAA4B,GAC7B,MAAM,gCAAgC,CAAC;AACxC,OAAO,EACL,qBAAqB,EACrB,yBAAyB,GAC1B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,0BAA0B,CAAC;
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GACvB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,OAAO,EAAE,qBAAqB,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AAC9F,OAAO,EAAE,qBAAqB,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AAC9F,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACrF,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EAAE,qBAAqB,EAAE,yBAAyB,EAAE,MAAM,6BAA6B,CAAC;AAC/F,OAAO,EACL,wBAAwB,EACxB,4BAA4B,GAC7B,MAAM,gCAAgC,CAAC;AACxC,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,sBAAsB,EACtB,0BAA0B,GAC3B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,6BAA6B,EAC7B,iCAAiC,GAClC,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EACL,sBAAsB,EACtB,0BAA0B,GAC3B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,uBAAuB,EACvB,2BAA2B,GAC5B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,wBAAwB,EACxB,4BAA4B,GAC7B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,yBAAyB,EACzB,6BAA6B,GAC9B,MAAM,gCAAgC,CAAC;AACxC,OAAO,EACL,4BAA4B,EAC5B,gCAAgC,GACjC,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EACL,gCAAgC,EAChC,oCAAoC,GACrC,MAAM,wCAAwC,CAAC;AAChD,OAAO,EACL,oBAAoB,EACpB,wBAAwB,GACzB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,2BAA2B,EAC3B,+BAA+B,GAChC,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EACL,uBAAuB,EACvB,2BAA2B,GAC5B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,sBAAsB,EACtB,0BAA0B,GAC3B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,uBAAuB,EACvB,2BAA2B,GAC5B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,wBAAwB,EACxB,4BAA4B,GAC7B,MAAM,gCAAgC,CAAC;AACxC,OAAO,EACL,sBAAsB,EACtB,0BAA0B,GAC3B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,oBAAoB,EACpB,wBAAwB,GACzB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,qBAAqB,EACrB,yBAAyB,GAC1B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,iBAAiB,EACjB,qBAAqB,GACtB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,4BAA4B,EAC5B,gCAAgC,GACjC,MAAM,qCAAqC,CAAC;AAE7C,MAAM,KAAK,GAAG;IACZ;QACE,IAAI,EAAE,0BAA0B;QAChC,WAAW,EACT,6EAA6E;YAC7E,mFAAmF;QACrF,MAAM,EAAE,yBAAyB;QACjC,OAAO,EAAE,qBAAqB;KAC/B;IACD;QACE,IAAI,EAAE,0BAA0B;QAChC,WAAW,EACT,qFAAqF;YACrF,2EAA2E;YAC3E,oFAAoF;YACpF,sFAAsF;YACtF,qFAAqF;YACrF,oFAAoF;YACpF,yDAAyD;QAC3D,MAAM,EAAE,yBAAyB;QACjC,OAAO,EAAE,qBAAqB;KAC/B;IACD;QACE,IAAI,EAAE,mBAAmB;QACzB,WAAW,EACT,iFAAiF;YACjF,wBAAwB;QAC1B,MAAM,EAAE,kBAAkB;QAC1B,OAAO,EAAE,cAAc;KACxB;IACD;QACE,IAAI,EAAE,uBAAuB;QAC7B,WAAW,EACT,8EAA8E;YAC9E,+BAA+B;QACjC,MAAM,EAAE,sBAAsB;QAC9B,OAAO,EAAE,kBAAkB;KAC5B;IACD;QACE,IAAI,EAAE,qBAAqB;QAC3B,WAAW,EAAE,8EAA8E;QAC3F,MAAM,EAAE,oBAAoB;QAC5B,OAAO,EAAE,gBAAgB;KAC1B;IACD;QACE,IAAI,EAAE,2BAA2B;QACjC,WAAW,EACT,6EAA6E;YAC7E,qBAAqB;QACvB,MAAM,EAAE,yBAAyB;QACjC,OAAO,EAAE,qBAAqB;KAC/B;IACD;QACE,IAAI,EAAE,8BAA8B;QACpC,WAAW,EACT,kFAAkF;YAClF,wBAAwB;QAC1B,MAAM,EAAE,4BAA4B;QACpC,OAAO,EAAE,wBAAwB;KAClC;IACD;QACE,IAAI,EAAE,uBAAuB;QAC7B,WAAW,EACT,2FAA2F;YAC3F,gGAAgG;YAChG,wFAAwF;YACxF,oDAAoD;QACtD,MAAM,EAAE,sBAAsB;QAC9B,OAAO,EAAE,kBAAkB;KAC5B;IACD;QACE,IAAI,EAAE,4BAA4B;QAClC,WAAW,EACT,qFAAqF;YACrF,uEAAuE;YACvE,yEAAyE;QAC3E,MAAM,EAAE,0BAA0B;QAClC,OAAO,EAAE,sBAAsB;KAChC;IACD;QACE,IAAI,EAAE,mCAAmC;QACzC,WAAW,EACT,mFAAmF;YACnF,sFAAsF;YACtF,yFAAyF;YACzF,iFAAiF;QACnF,MAAM,EAAE,iCAAiC;QACzC,OAAO,EAAE,6BAA6B;KACvC;IACD;QACE,IAAI,EAAE,2BAA2B;QACjC,WAAW,EACT,+EAA+E;YAC/E,2EAA2E;YAC3E,oDAAoD;QACtD,MAAM,EAAE,0BAA0B;QAClC,OAAO,EAAE,sBAAsB;KAChC;IACD;QACE,IAAI,EAAE,4BAA4B;QAClC,WAAW,EACT,kFAAkF;YAClF,kFAAkF;YAClF,uFAAuF;YACvF,6BAA6B;QAC/B,MAAM,EAAE,2BAA2B;QACnC,OAAO,EAAE,uBAAuB;KACjC;IACD;QACE,IAAI,EAAE,6BAA6B;QACnC,WAAW,EACT,mFAAmF;YACnF,wEAAwE;YACxE,+EAA+E;YAC/E,oEAAoE;QACtE,MAAM,EAAE,4BAA4B;QACpC,OAAO,EAAE,wBAAwB;KAClC;IACD;QACE,IAAI,EAAE,8BAA8B;QACpC,WAAW,EACT,kFAAkF;YAClF,oFAAoF;YACpF,gFAAgF;YAChF,+EAA+E;YAC/E,kEAAkE;QACpE,MAAM,EAAE,6BAA6B;QACrC,OAAO,EAAE,yBAAyB;KACnC;IACD;QACE,IAAI,EAAE,sCAAsC;QAC5C,WAAW,EACT,gFAAgF;YAChF,gFAAgF;QAClF,MAAM,EAAE,oCAAoC;QAC5C,OAAO,EAAE,gCAAgC;KAC1C;IACD;QACE,IAAI,EAAE,iCAAiC;QACvC,WAAW,EACT,qFAAqF;YACrF,gDAAgD;QAClD,MAAM,EAAE,gCAAgC;QACxC,OAAO,EAAE,4BAA4B;KACtC;IACD;QACE,IAAI,EAAE,yBAAyB;QAC/B,WAAW,EACT,8EAA8E;YAC9E,+EAA+E;YAC/E,2EAA2E;YAC3E,iFAAiF;YACjF,aAAa;QACf,MAAM,EAAE,wBAAwB;QAChC,OAAO,EAAE,oBAAoB;KAC9B;IACD;QACE,IAAI,EAAE,iCAAiC;QACvC,WAAW,EACT,0EAA0E;YAC1E,mEAAmE;QACrE,MAAM,EAAE,+BAA+B;QACvC,OAAO,EAAE,2BAA2B;KACrC;IACD;QACE,IAAI,EAAE,4BAA4B;QAClC,WAAW,EACT,mFAAmF;YACnF,mFAAmF;YACnF,iCAAiC;QACnC,MAAM,EAAE,2BAA2B;QACnC,OAAO,EAAE,uBAAuB;KACjC;IACD;QACE,IAAI,EAAE,2BAA2B;QACjC,WAAW,EACT,+EAA+E;YAC/E,iFAAiF;YACjF,gCAAgC;QAClC,MAAM,EAAE,0BAA0B;QAClC,OAAO,EAAE,sBAAsB;KAChC;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EACT,sFAAsF;YACtF,mDAAmD;QACrD,MAAM,EAAE,sBAAsB;QAC9B,OAAO,EAAE,kBAAkB;KAC5B;IACD;QACE,IAAI,EAAE,yBAAyB;QAC/B,WAAW,EACT,mFAAmF;YACnF,8EAA8E;YAC9E,kDAAkD;QACpD,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,mBAAmB;KAC7B;IACD;QACE,IAAI,EAAE,yBAAyB;QAC/B,WAAW,EACT,8EAA8E;YAC9E,oFAAoF;YACpF,mCAAmC;QACrC,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,mBAAmB;KAC7B;IACD;QACE,IAAI,EAAE,uBAAuB;QAC7B,WAAW,EACT,gFAAgF;YAChF,cAAc;QAChB,MAAM,EAAE,sBAAsB;QAC9B,OAAO,EAAE,kBAAkB;KAC5B;IACD;QACE,IAAI,EAAE,uBAAuB;QAC7B,WAAW,EACT,4EAA4E;YAC5E,0EAA0E;YAC1E,wEAAwE;YACxE,2EAA2E;YAC3E,oBAAoB;QACtB,MAAM,EAAE,sBAAsB;QAC9B,OAAO,EAAE,kBAAkB;KAC5B;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EACT,6EAA6E;YAC7E,4EAA4E;YAC5E,gFAAgF;YAChF,mBAAmB;QACrB,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,mBAAmB;KAC7B;IACD;QACE,IAAI,EAAE,6BAA6B;QACnC,WAAW,EACT,8EAA8E;YAC9E,6DAA6D;QAC/D,MAAM,EAAE,2BAA2B;QACnC,OAAO,EAAE,uBAAuB;KACjC;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EACT,gFAAgF;YAChF,+EAA+E;YAC/E,sBAAsB;QACxB,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,mBAAmB;KAC7B;IACD;QACE,IAAI,EAAE,4BAA4B;QAClC,WAAW,EACT,6EAA6E;YAC7E,wEAAwE;YACxE,yEAAyE;YACzE,2EAA2E;YAC3E,4EAA4E;YAC5E,8EAA8E;YAC9E,+EAA+E;YAC/E,gCAAgC;QAClC,MAAM,EAAE,0BAA0B;QAClC,OAAO,EAAE,sBAAsB;KAChC;IACD;QACE,IAAI,EAAE,0BAA0B;QAChC,WAAW,EACT,4EAA4E;YAC5E,uEAAuE;YACvE,yEAAyE;YACzE,sEAAsE;YACtE,yEAAyE;YACzE,iCAAiC;QACnC,MAAM,EAAE,wBAAwB;QAChC,OAAO,EAAE,oBAAoB;KAC9B;IACD;QACE,IAAI,EAAE,8BAA8B;QACpC,WAAW,EACT,+EAA+E;YAC/E,iFAAiF;YACjF,kFAAkF;YAClF,kFAAkF;YAClF,gFAAgF;YAChF,gFAAgF;YAChF,4CAA4C;QAC9C,MAAM,EAAE,4BAA4B;QACpC,OAAO,EAAE,wBAAwB;KAClC;IACD;QACE,IAAI,EAAE,2BAA2B;QACjC,WAAW,EACT,6EAA6E;YAC7E,8EAA8E;YAC9E,gFAAgF;YAChF,kCAAkC;QACpC,MAAM,EAAE,yBAAyB;QACjC,OAAO,EAAE,qBAAqB;KAC/B;IACD;QACE,IAAI,EAAE,gCAAgC;QACtC,WAAW,EACT,iFAAiF;YACjF,+EAA+E;YAC/E,+EAA+E;YAC/E,iFAAiF;YACjF,gFAAgF;QAClF,MAAM,EAAE,8BAA8B;QACtC,OAAO,EAAE,0BAA0B;KACpC;IACD;QACE,IAAI,EAAE,gCAAgC;QACtC,WAAW,EACT,8EAA8E;YAC9E,4EAA4E;YAC5E,8EAA8E;YAC9E,qDAAqD;QACvD,MAAM,EAAE,8BAA8B;QACtC,OAAO,EAAE,0BAA0B;KACpC;IACD;QACE,IAAI,EAAE,sBAAsB;QAC5B,WAAW,EACT,iFAAiF;YACjF,iFAAiF;YACjF,kFAAkF;YAClF,sFAAsF;YACtF,gFAAgF;YAChF,kEAAkE;QACpE,MAAM,EAAE,qBAAqB;QAC7B,OAAO,EAAE,iBAAiB;KAC3B;IACD;QACE,IAAI,EAAE,mCAAmC;QACzC,WAAW,EACT,6EAA6E;YAC7E,qEAAqE;YACrE,qCAAqC;QACvC,MAAM,EAAE,gCAAgC;QACxC,OAAO,EAAE,4BAA4B;KACtC;IACD;QACE,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EACT,6EAA6E;YAC7E,gFAAgF;YAChF,4EAA4E;YAC5E,kFAAkF;QACpF,MAAM,EAAE,sBAAsB;QAC9B,OAAO,EAAE,kBAAkB;KAC5B;CACO,CAAC;AAEX,SAAS,SAAS;IAChB,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,8BAA8B,CAAC;IAC9E,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CACb,4EAA4E;YAC1E,oEAAoE,CACvE,CAAC;IACJ,CAAC;IACD,OAAO,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;AAChD,CAAC;AAED,KAAK,UAAU,IAAI;IACjB,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,EAAE,EAC1C,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAChC,CAAC;IAEF,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC5D,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;YAC1B,WAAW,EAAE,eAAe,CAAC,CAAC,CAAC,MAAM,CAA4B;SAClE,CAAC,CAAC;KACJ,CAAC,CAAC,CAAC;IAEJ,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;QAC5D,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3D,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;aACtE,CAAC;QACJ,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,yBAAyB,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE;iBACtF;aACF,CAAC;QACJ,CAAC;QACD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,8DAA8D;QAC9D,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,IAAW,CAAQ,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,sCAAsC;IACtC,OAAO,CAAC,KAAK,CAAC,gBAAgB,OAAO,qBAAqB,CAAC,CAAC;AAC9D,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACnB,sCAAsC;IACtC,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* edgegate_attach_byo_role — Phase 2 of edgegate_setup_byo_storage.
|
|
3
|
+
*
|
|
4
|
+
* Pairs with `edgegate_setup_byo_storage`: that tool creates a pending
|
|
5
|
+
* grant and returns AWS CLI commands the agent runs to create the IAM
|
|
6
|
+
* role. This tool takes the resulting Role ARN, hands it to EdgeGate,
|
|
7
|
+
* and surfaces the readiness-probe outcome.
|
|
8
|
+
*
|
|
9
|
+
* Why a separate tool: keeps each tool's responsibility small and lets
|
|
10
|
+
* the agent recover from probe failures by re-running just this tool
|
|
11
|
+
* after fixing the IAM config (instead of re-creating the grant from
|
|
12
|
+
* scratch).
|
|
13
|
+
*/
|
|
14
|
+
import { z } from "zod";
|
|
15
|
+
import { EdgeGateClient } from "../client.js";
|
|
16
|
+
import type { ToolResult } from "./setup_workspace.js";
|
|
17
|
+
export declare const attachByoRoleInputSchema: z.ZodObject<{
|
|
18
|
+
workspace_id: z.ZodString;
|
|
19
|
+
role_arn: z.ZodString;
|
|
20
|
+
}, "strip", z.ZodTypeAny, {
|
|
21
|
+
workspace_id: string;
|
|
22
|
+
role_arn: string;
|
|
23
|
+
}, {
|
|
24
|
+
workspace_id: string;
|
|
25
|
+
role_arn: string;
|
|
26
|
+
}>;
|
|
27
|
+
export type AttachByoRoleInput = z.infer<typeof attachByoRoleInputSchema>;
|
|
28
|
+
export declare function attachByoRoleHandler(client: EdgeGateClient, input: AttachByoRoleInput): Promise<ToolResult>;
|
|
@@ -0,0 +1,96 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* edgegate_attach_byo_role — Phase 2 of edgegate_setup_byo_storage.
|
|
3
|
+
*
|
|
4
|
+
* Pairs with `edgegate_setup_byo_storage`: that tool creates a pending
|
|
5
|
+
* grant and returns AWS CLI commands the agent runs to create the IAM
|
|
6
|
+
* role. This tool takes the resulting Role ARN, hands it to EdgeGate,
|
|
7
|
+
* and surfaces the readiness-probe outcome.
|
|
8
|
+
*
|
|
9
|
+
* Why a separate tool: keeps each tool's responsibility small and lets
|
|
10
|
+
* the agent recover from probe failures by re-running just this tool
|
|
11
|
+
* after fixing the IAM config (instead of re-creating the grant from
|
|
12
|
+
* scratch).
|
|
13
|
+
*/
|
|
14
|
+
import { z } from "zod";
|
|
15
|
+
import { EdgeGateError } from "../client.js";
|
|
16
|
+
export const attachByoRoleInputSchema = z.object({
|
|
17
|
+
workspace_id: z.string().uuid(),
|
|
18
|
+
role_arn: z
|
|
19
|
+
.string()
|
|
20
|
+
.regex(/^arn:aws:iam::\d{12}:role\/.+$/)
|
|
21
|
+
.describe("ARN of the IAM role created in step 3 of edgegate_setup_byo_storage. " +
|
|
22
|
+
"Format: arn:aws:iam::<account-id>:role/<name>. EdgeGate will sts:AssumeRole " +
|
|
23
|
+
"this immediately to verify the trust + permission policies are correct."),
|
|
24
|
+
});
|
|
25
|
+
export async function attachByoRoleHandler(client, input) {
|
|
26
|
+
try {
|
|
27
|
+
const grant = await client.attachByoRole(input.workspace_id, input.role_arn);
|
|
28
|
+
return resultText(grant);
|
|
29
|
+
}
|
|
30
|
+
catch (err) {
|
|
31
|
+
if (err instanceof EdgeGateError) {
|
|
32
|
+
if (err.status === 404) {
|
|
33
|
+
return errorText("No pending BYO grant for this workspace. Call edgegate_setup_byo_storage first to create one.");
|
|
34
|
+
}
|
|
35
|
+
if (err.status === 409) {
|
|
36
|
+
return errorText(`Grant cannot accept a role right now: ${err.detail}. If the grant is already active, no further action is needed. If it's revoked, disconnect first.`);
|
|
37
|
+
}
|
|
38
|
+
return errorText(`EdgeGate returned ${err.status}: ${err.detail}`);
|
|
39
|
+
}
|
|
40
|
+
throw err;
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
function resultText(grant) {
|
|
44
|
+
if (grant.status === "active") {
|
|
45
|
+
return {
|
|
46
|
+
content: [
|
|
47
|
+
{
|
|
48
|
+
type: "text",
|
|
49
|
+
text: [
|
|
50
|
+
`BYO storage **active** for bucket \`s3://${grant.bucket}\`.`,
|
|
51
|
+
``,
|
|
52
|
+
`- role: \`${grant.role_arn}\``,
|
|
53
|
+
`- external_id: \`${grant.external_id}\``,
|
|
54
|
+
grant.kms_key_id ? `- kms key: \`${grant.kms_key_id}\`` : `- kms key: (none)`,
|
|
55
|
+
`- verified: \`${grant.last_verified_at ?? "just now"}\``,
|
|
56
|
+
``,
|
|
57
|
+
`Pipelines can now reference s3:// URIs in this bucket via \`edgegate_register_byo_artifact\`.`,
|
|
58
|
+
].join("\n"),
|
|
59
|
+
},
|
|
60
|
+
],
|
|
61
|
+
};
|
|
62
|
+
}
|
|
63
|
+
// status='failed' is recoverable — role_arn is persisted, customer can fix
|
|
64
|
+
// IAM and re-call this tool to retry the probe.
|
|
65
|
+
return {
|
|
66
|
+
isError: true,
|
|
67
|
+
content: [
|
|
68
|
+
{
|
|
69
|
+
type: "text",
|
|
70
|
+
text: [
|
|
71
|
+
`BYO storage readiness probe **failed**.`,
|
|
72
|
+
``,
|
|
73
|
+
`- role: \`${grant.role_arn ?? "(not attached)"}\``,
|
|
74
|
+
`- last error: \`${grant.last_verify_error ?? "unknown"}\``,
|
|
75
|
+
``,
|
|
76
|
+
`Common causes:`,
|
|
77
|
+
`- Trust policy is missing the External ID Condition or has the wrong value.`,
|
|
78
|
+
`- Permission policy doesn't grant s3:GetObject + s3:HeadObject on \`arn:aws:s3:::${grant.bucket}/*\`.`,
|
|
79
|
+
grant.kms_key_id
|
|
80
|
+
? `- Missing kms:Decrypt grant on \`${grant.kms_key_id}\`.`
|
|
81
|
+
: `- Bucket is encrypted but no kms_key_id was provided to edgegate_setup_byo_storage.`,
|
|
82
|
+
`- IAM eventual consistency — wait 10s and re-call this tool with the same role_arn.`,
|
|
83
|
+
``,
|
|
84
|
+
`Re-call this tool with the same role_arn after fixing the underlying issue.`,
|
|
85
|
+
].join("\n"),
|
|
86
|
+
},
|
|
87
|
+
],
|
|
88
|
+
};
|
|
89
|
+
}
|
|
90
|
+
function errorText(message) {
|
|
91
|
+
return {
|
|
92
|
+
isError: true,
|
|
93
|
+
content: [{ type: "text", text: message }],
|
|
94
|
+
};
|
|
95
|
+
}
|
|
96
|
+
//# sourceMappingURL=attach_byo_role.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"attach_byo_role.js","sourceRoot":"","sources":["../../src/tools/attach_byo_role.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAkB,aAAa,EAAE,MAAM,cAAc,CAAC;AAI7D,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/C,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;IAC/B,QAAQ,EAAE,CAAC;SACR,MAAM,EAAE;SACR,KAAK,CAAC,gCAAgC,CAAC;SACvC,QAAQ,CACP,uEAAuE;QACrE,8EAA8E;QAC9E,yEAAyE,CAC5E;CACJ,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,MAAsB,EACtB,KAAyB;IAEzB,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC7E,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,aAAa,EAAE,CAAC;YACjC,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO,SAAS,CACd,+FAA+F,CAChG,CAAC;YACJ,CAAC;YACD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO,SAAS,CACd,yCAAyC,GAAG,CAAC,MAAM,mGAAmG,CACvJ,CAAC;YACJ,CAAC;YACD,OAAO,SAAS,CAAC,qBAAqB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;QACrE,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,KAAe;IACjC,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE;wBACJ,4CAA4C,KAAK,CAAC,MAAM,KAAK;wBAC7D,EAAE;wBACF,aAAa,KAAK,CAAC,QAAQ,IAAI;wBAC/B,oBAAoB,KAAK,CAAC,WAAW,IAAI;wBACzC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,gBAAgB,KAAK,CAAC,UAAU,IAAI,CAAC,CAAC,CAAC,mBAAmB;wBAC7E,iBAAiB,KAAK,CAAC,gBAAgB,IAAI,UAAU,IAAI;wBACzD,EAAE;wBACF,+FAA+F;qBAChG,CAAC,IAAI,CAAC,IAAI,CAAC;iBACb;aACF;SACF,CAAC;IACJ,CAAC;IACD,2EAA2E;IAC3E,gDAAgD;IAChD,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE;oBACJ,yCAAyC;oBACzC,EAAE;oBACF,aAAa,KAAK,CAAC,QAAQ,IAAI,gBAAgB,IAAI;oBACnD,mBAAmB,KAAK,CAAC,iBAAiB,IAAI,SAAS,IAAI;oBAC3D,EAAE;oBACF,gBAAgB;oBAChB,6EAA6E;oBAC7E,oFAAoF,KAAK,CAAC,MAAM,OAAO;oBACvG,KAAK,CAAC,UAAU;wBACd,CAAC,CAAC,oCAAoC,KAAK,CAAC,UAAU,KAAK;wBAC3D,CAAC,CAAC,qFAAqF;oBACzF,qFAAqF;oBACrF,EAAE;oBACF,6EAA6E;iBAC9E,CAAC,IAAI,CAAC,IAAI,CAAC;aACb;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,OAAe;IAChC,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;KAC3C,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
import { EdgeGateClient } from "../client.js";
|
|
3
|
+
import type { ToolResult } from "./setup_workspace.js";
|
|
4
|
+
export declare const checkLLMCompileStatusInputSchema: z.ZodObject<{
|
|
5
|
+
workspace_id: z.ZodString;
|
|
6
|
+
compile_job_id: z.ZodString;
|
|
7
|
+
}, "strip", z.ZodTypeAny, {
|
|
8
|
+
workspace_id: string;
|
|
9
|
+
compile_job_id: string;
|
|
10
|
+
}, {
|
|
11
|
+
workspace_id: string;
|
|
12
|
+
compile_job_id: string;
|
|
13
|
+
}>;
|
|
14
|
+
export type CheckLLMCompileStatusInput = z.infer<typeof checkLLMCompileStatusInputSchema>;
|
|
15
|
+
export declare function checkLLMCompileStatusHandler(client: EdgeGateClient, input: CheckLLMCompileStatusInput): Promise<ToolResult>;
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
import { EdgeGateError } from "../client.js";
|
|
3
|
+
export const checkLLMCompileStatusInputSchema = z.object({
|
|
4
|
+
workspace_id: z.string().uuid(),
|
|
5
|
+
compile_job_id: z.string().uuid(),
|
|
6
|
+
});
|
|
7
|
+
export async function checkLLMCompileStatusHandler(client, input) {
|
|
8
|
+
try {
|
|
9
|
+
const job = await client.getLLMCompileJob(input.workspace_id, input.compile_job_id);
|
|
10
|
+
return { content: [{ type: "text", text: format(job) }] };
|
|
11
|
+
}
|
|
12
|
+
catch (err) {
|
|
13
|
+
if (err instanceof EdgeGateError) {
|
|
14
|
+
return {
|
|
15
|
+
isError: true,
|
|
16
|
+
content: [
|
|
17
|
+
{
|
|
18
|
+
type: "text",
|
|
19
|
+
text: err.status === 404
|
|
20
|
+
? `No LLM compile job ${input.compile_job_id} in workspace ${input.workspace_id}.`
|
|
21
|
+
: `EdgeGate returned ${err.status}: ${err.detail}`,
|
|
22
|
+
},
|
|
23
|
+
],
|
|
24
|
+
};
|
|
25
|
+
}
|
|
26
|
+
throw err;
|
|
27
|
+
}
|
|
28
|
+
}
|
|
29
|
+
function format(job) {
|
|
30
|
+
const lines = [
|
|
31
|
+
`LLM compile job **${job.compile_job_id}**: ${job.status.toUpperCase()}`,
|
|
32
|
+
`- progress: ${job.progress.compile_jobs_done}/${job.progress.compile_jobs_total} compile jobs done`,
|
|
33
|
+
];
|
|
34
|
+
if (job.device_id)
|
|
35
|
+
lines.push(`- device: ${job.device_id}`);
|
|
36
|
+
if (job.target_runtime)
|
|
37
|
+
lines.push(`- target_runtime: ${job.target_runtime}`);
|
|
38
|
+
if (job.status === "completed") {
|
|
39
|
+
lines.push(``, `Composite artifact ready:`, `- composite_artifact_id: ${job.composite_artifact_id}`, ``, `Reference it from a pipeline via:`, ` models: [{ name: "...", artifact_id: "${job.composite_artifact_id}" }]`);
|
|
40
|
+
}
|
|
41
|
+
else if (job.status === "failed") {
|
|
42
|
+
lines.push(``, `Error: ${job.error_detail ?? "unknown"}`);
|
|
43
|
+
}
|
|
44
|
+
else {
|
|
45
|
+
lines.push(``, `Still in flight. Re-poll in ~30s.`);
|
|
46
|
+
}
|
|
47
|
+
return lines.join("\n");
|
|
48
|
+
}
|
|
49
|
+
//# sourceMappingURL=check_llm_compile_status.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"check_llm_compile_status.js","sourceRoot":"","sources":["../../src/tools/check_llm_compile_status.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAkB,aAAa,EAAE,MAAM,cAAc,CAAC;AAI7D,MAAM,CAAC,MAAM,gCAAgC,GAAG,CAAC,CAAC,MAAM,CAAC;IACvD,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;IAC/B,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;CAClC,CAAC,CAAC;AAMH,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,MAAsB,EACtB,KAAiC;IAEjC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,gBAAgB,CACvC,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,cAAc,CACrB,CAAC;QACF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;IAC5D,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,aAAa,EAAE,CAAC;YACjC,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EACF,GAAG,CAAC,MAAM,KAAK,GAAG;4BAChB,CAAC,CAAC,sBAAsB,KAAK,CAAC,cAAc,iBAAiB,KAAK,CAAC,YAAY,GAAG;4BAClF,CAAC,CAAC,qBAAqB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE;qBACvD;iBACF;aACF,CAAC;QACJ,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,MAAM,CAAC,GAAkB;IAChC,MAAM,KAAK,GAAa;QACtB,qBAAqB,GAAG,CAAC,cAAc,OAAO,GAAG,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE;QACxE,eAAe,GAAG,CAAC,QAAQ,CAAC,iBAAiB,IAAI,GAAG,CAAC,QAAQ,CAAC,kBAAkB,oBAAoB;KACrG,CAAC;IACF,IAAI,GAAG,CAAC,SAAS;QAAE,KAAK,CAAC,IAAI,CAAC,aAAa,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;IAC5D,IAAI,GAAG,CAAC,cAAc;QAAE,KAAK,CAAC,IAAI,CAAC,qBAAqB,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC;IAE9E,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;QAC/B,KAAK,CAAC,IAAI,CACR,EAAE,EACF,2BAA2B,EAC3B,4BAA4B,GAAG,CAAC,qBAAqB,EAAE,EACvD,EAAE,EACF,mCAAmC,EACnC,2CAA2C,GAAG,CAAC,qBAAqB,MAAM,CAC3E,CAAC;IACJ,CAAC;SAAM,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QACnC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,GAAG,CAAC,YAAY,IAAI,SAAS,EAAE,CAAC,CAAC;IAC5D,CAAC;SAAM,CAAC;QACN,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,mCAAmC,CAAC,CAAC;IACtD,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}
|