@agent-os-sdk/client 0.3.11 → 0.3.14

package/dist/client/AgentOsClient.d.ts

@@ -59,6 +59,8 @@ import { MetricsModule } from "../modules/metrics.js";
 import { GraphsModule } from "../modules/graphs.js";
 import { CatalogModule } from "../modules/catalog.js";
 import { ApprovalsModule } from "../modules/approvals.js";
+import { ApiTokensModule } from "../modules/apiTokens.js";
+import { MembershipsModule } from "../modules/memberships.js";
 export type { AgentOsClientOptions, AuthProvider } from "./auth.js";
 export { isApiTokenAuth, isJwtAuth } from "./auth.js";
 export declare class AgentOsClient {
@@ -102,10 +104,16 @@ export declare class AgentOsClient {
         get: EvaluationModule["getExperiment"];
         create: EvaluationModule["createExperiment"];
     };
+    readonly apiTokens: ApiTokensModule;
+    readonly memberships: MembershipsModule;
     readonly agentVersions: {
         list: (agentId: string) => ReturnType<AgentsModule["listVersions"]>;
         get: (agentId: string, versionId: string) => ReturnType<AgentsModule["getVersion"]>;
         create: (agentId: string, body: Parameters<AgentsModule["createVersion"]>[1]) => ReturnType<AgentsModule["createVersion"]>;
+        createDraft: (agentId: string, body?: Parameters<AgentsModule["createDraft"]>[1]) => ReturnType<AgentsModule["createDraft"]>;
+        getBundle: (agentId: string, versionId: string) => ReturnType<AgentsModule["getBundle"]>;
+        update: (agentId: string, versionId: string, body: Parameters<AgentsModule["updateVersion"]>[2]) => ReturnType<AgentsModule["updateVersion"]>;
+        publish: (agentId: string, versionId: string) => ReturnType<AgentsModule["publishVersion"]>;
     };
     constructor(options: AgentOsClientOptions);
     /**

package/dist/client/AgentOsClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AgentOsClient.d.ts","sourceRoot":"","sources":["../../src/client/AgentOsClient.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AAEH,OAAO,EAAmB,KAAK,SAAS,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EACH,KAAK,oBAAoB,EAK5B,MAAM,WAAW,CAAC;AAEnB,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAG5D,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAG1D,YAAY,EAAE,oBAAoB,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACpE,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAEtD,qBAAa,aAAa;IACtB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAY;IACpC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAe;IACrC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAyB;IAGxD,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC;IAC9B,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;IAC1B,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,SAAS,EAAE,eAAe,CAAC;IACpC,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAC;IAClC,QAAQ,CAAC,WAAW,EAAE,iBAAiB,CAAC;IACxC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,UAAU,EAAE,gBAAgB,CAAC;IAGtC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC;IAC9B,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,YAAY,EAAE,kBAAkB,CAAC;IAC1C,QAAQ,CAAC,UAAU,EAAE,gBAAgB,CAAC;IACtC,QAAQ,CAAC,WAAW,EAAE,iBAAiB,CAAC;IACxC,QAAQ,CAAC,UAAU,EAAE,gBAAgB,CAAC;IACtC,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,GAAG,EAAE,SAAS,CAAC;IACxB,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,GAAG,EAAE,SAAS,CAAC;IACxB,QAAQ,CAAC,GAAG,EAAE,SAAS,CAAC;IACxB,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;IAC1B,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC;IAC9B,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,SAAS,EAAE,eAAe,CAAC;IAGpC,QAAQ,CAAC,WAAW,EAAE;QAClB,IAAI,EAAE,gBAAgB,CAAC,iBAAiB,CAAC,CAAC;QAC1C,GAAG,EAAE,gBAAgB,CAAC,eAAe,CAAC,CAAC;QACvC,MAAM,EAAE,gBAAgB,CAAC,kBAAkB,CAAC,CAAC;KAChD,CAAC;IAEF,QAAQ,CAAC,aAAa,EAAE;QACpB,IAAI,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,UAAU,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC;QACpE,GAAG,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,KAAK,UAAU,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;QACpF,MAAM,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC;KAC9H,CAAC;gBAEU,OAAO,EAAE,oBAAoB;IAoEzC;;OAEG;IACH,OAAO,CAAC,aAAa;IAYrB;;;;;;OAMG;YACW,eAAe;IA2C7B;;OAEG;IACG,eAAe,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAIxD,yBAAyB;IACzB,IAAI,QAAQ,IAAI,WAAW,GAAG,KAAK,CAElC;IAED,4CAA4C;IAC5C,IAAI,GAAG,IAAI,SAAS,CAEnB;CACJ"}
+{"version":3,"file":"AgentOsClient.d.ts","sourceRoot":"","sources":["../../src/client/AgentOsClient.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AAEH,OAAO,EAAmB,KAAK,SAAS,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EACH,KAAK,oBAAoB,EAK5B,MAAM,WAAW,CAAC;AAEnB,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAG5D,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAG9D,YAAY,EAAE,oBAAoB,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACpE,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAEtD,qBAAa,aAAa;IACtB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAY;IACpC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAe;IACrC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAyB;IAGxD,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC;IAC9B,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;IAC1B,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,SAAS,EAAE,eAAe,CAAC;IACpC,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAC;IAClC,QAAQ,CAAC,WAAW,EAAE,iBAAiB,CAAC;IACxC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,UAAU,EAAE,gBAAgB,CAAC;IAGtC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC;IAC9B,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,YAAY,EAAE,kBAAkB,CAAC;IAC1C,QAAQ,CAAC,UAAU,EAAE,gBAAgB,CAAC;IACtC,QAAQ,CAAC,WAAW,EAAE,iBAAiB,CAAC;IACxC,QAAQ,CAAC,UAAU,EAAE,gBAAgB,CAAC;IACtC,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,GAAG,EAAE,SAAS,CAAC;IACxB,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,QAAQ,CAAC,GAAG,EAAE,SAAS,CAAC;IACxB,QAAQ,CAAC,GAAG,EAAE,SAAS,CAAC;IACxB,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;IAC1B,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC;IAC9B,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC;IAChC,QAAQ,CAAC,SAAS,EAAE,eAAe,CAAC;IAGpC,QAAQ,CAAC,WAAW,EAAE;QAClB,IAAI,EAAE,gBAAgB,CAAC,iBAAiB,CAAC,CAAC;QAC1C,GAAG,EAAE,gBAAgB,CAAC,eAAe,CAAC,CAAC;QACvC,MAAM,EAAE,gBAAgB,CAAC,kBAAkB,CAAC,CAAC;KAChD,CAAC;IAEF,SAAgB,SAAS,EAAE,eAAe,CAAC;IAC3C,SAAgB,WAAW,EAAE,iBAAiB,CAAC;IAE/C,QAAQ,CAAC,aAAa,EAAE;QACpB,IAAI,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,UAAU,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC;QACpE,GAAG,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,KAAK,UAAU,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;QACpF,MAAM,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC;QAC3H,WAAW,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,UAAU,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;QAC7H,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,KAAK,UAAU,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;QACzF,MAAM,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC;QAC9I,OAAO,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,KAAK,UAAU,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC;KAC/F,CAAC;gBAEU,OAAO,EAAE,oBAAoB;IA4EzC;;OAEG;IACH,OAAO,CAAC,aAAa;IAYrB;;;;;;OAMG;YACW,eAAe;IA2C7B;;OAEG;IACG,eAAe,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAIxD,yBAAyB;IACzB,IAAI,QAAQ,IAAI,WAAW,GAAG,KAAK,CAElC;IAED,4CAA4C;IAC5C,IAAI,GAAG,IAAI,SAAS,CAEnB;CACJ"}

package/dist/client/AgentOsClient.js

@@ -60,6 +60,8 @@ import { MetricsModule } from "../modules/metrics.js";
 import { GraphsModule } from "../modules/graphs.js";
 import { CatalogModule } from "../modules/catalog.js";
 import { ApprovalsModule } from "../modules/approvals.js";
+import { ApiTokensModule } from "../modules/apiTokens.js";
+import { MembershipsModule } from "../modules/memberships.js";
 export { isApiTokenAuth, isJwtAuth } from "./auth.js";
 export class AgentOsClient {
     _client;
@@ -101,6 +103,8 @@ export class AgentOsClient {
     approvals;
     // Convenience aliases
     experiments;
+    apiTokens;
+    memberships;
     agentVersions;
     constructor(options) {
         this._baseUrl = options.baseUrl;
@@ -156,10 +160,16 @@ export class AgentOsClient {
             get: this.evaluation.getExperiment.bind(this.evaluation),
             create: this.evaluation.createExperiment.bind(this.evaluation),
         };
+        this.apiTokens = new ApiTokensModule(this._client);
+        this.memberships = new MembershipsModule(this._client);
         this.agentVersions = {
             list: (agentId) => this.agents.listVersions(agentId),
             get: (agentId, versionId) => this.agents.getVersion(agentId, versionId),
             create: (agentId, body) => this.agents.createVersion(agentId, body),
+            createDraft: (agentId, body) => this.agents.createDraft(agentId, body),
+            getBundle: (agentId, versionId) => this.agents.getBundle(agentId, versionId),
+            update: (agentId, versionId, body) => this.agents.updateVersion(agentId, versionId, body),
+            publish: (agentId, versionId) => this.agents.publishVersion(agentId, versionId),
         };
     }
     /**

package/dist/modules/agents.d.ts

@@ -6,9 +6,8 @@
  * - list* for collections: listAgents()
  * - create*, update*, delete* for mutations
  */
-import type { RawClient, APIResponse, components } from "../client/raw.js";
+import type { RawClient, APIResponse } from "../client/raw.js";
 import type { PaginationParams, PaginatedResponse } from "../client/helpers.js";
-type AgentBundle = components["schemas"]["AgentBundle"];
 export interface Agent {
     id: string;
     name: string;
@@ -28,6 +27,65 @@ export interface AgentVersion {
     format?: string;
     graph_spec?: unknown;
     created_at: string;
+    draft_state?: string | null;
+    base_version_id?: string | null;
+}
+export interface CreateDraftRequest {
+    /** Base type: "live" (current live version) or "version" (specific version) */
+    base?: "live" | "version";
+    /** Required when base="version" */
+    base_version_id?: string;
+}
+export interface CreateDraftResponse {
+    draft_version_id: string;
+    agent_id: string;
+    base_version_id?: string | null;
+    created_at: string;
+}
+export interface AgentBundle {
+    schema_version: string;
+    exported_at: string;
+    agent?: AgentBundleAgent;
+    versions?: AgentBundleVersion[];
+}
+export interface AgentBundleAgent {
+    id: string;
+    name: string;
+    live_version_id?: string | null;
+}
+export interface AgentBundleVersion {
+    id: string;
+    version_tag: string;
+    artifact_uri: string;
+    artifact_hash?: string;
+    format?: string;
+    created_at: string;
+}
+export interface ImportAgentResponse {
+    agent_id: string;
+    message: string;
+    versions_imported: number;
+}
+export interface AgentVersionMetadata {
+    id: string;
+    agent_id: string;
+    version_tag: string;
+    created_at: string;
+}
+export interface AgentVersionBundleResponse {
+    version: AgentVersionMetadata;
+    graph_spec: unknown;
+    artifact_hash: string;
+    format: string;
+    resolved_at: string;
+}
+export interface UpdateAgentVersionRequest {
+    graph_spec?: unknown;
+}
+export interface PublishAgentVersionResponse {
+    agent_id: string;
+    live_version_id: string;
+    status: string;
 }
 export interface AgentGraphResponse {
     mermaid?: string;
@@ -115,23 +173,46 @@ export declare class AgentsModule {
      */
     getVersion(agentId: string, versionId: string): Promise<APIResponse<AgentVersion>>;
     /**
-     * Get the graph for an agent (Mermaid or JSON).
+     * Create or retrieve an active draft version for editing.
+     * This endpoint is idempotent - calling it multiple times returns the same draft.
+     * @example
+     * ```ts
+     * const { data } = await client.agents.createDraft("agent-uuid", { base: "live" });
+     * console.log(data?.draft_version_id);
+     * // Then call getBundle(draftVersionId) to load the graph
+     * ```
      */
-    getGraph(agentId: string, format?: "mermaid" | "json"): Promise<APIResponse<AgentGraphResponse>>;
-    /** Alias for getGraph (convenience) */
-    graph: (agentId: string, format?: "mermaid" | "json") => Promise<APIResponse<AgentGraphResponse>>;
+    createDraft(agentId: string, body?: CreateDraftRequest): Promise<APIResponse<CreateDraftResponse>>;
     /**
-     * Validate the graph configuration.
+     * Get the full bundle for a version (including resolved graph_spec).
      */
-    validateGraph(agentId: string): Promise<APIResponse<GraphValidationResponse>>;
+    getBundle(agentId: string, versionId: string): Promise<APIResponse<AgentVersionBundleResponse>>;
+    /**
+     * Update a draft version.
+     * Only allowed if draft_state="active".
+     */
+    updateVersion(agentId: string, versionId: string, body: UpdateAgentVersionRequest): Promise<APIResponse<AgentVersion>>;
     /**
-     * Export an agent with all its versions.
+     * Publish a draft version to live.
+     */
+    publishVersion(agentId: string, versionId: string): Promise<APIResponse<PublishAgentVersionResponse>>;
+    /**
+     * Export an agent as a JSON bundle.
      */
     export(agentId: string): Promise<APIResponse<AgentBundle>>;
     /**
      * Import an agent bundle.
      */
-    import(bundle: AgentBundle): Promise<APIResponse<Agent>>;
+    import(bundle: AgentBundle): Promise<APIResponse<ImportAgentResponse>>;
+    /**
+     * Get the graph for an agent (Mermaid or JSON).
+     */
+    getGraph(agentId: string, format?: "mermaid" | "json"): Promise<APIResponse<AgentGraphResponse>>;
+    /** Alias for getGraph (convenience) */
+    graph: (agentId: string, format?: "mermaid" | "json") => Promise<APIResponse<AgentGraphResponse>>;
+    /**
+     * Validate the graph configuration.
+     */
+    validateGraph(agentId: string): Promise<APIResponse<GraphValidationResponse>>;
 }
-export {};
 //# sourceMappingURL=agents.d.ts.map

package/dist/modules/agents.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agents.d.ts","sourceRoot":"","sources":["../../src/modules/agents.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC3E,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAIhF,KAAK,WAAW,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,aAAa,CAAC,CAAC;AAIxD,MAAM,WAAW,KAAK;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACnC;AAED,MAAM,WAAW,YAAY;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,kBAAkB;IAC/B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,uBAAuB;IACpC,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;CACvB;AAED,MAAM,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;AACzD,MAAM,MAAM,wBAAwB,GAAG,iBAAiB,CAAC,YAAY,CAAC,CAAC;AAEvE,qBAAa,YAAY;IACT,OAAO,CAAC,MAAM;IAAa,OAAO,CAAC,OAAO;gBAAlC,MAAM,EAAE,SAAS,EAAU,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAIpF;;;;;;;OAOG;IACG,IAAI,CAAC,MAAM,CAAC,EAAE,gBAAgB,GAAG;QACnC,YAAY,CAAC,EAAE,MAAM,CAAC;KACzB,GAAG,OAAO,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAO3C;;;;;;OAMG;IACG,GAAG,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IAOvD;;;;;;;;;OASG;IACG,MAAM,CAAC,IAAI,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,UAAU,CAAC,EAAE,OAAO,CAAC;KACxB,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IAO/B;;OAEG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE;QAChC,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC5B,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IAQ/B;;OAEG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IASzD;;;;;;OAMG;IACG,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC,CAAC;IAOzE,2CAA2C;IAC3C,QAAQ,GAAI,SAAS,MAAM,0CAAgC;IAE3D;;OAEG;IACG,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE;QACvC,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,OAAO,CAAC;KACxB,GAAG,OAAO,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IAQtC;;OAEG;IACG,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IASxF;;OAEG;IACG,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,SAAS,GAAG,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAOtG,uCAAuC;IACvC,KAAK,GAAI,SAAS,MAAM,EAAE,SAAS,SAAS,GAAG,MAAM,8CAAoC;IAEzF;;OAEG;IACG,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,uBAAuB,CAAC,CAAC;IASnF;;OAEG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IAOhE;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;CAMjE"}
+{"version":3,"file":"agents.d.ts","sourceRoot":"","sources":["../../src/modules/agents.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAc,MAAM,kBAAkB,CAAC;AAC3E,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAKhF,MAAM,WAAW,KAAK;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACnC;AAED,MAAM,WAAW,YAAY;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACnC;AAED,MAAM,WAAW,kBAAkB;IAC/B,+EAA+E;IAC/E,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,mCAAmC;IACnC,eAAe,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,WAAW,mBAAmB;IAChC,gBAAgB,EAAE,MAAM,CAAC;IACzB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,WAAW;IACxB,cAAc,EAAE,MAAM,CAAC;IACvB,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,gBAAgB,CAAC;IACzB,QAAQ,CAAC,EAAE,kBAAkB,EAAE,CAAC;CACnC;AAED,MAAM,WAAW,gBAAgB;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACnC;AAED,MAAM,WAAW,kBAAkB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,mBAAmB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,iBAAiB,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,oBAAoB;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,0BAA0B;IACvC,OAAO,EAAE,oBAAoB,CAAC;IAC9B,UAAU,EAAE,OAAO,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,yBAAyB;IACtC,UAAU,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,2BAA2B;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,MAAM,CAAC;IACxB,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,kBAAkB;IAC/B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,uBAAuB;IACpC,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;CACvB;AAED,MAAM,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;AACzD,MAAM,MAAM,wBAAwB,GAAG,iBAAiB,CAAC,YAAY,CAAC,CAAC;AAEvE,qBAAa,YAAY;IACT,OAAO,CAAC,MAAM;IAAa,OAAO,CAAC,OAAO;gBAAlC,MAAM,EAAE,SAAS,EAAU,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAIpF;;;;;;;OAOG;IACG,IAAI,CAAC,MAAM,CAAC,EAAE,gBAAgB,GAAG;QACnC,YAAY,CAAC,EAAE,MAAM,CAAC;KACzB,GAAG,OAAO,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAO3C;;;;;;OAMG;IACG,GAAG,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IAOvD;;;;;;;;;OASG;IACG,MAAM,CAAC,IAAI,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,UAAU,CAAC,EAAE,OAAO,CAAC;KACxB,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IAO/B;;OAEG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE;QAChC,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC5B,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IAQ/B;;OAEG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IASzD;;;;;;OAMG;IACG,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC,CAAC;IAOzE,2CAA2C;IAC3C,QAAQ,GAAI,SAAS,MAAM,0CAAgC;IAE3D;;OAEG;IACG,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE;QACvC,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,OAAO,CAAC;KACxB,GAAG,OAAO,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IAQtC;;OAEG;IACG,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IAOxF;;;;;;;;;OASG;IACG,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;IAQxG;;OAEG;IACG,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,0BAA0B,CAAC,CAAC;IAOrG;;;OAGG;IACG,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,yBAAyB,GAAG,OAAO,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IAQ5H;;OAEG;IACG,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,2BAA2B,CAAC,CAAC;IAS3G;;OAEG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IAOhE;;OAEG;IACG,MAAM,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;IAS5E;;OAEG;IACG,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,SAAS,GAAG,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAOtG,uCAAuC;IACvC,KAAK,GAAI,SAAS,MAAM,EAAE,SAAS,SAAS,GAAG,MAAM,8CAAoC;IAEzF;;OAEG;IACG,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,uBAAuB,CAAC,CAAC;CAQtF"}

package/dist/modules/agents.js

@@ -111,30 +111,55 @@ export class AgentsModule {
             headers: this.headers(),
         });
     }
-    // ======================== Graph ========================
     /**
-     * Get the graph for an agent (Mermaid or JSON).
+     * Create or retrieve an active draft version for editing.
+     * This endpoint is idempotent - calling it multiple times returns the same draft.
+     * @example
+     * ```ts
+     * const { data } = await client.agents.createDraft("agent-uuid", { base: "live" });
+     * console.log(data?.draft_version_id);
+     * // Then call getBundle(draftVersionId) to load the graph
+     * ```
      */
-    async getGraph(agentId, format) {
-        return this.client.GET("/v1/api/agents/{agentId}/graph", {
-            params: { path: { agentId }, query: { format } },
+    async createDraft(agentId, body) {
+        return this.client.POST("/v1/api/agents/{agentId}/versions/draft", {
+            params: { path: { agentId } },
+            body: body ?? { base: "live" },
             headers: this.headers(),
         });
     }
-    /** Alias for getGraph (convenience) */
-    graph = (agentId, format) => this.getGraph(agentId, format);
     /**
-     * Validate the graph configuration.
+     * Get the full bundle for a version (including resolved graph_spec).
      */
-    async validateGraph(agentId) {
-        return this.client.POST("/v1/api/agents/{agentId}/graph/validate", {
-            params: { path: { agentId } },
+    async getBundle(agentId, versionId) {
+        return this.client.GET("/v1/api/agents/{agentId}/versions/{versionId}/bundle", {
+            params: { path: { agentId, versionId } },
+            headers: this.headers(),
+        });
+    }
+    /**
+     * Update a draft version.
+     * Only allowed if draft_state="active".
+     */
+    async updateVersion(agentId, versionId, body) {
+        return this.client.PATCH("/v1/api/agents/{agentId}/versions/{versionId}", {
+            params: { path: { agentId, versionId } },
+            body,
+            headers: this.headers(),
+        });
+    }
+    /**
+     * Publish a draft version to live.
+     */
+    async publishVersion(agentId, versionId) {
+        return this.client.POST("/v1/api/agents/{agentId}/versions/{versionId}/publish", {
+            params: { path: { agentId, versionId } },
             headers: this.headers(),
         });
     }
-    // ======================== Import/Export ========================
+    // ======================== Portability ========================
     /**
-     * Export an agent with all its versions.
+     * Export an agent as a JSON bundle.
      */
     async export(agentId) {
         return this.client.GET("/v1/api/agents/{id}/export", {
@@ -151,4 +176,25 @@ export class AgentsModule {
             headers: this.headers(),
         });
     }
+    // ======================== Graph ========================
+    /**
+     * Get the graph for an agent (Mermaid or JSON).
+     */
+    async getGraph(agentId, format) {
+        return this.client.GET("/v1/api/agents/{agentId}/graph", {
+            params: { path: { agentId }, query: { format } },
+            headers: this.headers(),
+        });
+    }
+    /** Alias for getGraph (convenience) */
+    graph = (agentId, format) => this.getGraph(agentId, format);
+    /**
+     * Validate the graph configuration.
+     */
+    async validateGraph(agentId) {
+        return this.client.POST("/v1/api/agents/{agentId}/graph/validate", {
+            params: { path: { agentId } },
+            headers: this.headers(),
+        });
+    }
 }

package/dist/modules/apiTokens.d.ts

@@ -0,0 +1,51 @@
+import type { RawClient, APIResponse } from "../client/raw.js";
+export interface ApiToken {
+    id: string;
+    name: string;
+    token_prefix: string;
+    scopes: string[];
+    created_at: string;
+    expires_at?: string | null;
+    last_used_at?: string | null;
+}
+export interface ApiTokenSecret extends ApiToken {
+    token: string;
+    workspace_id: string;
+}
+export interface CreateTokenRequest {
+    name: string;
+    scopes?: string[];
+    expires_at?: string | null;
+    environment?: "live" | "test";
+}
+export interface RotateTokenResponse extends ApiTokenSecret {
+    rotated_from: string;
+}
+export declare class ApiTokensModule {
+    private client;
+    constructor(client: RawClient);
+    private headers;
+    /**
+     * List all API tokens for the current workspace.
+     */
+    list(): Promise<APIResponse<{
+        tokens: ApiToken[];
+    }>>;
+    /**
+     * Create a new API token.
+     * The full token secret is only returned in this response.
+     */
+    create(body: CreateTokenRequest): Promise<APIResponse<ApiTokenSecret>>;
+    /**
+     * Revoke an API token immediately.
+     */
+    revoke(id: string): Promise<APIResponse<{
+        message: string;
+    }>>;
+    /**
+     * Rotate a token (revoke old, create new).
+     * Returns the new token secret.
+     */
+    rotate(id: string): Promise<APIResponse<RotateTokenResponse>>;
+}
+//# sourceMappingURL=apiTokens.d.ts.map

package/dist/modules/apiTokens.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"apiTokens.d.ts","sourceRoot":"","sources":["../../src/modules/apiTokens.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/D,MAAM,WAAW,QAAQ;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAChC;AAED,MAAM,WAAW,cAAe,SAAQ,QAAQ;IAC5C,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,kBAAkB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,WAAW,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;CACjC;AAED,MAAM,WAAW,mBAAoB,SAAQ,cAAc;IACvD,YAAY,EAAE,MAAM,CAAC;CACxB;AAED,qBAAa,eAAe;IACxB,OAAO,CAAC,MAAM,CAAY;gBAEd,MAAM,EAAE,SAAS;IAI7B,OAAO,CAAC,OAAO;IAMf;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,EAAE,CAAA;KAAE,CAAC,CAAC;IAM1D;;;OAGG;IACG,MAAM,CAAC,IAAI,EAAE,kBAAkB,GAAG,OAAO,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;IAO5E;;OAEG;IACG,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAOnE;;;OAGG;IACG,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;CAMtE"}

package/dist/modules/apiTokens.js

@@ -0,0 +1,48 @@
+export class ApiTokensModule {
+    client;
+    constructor(client) {
+        this.client = client;
+    }
+    headers() {
+        return {
+            "Content-Type": "application/json",
+        };
+    }
+    /**
+     * List all API tokens for the current workspace.
+     */
+    async list() {
+        return this.client.GET("/v1/api/tokens", {
+            headers: this.headers(),
+        });
+    }
+    /**
+     * Create a new API token.
+     * The full token secret is only returned in this response.
+     */
+    async create(body) {
+        return this.client.POST("/v1/api/tokens", {
+            body,
+            headers: this.headers(),
+        });
+    }
+    /**
+     * Revoke an API token immediately.
+     */
+    async revoke(id) {
+        return this.client.DELETE("/v1/api/tokens/{id}", {
+            params: { path: { id } },
+            headers: this.headers(),
+        });
+    }
+    /**
+     * Rotate a token (revoke old, create new).
+     * Returns the new token secret.
+     */
+    async rotate(id) {
+        return this.client.POST("/v1/api/tokens/{id}/rotate", {
+            params: { path: { id } },
+            headers: this.headers(),
+        });
+    }
+}

package/dist/modules/memberships.d.ts

@@ -0,0 +1,25 @@
+import type { RawClient, APIResponse } from "../client/raw.js";
+export interface MembershipResponse {
+    id: string;
+    user_id: string;
+    tenant_id: string;
+    role_id: string;
+    role_name?: string;
+    workspace_id?: string;
+    workspace_name?: string;
+    created_at: string;
+}
+export interface EnsureMembershipRequest {
+    tenant_id?: string;
+    workspace_id?: string;
+}
+export declare class MembershipsModule {
+    private client;
+    constructor(client: RawClient);
+    private headers;
+    /**
+     * Ensure the current user has a membership in a tenant (self-provisioning).
+     */
+    ensure(body: EnsureMembershipRequest): Promise<APIResponse<MembershipResponse>>;
+}
+//# sourceMappingURL=memberships.d.ts.map

package/dist/modules/memberships.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memberships.d.ts","sourceRoot":"","sources":["../../src/modules/memberships.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/D,MAAM,WAAW,kBAAkB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,uBAAuB;IACpC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,qBAAa,iBAAiB;IAC1B,OAAO,CAAC,MAAM,CAAY;gBAEd,MAAM,EAAE,SAAS;IAI7B,OAAO,CAAC,OAAO;IAMf;;OAEG;IACG,MAAM,CAAC,IAAI,EAAE,uBAAuB,GAAG,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;CAMxF"}

package/dist/modules/memberships.js

@@ -0,0 +1,20 @@
+export class MembershipsModule {
+    client;
+    constructor(client) {
+        this.client = client;
+    }
+    headers() {
+        return {
+            "Content-Type": "application/json",
+        };
+    }
+    /**
+     * Ensure the current user has a membership in a tenant (self-provisioning).
+     */
+    async ensure(body) {
+        return this.client.POST("/v1/api/memberships/ensure", {
+            body,
+            headers: this.headers(),
+        });
+    }
+}

package/dist/modules/workspaces.d.ts

@@ -48,5 +48,57 @@ export declare class WorkspacesModule {
      * Delete a workspace.
      */
     delete(workspaceId: string): Promise<APIResponse<void>>;
+    /**
+     * Get workspace quotas.
+     */
+    getQuotas(workspaceId: string): Promise<APIResponse<WorkspaceQuotaResponse>>;
+    /**
+     * Get workspace usage.
+     */
+    getUsage(workspaceId: string, params?: {
+        from?: string;
+        to?: string;
+    }): Promise<APIResponse<WorkspaceUsageResponse>>;
+}
+export interface WorkspaceQuotaResponse {
+    workspace_id: string;
+    quotas: {
+        runs_per_month: Quota;
+        tokens_per_month: Quota;
+        agents_max: Quota;
+        storage_mb: Quota;
+    };
+    reset_at: string;
+    source: string;
+}
+export interface Quota {
+    limit: number;
+    used: number;
+    remaining: number;
+}
+export interface WorkspaceUsageResponse {
+    workspace_id: string;
+    period: {
+        from: string;
+        to: string;
+    };
+    usage: {
+        total_runs: number;
+        completed_runs: number;
+        failed_runs: number;
+        total_tokens: number;
+        input_tokens: number;
+        output_tokens: number;
+        total_duration_ms: number;
+        estimated_count: number;
+    };
+    daily_breakdown: Array<{
+        date: string;
+        runs: number;
+        tokens_in: number;
+        tokens_out: number;
+        duration_ms: number;
+    }>;
+    source: string;
 }
 //# sourceMappingURL=workspaces.d.ts.map

package/dist/modules/workspaces.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"workspaces.d.ts","sourceRoot":"","sources":["../../src/modules/workspaces.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAc,MAAM,kBAAkB,CAAC;AAI3E,MAAM,WAAW,SAAS;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,qBAAqB;IAClC,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,gBAAgB;IAErB,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,OAAO;gBAFP,MAAM,EAAE,SAAS,EACjB,WAAW,EAAE,MAAM,MAAM,EACzB,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAGjD;;OAEG;IACG,IAAI,CAAC,MAAM,CAAC,EAAE;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,WAAW,CAAC,qBAAqB,CAAC,CAAC;IAO/C;;OAEG;IACG,GAAG,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAO/D;;OAEG;IACG,MAAM,CAAC,IAAI,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,CAAC,EAAE,OAAO,CAAC;KACxB,GAAG,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAOnC;;OAEG;IACG,MAAM,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE;QACpC,IAAI,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAQnC;;OAEG;IACG,MAAM,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;CAMhE"}
+{"version":3,"file":"workspaces.d.ts","sourceRoot":"","sources":["../../src/modules/workspaces.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAc,MAAM,kBAAkB,CAAC;AAI3E,MAAM,WAAW,SAAS;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,qBAAqB;IAClC,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,gBAAgB;IAErB,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,OAAO;gBAFP,MAAM,EAAE,SAAS,EACjB,WAAW,EAAE,MAAM,MAAM,EACzB,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAGjD;;OAEG;IACG,IAAI,CAAC,MAAM,CAAC,EAAE;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,WAAW,CAAC,qBAAqB,CAAC,CAAC;IAO/C;;OAEG;IACG,GAAG,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAO/D;;OAEG;IACG,MAAM,CAAC,IAAI,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,CAAC,EAAE,OAAO,CAAC;KACxB,GAAG,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAOnC;;OAEG;IACG,MAAM,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE;QACpC,IAAI,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAQnC;;OAEG;IACG,MAAM,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IAO7D;;OAEG;IACG,SAAS,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC;IAOlF;;OAEG;IACG,QAAQ,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC;CAM7H;AAED,MAAM,WAAW,sBAAsB;IACnC,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE;QACJ,cAAc,EAAE,KAAK,CAAC;QACtB,gBAAgB,EAAE,KAAK,CAAC;QACxB,UAAU,EAAE,KAAK,CAAC;QAClB,UAAU,EAAE,KAAK,CAAC;KACrB,CAAC;IACF,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,KAAK;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,sBAAsB;IACnC,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC;IACrC,KAAK,EAAE;QACH,UAAU,EAAE,MAAM,CAAC;QACnB,cAAc,EAAE,MAAM,CAAC;QACvB,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,eAAe,EAAE,MAAM,CAAC;KAC3B,CAAC;IACF,eAAe,EAAE,KAAK,CAAC;QACnB,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;KACvB,CAAC,CAAC;IACH,MAAM,EAAE,MAAM,CAAC;CAClB"}

package/dist/modules/workspaces.js

@@ -56,4 +56,22 @@ export class WorkspacesModule {
             headers: this.headers(),
         });
     }
+    /**
+     * Get workspace quotas.
+     */
+    async getQuotas(workspaceId) {
+        return this.client.GET("/v1/api/workspaces/{workspaceId}/quotas", {
+            params: { path: { workspaceId } },
+            headers: this.headers(),
+        });
+    }
+    /**
+     * Get workspace usage.
+     */
+    async getUsage(workspaceId, params) {
+        return this.client.GET("/v1/api/workspaces/{workspaceId}/usage", {
+            params: { path: { workspaceId }, query: params },
+            headers: this.headers(),
+        });
+    }
 }

package/package.json

@@ -1,52 +1,51 @@
 {
-    "name": "@agent-os-sdk/client",
-    "version": "0.3.11",
-    "description": "Official TypeScript SDK for Agent OS platform",
-    "type": "module",
-    "main": "./dist/index.js",
-    "module": "./dist/index.js",
-    "types": "./dist/index.d.ts",
-    "exports": {
-        ".": {
-            "import": "./dist/index.js",
-            "types": "./dist/index.d.ts"
-        }
-    },
-    "scripts": {
-        "generate": "tsx scripts/generate.ts",
-        "build": "tsc",
-        "dev": "tsc --watch",
-        "test": "vitest run",
-        "typecheck": "tsc --noEmit",
-        "prepublishOnly": "pnpm build"
-    },
-    "keywords": [
-        "agent-os",
-        "sdk",
-        "ai",
-        "agents",
-        "typescript"
-    ],
-    "author": "Agent OS Team",
-    "license": "MIT",
-    "repository": {
-        "type": "git",
-        "url": "https://github.com/yuri12344/agent-os.git"
-    },
-    "publishConfig": {
-        "access": "public"
-    },
-    "dependencies": {
-        "openapi-fetch": "^0.13.0"
-    },
-    "devDependencies": {
-        "openapi-typescript": "^7.4.0",
-        "tsx": "^4.19.0",
-        "typescript": "^5.5.0",
-        "vitest": "^4.0.18"
-    },
-    "files": [
-        "dist",
-        "src"
-    ]
-}
+  "name": "@agent-os-sdk/client",
+  "version": "0.3.14",
+  "description": "Official TypeScript SDK for Agent OS platform",
+  "type": "module",
+  "main": "./dist/index.js",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "keywords": [
+    "agent-os",
+    "sdk",
+    "ai",
+    "agents",
+    "typescript"
+  ],
+  "author": "Agent OS Team",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/yuri12344/agent-os.git"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "openapi-fetch": "^0.13.0"
+  },
+  "devDependencies": {
+    "openapi-typescript": "^7.4.0",
+    "tsx": "^4.19.0",
+    "typescript": "^5.5.0",
+    "vitest": "^4.0.18"
+  },
+  "files": [
+    "dist",
+    "src"
+  ],
+  "scripts": {
+    "generate": "tsx scripts/generate.ts",
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "test": "vitest run",
+    "typecheck": "tsc --noEmit"
+  }
+}

package/src/client/AgentOsClient.ts

@@ -69,6 +69,8 @@ import { MetricsModule } from "../modules/metrics.js";
 import { GraphsModule } from "../modules/graphs.js";
 import { CatalogModule } from "../modules/catalog.js";
 import { ApprovalsModule } from "../modules/approvals.js";
+import { ApiTokensModule } from "../modules/apiTokens.js";
+import { MembershipsModule } from "../modules/memberships.js";
 
 // Re-export auth types
 export type { AgentOsClientOptions, AuthProvider } from "./auth.js";
@@ -122,10 +124,17 @@ export class AgentOsClient {
         create: EvaluationModule["createExperiment"];
     };
 
+    public readonly apiTokens: ApiTokensModule;
+    public readonly memberships: MembershipsModule;
+
     readonly agentVersions: {
         list: (agentId: string) => ReturnType<AgentsModule["listVersions"]>;
         get: (agentId: string, versionId: string) => ReturnType<AgentsModule["getVersion"]>;
         create: (agentId: string, body: Parameters<AgentsModule["createVersion"]>[1]) => ReturnType<AgentsModule["createVersion"]>;
+        createDraft: (agentId: string, body?: Parameters<AgentsModule["createDraft"]>[1]) => ReturnType<AgentsModule["createDraft"]>;
+        getBundle: (agentId: string, versionId: string) => ReturnType<AgentsModule["getBundle"]>;
+        update: (agentId: string, versionId: string, body: Parameters<AgentsModule["updateVersion"]>[2]) => ReturnType<AgentsModule["updateVersion"]>;
+        publish: (agentId: string, versionId: string) => ReturnType<AgentsModule["publishVersion"]>;
     };
 
     constructor(options: AgentOsClientOptions) {
@@ -189,10 +198,18 @@ export class AgentOsClient {
             create: this.evaluation.createExperiment.bind(this.evaluation),
         };
 
+        this.apiTokens = new ApiTokensModule(this._client);
+        this.memberships = new MembershipsModule(this._client);
+
         this.agentVersions = {
             list: (agentId: string) => this.agents.listVersions(agentId),
             get: (agentId: string, versionId: string) => this.agents.getVersion(agentId, versionId),
             create: (agentId: string, body: any) => this.agents.createVersion(agentId, body),
+            createDraft: (agentId: string, body?: { base?: "live" | "version"; base_version_id?: string }) =>
+                this.agents.createDraft(agentId, body),
+            getBundle: (agentId: string, versionId: string) => this.agents.getBundle(agentId, versionId),
+            update: (agentId: string, versionId: string, body: any) => this.agents.updateVersion(agentId, versionId, body),
+            publish: (agentId: string, versionId: string) => this.agents.publishVersion(agentId, versionId),
         };
     }
 

package/src/modules/agents.ts

@@ -12,9 +12,6 @@ import type { PaginationParams, PaginatedResponse } from "../client/helpers.js";
 
 // Type aliases for this module
 type AgentSchema = components["schemas"]["AgentBundleAgent"];
-type AgentBundle = components["schemas"]["AgentBundle"];
-type AgentVersionSchema = components["schemas"]["AgentBundleVersion"];
-
 // Response types
 export interface Agent {
     id: string;
@@ -36,6 +33,75 @@ export interface AgentVersion {
     format?: string;
     graph_spec?: unknown;
     created_at: string;
+    draft_state?: string | null;
+    base_version_id?: string | null;
+}
+
+export interface CreateDraftRequest {
+    /** Base type: "live" (current live version) or "version" (specific version) */
+    base?: "live" | "version";
+    /** Required when base="version" */
+    base_version_id?: string;
+}
+
+export interface CreateDraftResponse {
+    draft_version_id: string;
+    agent_id: string;
+    base_version_id?: string | null;
+    created_at: string;
+}
+
+export interface AgentBundle {
+    schema_version: string;
+    exported_at: string;
+    agent?: AgentBundleAgent;
+    versions?: AgentBundleVersion[];
+}
+
+export interface AgentBundleAgent {
+    id: string;
+    name: string;
+    live_version_id?: string | null;
+}
+
+export interface AgentBundleVersion {
+    id: string;
+    version_tag: string;
+    artifact_uri: string;
+    artifact_hash?: string;
+    format?: string;
+    created_at: string;
+}
+
+export interface ImportAgentResponse {
+    agent_id: string;
+    message: string;
+    versions_imported: number;
+}
+
+export interface AgentVersionMetadata {
+    id: string;
+    agent_id: string;
+    version_tag: string;
+    created_at: string;
+}
+
+export interface AgentVersionBundleResponse {
+    version: AgentVersionMetadata;
+    graph_spec: unknown;
+    artifact_hash: string;
+    format: string;
+    resolved_at: string;
+}
+
+export interface UpdateAgentVersionRequest {
+    graph_spec?: unknown;
+}
+
+export interface PublishAgentVersionResponse {
+    agent_id: string;
+    live_version_id: string;
+    status: string;
 }
 
 export interface AgentGraphResponse {
@@ -180,35 +246,60 @@ export class AgentsModule {
         });
     }
 
-    // ======================== Graph ========================
+    /**
+     * Create or retrieve an active draft version for editing.
+     * This endpoint is idempotent - calling it multiple times returns the same draft.
+     * @example
+     * ```ts
+     * const { data } = await client.agents.createDraft("agent-uuid", { base: "live" });
+     * console.log(data?.draft_version_id);
+     * // Then call getBundle(draftVersionId) to load the graph
+     * ```
+     */
+    async createDraft(agentId: string, body?: CreateDraftRequest): Promise<APIResponse<CreateDraftResponse>> {
+        return this.client.POST<CreateDraftResponse>("/v1/api/agents/{agentId}/versions/draft", {
+            params: { path: { agentId } },
+            body: body ?? { base: "live" },
+            headers: this.headers(),
+        });
+    }
 
     /**
-     * Get the graph for an agent (Mermaid or JSON).
+     * Get the full bundle for a version (including resolved graph_spec).
      */
-    async getGraph(agentId: string, format?: "mermaid" | "json"): Promise<APIResponse<AgentGraphResponse>> {
-        return this.client.GET<AgentGraphResponse>("/v1/api/agents/{agentId}/graph", {
-            params: { path: { agentId }, query: { format } },
+    async getBundle(agentId: string, versionId: string): Promise<APIResponse<AgentVersionBundleResponse>> {
+        return this.client.GET<AgentVersionBundleResponse>("/v1/api/agents/{agentId}/versions/{versionId}/bundle", {
+            params: { path: { agentId, versionId } },
             headers: this.headers(),
         });
     }
 
-    /** Alias for getGraph (convenience) */
-    graph = (agentId: string, format?: "mermaid" | "json") => this.getGraph(agentId, format);
+    /**
+     * Update a draft version.
+     * Only allowed if draft_state="active".
+     */
+    async updateVersion(agentId: string, versionId: string, body: UpdateAgentVersionRequest): Promise<APIResponse<AgentVersion>> {
+        return this.client.PATCH<AgentVersion>("/v1/api/agents/{agentId}/versions/{versionId}", {
+            params: { path: { agentId, versionId } },
+            body,
+            headers: this.headers(),
+        });
+    }
 
     /**
-     * Validate the graph configuration.
+     * Publish a draft version to live.
      */
-    async validateGraph(agentId: string): Promise<APIResponse<GraphValidationResponse>> {
-        return this.client.POST<GraphValidationResponse>("/v1/api/agents/{agentId}/graph/validate", {
-            params: { path: { agentId } },
+    async publishVersion(agentId: string, versionId: string): Promise<APIResponse<PublishAgentVersionResponse>> {
+        return this.client.POST<PublishAgentVersionResponse>("/v1/api/agents/{agentId}/versions/{versionId}/publish", {
+            params: { path: { agentId, versionId } },
             headers: this.headers(),
         });
     }
 
-    // ======================== Import/Export ========================
+    // ======================== Portability ========================
 
     /**
-     * Export an agent with all its versions.
+     * Export an agent as a JSON bundle.
      */
     async export(agentId: string): Promise<APIResponse<AgentBundle>> {
         return this.client.GET<AgentBundle>("/v1/api/agents/{id}/export", {
@@ -220,10 +311,37 @@ export class AgentsModule {
     /**
      * Import an agent bundle.
      */
-    async import(bundle: AgentBundle): Promise<APIResponse<Agent>> {
-        return this.client.POST<Agent>("/v1/api/agents/import", {
+    async import(bundle: AgentBundle): Promise<APIResponse<ImportAgentResponse>> {
+        return this.client.POST<ImportAgentResponse>("/v1/api/agents/import", {
             body: bundle,
             headers: this.headers(),
         });
     }
+
+    // ======================== Graph ========================
+
+    /**
+     * Get the graph for an agent (Mermaid or JSON).
+     */
+    async getGraph(agentId: string, format?: "mermaid" | "json"): Promise<APIResponse<AgentGraphResponse>> {
+        return this.client.GET<AgentGraphResponse>("/v1/api/agents/{agentId}/graph", {
+            params: { path: { agentId }, query: { format } },
+            headers: this.headers(),
+        });
+    }
+
+    /** Alias for getGraph (convenience) */
+    graph = (agentId: string, format?: "mermaid" | "json") => this.getGraph(agentId, format);
+
+    /**
+     * Validate the graph configuration.
+     */
+    async validateGraph(agentId: string): Promise<APIResponse<GraphValidationResponse>> {
+        return this.client.POST<GraphValidationResponse>("/v1/api/agents/{agentId}/graph/validate", {
+            params: { path: { agentId } },
+            headers: this.headers(),
+        });
+    }
+
+
 }

package/src/modules/apiTokens.ts

@@ -0,0 +1,83 @@
+
+import type { RawClient, APIResponse } from "../client/raw.js";
+
+export interface ApiToken {
+    id: string;
+    name: string;
+    token_prefix: string;
+    scopes: string[];
+    created_at: string;
+    expires_at?: string | null;
+    last_used_at?: string | null;
+}
+
+export interface ApiTokenSecret extends ApiToken {
+    token: string; // The full secret, only returned once
+    workspace_id: string;
+}
+
+export interface CreateTokenRequest {
+    name: string;
+    scopes?: string[];
+    expires_at?: string | null;
+    environment?: "live" | "test";
+}
+
+export interface RotateTokenResponse extends ApiTokenSecret {
+    rotated_from: string;
+}
+
+export class ApiTokensModule {
+    private client: RawClient;
+
+    constructor(client: RawClient) {
+        this.client = client;
+    }
+
+    private headers() {
+        return {
+            "Content-Type": "application/json",
+        };
+    }
+
+    /**
+     * List all API tokens for the current workspace.
+     */
+    async list(): Promise<APIResponse<{ tokens: ApiToken[] }>> {
+        return this.client.GET<{ tokens: ApiToken[] }>("/v1/api/tokens", {
+            headers: this.headers(),
+        });
+    }
+
+    /**
+     * Create a new API token.
+     * The full token secret is only returned in this response.
+     */
+    async create(body: CreateTokenRequest): Promise<APIResponse<ApiTokenSecret>> {
+        return this.client.POST<ApiTokenSecret>("/v1/api/tokens", {
+            body,
+            headers: this.headers(),
+        });
+    }
+
+    /**
+     * Revoke an API token immediately.
+     */
+    async revoke(id: string): Promise<APIResponse<{ message: string }>> {
+        return this.client.DELETE<{ message: string }>("/v1/api/tokens/{id}", {
+            params: { path: { id } },
+            headers: this.headers(),
+        });
+    }
+
+    /**
+     * Rotate a token (revoke old, create new).
+     * Returns the new token secret.
+     */
+    async rotate(id: string): Promise<APIResponse<RotateTokenResponse>> {
+        return this.client.POST<RotateTokenResponse>("/v1/api/tokens/{id}/rotate", {
+            params: { path: { id } },
+            headers: this.headers(),
+        });
+    }
+}

package/src/modules/memberships.ts

@@ -0,0 +1,42 @@
+
+import type { RawClient, APIResponse } from "../client/raw.js";
+
+export interface MembershipResponse {
+    id: string;
+    user_id: string;
+    tenant_id: string;
+    role_id: string;
+    role_name?: string;
+    workspace_id?: string;
+    workspace_name?: string;
+    created_at: string;
+}
+
+export interface EnsureMembershipRequest {
+    tenant_id?: string;
+    workspace_id?: string;
+}
+
+export class MembershipsModule {
+    private client: RawClient;
+
+    constructor(client: RawClient) {
+        this.client = client;
+    }
+
+    private headers() {
+        return {
+            "Content-Type": "application/json",
+        };
+    }
+
+    /**
+     * Ensure the current user has a membership in a tenant (self-provisioning).
+     */
+    async ensure(body: EnsureMembershipRequest): Promise<APIResponse<MembershipResponse>> {
+        return this.client.POST<MembershipResponse>("/v1/api/memberships/ensure", {
+            body,
+            headers: this.headers(),
+        });
+    }
+}

package/src/modules/workspaces.ts

@@ -86,4 +86,65 @@ export class WorkspacesModule {
             headers: this.headers(),
         });
     }
+
+    /**
+     * Get workspace quotas.
+     */
+    async getQuotas(workspaceId: string): Promise<APIResponse<WorkspaceQuotaResponse>> {
+        return this.client.GET<WorkspaceQuotaResponse>("/v1/api/workspaces/{workspaceId}/quotas", {
+            params: { path: { workspaceId } },
+            headers: this.headers(),
+        });
+    }
+
+    /**
+     * Get workspace usage.
+     */
+    async getUsage(workspaceId: string, params?: { from?: string; to?: string }): Promise<APIResponse<WorkspaceUsageResponse>> {
+        return this.client.GET<WorkspaceUsageResponse>("/v1/api/workspaces/{workspaceId}/usage", {
+            params: { path: { workspaceId }, query: params },
+            headers: this.headers(),
+        });
+    }
+}
+
+export interface WorkspaceQuotaResponse {
+    workspace_id: string;
+    quotas: {
+        runs_per_month: Quota;
+        tokens_per_month: Quota;
+        agents_max: Quota;
+        storage_mb: Quota;
+    };
+    reset_at: string;
+    source: string;
+}
+
+export interface Quota {
+    limit: number;
+    used: number;
+    remaining: number;
+}
+
+export interface WorkspaceUsageResponse {
+    workspace_id: string;
+    period: { from: string; to: string };
+    usage: {
+        total_runs: number;
+        completed_runs: number;
+        failed_runs: number;
+        total_tokens: number;
+        input_tokens: number;
+        output_tokens: number;
+        total_duration_ms: number;
+        estimated_count: number;
+    };
+    daily_breakdown: Array<{
+        date: string;
+        runs: number;
+        tokens_in: number;
+        tokens_out: number;
+        duration_ms: number;
+    }>;
+    source: string;
 }