@integrity-labs/cloud-broker 0.4.1 → 0.4.4

package/dist/index.js

@@ -20995,6 +20995,7 @@ function makeBrokerError(status, message, detail) {
 var BrokerClient = class {
   host;
   agentId;
+  runId;
   apiKey;
   fetchImpl;
   token;
@@ -21003,6 +21004,7 @@ var BrokerClient = class {
   constructor(config2) {
     this.host = config2.host.replace(/\/+$/, "");
     this.agentId = config2.agentId;
+    this.runId = config2.runId;
     this.apiKey = config2.apiKey;
     this.fetchImpl = config2.fetchImpl ?? fetch;
     this.token = config2.initialToken ?? "";
@@ -21088,25 +21090,38 @@ var BrokerClient = class {
   }
   // ────────────────────────── tool-shaped wrappers ──────────────────────────
   describeScope(args) {
-    const query = { account_id: args.account_id };
-    if (this.agentId) query.agent_id = this.agentId;
+    if (!this.agentId) {
+      throw makeBrokerError(400, "BrokerClient.describeScope requires agentId \u2014 pass it in BrokerClientConfig");
+    }
     return this.request(
       "GET",
       "/aws/scope",
-      { query }
+      { query: { account_id: args.account_id, agent_id: this.agentId } }
     );
   }
   previewRequest(args) {
+    if (!this.agentId) {
+      throw makeBrokerError(400, "BrokerClient.previewRequest requires agentId \u2014 pass it in BrokerClientConfig");
+    }
     const body = { ...args };
-    if (this.agentId && body.agent_id === void 0) body.agent_id = this.agentId;
+    if (body.agent_id === void 0) body.agent_id = this.agentId;
     return this.request(
       "POST",
       "/aws/scope/preview",
       { body }
     );
   }
-  requestAccess(args) {
-    return this.request("POST", "/aws/grants", { body: args });
+  async requestAccess(args) {
+    const agentId = args.agent_id ?? this.agentId;
+    const runId = args.run_id ?? this.runId;
+    if (!agentId) {
+      throw makeBrokerError(400, "BrokerClient.requestAccess requires agent_id (pass it in args, or set agentId on BrokerClientConfig)");
+    }
+    if (!runId) {
+      throw makeBrokerError(400, "BrokerClient.requestAccess requires run_id (pass it in args, or set runId on BrokerClientConfig)");
+    }
+    const body = { ...args, agent_id: agentId, run_id: runId };
+    return this.request("POST", "/aws/grants", { body });
   }
   pollGrant(args) {
     return this.request("GET", `/aws/grants/${encodeURIComponent(args.grant_id)}`);
@@ -21136,8 +21151,16 @@ var previewRequestSchema = external_exports.object({
   ttl_seconds: ttlSecondsSchema
 });
 var requestAccessSchema = external_exports.object({
-  agent_id: external_exports.string().uuid("agent_id must be a UUID"),
-  run_id: external_exports.string().uuid("run_id must be a UUID"),
+  // ENG-4746: agent_id and run_id are optional. The MCP server fills them
+  // from AGT_AGENT_ID / AGT_RUN_ID env at provision time, so the agent
+  // doesn't need to know its own UUID. Pass them explicitly only to
+  // attribute the grant to a different agent or run (e.g. delegated work).
+  agent_id: external_exports.string().uuid("agent_id must be a UUID").optional().describe(
+    "Optional. Defaults to the host MCP env (AGT_AGENT_ID). Pass explicitly only to attribute the grant to a different agent."
+  ),
+  run_id: external_exports.string().uuid("run_id must be a UUID").optional().describe(
+    "Optional. Defaults to the host MCP env (AGT_RUN_ID). Pass explicitly only to attribute the grant to a different run."
+  ),
   account_id: accountIdSchema,
   actions: actionListSchema,
   resources: resourceListSchema,
@@ -21163,6 +21186,7 @@ var releaseAccessShape = releaseAccessSchema.shape;
 // src/index.ts
 var AGT_HOST = process.env.AGT_HOST;
 var AGT_AGENT_ID = process.env.AGT_AGENT_ID;
+var AGT_RUN_ID = process.env.AGT_RUN_ID ?? "";
 var AGT_TOKEN = process.env.AGT_TOKEN ?? "";
 var AGT_API_KEY = process.env.AGT_API_KEY ?? "";
 if (!AGT_HOST || !AGT_AGENT_ID || !AGT_TOKEN && !AGT_API_KEY) {
@@ -21179,6 +21203,10 @@ if (process.env.AGT_TEAM_SLUG) {
 var broker = new BrokerClient({
   host: AGT_HOST,
   agentId: AGT_AGENT_ID,
+  // ENG-4746: thread AGT_RUN_ID through so request_access can default it.
+  // Empty string is treated as absent — the broker client only fills runId
+  // on requestAccess if a non-empty value is provided.
+  runId: AGT_RUN_ID || void 0,
   initialToken: AGT_TOKEN || void 0,
   apiKey: AGT_API_KEY || void 0
 });
@@ -21221,7 +21249,7 @@ server.tool(
 );
 server.tool(
   "request_access",
-  'Request scoped, TTL-bounded AWS credentials for the current task. Returns { grant_id, status, secret_ref?, expires_at?, denial_reason? }. status="active" means credentials are ready; status="pending" means a human approver was paged and you should poll_grant; status="denied" means the request was rejected (denial_reason explains).',
+  'Request scoped, TTL-bounded AWS credentials for the current task. agent_id and run_id are optional \u2014 the broker fills them from the host MCP env (AGT_AGENT_ID / AGT_RUN_ID). Only pass them explicitly to attribute the grant to a different agent or run. Returns { grant_id, status, secret_ref?, expires_at?, denial_reason?, notification_status?, notification_failure_reason? }. status="active" means credentials are ready, use them now. status="denied" means the request was rejected (denial_reason explains). status="pending" means approval is still outstanding \u2014 DO NOT poll. The broker pushes the resolution to you via direct-chat the moment a human approves or denies, so save the grant_id, return control, and resume when the inbound message arrives. Only check notification_status if you need to flag a setup issue to the user: "sent" means a human was paged; "failed" or "not_attempted" means no human was paged (typically channel_not_found because the approval-bot is not a member of the configured channel) \u2014 quote grant_id + notification_failure_reason and recommend manual escalation. poll_grant exists as an escape hatch for explicit re-checks but the autonomous flow does not need it.',
   requestAccessShape,
   async (args) => {
     try {
@@ -21235,7 +21263,7 @@ server.tool(
 );
 server.tool(
   "poll_grant",
-  'Single-shot status check for a grant. Use this when request_access returned status="pending"; back-off ~10s between polls. Returns { grant_id, status, secret_ref?, expires_at?, denial_reason? }.',
+  "Single-shot status check for a grant. The broker pushes resolution updates to you via direct-chat automatically \u2014 you do NOT need to poll in the normal flow. Use this only as an escape hatch: explicit re-check after a notification, or if you suspect a notification was lost (e.g. you got `request_access` returning pending more than ~5 minutes ago and have heard nothing). Returns { grant_id, status, secret_ref?, expires_at?, denial_reason? }.",
   pollGrantShape,
   async (args) => {
     try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@integrity-labs/cloud-broker",
-  "version": "0.4.1",
+  "version": "0.4.4",
   "description": "Cloud Access Broker — MCP server that mints scoped, TTL-bounded cloud credentials per agent task. v1 ships AWS support (request_access, poll_grant, release_access, describe_scope, preview_request — STS AssumeRole under the hood); GCP, Azure, and Cloudflare land alongside in the same package as the broker grows.",
   "type": "module",
   "bin": {