@wildix/wim-knowledge-base-client 0.0.48 → 0.0.49

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (54) hide show
  1. package/dist-cjs/KnowledgeBase.js +16 -8
  2. package/dist-cjs/commands/{QueryKnowledgeBaseCommand.js → GenerateSearchAnswerCommand.js} +7 -7
  3. package/dist-cjs/commands/{SearchHistoryCommand.js → GetDataSourceSyncStatusCommand.js} +7 -7
  4. package/dist-cjs/commands/{GetSyncLogsCommand.js → GetDocumentChunksCommand.js} +7 -7
  5. package/dist-cjs/commands/{GetSyncStatusCommand.js → GetDocumentMarkdownCommand.js} +7 -7
  6. package/dist-cjs/commands/GetDocumentWithContentAndChunksCommand.js +21 -0
  7. package/dist-cjs/commands/ListDataSourceStatisticsCommand.js +21 -0
  8. package/dist-cjs/commands/ListDataSourceSynchronizationLogsCommand.js +21 -0
  9. package/dist-cjs/commands/ListSearchQueriesHistoryCommand.js +21 -0
  10. package/dist-cjs/commands/index.js +8 -4
  11. package/dist-cjs/models/models_0.js +15 -27
  12. package/dist-cjs/protocols/Aws_restJson1.js +228 -140
  13. package/dist-es/KnowledgeBase.js +16 -8
  14. package/dist-es/commands/GenerateSearchAnswerCommand.js +17 -0
  15. package/dist-es/commands/GetDataSourceSyncStatusCommand.js +17 -0
  16. package/dist-es/commands/GetDocumentChunksCommand.js +17 -0
  17. package/dist-es/commands/GetDocumentMarkdownCommand.js +17 -0
  18. package/dist-es/commands/GetDocumentWithContentAndChunksCommand.js +17 -0
  19. package/dist-es/commands/ListDataSourceStatisticsCommand.js +17 -0
  20. package/dist-es/commands/ListDataSourceSynchronizationLogsCommand.js +17 -0
  21. package/dist-es/commands/ListSearchQueriesHistoryCommand.js +17 -0
  22. package/dist-es/commands/index.js +8 -4
  23. package/dist-es/models/models_0.js +14 -26
  24. package/dist-es/protocols/Aws_restJson1.js +203 -123
  25. package/dist-types/KnowledgeBase.d.ts +54 -26
  26. package/dist-types/KnowledgeBaseClient.d.ts +10 -6
  27. package/dist-types/commands/ConfigureDataSourceCommand.d.ts +1 -36
  28. package/dist-types/commands/CreateDataSourceCommand.d.ts +2 -22
  29. package/dist-types/commands/GenerateSearchAnswerCommand.d.ts +105 -0
  30. package/dist-types/commands/GetDataSourceCommand.d.ts +1 -21
  31. package/dist-types/commands/GetDataSourceSyncStatusCommand.d.ts +88 -0
  32. package/dist-types/commands/GetDocumentChunksCommand.d.ts +94 -0
  33. package/dist-types/commands/GetDocumentCommand.d.ts +0 -2
  34. package/dist-types/commands/GetDocumentMarkdownCommand.d.ts +86 -0
  35. package/dist-types/commands/GetDocumentWithContentAndChunksCommand.d.ts +113 -0
  36. package/dist-types/commands/ListDataSourceStatisticsCommand.d.ts +86 -0
  37. package/dist-types/commands/ListDataSourceSynchronizationLogsCommand.d.ts +96 -0
  38. package/dist-types/commands/ListDataSourcesCommand.d.ts +1 -21
  39. package/dist-types/commands/ListSearchQueriesHistoryCommand.d.ts +97 -0
  40. package/dist-types/commands/SearchKnowledgeBaseCommand.d.ts +7 -9
  41. package/dist-types/commands/StartSyncDataSourceCommand.d.ts +1 -1
  42. package/dist-types/commands/UpdateDataSourceCommand.d.ts +1 -21
  43. package/dist-types/commands/index.d.ts +8 -4
  44. package/dist-types/models/models_0.d.ts +253 -336
  45. package/dist-types/protocols/Aws_restJson1.d.ts +64 -28
  46. package/package.json +1 -1
  47. package/dist-es/commands/GetSyncLogsCommand.js +0 -17
  48. package/dist-es/commands/GetSyncStatusCommand.js +0 -17
  49. package/dist-es/commands/QueryKnowledgeBaseCommand.js +0 -17
  50. package/dist-es/commands/SearchHistoryCommand.js +0 -17
  51. package/dist-types/commands/GetSyncLogsCommand.d.ts +0 -96
  52. package/dist-types/commands/GetSyncStatusCommand.d.ts +0 -94
  53. package/dist-types/commands/QueryKnowledgeBaseCommand.d.ts +0 -103
  54. package/dist-types/commands/SearchHistoryCommand.d.ts +0 -98
@@ -2,7 +2,7 @@ import { KnowledgeBaseServiceException as __BaseException } from "../models/Know
2
2
  import { DataSourceConfig, DataSourceNotFoundException, DocumentNotFoundException, ForbiddenException, KnowledgeBaseNotFoundException, UnauthorizedException, ValidationException, } from "../models/models_0";
3
3
  import { awsExpectUnion as __expectUnion, loadRestJsonErrorCode, parseJsonBody as parseBody, parseJsonErrorBody as parseErrorBody, } from "@aws-sdk/core";
4
4
  import { requestBuilder as rb } from "@smithy/core";
5
- import { decorateServiceException as __decorateServiceException, expectInt32 as __expectInt32, expectNonNull as __expectNonNull, expectObject as __expectObject, expectString as __expectString, limitedParseDouble as __limitedParseDouble, serializeFloat as __serializeFloat, _json, collectBody, isSerializableHeaderValue, map, take, withBaseException, } from "@smithy/smithy-client";
5
+ 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";
6
6
  export const se_CleanDataSourceCommand = async (input, context) => {
7
7
  const b = rb(input, context);
8
8
  const headers = {};
@@ -188,6 +188,30 @@ export const se_DescribeDataSourceCommand = async (input, context) => {
188
188
  .b(body);
189
189
  return b.build();
190
190
  };
191
+ export const se_GenerateSearchAnswerCommand = async (input, context) => {
192
+ const b = rb(input, context);
193
+ const headers = {
194
+ 'content-type': 'application/json',
195
+ };
196
+ b.bp("/api/v1/knowledge-bases/generate-answer");
197
+ const query = map({
198
+ [_c]: [, input[_cI]],
199
+ });
200
+ let body;
201
+ body = JSON.stringify(take(input, {
202
+ 'maxOutputTokens': [],
203
+ 'model': [],
204
+ 'searchQuery': [],
205
+ 'searchResults': _ => se_SearchKnowledgeBaseResults(_, context),
206
+ 'systemPrompt': [],
207
+ 'temperature': _ => __serializeFloat(_),
208
+ }));
209
+ b.m("POST")
210
+ .h(headers)
211
+ .q(query)
212
+ .b(body);
213
+ return b.build();
214
+ };
191
215
  export const se_GetDataSourceCommand = async (input, context) => {
192
216
  const b = rb(input, context);
193
217
  const headers = {};
@@ -203,6 +227,21 @@ export const se_GetDataSourceCommand = async (input, context) => {
203
227
  .b(body);
204
228
  return b.build();
205
229
  };
230
+ export const se_GetDataSourceSyncStatusCommand = async (input, context) => {
231
+ const b = rb(input, context);
232
+ const headers = {};
233
+ b.bp("/api/v1/data-sources/{dataSourceId}/sync-status");
234
+ b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
235
+ const query = map({
236
+ [_c]: [, input[_cI]],
237
+ });
238
+ let body;
239
+ b.m("GET")
240
+ .h(headers)
241
+ .q(query)
242
+ .b(body);
243
+ return b.build();
244
+ };
206
245
  export const se_GetDocumentCommand = async (input, context) => {
207
246
  const b = rb(input, context);
208
247
  const headers = {};
@@ -211,8 +250,23 @@ export const se_GetDocumentCommand = async (input, context) => {
211
250
  b.p('documentId', () => input.documentId, '{documentId}', false);
212
251
  const query = map({
213
252
  [_c]: [, input[_cI]],
214
- [_wC]: [() => input.withContent !== void 0, () => (input[_wC].toString())],
215
- [_wCi]: [() => input.withChunks !== void 0, () => (input[_wCi].toString())],
253
+ [_wC]: [() => input.withChunks !== void 0, () => (input[_wC].toString())],
254
+ });
255
+ let body;
256
+ b.m("GET")
257
+ .h(headers)
258
+ .q(query)
259
+ .b(body);
260
+ return b.build();
261
+ };
262
+ export const se_GetDocumentChunksCommand = async (input, context) => {
263
+ const b = rb(input, context);
264
+ const headers = {};
265
+ b.bp("/api/v1/data-sources/{dataSourceId}/documents/{documentId}/chunks");
266
+ b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
267
+ b.p('documentId', () => input.documentId, '{documentId}', false);
268
+ const query = map({
269
+ [_c]: [, input[_cI]],
216
270
  });
217
271
  let body;
218
272
  b.m("GET")
@@ -238,10 +292,10 @@ export const se_GetDocumentDownloadUrlCommand = async (input, context) => {
238
292
  .b(body);
239
293
  return b.build();
240
294
  };
241
- export const se_GetDocumentUploadUrlCommand = async (input, context) => {
295
+ export const se_GetDocumentMarkdownCommand = async (input, context) => {
242
296
  const b = rb(input, context);
243
297
  const headers = {};
244
- b.bp("/api/v1/data-sources/{dataSourceId}/documents/{documentId}/upload_url");
298
+ b.bp("/api/v1/data-sources/{dataSourceId}/documents/{documentId}/markdown");
245
299
  b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
246
300
  b.p('documentId', () => input.documentId, '{documentId}', false);
247
301
  const query = map({
@@ -254,11 +308,12 @@ export const se_GetDocumentUploadUrlCommand = async (input, context) => {
254
308
  .b(body);
255
309
  return b.build();
256
310
  };
257
- export const se_GetKnowledgeBaseCommand = async (input, context) => {
311
+ export const se_GetDocumentUploadUrlCommand = async (input, context) => {
258
312
  const b = rb(input, context);
259
313
  const headers = {};
260
- b.bp("/api/v1/knowledge-bases/{knowledgeBaseId}");
261
- b.p('knowledgeBaseId', () => input.knowledgeBaseId, '{knowledgeBaseId}', false);
314
+ b.bp("/api/v1/data-sources/{dataSourceId}/documents/{documentId}/upload_url");
315
+ b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
316
+ b.p('documentId', () => input.documentId, '{documentId}', false);
262
317
  const query = map({
263
318
  [_c]: [, input[_cI]],
264
319
  });
@@ -269,18 +324,14 @@ export const se_GetKnowledgeBaseCommand = async (input, context) => {
269
324
  .b(body);
270
325
  return b.build();
271
326
  };
272
- export const se_GetSyncLogsCommand = async (input, context) => {
327
+ export const se_GetDocumentWithContentAndChunksCommand = async (input, context) => {
273
328
  const b = rb(input, context);
274
329
  const headers = {};
275
- b.bp("/api/v1/data-sources/{dataSourceId}/sync-logs");
330
+ b.bp("/api/v1/data-sources/{dataSourceId}/documents/{documentId}/with-content-and-chunks");
276
331
  b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
332
+ b.p('documentId', () => input.documentId, '{documentId}', false);
277
333
  const query = map({
278
334
  [_c]: [, input[_cI]],
279
- [_l]: [() => input.limit !== void 0, () => (input[_l].toString())],
280
- [_o]: [() => input.offset !== void 0, () => (input[_o].toString())],
281
- [_s]: [, input[_s]],
282
- [_lL]: [, input[_lL]],
283
- [_se]: [, input[_se]],
284
335
  });
285
336
  let body;
286
337
  b.m("GET")
@@ -289,11 +340,11 @@ export const se_GetSyncLogsCommand = async (input, context) => {
289
340
  .b(body);
290
341
  return b.build();
291
342
  };
292
- export const se_GetSyncStatusCommand = async (input, context) => {
343
+ export const se_GetKnowledgeBaseCommand = async (input, context) => {
293
344
  const b = rb(input, context);
294
345
  const headers = {};
295
- b.bp("/api/v1/data-sources/{dataSourceId}/sync-status");
296
- b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
346
+ b.bp("/api/v1/knowledge-bases/{knowledgeBaseId}");
347
+ b.p('knowledgeBaseId', () => input.knowledgeBaseId, '{knowledgeBaseId}', false);
297
348
  const query = map({
298
349
  [_c]: [, input[_cI]],
299
350
  });
@@ -318,15 +369,33 @@ export const se_ListDataSourcesCommand = async (input, context) => {
318
369
  .b(body);
319
370
  return b.build();
320
371
  };
321
- export const se_ListDocumentsCommand = async (input, context) => {
372
+ export const se_ListDataSourceStatisticsCommand = async (input, context) => {
322
373
  const b = rb(input, context);
323
374
  const headers = {};
324
- b.bp("/api/v1/data-sources/{dataSourceId}/documents");
375
+ b.bp("/api/v1/data-sources/{dataSourceId}/statistics");
376
+ b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
377
+ const query = map({
378
+ [_c]: [, input[_cI]],
379
+ });
380
+ let body;
381
+ b.m("GET")
382
+ .h(headers)
383
+ .q(query)
384
+ .b(body);
385
+ return b.build();
386
+ };
387
+ export const se_ListDataSourceSynchronizationLogsCommand = async (input, context) => {
388
+ const b = rb(input, context);
389
+ const headers = {};
390
+ b.bp("/api/v1/data-sources/{dataSourceId}/sync-logs");
325
391
  b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
326
392
  const query = map({
327
393
  [_c]: [, input[_cI]],
328
394
  [_l]: [() => input.limit !== void 0, () => (input[_l].toString())],
329
395
  [_o]: [() => input.offset !== void 0, () => (input[_o].toString())],
396
+ [_s]: [, input[_s]],
397
+ [_lL]: [, input[_lL]],
398
+ [_se]: [, input[_se]],
330
399
  });
331
400
  let body;
332
401
  b.m("GET")
@@ -335,12 +404,15 @@ export const se_ListDocumentsCommand = async (input, context) => {
335
404
  .b(body);
336
405
  return b.build();
337
406
  };
338
- export const se_ListKnowledgeBasesCommand = async (input, context) => {
407
+ export const se_ListDocumentsCommand = async (input, context) => {
339
408
  const b = rb(input, context);
340
409
  const headers = {};
341
- b.bp("/api/v1/knowledge-bases");
410
+ b.bp("/api/v1/data-sources/{dataSourceId}/documents");
411
+ b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
342
412
  const query = map({
343
413
  [_c]: [, input[_cI]],
414
+ [_l]: [() => input.limit !== void 0, () => (input[_l].toString())],
415
+ [_o]: [() => input.offset !== void 0, () => (input[_o].toString())],
344
416
  });
345
417
  let body;
346
418
  b.m("GET")
@@ -349,36 +421,26 @@ export const se_ListKnowledgeBasesCommand = async (input, context) => {
349
421
  .b(body);
350
422
  return b.build();
351
423
  };
352
- export const se_QueryKnowledgeBaseCommand = async (input, context) => {
424
+ export const se_ListKnowledgeBasesCommand = async (input, context) => {
353
425
  const b = rb(input, context);
354
- const headers = map({}, isSerializableHeaderValue, {
355
- 'content-type': 'application/json',
356
- [_xsu]: input[_sU],
357
- [_xesi]: input[_eSI],
358
- });
359
- b.bp("/api/v1/knowledge-bases/{knowledgeBaseId}/query");
360
- b.p('knowledgeBaseId', () => input.knowledgeBaseId, '{knowledgeBaseId}', false);
426
+ const headers = {};
427
+ b.bp("/api/v1/knowledge-bases");
361
428
  const query = map({
362
429
  [_c]: [, input[_cI]],
363
430
  });
364
431
  let body;
365
- body = JSON.stringify(take(input, {
366
- 'llmConfig': _ => se_LlmConfig(_, context),
367
- 'query': [],
368
- 'searchConfig': _ => se_SearchConfig(_, context),
369
- }));
370
- b.m("POST")
432
+ b.m("GET")
371
433
  .h(headers)
372
434
  .q(query)
373
435
  .b(body);
374
436
  return b.build();
375
437
  };
376
- export const se_SearchHistoryCommand = async (input, context) => {
438
+ export const se_ListSearchQueriesHistoryCommand = async (input, context) => {
377
439
  const b = rb(input, context);
378
440
  const headers = {
379
441
  'content-type': 'application/json',
380
442
  };
381
- b.bp("/api/v1/history/search");
443
+ b.bp("/api/v1/history/search-queries");
382
444
  const query = map({
383
445
  [_c]: [, input[_cI]],
384
446
  [_l]: [() => input.limit !== void 0, () => (input[_l].toString())],
@@ -398,11 +460,9 @@ export const se_SearchHistoryCommand = async (input, context) => {
398
460
  };
399
461
  export const se_SearchKnowledgeBaseCommand = async (input, context) => {
400
462
  const b = rb(input, context);
401
- const headers = map({}, isSerializableHeaderValue, {
463
+ const headers = {
402
464
  'content-type': 'application/json',
403
- [_xsu]: input[_sU],
404
- [_xesi]: input[_eSI],
405
- });
465
+ };
406
466
  b.bp("/api/v1/knowledge-bases/{knowledgeBaseId}/search");
407
467
  b.p('knowledgeBaseId', () => input.knowledgeBaseId, '{knowledgeBaseId}', false);
408
468
  const query = map({
@@ -410,9 +470,12 @@ export const se_SearchKnowledgeBaseCommand = async (input, context) => {
410
470
  });
411
471
  let body;
412
472
  body = JSON.stringify(take(input, {
473
+ 'externalSessionId': [],
413
474
  'query': [],
414
- 'searchConfig': _ => se_SearchConfig(_, context),
415
- 'withMetadata': [],
475
+ 'searchLimit': [],
476
+ 'searchThreshold': [],
477
+ 'searchUser': [],
478
+ 'useQueryOptimization': [],
416
479
  }));
417
480
  b.m("POST")
418
481
  .h(headers)
@@ -427,7 +490,7 @@ export const se_StartSyncDataSourceCommand = async (input, context) => {
427
490
  b.p('dataSourceId', () => input.dataSourceId, '{dataSourceId}', false);
428
491
  const query = map({
429
492
  [_c]: [, input[_cI]],
430
- [_sT]: [, __expectNonNull(input[_sT], `syncType`)],
493
+ [_sT]: [, input[_sT]],
431
494
  });
432
495
  let body;
433
496
  b.m("POST")
@@ -644,6 +707,22 @@ export const de_DescribeDataSourceCommand = async (output, context) => {
644
707
  Object.assign(contents, doc);
645
708
  return contents;
646
709
  };
710
+ export const de_GenerateSearchAnswerCommand = async (output, context) => {
711
+ if (output.statusCode !== 200 && output.statusCode >= 300) {
712
+ return de_CommandError(output, context);
713
+ }
714
+ const contents = map({
715
+ $metadata: deserializeMetadata(output),
716
+ });
717
+ const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
718
+ const doc = take(data, {
719
+ 'answer': __expectString,
720
+ 'metadata': _ => de_Document(_, context),
721
+ 'sources': _ => de_Document(_, context),
722
+ });
723
+ Object.assign(contents, doc);
724
+ return contents;
725
+ };
647
726
  export const de_GetDataSourceCommand = async (output, context) => {
648
727
  if (output.statusCode !== 200 && output.statusCode >= 300) {
649
728
  return de_CommandError(output, context);
@@ -658,6 +737,23 @@ export const de_GetDataSourceCommand = async (output, context) => {
658
737
  Object.assign(contents, doc);
659
738
  return contents;
660
739
  };
740
+ export const de_GetDataSourceSyncStatusCommand = async (output, context) => {
741
+ if (output.statusCode !== 200 && output.statusCode >= 300) {
742
+ return de_CommandError(output, context);
743
+ }
744
+ const contents = map({
745
+ $metadata: deserializeMetadata(output),
746
+ });
747
+ const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
748
+ const doc = take(data, {
749
+ 'lastSyncErrorMessage': __expectString,
750
+ 'lastSyncMode': __expectString,
751
+ 'lastSyncedAt': __expectString,
752
+ 'syncStatus': __expectString,
753
+ });
754
+ Object.assign(contents, doc);
755
+ return contents;
756
+ };
661
757
  export const de_GetDocumentCommand = async (output, context) => {
662
758
  if (output.statusCode !== 200 && output.statusCode >= 300) {
663
759
  return de_CommandError(output, context);
@@ -668,12 +764,25 @@ export const de_GetDocumentCommand = async (output, context) => {
668
764
  const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
669
765
  const doc = take(data, {
670
766
  'chunks': _json,
671
- 'content': __expectString,
672
767
  'document': _json,
673
768
  });
674
769
  Object.assign(contents, doc);
675
770
  return contents;
676
771
  };
772
+ export const de_GetDocumentChunksCommand = async (output, context) => {
773
+ if (output.statusCode !== 200 && output.statusCode >= 300) {
774
+ return de_CommandError(output, context);
775
+ }
776
+ const contents = map({
777
+ $metadata: deserializeMetadata(output),
778
+ });
779
+ const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
780
+ const doc = take(data, {
781
+ 'chunks': _json,
782
+ });
783
+ Object.assign(contents, doc);
784
+ return contents;
785
+ };
677
786
  export const de_GetDocumentDownloadUrlCommand = async (output, context) => {
678
787
  if (output.statusCode !== 200 && output.statusCode >= 300) {
679
788
  return de_CommandError(output, context);
@@ -688,7 +797,7 @@ export const de_GetDocumentDownloadUrlCommand = async (output, context) => {
688
797
  Object.assign(contents, doc);
689
798
  return contents;
690
799
  };
691
- export const de_GetDocumentUploadUrlCommand = async (output, context) => {
800
+ export const de_GetDocumentMarkdownCommand = async (output, context) => {
692
801
  if (output.statusCode !== 200 && output.statusCode >= 300) {
693
802
  return de_CommandError(output, context);
694
803
  }
@@ -697,12 +806,12 @@ export const de_GetDocumentUploadUrlCommand = async (output, context) => {
697
806
  });
698
807
  const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
699
808
  const doc = take(data, {
700
- 'uploadUrl': __expectString,
809
+ 'markdown': __expectString,
701
810
  });
702
811
  Object.assign(contents, doc);
703
812
  return contents;
704
813
  };
705
- export const de_GetKnowledgeBaseCommand = async (output, context) => {
814
+ export const de_GetDocumentUploadUrlCommand = async (output, context) => {
706
815
  if (output.statusCode !== 200 && output.statusCode >= 300) {
707
816
  return de_CommandError(output, context);
708
817
  }
@@ -711,12 +820,12 @@ export const de_GetKnowledgeBaseCommand = async (output, context) => {
711
820
  });
712
821
  const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
713
822
  const doc = take(data, {
714
- 'knowledgeBase': _json,
823
+ 'uploadUrl': __expectString,
715
824
  });
716
825
  Object.assign(contents, doc);
717
826
  return contents;
718
827
  };
719
- export const de_GetSyncLogsCommand = async (output, context) => {
828
+ export const de_GetDocumentWithContentAndChunksCommand = async (output, context) => {
720
829
  if (output.statusCode !== 200 && output.statusCode >= 300) {
721
830
  return de_CommandError(output, context);
722
831
  }
@@ -725,12 +834,14 @@ export const de_GetSyncLogsCommand = async (output, context) => {
725
834
  });
726
835
  const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
727
836
  const doc = take(data, {
728
- 'logs': _json,
837
+ 'chunks': _json,
838
+ 'content': __expectString,
839
+ 'document': _json,
729
840
  });
730
841
  Object.assign(contents, doc);
731
842
  return contents;
732
843
  };
733
- export const de_GetSyncStatusCommand = async (output, context) => {
844
+ export const de_GetKnowledgeBaseCommand = async (output, context) => {
734
845
  if (output.statusCode !== 200 && output.statusCode >= 300) {
735
846
  return de_CommandError(output, context);
736
847
  }
@@ -739,10 +850,7 @@ export const de_GetSyncStatusCommand = async (output, context) => {
739
850
  });
740
851
  const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
741
852
  const doc = take(data, {
742
- 'lastSyncLogs': _json,
743
- 'progress': __expectInt32,
744
- 'syncErrorMessage': __expectString,
745
- 'syncStatus': __expectString,
853
+ 'knowledgeBase': _json,
746
854
  });
747
855
  Object.assign(contents, doc);
748
856
  return contents;
@@ -761,7 +869,7 @@ export const de_ListDataSourcesCommand = async (output, context) => {
761
869
  Object.assign(contents, doc);
762
870
  return contents;
763
871
  };
764
- export const de_ListDocumentsCommand = async (output, context) => {
872
+ export const de_ListDataSourceStatisticsCommand = async (output, context) => {
765
873
  if (output.statusCode !== 200 && output.statusCode >= 300) {
766
874
  return de_CommandError(output, context);
767
875
  }
@@ -770,12 +878,13 @@ export const de_ListDocumentsCommand = async (output, context) => {
770
878
  });
771
879
  const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
772
880
  const doc = take(data, {
773
- 'documents': _json,
881
+ 'chunksCount': __expectInt32,
882
+ 'documentsCount': __expectInt32,
774
883
  });
775
884
  Object.assign(contents, doc);
776
885
  return contents;
777
886
  };
778
- export const de_ListKnowledgeBasesCommand = async (output, context) => {
887
+ export const de_ListDataSourceSynchronizationLogsCommand = async (output, context) => {
779
888
  if (output.statusCode !== 200 && output.statusCode >= 300) {
780
889
  return de_CommandError(output, context);
781
890
  }
@@ -784,12 +893,12 @@ export const de_ListKnowledgeBasesCommand = async (output, context) => {
784
893
  });
785
894
  const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
786
895
  const doc = take(data, {
787
- 'knowledgeBases': _json,
896
+ 'logs': _json,
788
897
  });
789
898
  Object.assign(contents, doc);
790
899
  return contents;
791
900
  };
792
- export const de_QueryKnowledgeBaseCommand = async (output, context) => {
901
+ export const de_ListDocumentsCommand = async (output, context) => {
793
902
  if (output.statusCode !== 200 && output.statusCode >= 300) {
794
903
  return de_CommandError(output, context);
795
904
  }
@@ -798,14 +907,26 @@ export const de_QueryKnowledgeBaseCommand = async (output, context) => {
798
907
  });
799
908
  const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
800
909
  const doc = take(data, {
801
- 'answer': __expectString,
802
- 'metadata': _ => de_Document(_, context),
803
- 'sources': _ => de_Document(_, context),
910
+ 'documents': _json,
911
+ });
912
+ Object.assign(contents, doc);
913
+ return contents;
914
+ };
915
+ export const de_ListKnowledgeBasesCommand = async (output, context) => {
916
+ if (output.statusCode !== 200 && output.statusCode >= 300) {
917
+ return de_CommandError(output, context);
918
+ }
919
+ const contents = map({
920
+ $metadata: deserializeMetadata(output),
921
+ });
922
+ const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
923
+ const doc = take(data, {
924
+ 'knowledgeBases': _json,
804
925
  });
805
926
  Object.assign(contents, doc);
806
927
  return contents;
807
928
  };
808
- export const de_SearchHistoryCommand = async (output, context) => {
929
+ export const de_ListSearchQueriesHistoryCommand = async (output, context) => {
809
930
  if (output.statusCode !== 200 && output.statusCode >= 300) {
810
931
  return de_CommandError(output, context);
811
932
  }
@@ -814,7 +935,7 @@ export const de_SearchHistoryCommand = async (output, context) => {
814
935
  });
815
936
  const data = __expectNonNull((__expectObject(await parseBody(output.body, context))), "body");
816
937
  const doc = take(data, {
817
- 'results': _ => de_SearchHistoryResults(_, context),
938
+ 'results': _ => de_SearchHistoryResultsList(_, context),
818
939
  });
819
940
  Object.assign(contents, doc);
820
941
  return contents;
@@ -1017,7 +1138,6 @@ const se_DataSourceConfig = (input, context) => {
1017
1138
  confluence: value => ({ "confluence": _json(value) }),
1018
1139
  files: value => ({ "files": se_Document(value, context) }),
1019
1140
  gdrive: value => ({ "gdrive": _json(value) }),
1020
- proxy: value => ({ "proxy": se_ProxyConfig(value, context) }),
1021
1141
  _: (name, value) => ({ [name]: value })
1022
1142
  });
1023
1143
  };
@@ -1029,34 +1149,21 @@ const se_DescribeDataSourceParameters = (input, context) => {
1029
1149
  'proxy': _ => se_Document(_, context),
1030
1150
  });
1031
1151
  };
1032
- const se_LlmConfig = (input, context) => {
1033
- return take(input, {
1034
- 'maxOutputTokens': [],
1035
- 'model': [],
1036
- 'systemPrompt': [],
1037
- 'temperature': __serializeFloat,
1038
- });
1039
- };
1040
- const se_ProxyConfig = (input, context) => {
1152
+ const se_SearchKnowledgeBaseResult = (input, context) => {
1041
1153
  return take(input, {
1042
- 'body': _ => se_Document(_, context),
1043
- 'headers': _json,
1044
- 'method': [],
1045
- 'password': [],
1046
- 'responseMapping': _json,
1047
- 'timeoutMs': [],
1048
- 'token': [],
1049
- 'url': [],
1050
- 'username': [],
1154
+ 'content': [],
1155
+ 'dataSourceId': [],
1156
+ 'dataSourceType': [],
1157
+ 'documentHierarchy': _json,
1158
+ 'documentId': [],
1159
+ 'documentTitle': [],
1160
+ 'documentUrl': [],
1161
+ 'score': __serializeFloat,
1051
1162
  });
1052
1163
  };
1053
- const se_SearchConfig = (input, context) => {
1054
- return take(input, {
1055
- 'extendMetadata': [],
1056
- 'searchStrategy': [],
1057
- 'threshold': __serializeFloat,
1058
- 'topK': [],
1059
- 'useQueryOptimization': [],
1164
+ const se_SearchKnowledgeBaseResults = (input, context) => {
1165
+ return input.filter((e) => e != null).map(entry => {
1166
+ return se_SearchKnowledgeBaseResult(entry, context);
1060
1167
  });
1061
1168
  };
1062
1169
  const de_Document = (output, context) => {
@@ -1078,11 +1185,6 @@ const de_DataSourceConfig = (output, context) => {
1078
1185
  gdrive: _json(output.gdrive)
1079
1186
  };
1080
1187
  }
1081
- if (output.proxy != null) {
1082
- return {
1083
- proxy: de_ProxyConfig(output.proxy, context)
1084
- };
1085
- }
1086
1188
  return { $unknown: Object.entries(output)[0] };
1087
1189
  };
1088
1190
  const de_DataSourceItem = (output, context) => {
@@ -1091,14 +1193,9 @@ const de_DataSourceItem = (output, context) => {
1091
1193
  'config': (_) => de_DataSourceConfig(__expectUnion(_), context),
1092
1194
  'createdAt': __expectString,
1093
1195
  'description': __expectString,
1094
- 'documentsCount': __expectInt32,
1095
1196
  'id': __expectString,
1096
- 'lastSyncErrorMessage': __expectString,
1097
- 'lastSyncMode': __expectString,
1098
- 'lastSyncedAt': __expectString,
1099
1197
  'name': __expectString,
1100
1198
  'syncSchedule': __expectString,
1101
- 'syncStatus': __expectString,
1102
1199
  'type': __expectString,
1103
1200
  'updatedAt': __expectString,
1104
1201
  });
@@ -1117,20 +1214,7 @@ const de_DescribeDataSourceResult = (output, context) => {
1117
1214
  'proxy': (_) => de_Document(_, context),
1118
1215
  });
1119
1216
  };
1120
- const de_ProxyConfig = (output, context) => {
1121
- return take(output, {
1122
- 'body': (_) => de_Document(_, context),
1123
- 'headers': _json,
1124
- 'method': __expectString,
1125
- 'password': __expectString,
1126
- 'responseMapping': _json,
1127
- 'timeoutMs': __expectInt32,
1128
- 'token': __expectString,
1129
- 'url': __expectString,
1130
- 'username': __expectString,
1131
- });
1132
- };
1133
- const de_SearchHistoryResults = (output, context) => {
1217
+ const de_SearchHistoryResultsList = (output, context) => {
1134
1218
  const retVal = (output || []).filter((e) => e != null).map((entry) => {
1135
1219
  return de_Document(entry, context);
1136
1220
  });
@@ -1141,6 +1225,7 @@ const de_SearchKnowledgeBaseResult = (output, context) => {
1141
1225
  'content': __expectString,
1142
1226
  'dataSourceId': __expectString,
1143
1227
  'dataSourceType': __expectString,
1228
+ 'documentHierarchy': _json,
1144
1229
  'documentId': __expectString,
1145
1230
  'documentTitle': __expectString,
1146
1231
  'documentUrl': __expectString,
@@ -1163,15 +1248,10 @@ const collectBodyString = (streamBody, context) => collectBody(streamBody, conte
1163
1248
  const _c = "company";
1164
1249
  const _cI = "companyId";
1165
1250
  const _dT = "documentType";
1166
- const _eSI = "externalSessionId";
1167
1251
  const _l = "limit";
1168
1252
  const _lL = "logLevel";
1169
1253
  const _o = "offset";
1170
1254
  const _s = "sort";
1171
1255
  const _sT = "syncType";
1172
- const _sU = "searchUser";
1173
1256
  const _se = "search";
1174
- const _wC = "withContent";
1175
- const _wCi = "withChunks";
1176
- const _xesi = "x-external-session-id";
1177
- const _xsu = "x-search-user";
1257
+ const _wC = "withChunks";