@azure/search-documents 12.2.0 → 12.3.0-alpha.20251031.1

package/dist/commonjs/searchIndexClient.js

@@ -4,10 +4,10 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SearchIndexClient = void 0;
 const tslib_1 = require("tslib");
-/// <reference lib="esnext.asynciterable" />
 const core_auth_1 = require("@azure/core-auth");
 const core_rest_pipeline_1 = require("@azure/core-rest-pipeline");
 const searchServiceClient_js_1 = require("./generated/service/searchServiceClient.js");
+const knowledgeRetrievalClient_js_1 = require("./knowledgeRetrievalClient.js");
 const logger_js_1 = require("./logger.js");
 const odataMetadataPolicy_js_1 = require("./odataMetadataPolicy.js");
 const searchApiKeyCredentialPolicy_js_1 = require("./searchApiKeyCredentialPolicy.js");
@@ -39,6 +39,10 @@ class SearchIndexClient {
      * A reference to the auto-generated SearchServiceClient
      */
     client;
+    /**
+     * A reference to the internal HTTP pipeline for use with raw requests
+     */
+    pipeline;
     /**
      * Used to authenticate requests to the service.
      */
@@ -51,13 +55,10 @@ class SearchIndexClient {
      * Creates an instance of SearchIndexClient.
      *
      * Example usage:
-     * ```ts
-     * const { SearchIndexClient, AzureKeyCredential } = require("@azure/search-documents");
+     * ```ts snippet:ReadmeSampleSearchIndexClient
+     * import { SearchIndexClient, AzureKeyCredential } from "@azure/search-documents";
      *
-     * const client = new SearchIndexClient(
-     *   "<endpoint>",
-     *   new AzureKeyCredential("<Admin Key>");
-     * );
+     * const indexClient = new SearchIndexClient("<endpoint>", new AzureKeyCredential("<apiKey>"));
      * ```
      * @param endpoint - The endpoint of the search service
      * @param credential - Used to authenticate requests to the service.
@@ -87,6 +88,7 @@ class SearchIndexClient {
             this.options.serviceVersion ?? this.options.apiVersion ?? utils.defaultServiceVersion;
         this.apiVersion = this.serviceVersion;
         this.client = new searchServiceClient_js_1.SearchServiceClient(this.endpoint, this.serviceVersion, internalClientPipelineOptions);
+        this.pipeline = this.client.pipeline;
         if ((0, core_auth_1.isTokenCredential)(credential)) {
             const scope = this.options.audience
                 ? `${this.options.audience}/.default`
@@ -139,6 +141,46 @@ class SearchIndexClient {
             },
         };
     }
+    async *listAliasesPage(options = {}) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-listAliases", options);
+        try {
+            const result = await this.client.aliases.list(updatedOptions);
+            yield result.aliases;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    async *listAliasesAll(options = {}) {
+        for await (const page of this.listAliasesPage(options)) {
+            yield* page;
+        }
+    }
+    /**
+     * Lists all aliases available for a search service.
+     * @param options - The options parameters.
+     */
+    listAliases(options = {}) {
+        const iter = this.listAliasesAll(options);
+        return {
+            next() {
+                return iter.next();
+            },
+            [Symbol.asyncIterator]() {
+                return this;
+            },
+            byPage: () => {
+                return this.listAliasesPage(options);
+            },
+        };
+    }
     async *listIndexesNamesPage(options = {}) {
         const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-listIndexesNamesPage", options);
         try {
@@ -169,6 +211,7 @@ class SearchIndexClient {
      * Retrieves a list of names of existing indexes in the service.
      * @param options - Options to the list index operation.
      */
+    // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     listIndexesNames(options = {}) {
         const iter = this.listIndexesNamesAll(options);
         return {
@@ -208,6 +251,7 @@ class SearchIndexClient {
      * Retrieves a list of names of existing SynonymMaps in the service.
      * @param options - Options to the list SynonymMaps operation.
      */
+    // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     async listSynonymMapsNames(options = {}) {
         const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-listSynonymMapsNames", options);
         try {
@@ -255,7 +299,9 @@ class SearchIndexClient {
      * @param synonymMapName - The name of the SynonymMap.
      * @param options - Additional optional arguments.
      */
-    async getSynonymMap(synonymMapName, options = {}) {
+    async getSynonymMap(synonymMapName, 
+    // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
+    options = {}) {
         const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-getSynonymMaps", options);
         try {
             const result = await this.client.synonymMaps.get(synonymMapName, updatedOptions);
@@ -368,11 +414,6 @@ class SearchIndexClient {
             span.end();
         }
     }
-    /**
-     * Deletes an existing index.
-     * @param indexName - Index/Name of the index to delete.
-     * @param options - Additional optional arguments.
-     */
     async deleteIndex(index, options = {}) {
         const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-deleteIndex", options);
         try {
@@ -424,6 +465,97 @@ class SearchIndexClient {
             span.end();
         }
     }
+    /**
+     * Creates a new search alias or updates an alias if it already exists.
+     * @param alias - The definition of the alias to create or update.
+     * @param options - The options parameters.
+     */
+    async createOrUpdateAlias(alias, options = {}) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-createOrUpdateAlias", options);
+        try {
+            const etag = options.onlyIfUnchanged ? alias.etag : undefined;
+            const result = await this.client.aliases.createOrUpdate(alias.name, alias, {
+                ...updatedOptions,
+                ifMatch: etag,
+            });
+            return result;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    /**
+     * Creates a new search alias.
+     * @param alias - The definition of the alias to create.
+     * @param options - The options parameters.
+     */
+    async createAlias(alias, options = {}) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-createAlias", options);
+        try {
+            const result = await this.client.aliases.create(alias, updatedOptions);
+            return result;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    async deleteAlias(alias, options = {}) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-deleteAlias", options);
+        try {
+            const aliasName = typeof alias === "string" ? alias : alias.name;
+            const etag = typeof alias === "string" ? undefined : options.onlyIfUnchanged ? alias.etag : undefined;
+            await this.client.aliases.delete(aliasName, {
+                ...updatedOptions,
+                ifMatch: etag,
+            });
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    /**
+     * Retrieves an alias definition.
+     * @param aliasName - The name of the alias to retrieve.
+     * @param options - The options parameters.
+     */
+    async getAlias(aliasName, options = {}) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-getAlias", options);
+        try {
+            const result = await this.client.aliases.get(aliasName, updatedOptions);
+            return result;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
     /**
      * Retrieves statistics about an index, such as the count of documents and the size
      * of index storage.
@@ -497,18 +629,351 @@ class SearchIndexClient {
             span.end();
         }
     }
+    async *getIndexStatsSummaryPage(options = {}) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-getIndexStatsSummaryPage", options);
+        try {
+            const { indexesStatistics } = await this.client.getIndexStatsSummary(updatedOptions);
+            yield indexesStatistics;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    async *getIndexStatsSummaryAll(options = {}) {
+        for await (const page of this.getIndexStatsSummaryPage(options)) {
+            yield* page;
+        }
+    }
+    /**
+     * Retrieves a list of existing indexes in the service.
+     * @param options - Options to the list index operation.
+     */
+    getIndexStatsSummary(options = {}) {
+        const iter = this.getIndexStatsSummaryAll(options);
+        return {
+            next() {
+                return iter.next();
+            },
+            [Symbol.asyncIterator]() {
+                return this;
+            },
+            byPage: () => {
+                return this.getIndexStatsSummaryPage(options);
+            },
+        };
+    }
+    /**
+     * Creates a new knowledgebase.
+     * @param knowledgeBase - definition of the knowledgebase to create.
+     * @param options - options parameters.
+     */
+    async createKnowledgeBase(knowledgeBase, options) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-createKnowledgeBase", options);
+        try {
+            const result = await this.client.knowledgeBases.create(utils.convertKnowledgeBaseToGenerated(knowledgeBase), updatedOptions);
+            return utils.convertKnowledgeBaseToPublic(result);
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    /**
+     * Creates a new knowledge base or updates a knowledge base if it already exists.
+     * @param knowledgeBaseName - name of the knowledge base to create or update.
+     * @param knowledgeBase - definition of the knowledge base to create or update.
+     * @param options - options parameters.
+     */
+    async createOrUpdateKnowledgeBase(knowledgeBaseName, knowledgeBase, options) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-createOrUpdateKnowledgeBase", options);
+        try {
+            const etag = updatedOptions.onlyIfUnchanged ? knowledgeBase.etag : undefined;
+            const result = await this.client.knowledgeBases.createOrUpdate(knowledgeBaseName, utils.convertKnowledgeBaseToGenerated(knowledgeBase), {
+                ...updatedOptions,
+                ifMatch: etag,
+            });
+            return utils.convertKnowledgeBaseToPublic(result);
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    /**
+     * Retrieves a knowledge base definition.
+     * @param knowledgeBaseName - name of the knowledge base to retrieve.
+     * @param options - options parameters.
+     */
+    async getKnowledgeBase(knowledgeBaseName, options) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-getKnowledgeBase", options);
+        try {
+            const result = await this.client.knowledgeBases.get(knowledgeBaseName, updatedOptions);
+            return utils.convertKnowledgeBaseToPublic(result);
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    async *listKnowledgeBasesPage(options = {}) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-listKnowledgeBasesPage", options);
+        try {
+            const { knowledgeBases } = await this.client.knowledgeBases.list(updatedOptions);
+            const mapped = knowledgeBases.map((base) => utils.convertKnowledgeBaseToPublic(base));
+            yield mapped;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    async *listKnowledgeBasesAll(options = {}) {
+        for await (const page of this.listKnowledgeBasesPage(options)) {
+            yield* page;
+        }
+    }
+    /**
+     * Retrieves a list of existing KnowledgeBases in the service.
+     * @param options - Options to the list knowledge bases operation.
+     */
+    listKnowledgeBases(options = {}) {
+        const iter = this.listKnowledgeBasesAll(options);
+        return {
+            next() {
+                return iter.next();
+            },
+            [Symbol.asyncIterator]() {
+                return this;
+            },
+            byPage: () => {
+                return this.listKnowledgeBasesPage(options);
+            },
+        };
+    }
+    async deleteKnowledgeBase(knowledgeBase, options) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-deleteKnowledgeBase", options);
+        try {
+            const knowledgeBaseName = typeof knowledgeBase === "string" ? knowledgeBase : knowledgeBase.name;
+            const etag = typeof knowledgeBase !== "string" && updatedOptions.onlyIfUnchanged
+                ? knowledgeBase.etag
+                : undefined;
+            const result = await this.client.knowledgeBases.delete(knowledgeBaseName, {
+                ...updatedOptions,
+                ifMatch: etag,
+            });
+            return result;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    async createOrUpdateKnowledgeSource(sourceName, knowledgeSource, options) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-createOrUpdateKnowledgeSource", options);
+        try {
+            const etag = updatedOptions.onlyIfUnchanged ? knowledgeSource.etag : undefined;
+            const result = await this.client.knowledgeSources.createOrUpdate(sourceName, utils.convertKnowledgeSourceToGenerated(knowledgeSource), {
+                ...updatedOptions,
+                ifMatch: etag,
+            });
+            return utils.convertKnowledgeSourceToPublic(result);
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    async deleteKnowledgeSource(source, options) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-deleteKnowledgeSource", options);
+        try {
+            const sourceName = typeof source === "string" ? source : source.name;
+            const etag = typeof source !== "string" && updatedOptions.onlyIfUnchanged ? source.etag : undefined;
+            const result = await this.client.knowledgeSources.delete(sourceName, {
+                ...updatedOptions,
+                ifMatch: etag,
+            });
+            return result;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    /**
+     * Retrieves a knowledge source definition.
+     * @param sourceName - The name of the knowledge source to retrieve.
+     * @param options - The options parameters.
+     */
+    async getKnowledgeSource(sourceName, options) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-getKnowledgeSource", options);
+        try {
+            const result = await this.client.knowledgeSources.get(sourceName, updatedOptions);
+            return utils.convertKnowledgeSourceToPublic(result);
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    async *listKnowledgeSourcesPage(options = {}) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-listKnowledgeSourcesPage", options);
+        try {
+            const { knowledgeSources } = await this.client.knowledgeSources.list(updatedOptions);
+            const mapped = knowledgeSources.map((source) => utils.convertKnowledgeSourceToPublic(source));
+            yield mapped;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    async *listKnowledgeSourcesAll(options = {}) {
+        for await (const page of this.listKnowledgeSourcesPage(options)) {
+            yield* page;
+        }
+    }
+    /**
+     * Retrieves a list of existing KnowledgeSources in the service.
+     * @param options - Options to the list knowledge sources operation.
+     */
+    listKnowledgeSources(options = {}) {
+        const iter = this.listKnowledgeSourcesAll(options);
+        return {
+            next() {
+                return iter.next();
+            },
+            [Symbol.asyncIterator]() {
+                return this;
+            },
+            byPage: () => {
+                return this.listKnowledgeSourcesPage(options);
+            },
+        };
+    }
+    /**
+     * Creates a new knowledge source.
+     * @param knowledgeSource - The definition of the knowledge source to create.
+     * @param options - The options parameters.
+     */
+    async createKnowledgeSource(knowledgeSource, options) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-createKnowledgeSource", options);
+        try {
+            const result = await this.client.knowledgeSources.create(utils.convertKnowledgeSourceToGenerated(knowledgeSource), updatedOptions);
+            return utils.convertKnowledgeSourceToPublic(result);
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
+    /**
+     * Returns the current status and synchronization history of a knowledge source.
+     * @param sourceName - The name of the knowledge source for which to retrieve status.
+     * @param options - The options parameters.
+     */
+    async getKnowledgeSourceStatus(sourceName, options) {
+        const { span, updatedOptions } = (0, tracing_js_1.createSpan)("SearchIndexClient-getKnowledgeSourceStatus", options);
+        try {
+            const result = await this.client.knowledgeSources.getStatus(sourceName, updatedOptions);
+            return result;
+        }
+        catch (e) {
+            span.setStatus({
+                status: "error",
+                error: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    }
     /**
      * Retrieves the SearchClient corresponding to this SearchIndexClient
      * @param indexName - Name of the index
      * @param options - SearchClient Options
-     * @typeParam TModel - An optional type that represents the documents stored in
-     * the search index. For the best typing experience, all non-key fields should
-     * be marked optional and nullable, and the key property should have the
-     * non-nullable type `string`.
+     * @typeParam TModel - An optional type that represents the documents stored in the search index.
+     * For the best typing experience, all non-key fields should be marked optional and nullable, and
+     * the key property should have the non-nullable type `string`.
      */
     getSearchClient(indexName, options) {
         return new searchClient_js_1.SearchClient(this.endpoint, indexName, this.credential, options || this.options);
     }
+    /**
+     * Retrieves the KnowledgeRetrievalClient corresponding to this SearchIndexClient
+     * @param knowledgeBaseName - Name of the knowledge base
+     * @param options - KnowledgeRetrievalClient Options
+     */
+    getKnowledgeRetrievalClient(knowledgeBaseName, options) {
+        return new knowledgeRetrievalClient_js_1.KnowledgeRetrievalClient(this.endpoint, knowledgeBaseName, this.credential, options || this.options);
+    }
 }
 exports.SearchIndexClient = SearchIndexClient;
 //# sourceMappingURL=searchIndexClient.js.map