@agent-os-sdk/client 0.3.15 → 0.4.0

package/src/index.ts

@@ -64,6 +64,68 @@ export {
     type Unwrapped,
 } from "./client/helpers.js";
 
+// ============================================================================
+// Typed Errors (Enterprise)
+// ============================================================================
+export {
+    // Base class
+    AgentOsError,
+    type ErrorOptions,
+    // Error classes
+    UnauthorizedError,
+    ForbiddenError,
+    NotFoundError,
+    ConflictError,
+    ValidationError,
+    RateLimitError,
+    ServerError,
+    NetworkError,
+    TimeoutError,
+    // Field errors
+    type FieldError,
+    // Type guards
+    isAgentOsError,
+    isRetryableError,
+    isAuthError,
+    isClientError,
+    isServerError,
+} from "./errors/index.js";
+
+export { createErrorFromResponse } from "./errors/factory.js";
+
+// ============================================================================
+// Network Configuration
+// ============================================================================
+export {
+    type RetryConfig,
+    type NetworkConfig,
+    DEFAULT_NETWORK_CONFIG,
+    INTERACTIVE_NETWORK_CONFIG,
+    BACKGROUND_NETWORK_CONFIG,
+    mergeNetworkConfig,
+} from "./client/config.js";
+
+// ============================================================================
+// Retry & Timeout Utilities
+// ============================================================================
+export { withRetry, type RetryContext, sleep } from "./client/retry.js";
+export { withTimeout, createTimeoutController } from "./client/timeout.js";
+
+// ============================================================================
+// Pagination Utilities
+// ============================================================================
+export {
+    paginate,
+    collectAll,
+    getFirst,
+    type OffsetPaginatedResponse,
+    type CursorPaginatedResponse,
+    type OffsetParams,
+    type CursorParams,
+    type PaginateOptions,
+} from "./client/pagination.js";
+
+
 // ============================================================================
 // Raw Client & Core Types
 // ============================================================================
@@ -74,6 +136,11 @@ export {
     type TypedClient,
     type ClientOptions,
     type APIResponse,
+    // SDK Hooks for observability (OTEL, Sentry, etc.)
+    type SDKHooks,
+    type HookRequestContext,
+    type HookResponseContext,
+    type HookErrorContext,
 } from "./client/raw.js";
 
 // Export OpenAPI types
@@ -85,7 +152,7 @@ export type { paths, components } from "./client/raw.js";
 
 // Core modules
 export { AgentsModule, type Agent, type AgentVersion, type AgentListResponse, type AgentGraphResponse } from "./modules/agents.js";
-export { RunsModule, type Run, type RunStatus, type RunEvent, type RunListResponse, type RunEventsResponse, type CreateRunResponse, type RunEventsPollResponse, type RunEventDto } from "./modules/runs.js";
+export { RunsModule, type Run, type RunStatus, type RunEvent, type RunListResponse, type RunEventsResponse, type CreateRunResponse, type RunEventsPollResponse, type RunEventDto, type FollowOptions, type FollowEvent } from "./modules/runs.js";
 export { ThreadsModule, type Thread, type ThreadState, type ThreadMessage, type ThreadRun, type ThreadListResponse, type ThreadMessagesResponse } from "./modules/threads.js";
 export { ToolsModule, type Tool, type ToolListResponse } from "./modules/tools.js";
 export { KnowledgeModule, type KnowledgeDataset, type KnowledgeSearchResponse } from "./modules/knowledge.js";
@@ -112,7 +179,7 @@ export { UsageModule, type UsageResponse, type UsageQuota } from "./modules/usag
 export { McpModule, type McpServer } from "./modules/mcp.js";
 export { A2aModule, type JsonRpcRequest, type JsonRpcResponse, type A2aAgentCard } from "./modules/a2a.js";
 export { MeModule, type MeResponse } from "./modules/me.js";
-export { InfoModule, type ServerInfo } from "./modules/info.js";
+export { InfoModule, type ServerInfo, type ServerCapabilities } from "./modules/info.js";
 export { MetricsModule, type MetricsResponse } from "./modules/metrics.js";
 export { GraphsModule, type GraphValidationResult, type GraphIntrospectionResult } from "./modules/graphs.js";
 export { CatalogModule, type CatalogVersions, type NodeCatalogResponse, type ToolCatalogResponse, type TriggerCatalogResponse, type NodeDefinition, type ToolDefinition, type TriggerTemplate } from "./modules/catalog.js";
@@ -120,6 +187,9 @@ export { CatalogModule, type CatalogVersions, type NodeCatalogResponse, type Too
 // Approvals is real (has backend implementation)
 export { ApprovalsModule, type Approval, type ApprovalStatus, type ApprovalDecision, type ApprovalListResponse, type ApprovalStatusResponse } from "./modules/approvals.js";
 
+export { ApiTokensModule, type ApiToken, type ApiTokenSecret, type CreateTokenRequest, type RotateTokenResponse } from "./modules/apiTokens.js";
+export { MembershipsModule, type MembershipResponse, type EnsureMembershipRequest } from "./modules/memberships.js";
+
 // ============================================================================
 // SSE Streaming
 // ============================================================================

package/src/modules/audit.ts

@@ -37,11 +37,21 @@ export class AuditModule {
         workspace_id?: string;
         from?: string;
         to?: string;
-        skip?: number;
-        take?: number;
+        limit?: number;
+        offset?: number;
     }): Promise<APIResponse<AuditListResponse>> {
+        const queryParams: any = { ...params };
+        if (params?.limit !== undefined) {
+            queryParams.take = params.limit;
+            delete queryParams.limit;
+        }
+        if (params?.offset !== undefined) {
+            queryParams.skip = params.offset;
+            delete queryParams.offset;
+        }
+
         return this.client.GET<AuditListResponse>("/v1/api/audit", {
-            params: { query: params },
+            params: { query: queryParams },
             headers: this.headers(),
         });
     }
@@ -63,12 +73,73 @@ export class AuditModule {
         query?: string;
         from?: string;
         to?: string;
-        skip?: number;
-        take?: number;
+        limit?: number;
+        offset?: number;
     }): Promise<APIResponse<AuditListResponse>> {
+        const queryParams: any = { ...params };
+        if (params?.limit !== undefined) {
+            queryParams.take = params.limit;
+            delete queryParams.limit;
+        }
+        if (params?.offset !== undefined) {
+            queryParams.skip = params.offset;
+            delete queryParams.offset;
+        }
+
         return this.client.GET<AuditListResponse>("/v1/api/audit/search", {
-            params: { query: params },
+            params: { query: queryParams },
             headers: this.headers(),
         });
     }
+
+    /**
+     * Iterate through all audit logs with automatic pagination.
+     * 
+     * @example
+     * ```ts
+     * // Get all audit logs for a specific action
+     * for await (const entry of client.audit.iterate({ action: "agent.created" })) {
+     *     console.log(entry.actor, entry.action);
+     * }
+     * ```
+     */
+    async *iterate(
+        filters?: {
+            actor?: string;
+            action?: string;
+            resource?: string;
+            workspace_id?: string;
+            from?: string;
+            to?: string;
+        },
+        options?: { pageSize?: number; maxItems?: number; signal?: AbortSignal }
+    ): AsyncGenerator<AuditLogEntry, void, unknown> {
+        const pageSize = options?.pageSize ?? 100;
+        const maxItems = options?.maxItems ?? Infinity;
+        let offset = 0;
+        let yielded = 0;
+        let hasMore = true;
+
+        while (hasMore && yielded < maxItems) {
+            if (options?.signal?.aborted) return;
+
+            const response = await this.list({
+                ...filters,
+                limit: Math.min(pageSize, maxItems - yielded),
+                offset,
+            });
+
+            if (response.error) throw response.error;
+            const data = response.data!;
+
+            for (const entry of data.items) {
+                if (yielded >= maxItems) return;
+                yield entry;
+                yielded++;
+            }
+
+            offset += data.items.length;
+            hasMore = offset < data.total && data.items.length > 0;
+        }
+    }
 }

package/src/modules/checkpoints.ts

@@ -66,7 +66,7 @@ export class CheckpointsModule {
      * Replay from a checkpoint.
      */
     async replay(checkpointId: string, options?: {
-        mode?: "best_effort" | "deterministic";
+        mode?: "best_effort" | "strict";
         modified_state?: unknown;
     }): Promise<APIResponse<ReplayResponse>> {
         return this.client.POST<ReplayResponse>("/v1/api/checkpoints/{id}/replay", {

package/src/modules/info.ts

@@ -10,8 +10,44 @@ export interface ServerInfo {
     environment: string;
     features?: string[];
     uptime_seconds?: number;
+    commit_hash?: string;
+    api_version?: string;
 }
 
+/** Normalized server capabilities (camelCase, with defaults) */
+export interface ServerCapabilities {
+    /** Server version */
+    version?: string;
+    /** Git commit hash */
+    commitHash?: string;
+    /** API version */
+    apiVersion?: string;
+    /** Environment (production, staging, development) */
+    environment?: string;
+    /** Feature flags - all default to false if not present */
+    features: {
+        sseResume: boolean;
+        cronTriggers: boolean;
+        batchRuns: boolean;
+        hitl: boolean;
+        traces: boolean;
+        vectorStores: boolean;
+        evaluations: boolean;
+        [key: string]: boolean;
+    };
+    /** Server limits */
+    limits?: {
+        maxBatchSize?: number;
+        maxFileSize?: number;
+        [key: string]: number | undefined;
+    };
+    /** SDK version that fetched this (useful for support) */
+    sdkVersion: string;
+}
+
+// SDK version from package (will be replaced at build time or use constant)
+const SDK_VERSION = "0.4.0";
+
 export class InfoModule {
     constructor(private client: RawClient, private headers: () => Record<string, string>) { }
 
@@ -32,4 +68,76 @@ export class InfoModule {
             headers: this.headers(),
         });
     }
+
+    /**
+     * Get normalized server capabilities for feature detection.
+     * 
+     * This method never assumes feature availability - all features
+     * default to false unless explicitly enabled by the server.
+     * 
+     * @example
+     * ```ts
+     * const caps = await client.info.capabilities();
+     * if (caps.data?.features.sseResume) {
+     *     // Use follow() with resume
+     * } else {
+     *     // Fall back to polling
+     * }
+     * ```
+     */
+    async capabilities(): Promise<APIResponse<ServerCapabilities>> {
+        const response = await this.get();
+
+        if (response.error || !response.data) {
+            // Pass through error - construct proper response without data
+            return {
+                error: response.error,
+                response: response.response,
+                meta: response.meta,
+            };
+        }
+
+        const info = response.data;
+
+        // Transform snake_case features array to camelCase object
+        // CRITICAL: Never assume true - default everything to false
+        const featureArray = info.features ?? [];
+        const features: ServerCapabilities["features"] = {
+            sseResume: featureArray.includes("sse_resume"),
+            cronTriggers: featureArray.includes("cron_triggers"),
+            batchRuns: featureArray.includes("batch_runs"),
+            hitl: featureArray.includes("hitl"),
+            traces: featureArray.includes("traces"),
+            vectorStores: featureArray.includes("vector_stores"),
+            evaluations: featureArray.includes("evaluations"),
+        };
+
+        // Add any other features from array as snake_case → camelCase
+        for (const feature of featureArray) {
+            const camelCase = snakeToCamel(feature);
+            if (!(camelCase in features)) {
+                features[camelCase] = true;
+            }
+        }
+
+        const capabilities: ServerCapabilities = {
+            version: info.version,
+            commitHash: info.commit_hash,
+            apiVersion: info.api_version,
+            environment: info.environment,
+            features,
+            sdkVersion: SDK_VERSION,
+        };
+
+        return {
+            data: capabilities,
+            response: response.response,
+            meta: response.meta,
+        };
+    }
+}
+
+/** Convert snake_case to camelCase */
+function snakeToCamel(str: string): string {
+    return str.replace(/_([a-z])/g, (_, letter) => letter.toUpperCase());
 }