@agent-os-sdk/client 0.9.8 → 0.9.10

package/src/modules/triggers.ts

@@ -11,6 +11,9 @@ import type { RawClient, APIResponse } from "../client/raw.js";
 // Types - Wave 5 Aligned
 // =============================================================================
 
+export type JsonSchema = Record<string, unknown>;
+export type TriggerInputMappingSpec = Record<string, unknown>;
+
 export interface Trigger {
     id: string;
     name: string;
@@ -39,9 +42,19 @@ export interface TriggerRequestContract {
     invoke_url: string;
     headers: Record<string, string>;
     signing: SigningContract;
+    body_schema_json?: string;
+    example_payload_json?: string;
+    curl_example: string;
+    curl_command_parts?: {
+        method: string;
+        url: string;
+        headers: Record<string, string>;
+        body_json?: string | null;
+    } | null;
+    /** @deprecated Use body_schema_json */
     body_schema?: Record<string, unknown>;
+    /** @deprecated Use example_payload_json */
     example_payload?: Record<string, unknown>;
-    curl_example: string;
 }
 
 export interface SigningContract {
@@ -83,10 +96,10 @@ export interface TriggerListResponse {
 export interface TriggerInputContractResponse {
     trigger_id: string;
     template_slug?: string;
-    external_body_schema?: unknown;
-    run_input_schema?: unknown;
-    effective_input_mapping_spec?: unknown;
-    example_payload?: unknown;
+    external_body_schema?: JsonSchema;
+    run_input_schema?: JsonSchema;
+    effective_input_mapping_spec?: TriggerInputMappingSpec;
+    example_payload?: JsonSchema;
     has_input_mapping: boolean;
 }
 
@@ -104,6 +117,28 @@ export interface TestMappingResponse {
     validation_errors?: MappingValidationError[];
 }
 
+export interface InferSchemaPathInfo {
+    path: string;
+    kind: string;
+}
+
+export interface InferSchemaResponse {
+    inferred_schema: JsonSchema;
+    paths: InferSchemaPathInfo[];
+    warnings: string[];
+}
+
+export interface SaveInputMappingResponse {
+    trigger_id: string;
+    input_mapping_override: TriggerInputMappingSpec;
+}
+
+export interface RegeneratedSecretResponse {
+    id: string;
+    secret_key: string;
+    regenerated_at: string;
+}
+
 // =============================================================================
 // Module
 // =============================================================================
@@ -272,5 +307,51 @@ export class TriggersModule {
             headers: this.headers(),
         });
     }
-}
 
+    /**
+     * Regenerate the webhook secret for a trigger.
+     */
+    async regenerateSecret(triggerId: string): Promise<APIResponse<RegeneratedSecretResponse>> {
+        return this.client.POST<RegeneratedSecretResponse>("/v1/api/triggers/{id}/regenerate-secret", {
+            params: { path: { id: triggerId } },
+            headers: this.headers(),
+        });
+    }
+
+    /**
+     * Infer schema from a sample payload.
+     */
+    async inferSchema(triggerId: string, samplePayload: unknown): Promise<APIResponse<InferSchemaResponse>> {
+        return this.client.POST<InferSchemaResponse>("/v1/api/triggers/{id}/infer-schema", {
+            params: { path: { id: triggerId } },
+            body: { sample_payload: samplePayload },
+            headers: this.headers(),
+        });
+    }
+
+    /**
+     * Set/update the input mapping for a trigger.
+     */
+    async putInputMapping(triggerId: string, inputMappingOverride: TriggerInputMappingSpec): Promise<APIResponse<SaveInputMappingResponse>> {
+        return this.client.PUT<SaveInputMappingResponse>("/v1/api/triggers/{id}/input-mapping", {
+            params: { path: { id: triggerId } },
+            body: { input_mapping_override: inputMappingOverride },
+            headers: this.headers(),
+        });
+    }
+
+    /**
+     * Preview input mapping transformation result.
+     */
+    async previewInputMapping(
+        triggerId: string,
+        samplePayload: unknown,
+        inputMappingOverride: TriggerInputMappingSpec
+    ): Promise<APIResponse<TestMappingResponse>> {
+        return this.client.POST<TestMappingResponse>("/v1/api/triggers/{id}/preview-input-mapping", {
+            params: { path: { id: triggerId } },
+            body: { sample_payload: samplePayload, input_mapping_override: inputMappingOverride },
+            headers: this.headers(),
+        });
+    }
+}

package/src/modules/usage.ts

@@ -1,31 +1,116 @@
 /**
- * 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;
+// =============================================================================
+// Types
+// =============================================================================
+
+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;
 }
 
+// =============================================================================
+// Module
+// =============================================================================
+
 export class UsageModule {
     constructor(private client: RawClient, private headers: () => Record<string, string>) { }
 
     /**
-     * Get current usage and quotas.
+     * Get current quota status (tenant-level).
      */
-    async get(): Promise<APIResponse<UsageResponse>> {
+    async getQuotas(params?: {
+        workspaceId?: string;
+        /** @deprecated Use workspaceId */
+        workspace_id?: string;
+    }): Promise<APIResponse<QuotasResponse>> {
+        const query = {
+            workspaceId: params?.workspaceId ?? params?.workspace_id,
+        };
+
+        return this.client.GET<QuotasResponse>("/v1/api/quotas", {
+            params: { query },
+            headers: this.headers(),
+        });
+    }
+
+    /**
+     * Get usage metrics with real data from runtime.cost_events.
+     */
+    async getUsage(params?: {
+        workspaceId?: string;
+        /** @deprecated Use workspaceId */
+        workspace_id?: string;
+        from?: string;
+        to?: string;
+    }): Promise<APIResponse<UsageResponse>> {
+        const query = {
+            workspaceId: params?.workspaceId ?? params?.workspace_id,
+            from: params?.from,
+            to: params?.to,
+        };
+
         return this.client.GET<UsageResponse>("/v1/api/usage", {
+            params: { query },
             headers: this.headers(),
         });
     }

package/src/modules/vectorStores.ts

@@ -11,7 +11,6 @@ type VectorStoreFileResponse = components["schemas"]["VectorStoreFileResponse"];
 export interface VectorStore {
     id: string;
     name: string;
-    workspace_id: string;
     embedding_provider: string;
     embedding_model: string;
     dimension: number;
@@ -52,6 +51,42 @@ export interface VectorStoreFilesResponse {
     total: number;
 }
 
+type VectorStoreLike = VectorStoreResponse & {
+    embeddingProvider?: string;
+    embeddingModel?: string;
+    credentialBindingAlias?: string;
+    fileCount?: number;
+    createdAt?: string;
+};
+
+type VectorStoreFileLike = VectorStoreFileResponse & {
+    fileId?: string;
+    createdAt?: string;
+};
+
+function toVectorStore(item: VectorStoreLike): VectorStore {
+    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: VectorStoreFileLike): VectorStoreFile {
+    return {
+        id: item.id ?? "",
+        file_id: item.file_id ?? item.fileId ?? "",
+        filename: item.filename ?? "",
+        status: (item.status ?? "pending") as VectorStoreFile["status"],
+        created_at: item.created_at ?? item.createdAt ?? "",
+    };
+}
+
 export class VectorStoresModule {
     constructor(private client: RawClient, private headers: () => Record<string, string>) { }
 
@@ -63,20 +98,41 @@ export class VectorStoresModule {
         limit?: number;
         offset?: number;
     }): Promise<APIResponse<VectorStoreListResponse>> {
-        return this.client.GET<VectorStoreListResponse>("/v1/api/vector-stores", {
+        const response = await this.client.GET<VectorStoreResponse[]>("/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: string): Promise<APIResponse<VectorStore>> {
-        return this.client.GET<VectorStore>("/v1/api/vector-stores/{id}", {
-            params: { path: { id: storeId } },
+        const response = await this.client.GET<VectorStoreResponse>("/v1/api/vector-stores/{vectorStoreId}", {
+            params: { path: { vectorStoreId: storeId } },
             headers: this.headers(),
         });
+
+        if (!response.data) {
+            return response as APIResponse<VectorStore>;
+        }
+
+        return {
+            ...response,
+            data: toVectorStore(response.data),
+        };
     }
 
     /**
@@ -89,18 +145,27 @@ export class VectorStoresModule {
         dimension?: number;
         credential_binding_alias?: string;
     }): Promise<APIResponse<VectorStore>> {
-        return this.client.POST<VectorStore>("/v1/api/vector-stores", {
+        const response = await this.client.POST<VectorStoreResponse>("/v1/api/vector-stores", {
             body,
             headers: this.headers(),
         });
+
+        if (!response.data) {
+            return response as APIResponse<VectorStore>;
+        }
+
+        return {
+            ...response,
+            data: toVectorStore(response.data),
+        };
     }
 
     /**
      * Delete a vector store.
      */
     async delete(storeId: string): Promise<APIResponse<void>> {
-        return this.client.DELETE<void>("/v1/api/vector-stores/{id}", {
-            params: { path: { id: storeId } },
+        return this.client.DELETE<void>("/v1/api/vector-stores/{vectorStoreId}", {
+            params: { path: { vectorStoreId: storeId } },
             headers: this.headers(),
         });
     }
@@ -115,11 +180,11 @@ export class VectorStoresModule {
         top_k?: number;
         min_score?: number;
     }): Promise<APIResponse<VectorQueryResult>> {
-        return this.client.POST<VectorQueryResult>("/v1/api/vector-stores/{id}/query", {
-            params: { path: { id: storeId } },
+        return this.client.POST<VectorQueryResponse>("/v1/api/vector-stores/{vectorStoreId}/query", {
+            params: { path: { vectorStoreId: storeId } },
             body,
             headers: this.headers(),
-        });
+        }) as Promise<APIResponse<VectorQueryResult>>;
     }
 
     // ======================== Files ========================
@@ -128,32 +193,64 @@ export class VectorStoresModule {
      * List files in a vector store.
      */
     async listFiles(storeId: string, params?: {
-        limit?: number;
-        offset?: number;
+        skip?: number;
+        take?: number;
     }): Promise<APIResponse<VectorStoreFilesResponse>> {
-        return this.client.GET<VectorStoreFilesResponse>("/v1/api/vector-stores/{id}/files", {
-            params: { path: { id: storeId }, query: params },
+        const response = await this.client.GET<VectorStoreFileResponse[]>("/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: string, fileId: string): Promise<APIResponse<VectorStoreFile>> {
-        return this.client.POST<VectorStoreFile>("/v1/api/vector-stores/{id}/files", {
-            params: { path: { id: storeId } },
+        const response = await this.client.POST<VectorStoreFileResponse>("/v1/api/vector-stores/{vectorStoreId}/files", {
+            params: { path: { vectorStoreId: storeId } },
             body: { file_id: fileId },
             headers: this.headers(),
         });
+
+        if (!response.data) {
+            return response as APIResponse<VectorStoreFile>;
+        }
+
+        return {
+            ...response,
+            data: toVectorStoreFile(response.data),
+        };
     }
 
     /**
      * Remove a file from a vector store.
      */
     async removeFile(storeId: string, fileId: string): Promise<APIResponse<void>> {
-        return this.client.DELETE<void>("/v1/api/vector-stores/{id}/files/{fileId}", {
-            params: { path: { id: storeId, fileId } },
+        return this.client.DELETE<void>("/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: string, agentId: string): Promise<APIResponse<void>> {
+        return this.client.POST<void>("/v1/api/vector-stores/{vectorStoreId}/bind-agent", {
+            params: { path: { vectorStoreId: storeId } },
+            body: { agent_id: agentId },
             headers: this.headers(),
         });
     }

package/src/modules/knowledge.ts

@@ -1,147 +0,0 @@
-/**
- * Knowledge Module - Fully Typed
- */
-
-import type { RawClient, APIResponse } from "../client/raw.js";
-
-export interface KnowledgeDataset {
-    id: string;
-    name: string;
-    description?: string;
-    workspace_id: string;
-    file_count: number;
-    total_chunks: number;
-    created_at: string;
-    updated_at: string;
-}
-
-export interface KnowledgeListResponse {
-    items: KnowledgeDataset[];
-    total: number;
-}
-
-export interface KnowledgeFile {
-    id: string;
-    dataset_id: string;
-    filename: string;
-    size_bytes: number;
-    chunk_count: number;
-    status: "pending" | "processing" | "completed" | "failed";
-    created_at: string;
-}
-
-export interface KnowledgeFileListResponse {
-    items: KnowledgeFile[];
-    total: number;
-}
-
-export interface KnowledgeSearchResult {
-    chunk_id: string;
-    file_id: string;
-    content: string;
-    score: number;
-    metadata?: Record<string, unknown>;
-}
-
-export interface KnowledgeSearchResponse {
-    results: KnowledgeSearchResult[];
-    total: number;
-}
-
-export class KnowledgeModule {
-    constructor(
-        private client: RawClient,
-        private getWorkspaceId: () => string,
-        private headers: () => Record<string, string>
-    ) { }
-
-    /**
-     * List all knowledge datasets.
-     */
-    async list(params?: {
-        workspace_id?: string;
-        limit?: number;
-        offset?: number;
-    }): Promise<APIResponse<KnowledgeListResponse>> {
-        return this.client.GET<KnowledgeListResponse>("/v1/api/knowledge", {
-            params: { query: params },
-            headers: this.headers(),
-        });
-    }
-
-    /**
-     * Get a knowledge dataset by ID.
-     */
-    async get(datasetId: string): Promise<APIResponse<KnowledgeDataset>> {
-        return this.client.GET<KnowledgeDataset>("/v1/api/knowledge/{id}", {
-            params: { path: { id: datasetId } },
-            headers: this.headers(),
-        });
-    }
-
-    /**
-     * Create a new knowledge dataset.
-     */
-    async create(body: {
-        name: string;
-        description?: string;
-    }): Promise<APIResponse<KnowledgeDataset>> {
-        return this.client.POST<KnowledgeDataset>("/v1/api/knowledge", {
-            body,
-            headers: this.headers(),
-        });
-    }
-
-    /**
-     * Delete a knowledge dataset.
-     */
-    async delete(datasetId: string): Promise<APIResponse<void>> {
-        return this.client.DELETE<void>("/v1/api/knowledge/{id}", {
-            params: { path: { id: datasetId } },
-            headers: this.headers(),
-        });
-    }
-
-    // ======================== Files ========================
-
-    /**
-     * List files in a dataset.
-     */
-    async listFiles(datasetId: string, params?: {
-        limit?: number;
-        offset?: number;
-    }): Promise<APIResponse<KnowledgeFileListResponse>> {
-        return this.client.GET<KnowledgeFileListResponse>("/v1/api/knowledge/{id}/files", {
-            params: { path: { id: datasetId }, query: params },
-            headers: this.headers(),
-        });
-    }
-
-    /**
-     * Add a file to a dataset.
-     */
-    async addFile(datasetId: string, fileId: string): Promise<APIResponse<KnowledgeFile>> {
-        return this.client.POST<KnowledgeFile>("/v1/api/knowledge/{id}/files", {
-            params: { path: { id: datasetId } },
-            body: { file_id: fileId },
-            headers: this.headers(),
-        });
-    }
-
-    // ======================== Search ========================
-
-    /**
-     * Search within a knowledge dataset.
-     */
-    async search(datasetId: string, body: {
-        query: string;
-        top_k?: number;
-        min_score?: number;
-    }): Promise<APIResponse<KnowledgeSearchResponse>> {
-        return this.client.POST<KnowledgeSearchResponse>("/v1/api/knowledge/{id}/search", {
-            params: { path: { id: datasetId } },
-            body,
-            headers: this.headers(),
-        });
-    }
-}