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.
aws · Mar 7, 2025 · e77d1e3 · e77d1e3
1 parent f460bc1
commit e77d1e3
Show file tree

Hide file tree

Showing 11 changed files with 597 additions and 184 deletions.
diff --git a/clients/client-bedrock-agent/src/commands/CreateDataSourceCommand.ts b/clients/client-bedrock-agent/src/commands/CreateDataSourceCommand.ts
@@ -214,6 +214,15 @@ export interface CreateDataSourceCommandOutput extends CreateDataSourceResponse,
  *         parsingModality: "MULTIMODAL",
  *       },
  *     },
+ *     contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
+ *       type: "BEDROCK_FOUNDATION_MODEL", // required
+ *       bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
+ *         enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
+ *           method: "CHUNK_ENTITY_EXTRACTION", // required
+ *         },
+ *         modelArn: "STRING_VALUE", // required
+ *       },
+ *     },
  *   },
  * };
  * const command = new CreateDataSourceCommand(input);
@@ -390,6 +399,15 @@ export interface CreateDataSourceCommandOutput extends CreateDataSourceResponse,
  * //           parsingModality: "MULTIMODAL",
  * //         },
  * //       },
+ * //       contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
+ * //         type: "BEDROCK_FOUNDATION_MODEL", // required
+ * //         bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
+ * //           enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
+ * //             method: "CHUNK_ENTITY_EXTRACTION", // required
+ * //           },
+ * //           modelArn: "STRING_VALUE", // required
+ * //         },
+ * //       },
  * //     },
  * //     dataDeletionPolicy: "RETAIN" || "DELETE",
  * //     createdAt: new Date("TIMESTAMP"), // required

diff --git a/clients/client-bedrock-agent/src/commands/CreateKnowledgeBaseCommand.ts b/clients/client-bedrock-agent/src/commands/CreateKnowledgeBaseCommand.ts
@@ -163,7 +163,7 @@ export interface CreateKnowledgeBaseCommandOutput extends CreateKnowledgeBaseRes
  *     },
  *   },
  *   storageConfiguration: { // StorageConfiguration
- *     type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
+ *     type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
  *     opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
  *       collectionArn: "STRING_VALUE", // required
  *       vectorIndexName: "STRING_VALUE", // required
@@ -217,6 +217,13 @@ export interface CreateKnowledgeBaseCommandOutput extends CreateKnowledgeBaseRes
  *       },
  *       endpointServiceName: "STRING_VALUE",
  *     },
+ *     neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
+ *       graphArn: "STRING_VALUE", // required
+ *       fieldMapping: { // NeptuneAnalyticsFieldMapping
+ *         textField: "STRING_VALUE", // required
+ *         metadataField: "STRING_VALUE", // required
+ *       },
+ *     },
  *   },
  *   tags: { // TagsMap
  *     "<keys>": "STRING_VALUE",
@@ -318,7 +325,7 @@ export interface CreateKnowledgeBaseCommandOutput extends CreateKnowledgeBaseRes
  * //       },
  * //     },
  * //     storageConfiguration: { // StorageConfiguration
- * //       type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
+ * //       type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
  * //       opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
  * //         collectionArn: "STRING_VALUE", // required
  * //         vectorIndexName: "STRING_VALUE", // required
@@ -372,6 +379,13 @@ export interface CreateKnowledgeBaseCommandOutput extends CreateKnowledgeBaseRes
  * //         },
  * //         endpointServiceName: "STRING_VALUE",
  * //       },
+ * //       neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
+ * //         graphArn: "STRING_VALUE", // required
+ * //         fieldMapping: { // NeptuneAnalyticsFieldMapping
+ * //           textField: "STRING_VALUE", // required
+ * //           metadataField: "STRING_VALUE", // required
+ * //         },
+ * //       },
  * //     },
  * //     status: "CREATING" || "ACTIVE" || "DELETING" || "UPDATING" || "FAILED" || "DELETE_UNSUCCESSFUL", // required
  * //     createdAt: new Date("TIMESTAMP"), // required

diff --git a/clients/client-bedrock-agent/src/commands/DeleteKnowledgeBaseDocumentsCommand.ts b/clients/client-bedrock-agent/src/commands/DeleteKnowledgeBaseDocumentsCommand.ts
@@ -6,8 +6,7 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { BedrockAgentClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../BedrockAgentClient";
 import { commonParams } from "../endpoint/EndpointParameters";
-import { DeleteKnowledgeBaseDocumentsRequest } from "../models/models_0";
-import { DeleteKnowledgeBaseDocumentsResponse } from "../models/models_1";
+import { DeleteKnowledgeBaseDocumentsRequest, DeleteKnowledgeBaseDocumentsResponse } from "../models/models_1";
 import {
   de_DeleteKnowledgeBaseDocumentsCommand,
   se_DeleteKnowledgeBaseDocumentsCommand,

diff --git a/clients/client-bedrock-agent/src/commands/GetDataSourceCommand.ts b/clients/client-bedrock-agent/src/commands/GetDataSourceCommand.ts
@@ -217,6 +217,15 @@ export interface GetDataSourceCommandOutput extends GetDataSourceResponse, __Met
  * //           parsingModality: "MULTIMODAL",
  * //         },
  * //       },
+ * //       contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
+ * //         type: "BEDROCK_FOUNDATION_MODEL", // required
+ * //         bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
+ * //           enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
+ * //             method: "CHUNK_ENTITY_EXTRACTION", // required
+ * //           },
+ * //           modelArn: "STRING_VALUE", // required
+ * //         },
+ * //       },
  * //     },
  * //     dataDeletionPolicy: "RETAIN" || "DELETE",
  * //     createdAt: new Date("TIMESTAMP"), // required

diff --git a/clients/client-bedrock-agent/src/commands/GetKnowledgeBaseCommand.ts b/clients/client-bedrock-agent/src/commands/GetKnowledgeBaseCommand.ts
@@ -138,7 +138,7 @@ export interface GetKnowledgeBaseCommandOutput extends GetKnowledgeBaseResponse,
  * //       },
  * //     },
  * //     storageConfiguration: { // StorageConfiguration
- * //       type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
+ * //       type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
  * //       opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
  * //         collectionArn: "STRING_VALUE", // required
  * //         vectorIndexName: "STRING_VALUE", // required
@@ -192,6 +192,13 @@ export interface GetKnowledgeBaseCommandOutput extends GetKnowledgeBaseResponse,
  * //         },
  * //         endpointServiceName: "STRING_VALUE",
  * //       },
+ * //       neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
+ * //         graphArn: "STRING_VALUE", // required
+ * //         fieldMapping: { // NeptuneAnalyticsFieldMapping
+ * //           textField: "STRING_VALUE", // required
+ * //           metadataField: "STRING_VALUE", // required
+ * //         },
+ * //       },
  * //     },
  * //     status: "CREATING" || "ACTIVE" || "DELETING" || "UPDATING" || "FAILED" || "DELETE_UNSUCCESSFUL", // required
  * //     createdAt: new Date("TIMESTAMP"), // required

diff --git a/clients/client-bedrock-agent/src/commands/UpdateDataSourceCommand.ts b/clients/client-bedrock-agent/src/commands/UpdateDataSourceCommand.ts
@@ -214,6 +214,15 @@ export interface UpdateDataSourceCommandOutput extends UpdateDataSourceResponse,
  *         parsingModality: "MULTIMODAL",
  *       },
  *     },
+ *     contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
+ *       type: "BEDROCK_FOUNDATION_MODEL", // required
+ *       bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
+ *         enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
+ *           method: "CHUNK_ENTITY_EXTRACTION", // required
+ *         },
+ *         modelArn: "STRING_VALUE", // required
+ *       },
+ *     },
  *   },
  * };
  * const command = new UpdateDataSourceCommand(input);
@@ -390,6 +399,15 @@ export interface UpdateDataSourceCommandOutput extends UpdateDataSourceResponse,
  * //           parsingModality: "MULTIMODAL",
  * //         },
  * //       },
+ * //       contextEnrichmentConfiguration: { // ContextEnrichmentConfiguration
+ * //         type: "BEDROCK_FOUNDATION_MODEL", // required
+ * //         bedrockFoundationModelConfiguration: { // BedrockFoundationModelContextEnrichmentConfiguration
+ * //           enrichmentStrategyConfiguration: { // EnrichmentStrategyConfiguration
+ * //             method: "CHUNK_ENTITY_EXTRACTION", // required
+ * //           },
+ * //           modelArn: "STRING_VALUE", // required
+ * //         },
+ * //       },
  * //     },
  * //     dataDeletionPolicy: "RETAIN" || "DELETE",
  * //     createdAt: new Date("TIMESTAMP"), // required

diff --git a/clients/client-bedrock-agent/src/commands/UpdateKnowledgeBaseCommand.ts b/clients/client-bedrock-agent/src/commands/UpdateKnowledgeBaseCommand.ts
@@ -151,7 +151,7 @@ export interface UpdateKnowledgeBaseCommandOutput extends UpdateKnowledgeBaseRes
  *     },
  *   },
  *   storageConfiguration: { // StorageConfiguration
- *     type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
+ *     type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
  *     opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
  *       collectionArn: "STRING_VALUE", // required
  *       vectorIndexName: "STRING_VALUE", // required
@@ -205,6 +205,13 @@ export interface UpdateKnowledgeBaseCommandOutput extends UpdateKnowledgeBaseRes
  *       },
  *       endpointServiceName: "STRING_VALUE",
  *     },
+ *     neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
+ *       graphArn: "STRING_VALUE", // required
+ *       fieldMapping: { // NeptuneAnalyticsFieldMapping
+ *         textField: "STRING_VALUE", // required
+ *         metadataField: "STRING_VALUE", // required
+ *       },
+ *     },
  *   },
  * };
  * const command = new UpdateKnowledgeBaseCommand(input);
@@ -303,7 +310,7 @@ export interface UpdateKnowledgeBaseCommandOutput extends UpdateKnowledgeBaseRes
  * //       },
  * //     },
  * //     storageConfiguration: { // StorageConfiguration
- * //       type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS", // required
+ * //       type: "OPENSEARCH_SERVERLESS" || "PINECONE" || "REDIS_ENTERPRISE_CLOUD" || "RDS" || "MONGO_DB_ATLAS" || "NEPTUNE_ANALYTICS", // required
  * //       opensearchServerlessConfiguration: { // OpenSearchServerlessConfiguration
  * //         collectionArn: "STRING_VALUE", // required
  * //         vectorIndexName: "STRING_VALUE", // required
@@ -357,6 +364,13 @@ export interface UpdateKnowledgeBaseCommandOutput extends UpdateKnowledgeBaseRes
  * //         },
  * //         endpointServiceName: "STRING_VALUE",
  * //       },
+ * //       neptuneAnalyticsConfiguration: { // NeptuneAnalyticsConfiguration
+ * //         graphArn: "STRING_VALUE", // required
+ * //         fieldMapping: { // NeptuneAnalyticsFieldMapping
+ * //           textField: "STRING_VALUE", // required
+ * //           metadataField: "STRING_VALUE", // required
+ * //         },
+ * //       },
  * //     },
  * //     status: "CREATING" || "ACTIVE" || "DELETING" || "UPDATING" || "FAILED" || "DELETE_UNSUCCESSFUL", // required
  * //     createdAt: new Date("TIMESTAMP"), // required