@wildix/wim-knowledge-base-client 0.0.7 → 0.0.8

package/dist-cjs/KnowledgeBase.js

@@ -21,7 +21,8 @@ const ListDocumentsCommand_1 = require("./commands/ListDocumentsCommand");
 const ListKnowledgeBasesCommand_1 = require("./commands/ListKnowledgeBasesCommand");
 const QueryKnowledgeBaseCommand_1 = require("./commands/QueryKnowledgeBaseCommand");
 const SearchKnowledgeBaseCommand_1 = require("./commands/SearchKnowledgeBaseCommand");
-const SyncDataSourceCommand_1 = require("./commands/SyncDataSourceCommand");
+const StartSyncDataSourceCommand_1 = require("./commands/StartSyncDataSourceCommand");
+const StopSyncDataSourceCommand_1 = require("./commands/StopSyncDataSourceCommand");
 const UpdateDataSourceCommand_1 = require("./commands/UpdateDataSourceCommand");
 const UpdateDocumentCommand_1 = require("./commands/UpdateDocumentCommand");
 const UpdateKnowledgeBaseCommand_1 = require("./commands/UpdateKnowledgeBaseCommand");
@@ -46,7 +47,8 @@ const commands = {
     ListKnowledgeBasesCommand: ListKnowledgeBasesCommand_1.ListKnowledgeBasesCommand,
     QueryKnowledgeBaseCommand: QueryKnowledgeBaseCommand_1.QueryKnowledgeBaseCommand,
     SearchKnowledgeBaseCommand: SearchKnowledgeBaseCommand_1.SearchKnowledgeBaseCommand,
-    SyncDataSourceCommand: SyncDataSourceCommand_1.SyncDataSourceCommand,
+    StartSyncDataSourceCommand: StartSyncDataSourceCommand_1.StartSyncDataSourceCommand,
+    StopSyncDataSourceCommand: StopSyncDataSourceCommand_1.StopSyncDataSourceCommand,
     UpdateDataSourceCommand: UpdateDataSourceCommand_1.UpdateDataSourceCommand,
     UpdateDocumentCommand: UpdateDocumentCommand_1.UpdateDocumentCommand,
     UpdateKnowledgeBaseCommand: UpdateKnowledgeBaseCommand_1.UpdateKnowledgeBaseCommand,

package/dist-cjs/commands/StartSyncDataSourceCommand.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StartSyncDataSourceCommand = exports.$Command = void 0;
+const Aws_restJson1_1 = require("../protocols/Aws_restJson1");
+const middleware_serde_1 = require("@smithy/middleware-serde");
+const smithy_client_1 = require("@smithy/smithy-client");
+Object.defineProperty(exports, "$Command", { enumerable: true, get: function () { return smithy_client_1.Command; } });
+class StartSyncDataSourceCommand extends smithy_client_1.Command.classBuilder()
+    .m(function (Command, cs, config, o) {
+    return [
+        (0, middleware_serde_1.getSerdePlugin)(config, this.serialize, this.deserialize),
+    ];
+})
+    .s("KnowledgeBase", "StartSyncDataSource", {})
+    .n("KnowledgeBaseClient", "StartSyncDataSourceCommand")
+    .f(void 0, void 0)
+    .ser(Aws_restJson1_1.se_StartSyncDataSourceCommand)
+    .de(Aws_restJson1_1.de_StartSyncDataSourceCommand)
+    .build() {
+}
+exports.StartSyncDataSourceCommand = StartSyncDataSourceCommand;

package/dist-cjs/commands/{SyncDataSourceCommand.js → StopSyncDataSourceCommand.js}

@@ -1,21 +1,21 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SyncDataSourceCommand = exports.$Command = void 0;
+exports.StopSyncDataSourceCommand = exports.$Command = void 0;
 const Aws_restJson1_1 = require("../protocols/Aws_restJson1");
 const middleware_serde_1 = require("@smithy/middleware-serde");
 const smithy_client_1 = require("@smithy/smithy-client");
 Object.defineProperty(exports, "$Command", { enumerable: true, get: function () { return smithy_client_1.Command; } });
-class SyncDataSourceCommand extends smithy_client_1.Command.classBuilder()
+class StopSyncDataSourceCommand extends smithy_client_1.Command.classBuilder()
     .m(function (Command, cs, config, o) {
     return [
         (0, middleware_serde_1.getSerdePlugin)(config, this.serialize, this.deserialize),
     ];
 })
-    .s("KnowledgeBase", "SyncDataSource", {})
-    .n("KnowledgeBaseClient", "SyncDataSourceCommand")
+    .s("KnowledgeBase", "StopSyncDataSource", {})
+    .n("KnowledgeBaseClient", "StopSyncDataSourceCommand")
     .f(void 0, void 0)
-    .ser(Aws_restJson1_1.se_SyncDataSourceCommand)
-    .de(Aws_restJson1_1.de_SyncDataSourceCommand)
+    .ser(Aws_restJson1_1.se_StopSyncDataSourceCommand)
+    .de(Aws_restJson1_1.de_StopSyncDataSourceCommand)
     .build() {
 }
-exports.SyncDataSourceCommand = SyncDataSourceCommand;
+exports.StopSyncDataSourceCommand = StopSyncDataSourceCommand;

package/dist-cjs/commands/index.js

@@ -20,7 +20,8 @@ tslib_1.__exportStar(require("./ListDocumentsCommand"), exports);
 tslib_1.__exportStar(require("./ListKnowledgeBasesCommand"), exports);
 tslib_1.__exportStar(require("./QueryKnowledgeBaseCommand"), exports);
 tslib_1.__exportStar(require("./SearchKnowledgeBaseCommand"), exports);
-tslib_1.__exportStar(require("./SyncDataSourceCommand"), exports);
+tslib_1.__exportStar(require("./StartSyncDataSourceCommand"), exports);
+tslib_1.__exportStar(require("./StopSyncDataSourceCommand"), exports);
 tslib_1.__exportStar(require("./UpdateDataSourceCommand"), exports);
 tslib_1.__exportStar(require("./UpdateDocumentCommand"), exports);
 tslib_1.__exportStar(require("./UpdateKnowledgeBaseCommand"), exports);

package/dist-cjs/models/models_0.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SearchStrategy = exports.SyncLogLevel = exports.DownloadDocumentType = exports.DocumentStatus = exports.KnowledgeBaseNotFoundException = exports.DocumentNotFoundException = exports.DataSourceNotFoundException = exports.SyncDataSourceStatus = exports.SyncDataSourceMode = exports.DataSourceType = exports.DataSourceConfig = exports.UnauthorizedException = exports.ForbiddenException = exports.ValidationException = void 0;
+exports.SearchStrategy = exports.SyncLogLevel = exports.DownloadDocumentType = exports.DocumentStatus = exports.KnowledgeBaseNotFoundException = exports.DocumentNotFoundException = exports.DataSourceNotFoundException = exports.SyncDataSourceStatus = exports.SyncDataSourceMode = exports.DataSourceType = exports.DataSourceConfig = exports.ProxyMethod = exports.UnauthorizedException = exports.ForbiddenException = exports.ValidationException = void 0;
 const KnowledgeBaseServiceException_1 = require("./KnowledgeBaseServiceException");
 class ValidationException extends KnowledgeBaseServiceException_1.KnowledgeBaseServiceException {
     name = "ValidationException";
@@ -41,6 +41,10 @@ class UnauthorizedException extends KnowledgeBaseServiceException_1.KnowledgeBas
     }
 }
 exports.UnauthorizedException = UnauthorizedException;
+exports.ProxyMethod = {
+    GET: "GET",
+    POST: "POST",
+};
 var DataSourceConfig;
 (function (DataSourceConfig) {
     DataSourceConfig.visit = (value, visitor) => {
@@ -50,6 +54,8 @@ var DataSourceConfig;
             return visitor.gdrive(value.gdrive);
         if (value.files !== undefined)
             return visitor.files(value.files);
+        if (value.proxy !== undefined)
+            return visitor.proxy(value.proxy);
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(DataSourceConfig || (exports.DataSourceConfig = DataSourceConfig = {}));

package/dist-cjs/protocols/Aws_restJson1.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.de_UpdateKnowledgeBaseCommand = exports.de_UpdateDocumentCommand = exports.de_UpdateDataSourceCommand = exports.de_SyncDataSourceCommand = exports.de_SearchKnowledgeBaseCommand = exports.de_QueryKnowledgeBaseCommand = exports.de_ListKnowledgeBasesCommand = exports.de_ListDocumentsCommand = exports.de_ListDataSourcesCommand = exports.de_GetSyncStatusCommand = exports.de_GetSyncLogsCommand = exports.de_GetKnowledgeBaseCommand = exports.de_GetDocumentPresignedUploadUrlCommand = exports.de_GetDocumentPresignedDownloadUrlCommand = exports.de_GetDocumentCommand = exports.de_GetDataSourceCommand = exports.de_GetAuthorizationSessionForDataSourceCommand = exports.de_DeleteKnowledgeBaseCommand = exports.de_DeleteDocumentCommand = exports.de_DeleteDataSourceCommand = exports.de_CreateKnowledgeBaseCommand = exports.de_CreateDocumentCommand = exports.de_CreateDataSourceCommand = exports.se_UpdateKnowledgeBaseCommand = exports.se_UpdateDocumentCommand = exports.se_UpdateDataSourceCommand = exports.se_SyncDataSourceCommand = exports.se_SearchKnowledgeBaseCommand = exports.se_QueryKnowledgeBaseCommand = exports.se_ListKnowledgeBasesCommand = exports.se_ListDocumentsCommand = exports.se_ListDataSourcesCommand = exports.se_GetSyncStatusCommand = exports.se_GetSyncLogsCommand = exports.se_GetKnowledgeBaseCommand = exports.se_GetDocumentPresignedUploadUrlCommand = exports.se_GetDocumentPresignedDownloadUrlCommand = exports.se_GetDocumentCommand = exports.se_GetDataSourceCommand = exports.se_GetAuthorizationSessionForDataSourceCommand = exports.se_DeleteKnowledgeBaseCommand = exports.se_DeleteDocumentCommand = exports.se_DeleteDataSourceCommand = exports.se_CreateKnowledgeBaseCommand = exports.se_CreateDocumentCommand = exports.se_CreateDataSourceCommand = void 0;
+exports.de_UpdateKnowledgeBaseCommand = exports.de_UpdateDocumentCommand = exports.de_UpdateDataSourceCommand = exports.de_StopSyncDataSourceCommand = exports.de_StartSyncDataSourceCommand = exports.de_SearchKnowledgeBaseCommand = exports.de_QueryKnowledgeBaseCommand = exports.de_ListKnowledgeBasesCommand = exports.de_ListDocumentsCommand = exports.de_ListDataSourcesCommand = exports.de_GetSyncStatusCommand = exports.de_GetSyncLogsCommand = exports.de_GetKnowledgeBaseCommand = exports.de_GetDocumentPresignedUploadUrlCommand = exports.de_GetDocumentPresignedDownloadUrlCommand = exports.de_GetDocumentCommand = exports.de_GetDataSourceCommand = exports.de_GetAuthorizationSessionForDataSourceCommand = exports.de_DeleteKnowledgeBaseCommand = exports.de_DeleteDocumentCommand = exports.de_DeleteDataSourceCommand = exports.de_CreateKnowledgeBaseCommand = exports.de_CreateDocumentCommand = exports.de_CreateDataSourceCommand = exports.se_UpdateKnowledgeBaseCommand = exports.se_UpdateDocumentCommand = exports.se_UpdateDataSourceCommand = exports.se_StopSyncDataSourceCommand = exports.se_StartSyncDataSourceCommand = exports.se_SearchKnowledgeBaseCommand = exports.se_QueryKnowledgeBaseCommand = exports.se_ListKnowledgeBasesCommand = exports.se_ListDocumentsCommand = exports.se_ListDataSourcesCommand = exports.se_GetSyncStatusCommand = exports.se_GetSyncLogsCommand = exports.se_GetKnowledgeBaseCommand = exports.se_GetDocumentPresignedUploadUrlCommand = exports.se_GetDocumentPresignedDownloadUrlCommand = exports.se_GetDocumentCommand = exports.se_GetDataSourceCommand = exports.se_GetAuthorizationSessionForDataSourceCommand = exports.se_DeleteKnowledgeBaseCommand = exports.se_DeleteDocumentCommand = exports.se_DeleteDataSourceCommand = exports.se_CreateKnowledgeBaseCommand = exports.se_CreateDocumentCommand = exports.se_CreateDataSourceCommand = void 0;
 const KnowledgeBaseServiceException_1 = require("../models/KnowledgeBaseServiceException");
 const models_0_1 = require("../models/models_0");
 const core_1 = require("@aws-sdk/core");
@@ -17,7 +17,7 @@ const se_CreateDataSourceCommand = async (input, context) => {
     });
     let body;
     body = JSON.stringify((0, smithy_client_1.take)(input, {
-        'config': _ => (0, smithy_client_1._json)(_),
+        'config': _ => se_DataSourceConfig(_, context),
         'description': [],
         'enabled': [],
         'name': [],
@@ -351,10 +351,10 @@ const se_SearchKnowledgeBaseCommand = async (input, context) => {
     return b.build();
 };
 exports.se_SearchKnowledgeBaseCommand = se_SearchKnowledgeBaseCommand;
-const se_SyncDataSourceCommand = async (input, context) => {
+const se_StartSyncDataSourceCommand = async (input, context) => {
     const b = (0, core_2.requestBuilder)(input, context);
     const headers = {};
-    b.bp("/api/v1/data-sources/{dataSourceId}/sync");
+    b.bp("/api/v1/data-sources/{dataSourceId}/start-sync");
     b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
     const query = (0, smithy_client_1.map)({
         [_c]: [, input[_cI]],
@@ -367,7 +367,23 @@ const se_SyncDataSourceCommand = async (input, context) => {
         .b(body);
     return b.build();
 };
-exports.se_SyncDataSourceCommand = se_SyncDataSourceCommand;
+exports.se_StartSyncDataSourceCommand = se_StartSyncDataSourceCommand;
+const se_StopSyncDataSourceCommand = async (input, context) => {
+    const b = (0, core_2.requestBuilder)(input, context);
+    const headers = {};
+    b.bp("/api/v1/data-sources/{dataSourceId}/stop-sync");
+    b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
+    const query = (0, smithy_client_1.map)({
+        [_c]: [, input[_cI]],
+    });
+    let body;
+    b.m("POST")
+        .h(headers)
+        .q(query)
+        .b(body);
+    return b.build();
+};
+exports.se_StopSyncDataSourceCommand = se_StopSyncDataSourceCommand;
 const se_UpdateDataSourceCommand = async (input, context) => {
     const b = (0, core_2.requestBuilder)(input, context);
     const headers = {
@@ -380,7 +396,7 @@ const se_UpdateDataSourceCommand = async (input, context) => {
     });
     let body;
     body = JSON.stringify((0, smithy_client_1.take)(input, {
-        'config': _ => (0, smithy_client_1._json)(_),
+        'config': _ => se_DataSourceConfig(_, context),
         'description': [],
         'enabled': [],
         'name': [],
@@ -453,7 +469,7 @@ const de_CreateDataSourceCommand = async (output, context) => {
     });
     const data = (0, smithy_client_1.expectNonNull)(((0, smithy_client_1.expectObject)(await (0, core_1.parseJsonBody)(output.body, context))), "body");
     const doc = (0, smithy_client_1.take)(data, {
-        'dataSource': smithy_client_1._json,
+        'dataSource': _ => de_DataSourceItem(_, context),
     });
     Object.assign(contents, doc);
     return contents;
@@ -546,7 +562,7 @@ const de_GetDataSourceCommand = async (output, context) => {
     });
     const data = (0, smithy_client_1.expectNonNull)(((0, smithy_client_1.expectObject)(await (0, core_1.parseJsonBody)(output.body, context))), "body");
     const doc = (0, smithy_client_1.take)(data, {
-        'dataSource': smithy_client_1._json,
+        'dataSource': _ => de_DataSourceItem(_, context),
     });
     Object.assign(contents, doc);
     return contents;
@@ -653,7 +669,7 @@ const de_ListDataSourcesCommand = async (output, context) => {
     });
     const data = (0, smithy_client_1.expectNonNull)(((0, smithy_client_1.expectObject)(await (0, core_1.parseJsonBody)(output.body, context))), "body");
     const doc = (0, smithy_client_1.take)(data, {
-        'dataSources': smithy_client_1._json,
+        'dataSources': _ => de_DataSourceList(_, context),
     });
     Object.assign(contents, doc);
     return contents;
@@ -720,7 +736,7 @@ const de_SearchKnowledgeBaseCommand = async (output, context) => {
     return contents;
 };
 exports.de_SearchKnowledgeBaseCommand = de_SearchKnowledgeBaseCommand;
-const de_SyncDataSourceCommand = async (output, context) => {
+const de_StartSyncDataSourceCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
     }
@@ -730,7 +746,18 @@ const de_SyncDataSourceCommand = async (output, context) => {
     await (0, smithy_client_1.collectBody)(output.body, context);
     return contents;
 };
-exports.de_SyncDataSourceCommand = de_SyncDataSourceCommand;
+exports.de_StartSyncDataSourceCommand = de_StartSyncDataSourceCommand;
+const de_StopSyncDataSourceCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = (0, smithy_client_1.map)({
+        $metadata: deserializeMetadata(output),
+    });
+    await (0, smithy_client_1.collectBody)(output.body, context);
+    return contents;
+};
+exports.de_StopSyncDataSourceCommand = de_StopSyncDataSourceCommand;
 const de_UpdateDataSourceCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -740,7 +767,7 @@ const de_UpdateDataSourceCommand = async (output, context) => {
     });
     const data = (0, smithy_client_1.expectNonNull)(((0, smithy_client_1.expectObject)(await (0, core_1.parseJsonBody)(output.body, context))), "body");
     const doc = (0, smithy_client_1.take)(data, {
-        'dataSource': smithy_client_1._json,
+        'dataSource': _ => de_DataSourceItem(_, context),
     });
     Object.assign(contents, doc);
     return contents;
@@ -889,6 +916,18 @@ const de_KnowledgeBaseNotFoundExceptionRes = async (parsedOutput, context) => {
     });
     return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
 };
+const se_Document = (input, context) => {
+    return input;
+};
+const se_DataSourceConfig = (input, context) => {
+    return models_0_1.DataSourceConfig.visit(input, {
+        confluence: value => ({ "confluence": (0, smithy_client_1._json)(value) }),
+        files: value => ({ "files": (0, smithy_client_1._json)(value) }),
+        gdrive: value => ({ "gdrive": (0, smithy_client_1._json)(value) }),
+        proxy: value => ({ "proxy": se_ProxyConfig(value, context) }),
+        _: (name, value) => ({ [name]: value })
+    });
+};
 const se_LlmConfig = (input, context) => {
     return (0, smithy_client_1.take)(input, {
         'model': [],
@@ -897,6 +936,19 @@ const se_LlmConfig = (input, context) => {
         'temperature': smithy_client_1.serializeFloat,
     });
 };
+const se_ProxyConfig = (input, context) => {
+    return (0, smithy_client_1.take)(input, {
+        'body': _ => se_Document(_, context),
+        'headers': smithy_client_1._json,
+        'method': [],
+        'password': [],
+        'responseMapping': smithy_client_1._json,
+        'timeoutMs': [],
+        'token': [],
+        'url': [],
+        'username': [],
+    });
+};
 const se_SearchConfig = (input, context) => {
     return (0, smithy_client_1.take)(input, {
         'searchStrategy': [],
@@ -907,6 +959,66 @@ const se_SearchConfig = (input, context) => {
 const de_Document = (output, context) => {
     return output;
 };
+const de_DataSourceConfig = (output, context) => {
+    if (output.confluence != null) {
+        return {
+            confluence: (0, smithy_client_1._json)(output.confluence)
+        };
+    }
+    if (output.files != null) {
+        return {
+            files: (0, smithy_client_1._json)(output.files)
+        };
+    }
+    if (output.gdrive != null) {
+        return {
+            gdrive: (0, smithy_client_1._json)(output.gdrive)
+        };
+    }
+    if (output.proxy != null) {
+        return {
+            proxy: de_ProxyConfig(output.proxy, context)
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
+const de_DataSourceItem = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        'companyId': smithy_client_1.expectString,
+        'config': (_) => de_DataSourceConfig((0, core_1.awsExpectUnion)(_), context),
+        'createdAt': smithy_client_1.expectString,
+        'description': smithy_client_1.expectString,
+        'enabled': smithy_client_1.expectBoolean,
+        'id': smithy_client_1.expectString,
+        'lastSyncErrorMessage': smithy_client_1.expectString,
+        'lastSyncMode': smithy_client_1.expectString,
+        'lastSyncedAt': smithy_client_1.expectString,
+        'name': smithy_client_1.expectString,
+        'syncSchedule': smithy_client_1.expectString,
+        'syncStatus': smithy_client_1.expectString,
+        'type': smithy_client_1.expectString,
+        'updatedAt': smithy_client_1.expectString,
+    });
+};
+const de_DataSourceList = (output, context) => {
+    const retVal = (output || []).filter((e) => e != null).map((entry) => {
+        return de_DataSourceItem(entry, context);
+    });
+    return retVal;
+};
+const de_ProxyConfig = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        'body': (_) => de_Document(_, context),
+        'headers': smithy_client_1._json,
+        'method': smithy_client_1.expectString,
+        'password': smithy_client_1.expectString,
+        'responseMapping': smithy_client_1._json,
+        'timeoutMs': smithy_client_1.expectInt32,
+        'token': smithy_client_1.expectString,
+        'url': smithy_client_1.expectString,
+        'username': smithy_client_1.expectString,
+    });
+};
 const de_SearchKnowledgeBaseResult = (output, context) => {
     return (0, smithy_client_1.take)(output, {
         'content': smithy_client_1.expectString,

package/dist-es/KnowledgeBase.js

@@ -18,7 +18,8 @@ import { ListDocumentsCommand, } from "./commands/ListDocumentsCommand";
 import { ListKnowledgeBasesCommand, } from "./commands/ListKnowledgeBasesCommand";
 import { QueryKnowledgeBaseCommand, } from "./commands/QueryKnowledgeBaseCommand";
 import { SearchKnowledgeBaseCommand, } from "./commands/SearchKnowledgeBaseCommand";
-import { SyncDataSourceCommand, } from "./commands/SyncDataSourceCommand";
+import { StartSyncDataSourceCommand, } from "./commands/StartSyncDataSourceCommand";
+import { StopSyncDataSourceCommand, } from "./commands/StopSyncDataSourceCommand";
 import { UpdateDataSourceCommand, } from "./commands/UpdateDataSourceCommand";
 import { UpdateDocumentCommand, } from "./commands/UpdateDocumentCommand";
 import { UpdateKnowledgeBaseCommand, } from "./commands/UpdateKnowledgeBaseCommand";
@@ -43,7 +44,8 @@ const commands = {
     ListKnowledgeBasesCommand,
     QueryKnowledgeBaseCommand,
     SearchKnowledgeBaseCommand,
-    SyncDataSourceCommand,
+    StartSyncDataSourceCommand,
+    StopSyncDataSourceCommand,
     UpdateDataSourceCommand,
     UpdateDocumentCommand,
     UpdateKnowledgeBaseCommand,

package/dist-es/commands/StartSyncDataSourceCommand.js

@@ -0,0 +1,17 @@
+import { de_StartSyncDataSourceCommand, se_StartSyncDataSourceCommand, } from "../protocols/Aws_restJson1";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+export { $Command };
+export class StartSyncDataSourceCommand extends $Command.classBuilder()
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+    ];
+})
+    .s("KnowledgeBase", "StartSyncDataSource", {})
+    .n("KnowledgeBaseClient", "StartSyncDataSourceCommand")
+    .f(void 0, void 0)
+    .ser(se_StartSyncDataSourceCommand)
+    .de(de_StartSyncDataSourceCommand)
+    .build() {
+}

package/dist-es/commands/StopSyncDataSourceCommand.js

@@ -0,0 +1,17 @@
+import { de_StopSyncDataSourceCommand, se_StopSyncDataSourceCommand, } from "../protocols/Aws_restJson1";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+export { $Command };
+export class StopSyncDataSourceCommand extends $Command.classBuilder()
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+    ];
+})
+    .s("KnowledgeBase", "StopSyncDataSource", {})
+    .n("KnowledgeBaseClient", "StopSyncDataSourceCommand")
+    .f(void 0, void 0)
+    .ser(se_StopSyncDataSourceCommand)
+    .de(de_StopSyncDataSourceCommand)
+    .build() {
+}

package/dist-es/commands/index.js

@@ -17,7 +17,8 @@ export * from "./ListDocumentsCommand";
 export * from "./ListKnowledgeBasesCommand";
 export * from "./QueryKnowledgeBaseCommand";
 export * from "./SearchKnowledgeBaseCommand";
-export * from "./SyncDataSourceCommand";
+export * from "./StartSyncDataSourceCommand";
+export * from "./StopSyncDataSourceCommand";
 export * from "./UpdateDataSourceCommand";
 export * from "./UpdateDocumentCommand";
 export * from "./UpdateKnowledgeBaseCommand";

package/dist-es/models/models_0.js

@@ -35,6 +35,10 @@ export class UnauthorizedException extends __BaseException {
         Object.setPrototypeOf(this, UnauthorizedException.prototype);
     }
 }
+export const ProxyMethod = {
+    GET: "GET",
+    POST: "POST",
+};
 export var DataSourceConfig;
 (function (DataSourceConfig) {
     DataSourceConfig.visit = (value, visitor) => {
@@ -44,6 +48,8 @@ export var DataSourceConfig;
             return visitor.gdrive(value.gdrive);
         if (value.files !== undefined)
             return visitor.files(value.files);
+        if (value.proxy !== undefined)
+            return visitor.proxy(value.proxy);
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(DataSourceConfig || (DataSourceConfig = {}));

package/dist-es/protocols/Aws_restJson1.js

@@ -1,8 +1,8 @@
 import { KnowledgeBaseServiceException as __BaseException } from "../models/KnowledgeBaseServiceException";
-import { DataSourceNotFoundException, DocumentNotFoundException, ForbiddenException, KnowledgeBaseNotFoundException, UnauthorizedException, ValidationException, } from "../models/models_0";
-import { loadRestJsonErrorCode, parseJsonBody as parseBody, parseJsonErrorBody as parseErrorBody, } from "@aws-sdk/core";
+import { DataSourceConfig, DataSourceNotFoundException, DocumentNotFoundException, ForbiddenException, KnowledgeBaseNotFoundException, UnauthorizedException, ValidationException, } from "../models/models_0";
+import { awsExpectUnion as __expectUnion, loadRestJsonErrorCode, parseJsonBody as parseBody, parseJsonErrorBody as parseErrorBody, } from "@aws-sdk/core";
 import { requestBuilder as rb } from "@smithy/core";
-import { decorateServiceException as __decorateServiceException, expectInt32 as __expectInt32, expectNonNull as __expectNonNull, expectObject as __expectObject, expectString as __expectString, limitedParseDouble as __limitedParseDouble, serializeFloat as __serializeFloat, _json, collectBody, map, take, withBaseException, } from "@smithy/smithy-client";
+import { decorateServiceException as __decorateServiceException, expectBoolean as __expectBoolean, expectInt32 as __expectInt32, expectNonNull as __expectNonNull, expectObject as __expectObject, expectString as __expectString, limitedParseDouble as __limitedParseDouble, serializeFloat as __serializeFloat, _json, collectBody, map, take, withBaseException, } from "@smithy/smithy-client";
 export const se_CreateDataSourceCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -14,7 +14,7 @@ export const se_CreateDataSourceCommand = async (input, context) => {
     });
     let body;
     body = JSON.stringify(take(input, {
-        'config': _ => _json(_),
+        'config': _ => se_DataSourceConfig(_, context),
         'description': [],
         'enabled': [],
         'name': [],
@@ -329,10 +329,10 @@ export const se_SearchKnowledgeBaseCommand = async (input, context) => {
         .b(body);
     return b.build();
 };
-export const se_SyncDataSourceCommand = async (input, context) => {
+export const se_StartSyncDataSourceCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
-    b.bp("/api/v1/data-sources/{dataSourceId}/sync");
+    b.bp("/api/v1/data-sources/{dataSourceId}/start-sync");
     b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
     const query = map({
         [_c]: [, input[_cI]],
@@ -345,6 +345,21 @@ export const se_SyncDataSourceCommand = async (input, context) => {
         .b(body);
     return b.build();
 };
+export const se_StopSyncDataSourceCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {};
+    b.bp("/api/v1/data-sources/{dataSourceId}/stop-sync");
+    b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
+    const query = map({
+        [_c]: [, input[_cI]],
+    });
+    let body;
+    b.m("POST")
+        .h(headers)
+        .q(query)
+        .b(body);
+    return b.build();
+};
 export const se_UpdateDataSourceCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -357,7 +372,7 @@ export const se_UpdateDataSourceCommand = async (input, context) => {
     });
     let body;
     body = JSON.stringify(take(input, {
-        'config': _ => _json(_),
+        'config': _ => se_DataSourceConfig(_, context),
         'description': [],
         'enabled': [],
         'name': [],
@@ -427,7 +442,7 @@ export const de_CreateDataSourceCommand = async (output, context) => {
     });
     const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
     const doc = take(data, {
-        'dataSource': _json,
+        'dataSource': _ => de_DataSourceItem(_, context),
     });
     Object.assign(contents, doc);
     return contents;
@@ -513,7 +528,7 @@ export const de_GetDataSourceCommand = async (output, context) => {
     });
     const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
     const doc = take(data, {
-        'dataSource': _json,
+        'dataSource': _ => de_DataSourceItem(_, context),
     });
     Object.assign(contents, doc);
     return contents;
@@ -613,7 +628,7 @@ export const de_ListDataSourcesCommand = async (output, context) => {
     });
     const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
     const doc = take(data, {
-        'dataSources': _json,
+        'dataSources': _ => de_DataSourceList(_, context),
     });
     Object.assign(contents, doc);
     return contents;
@@ -675,7 +690,17 @@ export const de_SearchKnowledgeBaseCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
-export const de_SyncDataSourceCommand = async (output, context) => {
+export const de_StartSyncDataSourceCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    await collectBody(output.body, context);
+    return contents;
+};
+export const de_StopSyncDataSourceCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
     }
@@ -694,7 +719,7 @@ export const de_UpdateDataSourceCommand = async (output, context) => {
     });
     const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
     const doc = take(data, {
-        'dataSource': _json,
+        'dataSource': _ => de_DataSourceItem(_, context),
     });
     Object.assign(contents, doc);
     return contents;
@@ -840,6 +865,18 @@ const de_KnowledgeBaseNotFoundExceptionRes = async (parsedOutput, context) => {
     });
     return __decorateServiceException(exception, parsedOutput.body);
 };
+const se_Document = (input, context) => {
+    return input;
+};
+const se_DataSourceConfig = (input, context) => {
+    return DataSourceConfig.visit(input, {
+        confluence: value => ({ "confluence": _json(value) }),
+        files: value => ({ "files": _json(value) }),
+        gdrive: value => ({ "gdrive": _json(value) }),
+        proxy: value => ({ "proxy": se_ProxyConfig(value, context) }),
+        _: (name, value) => ({ [name]: value })
+    });
+};
 const se_LlmConfig = (input, context) => {
     return take(input, {
         'model': [],
@@ -848,6 +885,19 @@ const se_LlmConfig = (input, context) => {
         'temperature': __serializeFloat,
     });
 };
+const se_ProxyConfig = (input, context) => {
+    return take(input, {
+        'body': _ => se_Document(_, context),
+        'headers': _json,
+        'method': [],
+        'password': [],
+        'responseMapping': _json,
+        'timeoutMs': [],
+        'token': [],
+        'url': [],
+        'username': [],
+    });
+};
 const se_SearchConfig = (input, context) => {
     return take(input, {
         'searchStrategy': [],
@@ -858,6 +908,66 @@ const se_SearchConfig = (input, context) => {
 const de_Document = (output, context) => {
     return output;
 };
+const de_DataSourceConfig = (output, context) => {
+    if (output.confluence != null) {
+        return {
+            confluence: _json(output.confluence)
+        };
+    }
+    if (output.files != null) {
+        return {
+            files: _json(output.files)
+        };
+    }
+    if (output.gdrive != null) {
+        return {
+            gdrive: _json(output.gdrive)
+        };
+    }
+    if (output.proxy != null) {
+        return {
+            proxy: de_ProxyConfig(output.proxy, context)
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
+const de_DataSourceItem = (output, context) => {
+    return take(output, {
+        'companyId': __expectString,
+        'config': (_) => de_DataSourceConfig(__expectUnion(_), context),
+        'createdAt': __expectString,
+        'description': __expectString,
+        'enabled': __expectBoolean,
+        'id': __expectString,
+        'lastSyncErrorMessage': __expectString,
+        'lastSyncMode': __expectString,
+        'lastSyncedAt': __expectString,
+        'name': __expectString,
+        'syncSchedule': __expectString,
+        'syncStatus': __expectString,
+        'type': __expectString,
+        'updatedAt': __expectString,
+    });
+};
+const de_DataSourceList = (output, context) => {
+    const retVal = (output || []).filter((e) => e != null).map((entry) => {
+        return de_DataSourceItem(entry, context);
+    });
+    return retVal;
+};
+const de_ProxyConfig = (output, context) => {
+    return take(output, {
+        'body': (_) => de_Document(_, context),
+        'headers': _json,
+        'method': __expectString,
+        'password': __expectString,
+        'responseMapping': _json,
+        'timeoutMs': __expectInt32,
+        'token': __expectString,
+        'url': __expectString,
+        'username': __expectString,
+    });
+};
 const de_SearchKnowledgeBaseResult = (output, context) => {
     return take(output, {
         'content': __expectString,