@agent-os-sdk/client 0.9.8 → 0.9.10

package/dist/modules/usage.d.ts

@@ -1,26 +1,85 @@
 /**
- * Usage Module - Fully Typed
+ * Usage Module — Tenant-level usage metrics and quotas.
+ *
+ * Endpoints: GET /v1/api/quotas, GET /v1/api/usage
+ * Source: runtime.cost_events aggregation
  */
 import type { RawClient, APIResponse } from "../client/raw.js";
-export interface UsageQuota {
-    resource: string;
+export interface QuotaItem {
     limit: number;
     used: number;
     remaining: number;
 }
+export interface QuotasResponse {
+    tenant_id: string;
+    workspace_id?: string;
+    quotas: {
+        runs_per_month: QuotaItem;
+        tokens_per_month: QuotaItem;
+        agents_max: QuotaItem;
+        storage_mb: QuotaItem;
+    };
+    reset_at: string;
+    source: string;
+}
+export interface UsageAggregated {
+    total_runs: number;
+    completed_runs: number;
+    failed_runs: number;
+    llm_events: number;
+    total_tokens: number;
+    input_tokens: number;
+    output_tokens: number;
+    total_duration_ms: number;
+    estimated_count: number;
+}
+export interface DailyBreakdownItem {
+    date: string;
+    runs: number;
+    llm_events: number;
+    tokens_in: number;
+    tokens_out: number;
+    duration_ms: number;
+}
+export interface ProviderBreakdownItem {
+    provider: string;
+    runs: number;
+    llm_events: number;
+    tokens: number;
+}
 export interface UsageResponse {
-    workspace_id: string;
-    period_start: string;
-    period_end: string;
-    quotas: UsageQuota[];
+    tenant_id: string;
+    workspace_id?: string;
+    period: {
+        from: string;
+        to: string;
+    };
+    usage: UsageAggregated;
+    daily_breakdown: DailyBreakdownItem[];
+    provider_breakdown: ProviderBreakdownItem[];
+    source: string;
 }
 export declare class UsageModule {
     private client;
     private headers;
     constructor(client: RawClient, headers: () => Record<string, string>);
     /**
-     * Get current usage and quotas.
+     * Get current quota status (tenant-level).
+     */
+    getQuotas(params?: {
+        workspaceId?: string;
+        /** @deprecated Use workspaceId */
+        workspace_id?: string;
+    }): Promise<APIResponse<QuotasResponse>>;
+    /**
+     * Get usage metrics with real data from runtime.cost_events.
      */
-    get(): Promise<APIResponse<UsageResponse>>;
+    getUsage(params?: {
+        workspaceId?: string;
+        /** @deprecated Use workspaceId */
+        workspace_id?: string;
+        from?: string;
+        to?: string;
+    }): Promise<APIResponse<UsageResponse>>;
 }
 //# sourceMappingURL=usage.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../../src/modules/usage.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/D,MAAM,WAAW,UAAU;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,UAAU,EAAE,CAAC;CACxB;AAED,qBAAa,WAAW;IACR,OAAO,CAAC,MAAM;IAAa,OAAO,CAAC,OAAO;gBAAlC,MAAM,EAAE,SAAS,EAAU,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAEpF;;OAEG;IACG,GAAG,IAAI,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;CAKnD"}
+{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../../src/modules/usage.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAM/D,MAAM,WAAW,SAAS;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,cAAc;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE;QACJ,cAAc,EAAE,SAAS,CAAC;QAC1B,gBAAgB,EAAE,SAAS,CAAC;QAC5B,UAAU,EAAE,SAAS,CAAC;QACtB,UAAU,EAAE,SAAS,CAAC;KACzB,CAAC;IACF,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,eAAe;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;IACvB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,eAAe,EAAE,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,kBAAkB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,qBAAqB;IAClC,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,aAAa;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC;IACrC,KAAK,EAAE,eAAe,CAAC;IACvB,eAAe,EAAE,kBAAkB,EAAE,CAAC;IACtC,kBAAkB,EAAE,qBAAqB,EAAE,CAAC;IAC5C,MAAM,EAAE,MAAM,CAAC;CAClB;AAMD,qBAAa,WAAW;IACR,OAAO,CAAC,MAAM;IAAa,OAAO,CAAC,OAAO;gBAAlC,MAAM,EAAE,SAAS,EAAU,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAEpF;;OAEG;IACG,SAAS,CAAC,MAAM,CAAC,EAAE;QACrB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,kCAAkC;QAClC,YAAY,CAAC,EAAE,MAAM,CAAC;KACzB,GAAG,OAAO,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;IAWxC;;OAEG;IACG,QAAQ,CAAC,MAAM,CAAC,EAAE;QACpB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,kCAAkC;QAClC,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,EAAE,CAAC,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;CAY1C"}

package/dist/modules/usage.js

@@ -1,6 +1,12 @@
 /**
- * Usage Module - Fully Typed
+ * Usage Module — Tenant-level usage metrics and quotas.
+ *
+ * Endpoints: GET /v1/api/quotas, GET /v1/api/usage
+ * Source: runtime.cost_events aggregation
  */
+// =============================================================================
+// Module
+// =============================================================================
 export class UsageModule {
     client;
     headers;
@@ -9,10 +15,28 @@ export class UsageModule {
         this.headers = headers;
     }
     /**
-     * Get current usage and quotas.
+     * Get current quota status (tenant-level).
      */
-    async get() {
+    async getQuotas(params) {
+        const query = {
+            workspaceId: params?.workspaceId ?? params?.workspace_id,
+        };
+        return this.client.GET("/v1/api/quotas", {
+            params: { query },
+            headers: this.headers(),
+        });
+    }
+    /**
+     * Get usage metrics with real data from runtime.cost_events.
+     */
+    async getUsage(params) {
+        const query = {
+            workspaceId: params?.workspaceId ?? params?.workspace_id,
+            from: params?.from,
+            to: params?.to,
+        };
         return this.client.GET("/v1/api/usage", {
+            params: { query },
             headers: this.headers(),
         });
     }

package/dist/modules/vectorStores.d.ts

@@ -5,7 +5,6 @@ import type { RawClient, APIResponse } from "../client/raw.js";
 export interface VectorStore {
     id: string;
     name: string;
-    workspace_id: string;
     embedding_provider: string;
     embedding_model: string;
     dimension: number;
@@ -82,8 +81,8 @@ export declare class VectorStoresModule {
      * List files in a vector store.
      */
     listFiles(storeId: string, params?: {
-        limit?: number;
-        offset?: number;
+        skip?: number;
+        take?: number;
     }): Promise<APIResponse<VectorStoreFilesResponse>>;
     /**
      * Attach a file to a vector store.
@@ -93,5 +92,9 @@ export declare class VectorStoresModule {
      * Remove a file from a vector store.
      */
     removeFile(storeId: string, fileId: string): Promise<APIResponse<void>>;
+    /**
+     * Bind a vector store to an agent.
+     */
+    bindAgent(storeId: string, agentId: string): Promise<APIResponse<void>>;
 }
 //# sourceMappingURL=vectorStores.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"vectorStores.d.ts","sourceRoot":"","sources":["../../src/modules/vectorStores.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAc,MAAM,kBAAkB,CAAC;AAM3E,MAAM,WAAW,WAAW;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,eAAe,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,uBAAuB;IACpC,KAAK,EAAE,WAAW,EAAE,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,kBAAkB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,kBAAkB,EAAE,CAAC;IAC9B,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,eAAe;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,SAAS,GAAG,YAAY,GAAG,WAAW,GAAG,QAAQ,CAAC;IAC1D,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,wBAAwB;IACrC,KAAK,EAAE,eAAe,EAAE,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,kBAAkB;IACf,OAAO,CAAC,MAAM;IAAa,OAAO,CAAC,OAAO;gBAAlC,MAAM,EAAE,SAAS,EAAU,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAEpF;;OAEG;IACG,IAAI,CAAC,MAAM,CAAC,EAAE;QAChB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,WAAW,CAAC,uBAAuB,CAAC,CAAC;IAOjD;;OAEG;IACG,GAAG,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IAO7D;;OAEG;IACG,MAAM,CAAC,IAAI,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,wBAAwB,CAAC,EAAE,MAAM,CAAC;KACrC,GAAG,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IAOrC;;OAEG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IASzD;;OAEG;IACG,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE;QAC/B,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,SAAS,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAU3C;;OAEG;IACG,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE;QACtC,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,WAAW,CAAC,wBAAwB,CAAC,CAAC;IAOlD;;OAEG;IACG,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;IAQxF;;OAEG;IACG,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;CAMhF"}
+{"version":3,"file":"vectorStores.d.ts","sourceRoot":"","sources":["../../src/modules/vectorStores.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAc,MAAM,kBAAkB,CAAC;AAM3E,MAAM,WAAW,WAAW;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,kBAAkB,EAAE,MAAM,CAAC;IAC3B,eAAe,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,uBAAuB;IACpC,KAAK,EAAE,WAAW,EAAE,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,kBAAkB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,kBAAkB,EAAE,CAAC;IAC9B,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,eAAe;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,SAAS,GAAG,YAAY,GAAG,WAAW,GAAG,QAAQ,CAAC;IAC1D,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,wBAAwB;IACrC,KAAK,EAAE,eAAe,EAAE,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;CACjB;AAsCD,qBAAa,kBAAkB;IACf,OAAO,CAAC,MAAM;IAAa,OAAO,CAAC,OAAO;gBAAlC,MAAM,EAAE,SAAS,EAAU,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAEpF;;OAEG;IACG,IAAI,CAAC,MAAM,CAAC,EAAE;QAChB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,WAAW,CAAC,uBAAuB,CAAC,CAAC;IAmBjD;;OAEG;IACG,GAAG,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IAgB7D;;OAEG;IACG,MAAM,CAAC,IAAI,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,wBAAwB,CAAC,EAAE,MAAM,CAAC;KACrC,GAAG,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IAgBrC;;OAEG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IASzD;;OAEG;IACG,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE;QAC/B,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,SAAS,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAU3C;;OAEG;IACG,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE;QACtC,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,WAAW,CAAC,wBAAwB,CAAC,CAAC;IAmBlD;;OAEG;IACG,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;IAiBxF;;OAEG;IACG,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IAO7E;;OAEG;IACG,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;CAOhF"}

package/dist/modules/vectorStores.js

@@ -1,6 +1,27 @@
 /**
  * Vector Stores Module - Fully Typed
  */
+function toVectorStore(item) {
+    return {
+        id: item.id ?? "",
+        name: item.name ?? "",
+        embedding_provider: item.embedding_provider ?? item.embeddingProvider ?? "",
+        embedding_model: item.embedding_model ?? item.embeddingModel ?? "",
+        dimension: item.dimension ?? 0,
+        credential_binding_alias: item.credential_binding_alias ?? item.credentialBindingAlias ?? undefined,
+        file_count: item.file_count ?? item.fileCount ?? 0,
+        created_at: item.created_at ?? item.createdAt ?? "",
+    };
+}
+function toVectorStoreFile(item) {
+    return {
+        id: item.id ?? "",
+        file_id: item.file_id ?? item.fileId ?? "",
+        filename: item.filename ?? "",
+        status: (item.status ?? "pending"),
+        created_at: item.created_at ?? item.createdAt ?? "",
+    };
+}
 export class VectorStoresModule {
     client;
     headers;
@@ -12,35 +33,59 @@ export class VectorStoresModule {
      * List all vector stores.
      */
     async list(params) {
-        return this.client.GET("/v1/api/vector-stores", {
+        const response = await this.client.GET("/v1/api/vector-stores", {
             params: { query: params },
             headers: this.headers(),
         });
+        const items = Array.isArray(response.data)
+            ? response.data.map((item) => toVectorStore(item))
+            : [];
+        return {
+            ...response,
+            data: {
+                items,
+                total: items.length,
+            },
+        };
     }
     /**
      * Get a vector store by ID.
      */
     async get(storeId) {
-        return this.client.GET("/v1/api/vector-stores/{id}", {
-            params: { path: { id: storeId } },
+        const response = await this.client.GET("/v1/api/vector-stores/{vectorStoreId}", {
+            params: { path: { vectorStoreId: storeId } },
             headers: this.headers(),
         });
+        if (!response.data) {
+            return response;
+        }
+        return {
+            ...response,
+            data: toVectorStore(response.data),
+        };
     }
     /**
      * Create a new vector store.
      */
     async create(body) {
-        return this.client.POST("/v1/api/vector-stores", {
+        const response = await this.client.POST("/v1/api/vector-stores", {
             body,
             headers: this.headers(),
         });
+        if (!response.data) {
+            return response;
+        }
+        return {
+            ...response,
+            data: toVectorStore(response.data),
+        };
     }
     /**
      * Delete a vector store.
      */
     async delete(storeId) {
-        return this.client.DELETE("/v1/api/vector-stores/{id}", {
-            params: { path: { id: storeId } },
+        return this.client.DELETE("/v1/api/vector-stores/{vectorStoreId}", {
+            params: { path: { vectorStoreId: storeId } },
             headers: this.headers(),
         });
     }
@@ -49,8 +94,8 @@ export class VectorStoresModule {
      * Query a vector store for similar content.
      */
     async query(storeId, body) {
-        return this.client.POST("/v1/api/vector-stores/{id}/query", {
-            params: { path: { id: storeId } },
+        return this.client.POST("/v1/api/vector-stores/{vectorStoreId}/query", {
+            params: { path: { vectorStoreId: storeId } },
             body,
             headers: this.headers(),
         });
@@ -60,27 +105,54 @@ export class VectorStoresModule {
      * List files in a vector store.
      */
     async listFiles(storeId, params) {
-        return this.client.GET("/v1/api/vector-stores/{id}/files", {
-            params: { path: { id: storeId }, query: params },
+        const response = await this.client.GET("/v1/api/vector-stores/{vectorStoreId}/files", {
+            params: { path: { vectorStoreId: storeId }, query: params },
             headers: this.headers(),
         });
+        const items = Array.isArray(response.data)
+            ? response.data.map((item) => toVectorStoreFile(item))
+            : [];
+        return {
+            ...response,
+            data: {
+                items,
+                total: items.length,
+            },
+        };
     }
     /**
      * Attach a file to a vector store.
      */
     async attachFile(storeId, fileId) {
-        return this.client.POST("/v1/api/vector-stores/{id}/files", {
-            params: { path: { id: storeId } },
+        const response = await this.client.POST("/v1/api/vector-stores/{vectorStoreId}/files", {
+            params: { path: { vectorStoreId: storeId } },
             body: { file_id: fileId },
             headers: this.headers(),
         });
+        if (!response.data) {
+            return response;
+        }
+        return {
+            ...response,
+            data: toVectorStoreFile(response.data),
+        };
     }
     /**
      * Remove a file from a vector store.
      */
     async removeFile(storeId, fileId) {
-        return this.client.DELETE("/v1/api/vector-stores/{id}/files/{fileId}", {
-            params: { path: { id: storeId, fileId } },
+        return this.client.DELETE("/v1/api/vector-stores/{vectorStoreId}/files/{fileId}", {
+            params: { path: { vectorStoreId: storeId, fileId } },
+            headers: this.headers(),
+        });
+    }
+    /**
+     * Bind a vector store to an agent.
+     */
+    async bindAgent(storeId, agentId) {
+        return this.client.POST("/v1/api/vector-stores/{vectorStoreId}/bind-agent", {
+            params: { path: { vectorStoreId: storeId } },
+            body: { agent_id: agentId },
             headers: this.headers(),
         });
     }

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@agent-os-sdk/client",
-    "version": "0.9.8",
+    "version": "0.9.10",
     "description": "Official TypeScript SDK for Agent OS platform",
     "type": "module",
     "main": "./dist/index.js",

package/src/client/AgentOsClient.ts

@@ -40,7 +40,8 @@ import { AgentsModule } from "../modules/agents.js";
 import { BuilderModule } from "../modules/builder.js";
 import { ChatwootModule } from "../modules/chatwoot.js";
 import { CredentialsModule } from "../modules/credentials.js";
-import { KnowledgeModule } from "../modules/knowledge.js";
+import { DatasetsModule } from "../modules/datasets.js";
+import { RolesModule } from "../modules/roles.js";
 import { MembersModule } from "../modules/members.js";
 import { RunsModule } from "../modules/runs.js";
 import { TenantsModule } from "../modules/tenants.js";
@@ -65,6 +66,7 @@ import { InfoModule } from "../modules/info.js";
 import { MeModule } from "../modules/me.js";
 import { MembershipsModule } from "../modules/memberships.js";
 import { MetricsModule } from "../modules/metrics.js";
+import { ObservabilityModule } from "../modules/observability.js";
 import { PlaygroundModule } from "../modules/playground.js";
 import { PresetsModule } from "../modules/presets.js";
 import { PromptsModule } from "../modules/prompts.js";
@@ -88,7 +90,8 @@ export class AgentOsClient {
     readonly runs: RunsModule;
     readonly threads: ThreadsModule;
     readonly tools: ToolsModule;
-    readonly knowledge: KnowledgeModule;
+    readonly datasets: DatasetsModule;
+    readonly roles: RolesModule;
     readonly triggers: TriggersModule;
     readonly credentials: CredentialsModule;
     readonly builder: BuilderModule;
@@ -117,6 +120,7 @@ export class AgentOsClient {
     readonly catalog: CatalogModule;
     readonly approvals: ApprovalsModule;
     readonly auth: AuthModule;
+    readonly observability: ObservabilityModule;
 
     // Convenience aliases
     readonly experiments: {
@@ -143,6 +147,7 @@ export class AgentOsClient {
         this._client = createRawClient({
             baseUrl: options.baseUrl,
             headerProvider: () => this._resolveHeaders(),
+            contextProvider: options.contextProvider,
             network: options.network,
             hooks: options.hooks,
         });
@@ -157,7 +162,8 @@ export class AgentOsClient {
         this.runs = new RunsModule(this._client);
         this.threads = new ThreadsModule(this._client, getHeaders);
         this.tools = new ToolsModule(this._client, getHeaders);
-        this.knowledge = new KnowledgeModule(this._client, getWorkspaceId, getHeaders);
+        this.datasets = new DatasetsModule(this._client, getHeaders);
+        this.roles = new RolesModule(this._client, getHeaders);
         this.triggers = new TriggersModule(this._client, getHeaders);
         this.credentials = new CredentialsModule(this._client, getWorkspaceId, getHeaders);
         this.builder = new BuilderModule(this._client);
@@ -186,6 +192,7 @@ export class AgentOsClient {
         this.catalog = new CatalogModule(this._client);
         this.approvals = new ApprovalsModule(this._client, getHeaders);
         this.auth = new AuthModule(this._client, getHeaders);
+        this.observability = new ObservabilityModule(this._client, getHeaders);
 
         // Initialize convenience aliases
         this.experiments = {

package/src/client/HttpRequestBuilder.ts

@@ -23,6 +23,11 @@ declare const __SDK_VERSION__: string | undefined;
 const SDK_VERSION = typeof __SDK_VERSION__ !== "undefined" ? __SDK_VERSION__ : "0.7.12-dev";
 const CORRELATION_ID_PATTERN = /^[A-Za-z0-9._:-]{8,128}$/;
 
+export function sanitizeValidCorrelationId(raw?: string): string | null {
+    const candidate = sanitizeHeader((raw ?? "").trim());
+    return CORRELATION_ID_PATTERN.test(candidate) ? candidate : null;
+}
+
 /**
  * Options for building request headers
  */
@@ -81,8 +86,8 @@ export class HttpRequestBuilder {
     }
 
     private resolveCorrelationId(raw?: string): string {
-        const candidate = sanitizeHeader((raw ?? "").trim());
-        if (CORRELATION_ID_PATTERN.test(candidate)) {
+        const candidate = sanitizeValidCorrelationId(raw);
+        if (candidate) {
             return candidate;
         }
 

package/src/client/auth.ts

@@ -8,6 +8,7 @@
 
 import type { NetworkConfig } from "./config.js";
 import type { SDKHooks } from "./raw.js";
+import type { OperationContextProvider } from "./OperationContextProvider.js";
 
 // ============================================================================
 // Auth Provider Types
@@ -89,6 +90,9 @@ export type AgentOsClientOptions = {
     network?: Partial<NetworkConfig>
     /** Optional request lifecycle hooks for observability */
     hooks?: SDKHooks
+    /** Optional flow context provider (ALS/manual) for sticky correlation per operation. */
+    /** Recommended for long-lived server clients to avoid process-wide sticky fallback correlation. */
+    contextProvider?: OperationContextProvider
     /** Custom headers to add to all requests */
     headers?: Record<string, string>
 }

package/src/client/raw.ts

@@ -9,8 +9,9 @@ import createClient, { type Client } from "openapi-fetch";
 import type { components, paths } from "../generated/openapi.js";
 import { createErrorFromResponse } from "../errors/factory.js";
 import { AgentOsError, NetworkError, TimeoutError } from "../errors/index.js";
-import { httpRequestBuilder } from "./HttpRequestBuilder.js";
+import { httpRequestBuilder, sanitizeValidCorrelationId } from "./HttpRequestBuilder.js";
 import type { OperationContext } from "./OperationContext.js";
+import type { OperationContextProvider } from "./OperationContextProvider.js";
 import { mergeNetworkConfig, type NetworkConfig } from "./config.js";
 import { withRetry } from "./retry.js";
 import { withTimeout } from "./timeout.js";
@@ -22,6 +23,8 @@ export type ClientOptions = {
     baseUrl: string;
     headers?: Record<string, string>;
     headerProvider?: () => Promise<Record<string, string>>;
+    /** Optional flow context provider (ALS/manual). */
+    contextProvider?: OperationContextProvider;
     /** Network behavior (timeouts/retries). Defaults to DEFAULT_NETWORK_CONFIG. */
     network?: Partial<NetworkConfig>;
     /** Hooks for observability (OTEL, Sentry, etc.) */
@@ -39,6 +42,8 @@ export interface SDKHooks {
     onResponse?: (context: HookResponseContext) => void | Promise<void>;
     /** Called on any error (network, timeout, HTTP error) */
     onError?: (context: HookErrorContext) => void | Promise<void>;
+    /** Optional logger bridge used by SDK internal warnings. */
+    logger?: { warn?: (message: string) => void };
 }
 
 export interface HookRequestContext {
@@ -106,8 +111,67 @@ export function createTypedClient(options: ClientOptions): TypedClient {
  * Wraps openapi-fetch to provide the old interface while maintaining types.
  */
 export function createRawClient(options: ClientOptions) {
-    const { baseUrl, headers: defaultHeaders = {}, headerProvider } = options;
+    const { baseUrl, headers: defaultHeaders = {}, headerProvider, contextProvider } = options;
     const networkConfig = mergeNetworkConfig(options.network);
+    const warn = (message: string): void => {
+        const hookWarn = options.hooks?.logger?.warn;
+        if (typeof hookWarn === "function") {
+            hookWarn(message);
+            return;
+        }
+        if (typeof console !== "undefined" && typeof console.warn === "function") {
+            console.warn(message);
+        }
+    };
+    // Sticky fallback context for clients that don't provide per-operation context.
+    const fallbackContext: OperationContext = Object.freeze({
+        correlationId: crypto.randomUUID(),
+    });
+    let warnedStickyFallbackContext = false;
+    let warnedInvalidContextCorrelation = false;
+    const resolveEffectiveContext = (optsContext?: OperationContext): OperationContext => {
+        const providerContext = contextProvider?.get();
+        const usingStickyFallback = !optsContext && !providerContext;
+        if (usingStickyFallback && !contextProvider && !warnedStickyFallbackContext) {
+            warnedStickyFallbackContext = true;
+            warn(
+                "[AgentOS SDK] Using sticky fallback correlation context per client instance. " +
+                "For long-lived server clients, configure contextProvider (ALS/manual) to scope correlation per request."
+            );
+        }
+
+        const effectiveContext = optsContext ?? providerContext ?? fallbackContext;
+        const correlationProvidedByCaller =
+            effectiveContext !== fallbackContext
+            && typeof effectiveContext?.correlationId === "string";
+
+        if (correlationProvidedByCaller) {
+            const validCorrelation = sanitizeValidCorrelationId(effectiveContext.correlationId);
+            if (!validCorrelation) {
+                if (!warnedInvalidContextCorrelation) {
+                    warnedInvalidContextCorrelation = true;
+                    warn(
+                        "[AgentOS SDK] Invalid correlationId in operation context ignored; generated fallback correlation id. " +
+                        "Provide a valid correlationId (8-128 chars: A-Za-z0-9._:-)."
+                    );
+                }
+                // Caller provided invalid flow id: use per-request fallback (not sticky fallback context).
+                return {
+                    ...effectiveContext,
+                    correlationId: crypto.randomUUID(),
+                };
+            }
+
+            if (validCorrelation !== effectiveContext.correlationId) {
+                return {
+                    ...effectiveContext,
+                    correlationId: validCorrelation,
+                };
+            }
+        }
+
+        return effectiveContext;
+    };
     type ErrorPayload = { code?: string; message?: string; details?: unknown } & Record<string, unknown>;
     type HttpAgentOsError = AgentOsError & {
         response?: Response;
@@ -173,9 +237,11 @@ export function createRawClient(options: ClientOptions) {
             headers["Content-Type"] = "application/json";
         }
 
+        const effectiveContext = resolveEffectiveContext(opts?.context);
+
         // Build standard headers via HttpRequestBuilder (single source of truth)
         const builderHeaders = httpRequestBuilder.build({
-            context: opts?.context,
+            context: effectiveContext,
             idempotencyKey: opts?.idempotencyKey,
             requireFlow: opts?.requireFlow,
         });
@@ -357,7 +423,12 @@ export function createRawClient(options: ClientOptions) {
          */
         streamGet: async (
             path: string,
-            opts?: { params?: { path?: Record<string, string>; query?: Record<string, unknown> }; headers?: Record<string, string> }
+            opts?: {
+                params?: { path?: Record<string, string>; query?: Record<string, unknown> };
+                headers?: Record<string, string>;
+                context?: OperationContext;
+                requireFlow?: boolean;
+            }
         ): Promise<Response> => {
             let url = path;
             if (opts?.params?.path) {
@@ -377,9 +448,15 @@ export function createRawClient(options: ClientOptions) {
             }
             const fullUrl = `${baseUrl}${url}`;
             const dynamicHeaders = headerProvider ? await headerProvider() : {};
+            const effectiveContext = resolveEffectiveContext(opts?.context);
+            const builderHeaders = httpRequestBuilder.build({
+                context: effectiveContext,
+                requireFlow: opts?.requireFlow,
+            });
             const headers: Record<string, string> = {
                 ...defaultHeaders,
                 ...dynamicHeaders,
+                ...builderHeaders,
                 Accept: "text/event-stream",
                 ...opts?.headers,
             };
@@ -397,6 +474,9 @@ export function createRawClient(options: ClientOptions) {
                 body?: unknown;
                 headers?: Record<string, string>;
                 signal?: AbortSignal;
+                context?: OperationContext;
+                idempotencyKey?: string;
+                requireFlow?: boolean;
             }
         ): Promise<Response> => {
             let url = path;
@@ -407,9 +487,16 @@ export function createRawClient(options: ClientOptions) {
             }
             const fullUrl = `${baseUrl}${url}`;
             const dynamicHeaders = headerProvider ? await headerProvider() : {};
+            const effectiveContext = resolveEffectiveContext(opts?.context);
+            const builderHeaders = httpRequestBuilder.build({
+                context: effectiveContext,
+                idempotencyKey: opts?.idempotencyKey,
+                requireFlow: opts?.requireFlow,
+            });
             const headers: Record<string, string> = {
                 ...defaultHeaders,
                 ...dynamicHeaders,
+                ...builderHeaders,
                 "Content-Type": "application/json",
                 Accept: "text/event-stream",
                 ...opts?.headers,