@airweave/sdk 0.9.49 → 0.9.51

package/dist/esm/api/resources/collections/client/Client.mjs

@@ -15,6 +15,7 @@ import * as core from "../../../../core/index.mjs";
 import * as AirweaveSDK from "../../../index.mjs";
 import { mergeHeaders, mergeOnlyDefinedHeaders } from "../../../../core/headers.mjs";
 import * as errors from "../../../../errors/index.mjs";
+import { Search } from "../resources/search/client/Client.mjs";
 /**
  * API endpoints for managing collections - logical groups of data sources that provide unified search capabilities
  */
@@ -22,6 +23,10 @@ export class Collections {
     constructor(_options) {
         this._options = _options;
     }
+    get search() {
+        var _a;
+        return ((_a = this._search) !== null && _a !== void 0 ? _a : (this._search = new Search(this._options)));
+    }
     /**
      * Retrieve all collections belonging to your organization.
      *
@@ -402,267 +407,6 @@ export class Collections {
             }
         });
     }
-    /**
-     * Direct vector search.
-     *
-     * @param {string} readableId
-     * @param {AirweaveSDK.InstantSearchRequest} request
-     * @param {Collections.RequestOptions} requestOptions - Request-specific configuration.
-     *
-     * @throws {@link AirweaveSDK.UnprocessableEntityError}
-     *
-     * @example
-     *     await client.collections.instantSearch("readable_id", {
-     *         query: "query"
-     *     })
-     */
-    instantSearch(readableId, request, requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__instantSearch(readableId, request, requestOptions));
-    }
-    __instantSearch(readableId, request, requestOptions) {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c;
-            let _headers = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, mergeOnlyDefinedHeaders(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
-            const _response = yield core.fetcher({
-                url: core.url.join((_c = (_b = (yield core.Supplier.get(this._options.baseUrl))) !== null && _b !== void 0 ? _b : (yield core.Supplier.get(this._options.environment))) !== null && _c !== void 0 ? _c : environments.AirweaveSDKEnvironment.Production, `collections/${encodeURIComponent(readableId)}/search/instant`),
-                method: "POST",
-                headers: _headers,
-                contentType: "application/json",
-                queryParameters: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.queryParams,
-                requestType: "json",
-                body: request,
-                timeoutMs: (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) != null ? requestOptions.timeoutInSeconds * 1000 : 60000,
-                maxRetries: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries,
-                abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
-            });
-            if (_response.ok) {
-                return { data: _response.body, rawResponse: _response.rawResponse };
-            }
-            if (_response.error.reason === "status-code") {
-                switch (_response.error.statusCode) {
-                    case 422:
-                        throw new AirweaveSDK.UnprocessableEntityError(_response.error.body, _response.rawResponse);
-                    default:
-                        throw new errors.AirweaveSDKError({
-                            statusCode: _response.error.statusCode,
-                            body: _response.error.body,
-                            rawResponse: _response.rawResponse,
-                        });
-                }
-            }
-            switch (_response.error.reason) {
-                case "non-json":
-                    throw new errors.AirweaveSDKError({
-                        statusCode: _response.error.statusCode,
-                        body: _response.error.rawBody,
-                        rawResponse: _response.rawResponse,
-                    });
-                case "timeout":
-                    throw new errors.AirweaveSDKTimeoutError("Timeout exceeded when calling POST /collections/{readable_id}/search/instant.");
-                case "unknown":
-                    throw new errors.AirweaveSDKError({
-                        message: _response.error.errorMessage,
-                        rawResponse: _response.rawResponse,
-                    });
-            }
-        });
-    }
-    /**
-     * AI-optimized search.
-     *
-     * @param {string} readableId
-     * @param {AirweaveSDK.ClassicSearchRequest} request
-     * @param {Collections.RequestOptions} requestOptions - Request-specific configuration.
-     *
-     * @throws {@link AirweaveSDK.UnprocessableEntityError}
-     *
-     * @example
-     *     await client.collections.classicSearch("readable_id", {
-     *         query: "query"
-     *     })
-     */
-    classicSearch(readableId, request, requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__classicSearch(readableId, request, requestOptions));
-    }
-    __classicSearch(readableId, request, requestOptions) {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c;
-            let _headers = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, mergeOnlyDefinedHeaders(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
-            const _response = yield core.fetcher({
-                url: core.url.join((_c = (_b = (yield core.Supplier.get(this._options.baseUrl))) !== null && _b !== void 0 ? _b : (yield core.Supplier.get(this._options.environment))) !== null && _c !== void 0 ? _c : environments.AirweaveSDKEnvironment.Production, `collections/${encodeURIComponent(readableId)}/search/classic`),
-                method: "POST",
-                headers: _headers,
-                contentType: "application/json",
-                queryParameters: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.queryParams,
-                requestType: "json",
-                body: request,
-                timeoutMs: (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) != null ? requestOptions.timeoutInSeconds * 1000 : 60000,
-                maxRetries: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries,
-                abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
-            });
-            if (_response.ok) {
-                return { data: _response.body, rawResponse: _response.rawResponse };
-            }
-            if (_response.error.reason === "status-code") {
-                switch (_response.error.statusCode) {
-                    case 422:
-                        throw new AirweaveSDK.UnprocessableEntityError(_response.error.body, _response.rawResponse);
-                    default:
-                        throw new errors.AirweaveSDKError({
-                            statusCode: _response.error.statusCode,
-                            body: _response.error.body,
-                            rawResponse: _response.rawResponse,
-                        });
-                }
-            }
-            switch (_response.error.reason) {
-                case "non-json":
-                    throw new errors.AirweaveSDKError({
-                        statusCode: _response.error.statusCode,
-                        body: _response.error.rawBody,
-                        rawResponse: _response.rawResponse,
-                    });
-                case "timeout":
-                    throw new errors.AirweaveSDKTimeoutError("Timeout exceeded when calling POST /collections/{readable_id}/search/classic.");
-                case "unknown":
-                    throw new errors.AirweaveSDKError({
-                        message: _response.error.errorMessage,
-                        rawResponse: _response.rawResponse,
-                    });
-            }
-        });
-    }
-    /**
-     * Agent that iteratively searches, reads, navigates hierarchies, and collects results.
-     *
-     * @param {string} readableId
-     * @param {AirweaveSDK.AgenticSearchRequest} request
-     * @param {Collections.RequestOptions} requestOptions - Request-specific configuration.
-     *
-     * @throws {@link AirweaveSDK.UnprocessableEntityError}
-     *
-     * @example
-     *     await client.collections.agenticSearch("readable_id", {
-     *         query: "query"
-     *     })
-     */
-    agenticSearch(readableId, request, requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__agenticSearch(readableId, request, requestOptions));
-    }
-    __agenticSearch(readableId, request, requestOptions) {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c;
-            let _headers = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, mergeOnlyDefinedHeaders(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
-            const _response = yield core.fetcher({
-                url: core.url.join((_c = (_b = (yield core.Supplier.get(this._options.baseUrl))) !== null && _b !== void 0 ? _b : (yield core.Supplier.get(this._options.environment))) !== null && _c !== void 0 ? _c : environments.AirweaveSDKEnvironment.Production, `collections/${encodeURIComponent(readableId)}/search/agentic`),
-                method: "POST",
-                headers: _headers,
-                contentType: "application/json",
-                queryParameters: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.queryParams,
-                requestType: "json",
-                body: request,
-                timeoutMs: (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) != null ? requestOptions.timeoutInSeconds * 1000 : 60000,
-                maxRetries: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries,
-                abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
-            });
-            if (_response.ok) {
-                return { data: _response.body, rawResponse: _response.rawResponse };
-            }
-            if (_response.error.reason === "status-code") {
-                switch (_response.error.statusCode) {
-                    case 422:
-                        throw new AirweaveSDK.UnprocessableEntityError(_response.error.body, _response.rawResponse);
-                    default:
-                        throw new errors.AirweaveSDKError({
-                            statusCode: _response.error.statusCode,
-                            body: _response.error.body,
-                            rawResponse: _response.rawResponse,
-                        });
-                }
-            }
-            switch (_response.error.reason) {
-                case "non-json":
-                    throw new errors.AirweaveSDKError({
-                        statusCode: _response.error.statusCode,
-                        body: _response.error.rawBody,
-                        rawResponse: _response.rawResponse,
-                    });
-                case "timeout":
-                    throw new errors.AirweaveSDKTimeoutError("Timeout exceeded when calling POST /collections/{readable_id}/search/agentic.");
-                case "unknown":
-                    throw new errors.AirweaveSDKError({
-                        message: _response.error.errorMessage,
-                        rawResponse: _response.rawResponse,
-                    });
-            }
-        });
-    }
-    /**
-     * Streaming agentic search via Server-Sent Events. Returns real-time events as the agent searches.
-     */
-    streamAgenticSearch(readableId, request, requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__streamAgenticSearch(readableId, request, requestOptions));
-    }
-    __streamAgenticSearch(readableId, request, requestOptions) {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c;
-            let _headers = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, mergeOnlyDefinedHeaders(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
-            const _response = yield core.fetcher({
-                url: core.url.join((_c = (_b = (yield core.Supplier.get(this._options.baseUrl))) !== null && _b !== void 0 ? _b : (yield core.Supplier.get(this._options.environment))) !== null && _c !== void 0 ? _c : environments.AirweaveSDKEnvironment.Production, `collections/${encodeURIComponent(readableId)}/search/agentic/stream`),
-                method: "POST",
-                headers: _headers,
-                contentType: "application/json",
-                queryParameters: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.queryParams,
-                requestType: "json",
-                body: request,
-                responseType: "sse",
-                timeoutMs: (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) != null ? requestOptions.timeoutInSeconds * 1000 : 60000,
-                maxRetries: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries,
-                abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
-            });
-            if (_response.ok) {
-                return {
-                    data: new core.Stream({
-                        stream: _response.body,
-                        parse: (data) => data,
-                        signal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
-                        eventShape: {
-                            type: "sse",
-                            streamTerminator: "[DONE]",
-                        },
-                    }),
-                    rawResponse: _response.rawResponse,
-                };
-            }
-            if (_response.error.reason === "status-code") {
-                switch (_response.error.statusCode) {
-                    case 422:
-                        throw new AirweaveSDK.UnprocessableEntityError(_response.error.body, _response.rawResponse);
-                    default:
-                        throw new errors.AirweaveSDKError({
-                            statusCode: _response.error.statusCode,
-                            body: _response.error.body,
-                            rawResponse: _response.rawResponse,
-                        });
-                }
-            }
-            switch (_response.error.reason) {
-                case "non-json":
-                    throw new errors.AirweaveSDKError({
-                        statusCode: _response.error.statusCode,
-                        body: _response.error.rawBody,
-                        rawResponse: _response.rawResponse,
-                    });
-                case "timeout":
-                    throw new errors.AirweaveSDKTimeoutError("Timeout exceeded when calling POST /collections/{readable_id}/search/agentic/stream.");
-                case "unknown":
-                    throw new errors.AirweaveSDKError({
-                        message: _response.error.errorMessage,
-                        rawResponse: _response.rawResponse,
-                    });
-            }
-        });
-    }
     _getCustomAuthorizationHeaders() {
         return __awaiter(this, void 0, void 0, function* () {
             const apiKeyValue = yield core.Supplier.get(this._options.apiKey);

package/dist/esm/api/resources/collections/client/requests/index.d.mts

@@ -1,5 +1,3 @@
 export { type ListCollectionsGetRequest } from "./ListCollectionsGetRequest.mjs";
 export { type CollectionCreate } from "./CollectionCreate.mjs";
 export { type CollectionUpdate } from "./CollectionUpdate.mjs";
-export { type InstantSearchRequest } from "./InstantSearchRequest.mjs";
-export { type ClassicSearchRequest } from "./ClassicSearchRequest.mjs";

package/dist/esm/api/resources/collections/index.d.mts

@@ -1 +1,2 @@
 export * from "./client/index.mjs";
+export * from "./resources/index.mjs";

package/dist/esm/api/resources/collections/index.mjs

@@ -1 +1,2 @@
 export * from "./client/index.mjs";
+export * from "./resources/index.mjs";

package/dist/esm/api/resources/collections/resources/index.d.mts

@@ -0,0 +1,2 @@
+export * as search from "./search/index.mjs";
+export * from "./search/client/requests/index.mjs";

package/dist/esm/api/resources/collections/resources/index.mjs

@@ -0,0 +1,2 @@
+export * as search from "./search/index.mjs";
+export * from "./search/client/requests/index.mjs";

package/dist/esm/api/resources/collections/resources/search/client/Client.d.mts

@@ -0,0 +1,90 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+import * as environments from "../../../../../../environments.mjs";
+import * as core from "../../../../../../core/index.mjs";
+import * as AirweaveSDK from "../../../../../index.mjs";
+export declare namespace Search {
+    interface Options {
+        environment?: core.Supplier<environments.AirweaveSDKEnvironment | string>;
+        /** Specify a custom URL to connect the client to. */
+        baseUrl?: core.Supplier<string>;
+        apiKey: core.Supplier<string>;
+        /** Additional headers to include in requests. */
+        headers?: Record<string, string | core.Supplier<string | null | undefined> | null | 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;
+        /** Additional query string parameters to include in the request. */
+        queryParams?: Record<string, unknown>;
+        /** Additional headers to include in the request. */
+        headers?: Record<string, string | core.Supplier<string | null | undefined> | null | undefined>;
+    }
+}
+export declare class Search {
+    protected readonly _options: Search.Options;
+    constructor(_options: Search.Options);
+    /**
+     * Direct vector search.
+     *
+     * @param {string} readableId
+     * @param {AirweaveSDK.collections.InstantSearchRequest} request
+     * @param {Search.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link AirweaveSDK.UnprocessableEntityError}
+     *
+     * @example
+     *     await client.collections.search.instant("customer-support-tickets-x7k9m", {
+     *         query: "How do I reset my password?",
+     *         retrieval_strategy: "hybrid",
+     *         limit: 10
+     *     })
+     */
+    instant(readableId: string, request: AirweaveSDK.collections.InstantSearchRequest, requestOptions?: Search.RequestOptions): core.HttpResponsePromise<AirweaveSDK.SearchV2Response>;
+    private __instant;
+    /**
+     * AI-optimized search.
+     *
+     * @param {string} readableId
+     * @param {AirweaveSDK.collections.ClassicSearchRequest} request
+     * @param {Search.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link AirweaveSDK.UnprocessableEntityError}
+     *
+     * @example
+     *     await client.collections.search.classic("customer-support-tickets-x7k9m", {
+     *         query: "quarterly revenue report",
+     *         limit: 10
+     *     })
+     */
+    classic(readableId: string, request: AirweaveSDK.collections.ClassicSearchRequest, requestOptions?: Search.RequestOptions): core.HttpResponsePromise<AirweaveSDK.SearchV2Response>;
+    private __classic;
+    /**
+     * Agent that iteratively searches, reads, navigates hierarchies, and collects results.
+     *
+     * @param {string} readableId
+     * @param {AirweaveSDK.AgenticSearchRequest} request
+     * @param {Search.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link AirweaveSDK.UnprocessableEntityError}
+     *
+     * @example
+     *     await client.collections.search.agentic("customer-support-tickets-x7k9m", {
+     *         query: "find all deployment-related docs from last month",
+     *         thinking: true
+     *     })
+     */
+    agentic(readableId: string, request: AirweaveSDK.AgenticSearchRequest, requestOptions?: Search.RequestOptions): core.HttpResponsePromise<AirweaveSDK.SearchV2Response>;
+    private __agentic;
+    /**
+     * Streaming agentic search via Server-Sent Events. Returns real-time events as the agent searches.
+     */
+    streamAgentic(readableId: string, request: AirweaveSDK.AgenticSearchRequest, requestOptions?: Search.RequestOptions): core.HttpResponsePromise<core.Stream<AirweaveSDK.AgenticSearchEvent>>;
+    private __streamAgentic;
+    protected _getCustomAuthorizationHeaders(): Promise<Record<string, string | undefined>>;
+}