@airweave/sdk 0.2.26 → 0.2.52

package/api/types/SourceConnectionStatus.d.ts

@@ -0,0 +1,12 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+/**
+ * Source connection status enum.
+ */
+export type SourceConnectionStatus = "active" | "in_progress" | "failing";
+export declare const SourceConnectionStatus: {
+    readonly Active: "active";
+    readonly InProgress: "in_progress";
+    readonly Failing: "failing";
+};

package/api/types/SourceConnectionStatus.js

@@ -0,0 +1,11 @@
+"use strict";
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SourceConnectionStatus = void 0;
+exports.SourceConnectionStatus = {
+    Active: "active",
+    InProgress: "in_progress",
+    Failing: "failing",
+};

package/api/types/Sync.d.ts

@@ -13,8 +13,6 @@ export interface Sync {
     description?: string;
     cronSchedule?: string;
     nextScheduledRun?: Date;
-    whiteLabelId?: string;
-    whiteLabelUserIdentifier?: string;
     syncMetadata?: Record<string, unknown>;
     status: AirweaveSDK.SyncStatus;
     id: string;

package/{dist/api/resources/sync/client/requests → api/types}/SyncCreate.d.ts

@@ -1,14 +1,9 @@
 /**
  * This file was auto-generated by Fern from our API Definition.
  */
-import * as AirweaveSDK from "../../../../index";
+import * as AirweaveSDK from "../index";
 /**
- * @example
- *     {
- *         name: "name",
- *         sourceConnectionId: "source_connection_id",
- *         destinationConnectionIds: ["destination_connection_ids"]
- *     }
+ * Schema for creating a Sync object.
  */
 export interface SyncCreate {
     name: string;
@@ -18,8 +13,6 @@ export interface SyncCreate {
     description?: string;
     cronSchedule?: string;
     nextScheduledRun?: Date;
-    whiteLabelId?: string;
-    whiteLabelUserIdentifier?: string;
     syncMetadata?: Record<string, unknown>;
     status?: AirweaveSDK.SyncStatus;
     runImmediately?: boolean;

package/api/types/SyncJob.d.ts

@@ -8,20 +8,23 @@ import * as AirweaveSDK from "../index";
 export interface SyncJob {
     syncId: string;
     status?: AirweaveSDK.SyncJobStatus;
-    entitiesDetected?: number;
     entitiesInserted?: number;
+    entitiesUpdated?: number;
     entitiesDeleted?: number;
+    entitiesKept?: number;
     entitiesSkipped?: number;
+    entitiesEncountered?: Record<string, number | undefined>;
+    startedAt?: Date;
+    completedAt?: Date;
+    failedAt?: Date;
     error?: string;
+    accessToken?: string;
     id: string;
     organizationId: string;
     createdByEmail: string;
     modifiedByEmail: string;
     createdAt: Date;
     modifiedAt: Date;
-    startedAt?: Date;
-    completedAt?: Date;
-    failedAt?: Date;
     /** Name of the sync, populated from join query */
     syncName?: string;
 }

package/api/{resources/sync/client/requests → types}/SyncUpdate.d.ts

@@ -1,17 +1,14 @@
 /**
  * This file was auto-generated by Fern from our API Definition.
  */
-import * as AirweaveSDK from "../../../../index";
+import * as AirweaveSDK from "../index";
 /**
- * @example
- *     {}
+ * Schema for updating a Sync object.
  */
 export interface SyncUpdate {
     name?: string;
     cronSchedule?: string;
     nextScheduledRun?: Date;
-    whiteLabelId?: string;
-    whiteLabelUserIdentifier?: string;
     syncMetadata?: Record<string, unknown>;
     status?: AirweaveSDK.SyncStatus;
 }

package/api/types/SyncWithSourceConnection.d.ts

@@ -13,8 +13,6 @@ export interface SyncWithSourceConnection {
     description?: string;
     cronSchedule?: string;
     nextScheduledRun?: Date;
-    whiteLabelId?: string;
-    whiteLabelUserIdentifier?: string;
     syncMetadata?: Record<string, unknown>;
     status: AirweaveSDK.SyncStatus;
     id: string;

package/api/types/User.d.ts

@@ -7,7 +7,7 @@
 export interface User {
     email: string;
     fullName?: string;
-    organizationId: string;
+    organizationId?: string;
     id: string;
     permissions?: string[];
 }

package/api/types/UserCreate.d.ts

@@ -0,0 +1,11 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+/**
+ * Schema for creating a User object.
+ */
+export interface UserCreate {
+    email: string;
+    fullName?: string;
+    organizationId?: string;
+}

package/api/types/WhiteLabel.d.ts

@@ -10,6 +10,7 @@ export interface WhiteLabel {
     redirectUrl: string;
     clientId: string;
     clientSecret: string;
+    allowedOrigins: string;
     id: string;
     organizationId: string;
     createdAt: Date;

package/api/{resources/whiteLabels/client/requests → types}/WhiteLabelCreate.d.ts

@@ -2,14 +2,7 @@
  * This file was auto-generated by Fern from our API Definition.
  */
 /**
- * @example
- *     {
- *         name: "name",
- *         sourceShortName: "source_short_name",
- *         redirectUrl: "redirect_url",
- *         clientId: "client_id",
- *         clientSecret: "client_secret"
- *     }
+ * Schema for creating a WhiteLabel object.
  */
 export interface WhiteLabelCreate {
     name: string;
@@ -17,4 +10,5 @@ export interface WhiteLabelCreate {
     redirectUrl: string;
     clientId: string;
     clientSecret: string;
+    allowedOrigins: string;
 }

package/api/{resources/whiteLabels/client/requests → types}/WhiteLabelUpdate.d.ts

@@ -2,12 +2,12 @@
  * This file was auto-generated by Fern from our API Definition.
  */
 /**
- * @example
- *     {}
+ * Schema for updating a WhiteLabel object.
  */
 export interface WhiteLabelUpdate {
     name?: string;
     redirectUrl?: string;
     clientId?: string;
     clientSecret?: string;
+    allowedOrigins?: string;
 }

package/api/types/index.d.ts

@@ -1,14 +1,19 @@
 export * from "./ApiKey";
-export * from "./ApiKeyWithPlainKey";
+export * from "./ApiKeyCreate";
 export * from "./AuthType";
 export * from "./BodyConnectSlackWithTokenConnectionsDirectTokenSlackPost";
-export * from "./BodySendOauth2CodeConnectionsOauth2SourceCodePost";
+export * from "./BodyCreateCredentialsFromAuthorizationCodeSourceConnectionsSourceShortNameCodeToTokenCredentialsPost";
+export * from "./BodyExchangeWhiteLabelOauth2CodeWhiteLabelsWhiteLabelIdOauth2CodeOptions";
 export * from "./Chat";
 export * from "./ChatCreate";
 export * from "./ChatMessage";
 export * from "./ChatMessageCreate";
 export * from "./ChatUpdate";
+export * from "./Collection";
+export * from "./CollectionStatus";
+export * from "./CollectionUpdate";
 export * from "./ConfigField";
+export * from "./ConfigValues";
 export * from "./Connection";
 export * from "./ConnectionStatus";
 export * from "./DagEdge";
@@ -16,9 +21,10 @@ export * from "./DagEdgeCreate";
 export * from "./DagNode";
 export * from "./DagNodeCreate";
 export * from "./Destination";
-export * from "./DestinationWithConfigFields";
+export * from "./DestinationWithAuthenticationFields";
 export * from "./EmbeddingModel";
-export * from "./EmbeddingModelWithConfigFields";
+export * from "./EmbeddingModelWithAuthenticationFields";
+export * from "./EntityCount";
 export * from "./EntityDefinitionEntitySchema";
 export * from "./EntityDefinition";
 export * from "./EntityDefinitionCreateEntitySchema";
@@ -31,22 +37,39 @@ export * from "./EntityRelationUpdate";
 export * from "./EntityType";
 export * from "./Fields";
 export * from "./HttpValidationError";
+export * from "./IntegrationCredentialInDb";
+export * from "./IntegrationCredentialRawCreate";
 export * from "./IntegrationType";
 export * from "./NodeType";
+export * from "./OAuth2AuthUrl";
+export * from "./Organization";
+export * from "./ResponseType";
+export * from "./SearchResponse";
+export * from "./SearchStatus";
 export * from "./Source";
-export * from "./SourceWithConfigFields";
+export * from "./SourceConnectionAuthFields";
+export * from "./SourceConnection";
+export * from "./SourceConnectionCreate";
+export * from "./SourceConnectionJob";
+export * from "./SourceConnectionListItem";
+export * from "./SourceConnectionStatus";
 export * from "./Sync";
+export * from "./SyncCreate";
 export * from "./SyncDag";
 export * from "./SyncDagCreate";
 export * from "./SyncDagUpdate";
 export * from "./SyncJob";
 export * from "./SyncJobStatus";
 export * from "./SyncStatus";
+export * from "./SyncUpdate";
 export * from "./SyncWithSourceConnection";
 export * from "./Transformer";
 export * from "./TransformerCreate";
 export * from "./TransformerUpdate";
 export * from "./User";
+export * from "./UserCreate";
 export * from "./ValidationErrorLocItem";
 export * from "./ValidationError";
 export * from "./WhiteLabel";
+export * from "./WhiteLabelCreate";
+export * from "./WhiteLabelUpdate";

package/api/types/index.js

@@ -15,16 +15,21 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./ApiKey"), exports);
-__exportStar(require("./ApiKeyWithPlainKey"), exports);
+__exportStar(require("./ApiKeyCreate"), exports);
 __exportStar(require("./AuthType"), exports);
 __exportStar(require("./BodyConnectSlackWithTokenConnectionsDirectTokenSlackPost"), exports);
-__exportStar(require("./BodySendOauth2CodeConnectionsOauth2SourceCodePost"), exports);
+__exportStar(require("./BodyCreateCredentialsFromAuthorizationCodeSourceConnectionsSourceShortNameCodeToTokenCredentialsPost"), exports);
+__exportStar(require("./BodyExchangeWhiteLabelOauth2CodeWhiteLabelsWhiteLabelIdOauth2CodeOptions"), exports);
 __exportStar(require("./Chat"), exports);
 __exportStar(require("./ChatCreate"), exports);
 __exportStar(require("./ChatMessage"), exports);
 __exportStar(require("./ChatMessageCreate"), exports);
 __exportStar(require("./ChatUpdate"), exports);
+__exportStar(require("./Collection"), exports);
+__exportStar(require("./CollectionStatus"), exports);
+__exportStar(require("./CollectionUpdate"), exports);
 __exportStar(require("./ConfigField"), exports);
+__exportStar(require("./ConfigValues"), exports);
 __exportStar(require("./Connection"), exports);
 __exportStar(require("./ConnectionStatus"), exports);
 __exportStar(require("./DagEdge"), exports);
@@ -32,9 +37,10 @@ __exportStar(require("./DagEdgeCreate"), exports);
 __exportStar(require("./DagNode"), exports);
 __exportStar(require("./DagNodeCreate"), exports);
 __exportStar(require("./Destination"), exports);
-__exportStar(require("./DestinationWithConfigFields"), exports);
+__exportStar(require("./DestinationWithAuthenticationFields"), exports);
 __exportStar(require("./EmbeddingModel"), exports);
-__exportStar(require("./EmbeddingModelWithConfigFields"), exports);
+__exportStar(require("./EmbeddingModelWithAuthenticationFields"), exports);
+__exportStar(require("./EntityCount"), exports);
 __exportStar(require("./EntityDefinitionEntitySchema"), exports);
 __exportStar(require("./EntityDefinition"), exports);
 __exportStar(require("./EntityDefinitionCreateEntitySchema"), exports);
@@ -47,22 +53,39 @@ __exportStar(require("./EntityRelationUpdate"), exports);
 __exportStar(require("./EntityType"), exports);
 __exportStar(require("./Fields"), exports);
 __exportStar(require("./HttpValidationError"), exports);
+__exportStar(require("./IntegrationCredentialInDb"), exports);
+__exportStar(require("./IntegrationCredentialRawCreate"), exports);
 __exportStar(require("./IntegrationType"), exports);
 __exportStar(require("./NodeType"), exports);
+__exportStar(require("./OAuth2AuthUrl"), exports);
+__exportStar(require("./Organization"), exports);
+__exportStar(require("./ResponseType"), exports);
+__exportStar(require("./SearchResponse"), exports);
+__exportStar(require("./SearchStatus"), exports);
 __exportStar(require("./Source"), exports);
-__exportStar(require("./SourceWithConfigFields"), exports);
+__exportStar(require("./SourceConnectionAuthFields"), exports);
+__exportStar(require("./SourceConnection"), exports);
+__exportStar(require("./SourceConnectionCreate"), exports);
+__exportStar(require("./SourceConnectionJob"), exports);
+__exportStar(require("./SourceConnectionListItem"), exports);
+__exportStar(require("./SourceConnectionStatus"), exports);
 __exportStar(require("./Sync"), exports);
+__exportStar(require("./SyncCreate"), exports);
 __exportStar(require("./SyncDag"), exports);
 __exportStar(require("./SyncDagCreate"), exports);
 __exportStar(require("./SyncDagUpdate"), exports);
 __exportStar(require("./SyncJob"), exports);
 __exportStar(require("./SyncJobStatus"), exports);
 __exportStar(require("./SyncStatus"), exports);
+__exportStar(require("./SyncUpdate"), exports);
 __exportStar(require("./SyncWithSourceConnection"), exports);
 __exportStar(require("./Transformer"), exports);
 __exportStar(require("./TransformerCreate"), exports);
 __exportStar(require("./TransformerUpdate"), exports);
 __exportStar(require("./User"), exports);
+__exportStar(require("./UserCreate"), exports);
 __exportStar(require("./ValidationErrorLocItem"), exports);
 __exportStar(require("./ValidationError"), exports);
 __exportStar(require("./WhiteLabel"), exports);
+__exportStar(require("./WhiteLabelCreate"), exports);
+__exportStar(require("./WhiteLabelUpdate"), exports);

package/dist/Client.d.ts

@@ -3,13 +3,9 @@
  */
 import * as environments from "./environments";
 import * as core from "./core";
-import { ApiKeys } from "./api/resources/apiKeys/client/Client";
 import { Sources } from "./api/resources/sources/client/Client";
-import { EmbeddingModels } from "./api/resources/embeddingModels/client/Client";
-import { Connections } from "./api/resources/connections/client/Client";
-import { Sync } from "./api/resources/sync/client/Client";
-import { Search } from "./api/resources/search/client/Client";
-import { WhiteLabels } from "./api/resources/whiteLabels/client/Client";
+import { Collections } from "./api/resources/collections/client/Client";
+import { SourceConnections } from "./api/resources/sourceConnections/client/Client";
 export declare namespace AirweaveSDKClient {
     interface Options {
         environment?: core.Supplier<environments.AirweaveSDKEnvironment | string>;
@@ -31,19 +27,11 @@ export declare namespace AirweaveSDKClient {
 }
 export declare class AirweaveSDKClient {
     protected readonly _options: AirweaveSDKClient.Options;
-    protected _apiKeys: ApiKeys | undefined;
     protected _sources: Sources | undefined;
-    protected _embeddingModels: EmbeddingModels | undefined;
-    protected _connections: Connections | undefined;
-    protected _sync: Sync | undefined;
-    protected _search: Search | undefined;
-    protected _whiteLabels: WhiteLabels | undefined;
+    protected _collections: Collections | undefined;
+    protected _sourceConnections: SourceConnections | undefined;
     constructor(_options?: AirweaveSDKClient.Options);
-    get apiKeys(): ApiKeys;
     get sources(): Sources;
-    get embeddingModels(): EmbeddingModels;
-    get connections(): Connections;
-    get sync(): Sync;
-    get search(): Search;
-    get whiteLabels(): WhiteLabels;
+    get collections(): Collections;
+    get sourceConnections(): SourceConnections;
 }

package/dist/Client.js

@@ -4,44 +4,24 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AirweaveSDKClient = void 0;
-const Client_1 = require("./api/resources/apiKeys/client/Client");
-const Client_2 = require("./api/resources/sources/client/Client");
-const Client_3 = require("./api/resources/embeddingModels/client/Client");
-const Client_4 = require("./api/resources/connections/client/Client");
-const Client_5 = require("./api/resources/sync/client/Client");
-const Client_6 = require("./api/resources/search/client/Client");
-const Client_7 = require("./api/resources/whiteLabels/client/Client");
+const Client_1 = require("./api/resources/sources/client/Client");
+const Client_2 = require("./api/resources/collections/client/Client");
+const Client_3 = require("./api/resources/sourceConnections/client/Client");
 class AirweaveSDKClient {
     constructor(_options = {}) {
         this._options = _options;
     }
-    get apiKeys() {
-        var _a;
-        return ((_a = this._apiKeys) !== null && _a !== void 0 ? _a : (this._apiKeys = new Client_1.ApiKeys(this._options)));
-    }
     get sources() {
         var _a;
-        return ((_a = this._sources) !== null && _a !== void 0 ? _a : (this._sources = new Client_2.Sources(this._options)));
-    }
-    get embeddingModels() {
-        var _a;
-        return ((_a = this._embeddingModels) !== null && _a !== void 0 ? _a : (this._embeddingModels = new Client_3.EmbeddingModels(this._options)));
-    }
-    get connections() {
-        var _a;
-        return ((_a = this._connections) !== null && _a !== void 0 ? _a : (this._connections = new Client_4.Connections(this._options)));
-    }
-    get sync() {
-        var _a;
-        return ((_a = this._sync) !== null && _a !== void 0 ? _a : (this._sync = new Client_5.Sync(this._options)));
+        return ((_a = this._sources) !== null && _a !== void 0 ? _a : (this._sources = new Client_1.Sources(this._options)));
     }
-    get search() {
+    get collections() {
         var _a;
-        return ((_a = this._search) !== null && _a !== void 0 ? _a : (this._search = new Client_6.Search(this._options)));
+        return ((_a = this._collections) !== null && _a !== void 0 ? _a : (this._collections = new Client_2.Collections(this._options)));
     }
-    get whiteLabels() {
+    get sourceConnections() {
         var _a;
-        return ((_a = this._whiteLabels) !== null && _a !== void 0 ? _a : (this._whiteLabels = new Client_7.WhiteLabels(this._options)));
+        return ((_a = this._sourceConnections) !== null && _a !== void 0 ? _a : (this._sourceConnections = new Client_3.SourceConnections(this._options)));
     }
 }
 exports.AirweaveSDKClient = AirweaveSDKClient;

package/dist/api/resources/collections/client/Client.d.ts

@@ -0,0 +1,135 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+import * as environments from "../../../../environments";
+import * as core from "../../../../core";
+import * as AirweaveSDK from "../../../index";
+export declare namespace Collections {
+    interface Options {
+        environment?: core.Supplier<environments.AirweaveSDKEnvironment | string>;
+        /** Override the x-api-key header */
+        apiKey?: core.Supplier<string | undefined>;
+    }
+    interface RequestOptions {
+        /** The maximum time to wait for a response in seconds. */
+        timeoutInSeconds?: number;
+        /** The number of times to retry the request. Defaults to 2. */
+        maxRetries?: number;
+        /** A hook to abort the request. */
+        abortSignal?: AbortSignal;
+        /** Override the x-api-key header */
+        apiKey?: string | undefined;
+        /** Additional headers to include in the request. */
+        headers?: Record<string, string>;
+    }
+}
+export declare class Collections {
+    protected readonly _options: Collections.Options;
+    constructor(_options?: Collections.Options);
+    /**
+     * List all collections for the current user's organization.
+     *
+     * @param {AirweaveSDK.ListCollectionsCollectionsGetRequest} request
+     * @param {Collections.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link AirweaveSDK.UnprocessableEntityError}
+     *
+     * @example
+     *     await client.collections.listCollections()
+     */
+    listCollections(request?: AirweaveSDK.ListCollectionsCollectionsGetRequest, requestOptions?: Collections.RequestOptions): Promise<AirweaveSDK.Collection[]>;
+    /**
+     * Create a new collection.
+     *
+     * @param {AirweaveSDK.CollectionCreate} request
+     * @param {Collections.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link AirweaveSDK.UnprocessableEntityError}
+     *
+     * @example
+     *     await client.collections.createCollection({
+     *         name: "name"
+     *     })
+     */
+    createCollection(request: AirweaveSDK.CollectionCreate, requestOptions?: Collections.RequestOptions): Promise<AirweaveSDK.Collection>;
+    /**
+     * Get a specific collection by its readable ID.
+     *
+     * @param {string} readableId
+     * @param {Collections.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link AirweaveSDK.UnprocessableEntityError}
+     *
+     * @example
+     *     await client.collections.getCollection("readable_id")
+     */
+    getCollection(readableId: string, requestOptions?: Collections.RequestOptions): Promise<AirweaveSDK.Collection>;
+    /**
+     * Delete a collection by its readable ID.
+     *
+     * Args:
+     *     readable_id: The readable ID of the collection to delete
+     *     delete_data: Whether to delete the data in destinations
+     *     db: The database session
+     *     current_user: The current user
+     *
+     * Returns:
+     *     The deleted collection
+     *
+     * @param {string} readableId
+     * @param {AirweaveSDK.DeleteCollectionCollectionsReadableIdDeleteRequest} request
+     * @param {Collections.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link AirweaveSDK.UnprocessableEntityError}
+     *
+     * @example
+     *     await client.collections.deleteCollection("readable_id")
+     */
+    deleteCollection(readableId: string, request?: AirweaveSDK.DeleteCollectionCollectionsReadableIdDeleteRequest, requestOptions?: Collections.RequestOptions): Promise<AirweaveSDK.Collection>;
+    /**
+     * Search within a collection identified by readable ID.
+     *
+     * Args:
+     *     readable_id: The readable ID of the collection to search
+     *     query: The search query
+     *     response_type: Type of response (raw results or AI completion)
+     *     db: The database session
+     *     current_user: The current user
+     *
+     * Returns:
+     *     dict: Search results or AI completion response
+     *
+     * @param {string} readableId
+     * @param {AirweaveSDK.SearchCollectionCollectionsReadableIdSearchGetRequest} request
+     * @param {Collections.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link AirweaveSDK.UnprocessableEntityError}
+     *
+     * @example
+     *     await client.collections.searchCollection("readable_id", {
+     *         query: "query"
+     *     })
+     */
+    searchCollection(readableId: string, request: AirweaveSDK.SearchCollectionCollectionsReadableIdSearchGetRequest, requestOptions?: Collections.RequestOptions): Promise<AirweaveSDK.SearchResponse>;
+    /**
+     * Start sync jobs for all source connections in the collection.
+     *
+     * Args:
+     *     readable_id: The readable ID of the collection
+     *     db: The database session
+     *     current_user: The current user
+     *     background_tasks: Background tasks for async operations
+     *
+     * Returns:
+     *     A list of created sync jobs
+     *
+     * @param {string} readableId
+     * @param {Collections.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link AirweaveSDK.UnprocessableEntityError}
+     *
+     * @example
+     *     await client.collections.refreshAllSourceConnections("readable_id")
+     */
+    refreshAllSourceConnections(readableId: string, requestOptions?: Collections.RequestOptions): Promise<AirweaveSDK.SourceConnectionJob[]>;
+}