Skip to content

Commit e77d1e3

Browse files
author
awstools
committed
feat(client-bedrock-agent): Introduces support for Neptune Analytics as a vector data store and adds Context Enrichment Configurations, enabling use cases such as GraphRAG.
1 parent f460bc1 commit e77d1e3

11 files changed

+597
-184
lines changed

‎clients/client-bedrock-agent/src/commands/CreateDataSourceCommand.ts

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,15 @@ export interface CreateDataSourceCommandOutput extends CreateDataSourceResponse,
214214
* parsingModality: "MULTIMODAL",
215215
* },
216216
* },
217+
* contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
218+
* type: "BEDROCK_FOUNDATION_MODEL", // required
219+
* bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
220+
* enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
221+
* method: "CHUNK_ENTITY_EXTRACTION", // required
222+
* },
223+
* modelArn: "STRING_VALUE", // required
224+
* },
225+
* },
217226
* },
218227
* };
219228
* const command = new CreateDataSourceCommand(input);
@@ -390,6 +399,15 @@ export interface CreateDataSourceCommandOutput extends CreateDataSourceResponse,
390399
* // parsingModality: "MULTIMODAL",
391400
* // },
392401
* // },
402+
* // contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
403+
* // type: "BEDROCK_FOUNDATION_MODEL", // required
404+
* // bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
405+
* // enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
406+
* // method: "CHUNK_ENTITY_EXTRACTION", // required
407+
* // },
408+
* // modelArn: "STRING_VALUE", // required
409+
* // },
410+
* // },
393411
* // },
394412
* // dataDeletionPolicy: "RETAIN" || "DELETE",
395413
* // createdAt: new Date("TIMESTAMP"), // required

‎clients/client-bedrock-agent/src/commands/CreateKnowledgeBaseCommand.ts

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ export interface CreateKnowledgeBaseCommandOutput extends CreateKnowledgeBaseRes
163163
* },
164164
* },
165165
* storageConfiguration: { // StorageConfiguration
166-
* type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
166+
* type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
167167
* opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
168168
* collectionArn: "STRING_VALUE", // required
169169
* vectorIndexName: "STRING_VALUE", // required
@@ -217,6 +217,13 @@ export interface CreateKnowledgeBaseCommandOutput extends CreateKnowledgeBaseRes
217217
* },
218218
* endpointServiceName: "STRING_VALUE",
219219
* },
220+
* neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
221+
* graphArn: "STRING_VALUE", // required
222+
* fieldMapping: { // NeptuneAnalyticsFieldMapping
223+
* textField: "STRING_VALUE", // required
224+
* metadataField: "STRING_VALUE", // required
225+
* },
226+
* },
220227
* },
221228
* tags: { // TagsMap
222229
* "<keys>": "STRING_VALUE",
@@ -318,7 +325,7 @@ export interface CreateKnowledgeBaseCommandOutput extends CreateKnowledgeBaseRes
318325
* // },
319326
* // },
320327
* // storageConfiguration: { // StorageConfiguration
321-
* // type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
328+
* // type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
322329
* // opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
323330
* // collectionArn: "STRING_VALUE", // required
324331
* // vectorIndexName: "STRING_VALUE", // required
@@ -372,6 +379,13 @@ export interface CreateKnowledgeBaseCommandOutput extends CreateKnowledgeBaseRes
372379
* // },
373380
* // endpointServiceName: "STRING_VALUE",
374381
* // },
382+
* // neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
383+
* // graphArn: "STRING_VALUE", // required
384+
* // fieldMapping: { // NeptuneAnalyticsFieldMapping
385+
* // textField: "STRING_VALUE", // required
386+
* // metadataField: "STRING_VALUE", // required
387+
* // },
388+
* // },
375389
* // },
376390
* // status: "CREATING" || "ACTIVE" || "DELETING" || "UPDATING" || "FAILED" || "DELETE_UNSUCCESSFUL", // required
377391
* // createdAt: new Date("TIMESTAMP"), // required

‎clients/client-bedrock-agent/src/commands/DeleteKnowledgeBaseDocumentsCommand.ts

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,7 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { BedrockAgentClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../BedrockAgentClient";
88
import { commonParams } from "../endpoint/EndpointParameters";
9-
import { DeleteKnowledgeBaseDocumentsRequest } from "../models/models_0";
10-
import { DeleteKnowledgeBaseDocumentsResponse } from "../models/models_1";
9+
import { DeleteKnowledgeBaseDocumentsRequest, DeleteKnowledgeBaseDocumentsResponse } from "../models/models_1";
1110
import {
1211
de_DeleteKnowledgeBaseDocumentsCommand,
1312
se_DeleteKnowledgeBaseDocumentsCommand,

‎clients/client-bedrock-agent/src/commands/GetDataSourceCommand.ts

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -217,6 +217,15 @@ export interface GetDataSourceCommandOutput extends GetDataSourceResponse, __Met
217217
* // parsingModality: "MULTIMODAL",
218218
* // },
219219
* // },
220+
* // contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
221+
* // type: "BEDROCK_FOUNDATION_MODEL", // required
222+
* // bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
223+
* // enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
224+
* // method: "CHUNK_ENTITY_EXTRACTION", // required
225+
* // },
226+
* // modelArn: "STRING_VALUE", // required
227+
* // },
228+
* // },
220229
* // },
221230
* // dataDeletionPolicy: "RETAIN" || "DELETE",
222231
* // createdAt: new Date("TIMESTAMP"), // required

‎clients/client-bedrock-agent/src/commands/GetKnowledgeBaseCommand.ts

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ export interface GetKnowledgeBaseCommandOutput extends GetKnowledgeBaseResponse,
138138
* // },
139139
* // },
140140
* // storageConfiguration: { // StorageConfiguration
141-
* // type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
141+
* // type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
142142
* // opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
143143
* // collectionArn: "STRING_VALUE", // required
144144
* // vectorIndexName: "STRING_VALUE", // required
@@ -192,6 +192,13 @@ export interface GetKnowledgeBaseCommandOutput extends GetKnowledgeBaseResponse,
192192
* // },
193193
* // endpointServiceName: "STRING_VALUE",
194194
* // },
195+
* // neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
196+
* // graphArn: "STRING_VALUE", // required
197+
* // fieldMapping: { // NeptuneAnalyticsFieldMapping
198+
* // textField: "STRING_VALUE", // required
199+
* // metadataField: "STRING_VALUE", // required
200+
* // },
201+
* // },
195202
* // },
196203
* // status: "CREATING" || "ACTIVE" || "DELETING" || "UPDATING" || "FAILED" || "DELETE_UNSUCCESSFUL", // required
197204
* // createdAt: new Date("TIMESTAMP"), // required

‎clients/client-bedrock-agent/src/commands/UpdateDataSourceCommand.ts

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,15 @@ export interface UpdateDataSourceCommandOutput extends UpdateDataSourceResponse,
214214
* parsingModality: "MULTIMODAL",
215215
* },
216216
* },
217+
* contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
218+
* type: "BEDROCK_FOUNDATION_MODEL", // required
219+
* bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
220+
* enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
221+
* method: "CHUNK_ENTITY_EXTRACTION", // required
222+
* },
223+
* modelArn: "STRING_VALUE", // required
224+
* },
225+
* },
217226
* },
218227
* };
219228
* const command = new UpdateDataSourceCommand(input);
@@ -390,6 +399,15 @@ export interface UpdateDataSourceCommandOutput extends UpdateDataSourceResponse,
390399
* // parsingModality: "MULTIMODAL",
391400
* // },
392401
* // },
402+
* // contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
403+
* // type: "BEDROCK_FOUNDATION_MODEL", // required
404+
* // bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
405+
* // enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
406+
* // method: "CHUNK_ENTITY_EXTRACTION", // required
407+
* // },
408+
* // modelArn: "STRING_VALUE", // required
409+
* // },
410+
* // },
393411
* // },
394412
* // dataDeletionPolicy: "RETAIN" || "DELETE",
395413
* // createdAt: new Date("TIMESTAMP"), // required

‎clients/client-bedrock-agent/src/commands/UpdateKnowledgeBaseCommand.ts

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ export interface UpdateKnowledgeBaseCommandOutput extends UpdateKnowledgeBaseRes
151151
* },
152152
* },
153153
* storageConfiguration: { // StorageConfiguration
154-
* type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
154+
* type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
155155
* opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
156156
* collectionArn: "STRING_VALUE", // required
157157
* vectorIndexName: "STRING_VALUE", // required
@@ -205,6 +205,13 @@ export interface UpdateKnowledgeBaseCommandOutput extends UpdateKnowledgeBaseRes
205205
* },
206206
* endpointServiceName: "STRING_VALUE",
207207
* },
208+
* neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
209+
* graphArn: "STRING_VALUE", // required
210+
* fieldMapping: { // NeptuneAnalyticsFieldMapping
211+
* textField: "STRING_VALUE", // required
212+
* metadataField: "STRING_VALUE", // required
213+
* },
214+
* },
208215
* },
209216
* };
210217
* const command = new UpdateKnowledgeBaseCommand(input);
@@ -303,7 +310,7 @@ export interface UpdateKnowledgeBaseCommandOutput extends UpdateKnowledgeBaseRes
303310
* // },
304311
* // },
305312
* // storageConfiguration: { // StorageConfiguration
306-
* // type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
313+
* // type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
307314
* // opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
308315
* // collectionArn: "STRING_VALUE", // required
309316
* // vectorIndexName: "STRING_VALUE", // required
@@ -357,6 +364,13 @@ export interface UpdateKnowledgeBaseCommandOutput extends UpdateKnowledgeBaseRes
357364
* // },
358365
* // endpointServiceName: "STRING_VALUE",
359366
* // },
367+
* // neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
368+
* // graphArn: "STRING_VALUE", // required
369+
* // fieldMapping: { // NeptuneAnalyticsFieldMapping
370+
* // textField: "STRING_VALUE", // required
371+
* // metadataField: "STRING_VALUE", // required
372+
* // },
373+
* // },
360374
* // },
361375
* // status: "CREATING" || "ACTIVE" || "DELETING" || "UPDATING" || "FAILED" || "DELETE_UNSUCCESSFUL", // required
362376
* // createdAt: new Date("TIMESTAMP"), // required

0 commit comments

Comments
 (0)