types-boto3-full 1.35.74__py3-none-any.whl → 1.35.76__py3-none-any.whl

types_boto3_bedrock_agent/type_defs.py

@@ -43,18 +43,25 @@ from .literals import (
     FlowStatusType,
     FlowValidationSeverityType,
     FlowValidationTypeType,
+    IncludeExcludeType,
     IngestionJobSortByAttributeType,
     IngestionJobStatusType,
     InlineContentTypeType,
     KnowledgeBaseStateType,
     KnowledgeBaseStatusType,
     KnowledgeBaseStorageTypeType,
+    KnowledgeBaseTypeType,
     MetadataSourceTypeType,
     MetadataValueTypeType,
     OrchestrationTypeType,
+    ParsingStrategyType,
     PromptStateType,
     PromptTemplateTypeType,
     PromptTypeType,
+    RedshiftProvisionedAuthTypeType,
+    RedshiftQueryEngineStorageTypeType,
+    RedshiftQueryEngineTypeType,
+    RedshiftServerlessAuthTypeType,
     RelayConversationHistoryType,
     RequireConfirmationType,
     SortOrderType,
@@ -91,6 +98,7 @@ __all__ = (
     "AssociateAgentCollaboratorResponseTypeDef",
     "AssociateAgentKnowledgeBaseRequestRequestTypeDef",
     "AssociateAgentKnowledgeBaseResponseTypeDef",
+    "BedrockDataAutomationConfigurationTypeDef",
     "BedrockEmbeddingModelConfigurationTypeDef",
     "BedrockFoundationModelConfigurationTypeDef",
     "BlobTypeDef",
@@ -135,6 +143,7 @@ __all__ = (
     "CreatePromptResponseTypeDef",
     "CreatePromptVersionRequestRequestTypeDef",
     "CreatePromptVersionResponseTypeDef",
+    "CuratedQueryTypeDef",
     "CustomContentTypeDef",
     "CustomDocumentIdentifierTypeDef",
     "CustomOrchestrationTypeDef",
@@ -249,6 +258,8 @@ __all__ = (
     "IngestionJobTypeDef",
     "InlineContentTypeDef",
     "IntermediateStorageTypeDef",
+    "KendraKnowledgeBaseConfigurationTypeDef",
+    "KnowledgeBaseConfigurationOutputTypeDef",
     "KnowledgeBaseConfigurationTypeDef",
     "KnowledgeBaseDocumentDetailTypeDef",
     "KnowledgeBaseDocumentTypeDef",
@@ -371,10 +382,35 @@ __all__ = (
     "PromptVariantOutputTypeDef",
     "PromptVariantTypeDef",
     "PromptVariantUnionTypeDef",
+    "QueryGenerationColumnTypeDef",
+    "QueryGenerationConfigurationOutputTypeDef",
+    "QueryGenerationConfigurationTypeDef",
+    "QueryGenerationConfigurationUnionTypeDef",
+    "QueryGenerationContextOutputTypeDef",
+    "QueryGenerationContextTypeDef",
+    "QueryGenerationContextUnionTypeDef",
+    "QueryGenerationTableOutputTypeDef",
+    "QueryGenerationTableTypeDef",
+    "QueryGenerationTableUnionTypeDef",
     "RdsConfigurationTypeDef",
     "RdsFieldMappingTypeDef",
     "RedisEnterpriseCloudConfigurationTypeDef",
     "RedisEnterpriseCloudFieldMappingTypeDef",
+    "RedshiftConfigurationOutputTypeDef",
+    "RedshiftConfigurationTypeDef",
+    "RedshiftConfigurationUnionTypeDef",
+    "RedshiftProvisionedAuthConfigurationTypeDef",
+    "RedshiftProvisionedConfigurationTypeDef",
+    "RedshiftQueryEngineAwsDataCatalogStorageConfigurationOutputTypeDef",
+    "RedshiftQueryEngineAwsDataCatalogStorageConfigurationTypeDef",
+    "RedshiftQueryEngineAwsDataCatalogStorageConfigurationUnionTypeDef",
+    "RedshiftQueryEngineConfigurationTypeDef",
+    "RedshiftQueryEngineRedshiftStorageConfigurationTypeDef",
+    "RedshiftQueryEngineStorageConfigurationOutputTypeDef",
+    "RedshiftQueryEngineStorageConfigurationTypeDef",
+    "RedshiftQueryEngineStorageConfigurationUnionTypeDef",
+    "RedshiftServerlessAuthConfigurationTypeDef",
+    "RedshiftServerlessConfigurationTypeDef",
     "ResponseMetadataTypeDef",
     "RetrievalFlowNodeConfigurationTypeDef",
     "RetrievalFlowNodeS3ConfigurationTypeDef",
@@ -405,6 +441,9 @@ __all__ = (
     "SharePointSourceConfigurationTypeDef",
     "SharePointSourceConfigurationUnionTypeDef",
     "SpecificToolChoiceTypeDef",
+    "SqlKnowledgeBaseConfigurationOutputTypeDef",
+    "SqlKnowledgeBaseConfigurationTypeDef",
+    "SqlKnowledgeBaseConfigurationUnionTypeDef",
     "StartIngestionJobRequestRequestTypeDef",
     "StartIngestionJobResponseTypeDef",
     "StopIngestionJobRequestRequestTypeDef",
@@ -413,6 +452,10 @@ __all__ = (
     "StorageFlowNodeConfigurationTypeDef",
     "StorageFlowNodeS3ConfigurationTypeDef",
     "StorageFlowNodeServiceConfigurationTypeDef",
+    "SupplementalDataStorageConfigurationOutputTypeDef",
+    "SupplementalDataStorageConfigurationTypeDef",
+    "SupplementalDataStorageConfigurationUnionTypeDef",
+    "SupplementalDataStorageLocationTypeDef",
     "SystemContentBlockTypeDef",
     "TagResourceRequestRequestTypeDef",
     "TextContentDocTypeDef",
@@ -473,7 +516,9 @@ __all__ = (
     "ValidateFlowDefinitionResponseTypeDef",
     "VectorIngestionConfigurationOutputTypeDef",
     "VectorIngestionConfigurationTypeDef",
+    "VectorKnowledgeBaseConfigurationOutputTypeDef",
     "VectorKnowledgeBaseConfigurationTypeDef",
+    "VectorKnowledgeBaseConfigurationUnionTypeDef",
     "WebCrawlerConfigurationOutputTypeDef",
     "WebCrawlerConfigurationTypeDef",
     "WebCrawlerConfigurationUnionTypeDef",
@@ -565,6 +610,10 @@ class AssociateAgentKnowledgeBaseRequestRequestTypeDef(TypedDict):
     knowledgeBaseState: NotRequired[KnowledgeBaseStateType]
 
 
+class BedrockDataAutomationConfigurationTypeDef(TypedDict):
+    parsingModality: NotRequired[Literal["MULTIMODAL"]]
+
+
 class BedrockEmbeddingModelConfigurationTypeDef(TypedDict):
     dimensions: NotRequired[int]
     embeddingDataType: NotRequired[EmbeddingDataTypeType]
@@ -638,6 +687,11 @@ class CreatePromptVersionRequestRequestTypeDef(TypedDict):
     tags: NotRequired[Mapping[str, str]]
 
 
+class CuratedQueryTypeDef(TypedDict):
+    naturalLanguage: str
+    sql: str
+
+
 CustomDocumentIdentifierTypeDef = TypedDict(
     "CustomDocumentIdentifierTypeDef",
     {
@@ -1044,6 +1098,10 @@ class TextContentDocTypeDef(TypedDict):
     data: str
 
 
+class KendraKnowledgeBaseConfigurationTypeDef(TypedDict):
+    kendraIndexArn: str
+
+
 class KnowledgeBaseSummaryTypeDef(TypedDict):
     knowledgeBaseId: str
     name: str
@@ -1231,6 +1289,12 @@ class PromptModelInferenceConfigurationTypeDef(TypedDict):
     topP: NotRequired[float]
 
 
+class QueryGenerationColumnTypeDef(TypedDict):
+    description: NotRequired[str]
+    inclusion: NotRequired[IncludeExcludeType]
+    name: NotRequired[str]
+
+
 class RdsFieldMappingTypeDef(TypedDict):
     metadataField: str
     primaryKeyField: str
@@ -1244,6 +1308,37 @@ class RedisEnterpriseCloudFieldMappingTypeDef(TypedDict):
     vectorField: str
 
 
+RedshiftProvisionedAuthConfigurationTypeDef = TypedDict(
+    "RedshiftProvisionedAuthConfigurationTypeDef",
+    {
+        "type": RedshiftProvisionedAuthTypeType,
+        "databaseUser": NotRequired[str],
+        "usernamePasswordSecretArn": NotRequired[str],
+    },
+)
+
+
+class RedshiftQueryEngineAwsDataCatalogStorageConfigurationOutputTypeDef(TypedDict):
+    tableNames: List[str]
+
+
+class RedshiftQueryEngineAwsDataCatalogStorageConfigurationTypeDef(TypedDict):
+    tableNames: Sequence[str]
+
+
+class RedshiftQueryEngineRedshiftStorageConfigurationTypeDef(TypedDict):
+    databaseName: str
+
+
+RedshiftServerlessAuthConfigurationTypeDef = TypedDict(
+    "RedshiftServerlessAuthConfigurationTypeDef",
+    {
+        "type": RedshiftServerlessAuthTypeType,
+        "usernamePasswordSecretArn": NotRequired[str],
+    },
+)
+
+
 class RetrievalFlowNodeS3ConfigurationTypeDef(TypedDict):
     bucketName: str
 
@@ -1570,6 +1665,7 @@ class EmbeddingModelConfigurationTypeDef(TypedDict):
 
 class BedrockFoundationModelConfigurationTypeDef(TypedDict):
     modelArn: str
+    parsingModality: NotRequired[Literal["MULTIMODAL"]]
     parsingPrompt: NotRequired[ParsingPromptTypeDef]
 
 
@@ -1706,6 +1802,15 @@ class S3ContentTypeDef(TypedDict):
     s3Location: S3LocationTypeDef
 
 
+SupplementalDataStorageLocationTypeDef = TypedDict(
+    "SupplementalDataStorageLocationTypeDef",
+    {
+        "type": Literal["S3"],
+        "s3Location": NotRequired[S3LocationTypeDef],
+    },
+)
+
+
 class FlowConnectionConfigurationTypeDef(TypedDict):
     conditional: NotRequired[FlowConditionalConnectionConfigurationTypeDef]
     data: NotRequired[FlowDataConnectionConfigurationTypeDef]
@@ -1976,6 +2081,20 @@ PromptModelInferenceConfigurationUnionTypeDef = Union[
 ]
 
 
+class QueryGenerationTableOutputTypeDef(TypedDict):
+    name: str
+    columns: NotRequired[List[QueryGenerationColumnTypeDef]]
+    description: NotRequired[str]
+    inclusion: NotRequired[IncludeExcludeType]
+
+
+class QueryGenerationTableTypeDef(TypedDict):
+    name: str
+    columns: NotRequired[Sequence[QueryGenerationColumnTypeDef]]
+    description: NotRequired[str]
+    inclusion: NotRequired[IncludeExcludeType]
+
+
 class RdsConfigurationTypeDef(TypedDict):
     credentialsSecretArn: str
     databaseName: str
@@ -1991,6 +2110,34 @@ class RedisEnterpriseCloudConfigurationTypeDef(TypedDict):
     vectorIndexName: str
 
 
+class RedshiftProvisionedConfigurationTypeDef(TypedDict):
+    authConfiguration: RedshiftProvisionedAuthConfigurationTypeDef
+    clusterIdentifier: str
+
+
+RedshiftQueryEngineAwsDataCatalogStorageConfigurationUnionTypeDef = Union[
+    RedshiftQueryEngineAwsDataCatalogStorageConfigurationTypeDef,
+    RedshiftQueryEngineAwsDataCatalogStorageConfigurationOutputTypeDef,
+]
+RedshiftQueryEngineStorageConfigurationOutputTypeDef = TypedDict(
+    "RedshiftQueryEngineStorageConfigurationOutputTypeDef",
+    {
+        "type": RedshiftQueryEngineStorageTypeType,
+        "awsDataCatalogConfiguration": NotRequired[
+            RedshiftQueryEngineAwsDataCatalogStorageConfigurationOutputTypeDef
+        ],
+        "redshiftConfiguration": NotRequired[
+            RedshiftQueryEngineRedshiftStorageConfigurationTypeDef
+        ],
+    },
+)
+
+
+class RedshiftServerlessConfigurationTypeDef(TypedDict):
+    authConfiguration: RedshiftServerlessAuthConfigurationTypeDef
+    workgroupArn: str
+
+
 class RetrievalFlowNodeServiceConfigurationTypeDef(TypedDict):
     s3: NotRequired[RetrievalFlowNodeS3ConfigurationTypeDef]
 
@@ -2114,13 +2261,9 @@ class ListAgentVersionsResponseTypeDef(TypedDict):
     nextToken: NotRequired[str]
 
 
-class VectorKnowledgeBaseConfigurationTypeDef(TypedDict):
-    embeddingModelArn: str
-    embeddingModelConfiguration: NotRequired[EmbeddingModelConfigurationTypeDef]
-
-
 class ParsingConfigurationTypeDef(TypedDict):
-    parsingStrategy: Literal["BEDROCK_FOUNDATION_MODEL"]
+    parsingStrategy: ParsingStrategyType
+    bedrockDataAutomationConfiguration: NotRequired[BedrockDataAutomationConfigurationTypeDef]
     bedrockFoundationModelConfiguration: NotRequired[BedrockFoundationModelConfigurationTypeDef]
 
 
@@ -2169,6 +2312,14 @@ class KnowledgeBaseDocumentDetailTypeDef(TypedDict):
     updatedAt: NotRequired[datetime]
 
 
+class SupplementalDataStorageConfigurationOutputTypeDef(TypedDict):
+    storageLocations: List[SupplementalDataStorageLocationTypeDef]
+
+
+class SupplementalDataStorageConfigurationTypeDef(TypedDict):
+    storageLocations: Sequence[SupplementalDataStorageLocationTypeDef]
+
+
 FlowConnectionTypeDef = TypedDict(
     "FlowConnectionTypeDef",
     {
@@ -2270,6 +2421,14 @@ class PromptInferenceConfigurationTypeDef(TypedDict):
     text: NotRequired[PromptModelInferenceConfigurationUnionTypeDef]
 
 
+class QueryGenerationContextOutputTypeDef(TypedDict):
+    curatedQueries: NotRequired[List[CuratedQueryTypeDef]]
+    tables: NotRequired[List[QueryGenerationTableOutputTypeDef]]
+
+
+QueryGenerationTableUnionTypeDef = Union[
+    QueryGenerationTableTypeDef, QueryGenerationTableOutputTypeDef
+]
 StorageConfigurationTypeDef = TypedDict(
     "StorageConfigurationTypeDef",
     {
@@ -2281,6 +2440,26 @@ StorageConfigurationTypeDef = TypedDict(
         "redisEnterpriseCloudConfiguration": NotRequired[RedisEnterpriseCloudConfigurationTypeDef],
     },
 )
+RedshiftQueryEngineStorageConfigurationTypeDef = TypedDict(
+    "RedshiftQueryEngineStorageConfigurationTypeDef",
+    {
+        "type": RedshiftQueryEngineStorageTypeType,
+        "awsDataCatalogConfiguration": NotRequired[
+            RedshiftQueryEngineAwsDataCatalogStorageConfigurationUnionTypeDef
+        ],
+        "redshiftConfiguration": NotRequired[
+            RedshiftQueryEngineRedshiftStorageConfigurationTypeDef
+        ],
+    },
+)
+RedshiftQueryEngineConfigurationTypeDef = TypedDict(
+    "RedshiftQueryEngineConfigurationTypeDef",
+    {
+        "type": RedshiftQueryEngineTypeType,
+        "provisionedConfiguration": NotRequired[RedshiftProvisionedConfigurationTypeDef],
+        "serverlessConfiguration": NotRequired[RedshiftServerlessConfigurationTypeDef],
+    },
+)
 
 
 class RetrievalFlowNodeConfigurationTypeDef(TypedDict):
@@ -2334,15 +2513,6 @@ class UpdateAgentAliasResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
-KnowledgeBaseConfigurationTypeDef = TypedDict(
-    "KnowledgeBaseConfigurationTypeDef",
-    {
-        "type": Literal["VECTOR"],
-        "vectorKnowledgeBaseConfiguration": NotRequired[VectorKnowledgeBaseConfigurationTypeDef],
-    },
-)
-
-
 class CustomContentTypeDef(TypedDict):
     customDocumentIdentifier: CustomDocumentIdentifierTypeDef
     sourceType: CustomSourceTypeType
@@ -2371,6 +2541,19 @@ class ListKnowledgeBaseDocumentsResponseTypeDef(TypedDict):
     nextToken: NotRequired[str]
 
 
+class VectorKnowledgeBaseConfigurationOutputTypeDef(TypedDict):
+    embeddingModelArn: str
+    embeddingModelConfiguration: NotRequired[EmbeddingModelConfigurationTypeDef]
+    supplementalDataStorageConfiguration: NotRequired[
+        SupplementalDataStorageConfigurationOutputTypeDef
+    ]
+
+
+SupplementalDataStorageConfigurationUnionTypeDef = Union[
+    SupplementalDataStorageConfigurationTypeDef, SupplementalDataStorageConfigurationOutputTypeDef
+]
+
+
 class ValidateFlowDefinitionResponseTypeDef(TypedDict):
     validations: List[FlowValidationTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -2476,6 +2659,22 @@ PromptInferenceConfigurationUnionTypeDef = Union[
 ]
 
 
+class QueryGenerationConfigurationOutputTypeDef(TypedDict):
+    executionTimeoutSeconds: NotRequired[int]
+    generationContext: NotRequired[QueryGenerationContextOutputTypeDef]
+
+
+class QueryGenerationContextTypeDef(TypedDict):
+    curatedQueries: NotRequired[Sequence[CuratedQueryTypeDef]]
+    tables: NotRequired[Sequence[QueryGenerationTableUnionTypeDef]]
+
+
+RedshiftQueryEngineStorageConfigurationUnionTypeDef = Union[
+    RedshiftQueryEngineStorageConfigurationTypeDef,
+    RedshiftQueryEngineStorageConfigurationOutputTypeDef,
+]
+
+
 class WebDataSourceConfigurationOutputTypeDef(TypedDict):
     sourceConfiguration: WebSourceConfigurationOutputTypeDef
     crawlerConfiguration: NotRequired[WebCrawlerConfigurationOutputTypeDef]
@@ -2503,45 +2702,20 @@ class CustomTransformationConfigurationTypeDef(TypedDict):
     transformations: Sequence[TransformationTypeDef]
 
 
-class CreateKnowledgeBaseRequestRequestTypeDef(TypedDict):
-    knowledgeBaseConfiguration: KnowledgeBaseConfigurationTypeDef
-    name: str
-    roleArn: str
-    storageConfiguration: StorageConfigurationTypeDef
-    clientToken: NotRequired[str]
-    description: NotRequired[str]
-    tags: NotRequired[Mapping[str, str]]
-
-
-class KnowledgeBaseTypeDef(TypedDict):
-    createdAt: datetime
-    knowledgeBaseArn: str
-    knowledgeBaseConfiguration: KnowledgeBaseConfigurationTypeDef
-    knowledgeBaseId: str
-    name: str
-    roleArn: str
-    status: KnowledgeBaseStatusType
-    storageConfiguration: StorageConfigurationTypeDef
-    updatedAt: datetime
-    description: NotRequired[str]
-    failureReasons: NotRequired[List[str]]
-
-
-class UpdateKnowledgeBaseRequestRequestTypeDef(TypedDict):
-    knowledgeBaseConfiguration: KnowledgeBaseConfigurationTypeDef
-    knowledgeBaseId: str
-    name: str
-    roleArn: str
-    storageConfiguration: StorageConfigurationTypeDef
-    description: NotRequired[str]
-
-
 class DocumentContentTypeDef(TypedDict):
     dataSourceType: ContentDataSourceTypeType
     custom: NotRequired[CustomContentTypeDef]
     s3: NotRequired[S3ContentTypeDef]
 
 
+class VectorKnowledgeBaseConfigurationTypeDef(TypedDict):
+    embeddingModelArn: str
+    embeddingModelConfiguration: NotRequired[EmbeddingModelConfigurationTypeDef]
+    supplementalDataStorageConfiguration: NotRequired[
+        SupplementalDataStorageConfigurationUnionTypeDef
+    ]
+
+
 class CreateAgentActionGroupResponseTypeDef(TypedDict):
     agentActionGroup: AgentActionGroupTypeDef
     ResponseMetadata: ResponseMetadataTypeDef
@@ -2635,6 +2809,17 @@ CrawlFilterConfigurationTypeDef = TypedDict(
         "patternObjectFilter": NotRequired[PatternObjectFilterConfigurationUnionTypeDef],
     },
 )
+
+
+class RedshiftConfigurationOutputTypeDef(TypedDict):
+    queryEngineConfiguration: RedshiftQueryEngineConfigurationTypeDef
+    storageConfigurations: List[RedshiftQueryEngineStorageConfigurationOutputTypeDef]
+    queryGenerationConfiguration: NotRequired[QueryGenerationConfigurationOutputTypeDef]
+
+
+QueryGenerationContextUnionTypeDef = Union[
+    QueryGenerationContextTypeDef, QueryGenerationContextOutputTypeDef
+]
 WebSourceConfigurationUnionTypeDef = Union[
     WebSourceConfigurationTypeDef, WebSourceConfigurationOutputTypeDef
 ]
@@ -2662,26 +2847,16 @@ CustomTransformationConfigurationUnionTypeDef = Union[
 ]
 
 
-class CreateKnowledgeBaseResponseTypeDef(TypedDict):
-    knowledgeBase: KnowledgeBaseTypeDef
-    ResponseMetadata: ResponseMetadataTypeDef
-
-
-class GetKnowledgeBaseResponseTypeDef(TypedDict):
-    knowledgeBase: KnowledgeBaseTypeDef
-    ResponseMetadata: ResponseMetadataTypeDef
-
-
-class UpdateKnowledgeBaseResponseTypeDef(TypedDict):
-    knowledgeBase: KnowledgeBaseTypeDef
-    ResponseMetadata: ResponseMetadataTypeDef
-
-
 class KnowledgeBaseDocumentTypeDef(TypedDict):
     content: DocumentContentTypeDef
     metadata: NotRequired[DocumentMetadataTypeDef]
 
 
+VectorKnowledgeBaseConfigurationUnionTypeDef = Union[
+    VectorKnowledgeBaseConfigurationTypeDef, VectorKnowledgeBaseConfigurationOutputTypeDef
+]
+
+
 class CreateAgentRequestRequestTypeDef(TypedDict):
     agentName: str
     agentCollaboration: NotRequired[AgentCollaborationType]
@@ -2731,6 +2906,18 @@ DataSourceConfigurationOutputTypeDef = TypedDict(
 CrawlFilterConfigurationUnionTypeDef = Union[
     CrawlFilterConfigurationTypeDef, CrawlFilterConfigurationOutputTypeDef
 ]
+SqlKnowledgeBaseConfigurationOutputTypeDef = TypedDict(
+    "SqlKnowledgeBaseConfigurationOutputTypeDef",
+    {
+        "type": Literal["REDSHIFT"],
+        "redshiftConfiguration": NotRequired[RedshiftConfigurationOutputTypeDef],
+    },
+)
+
+
+class QueryGenerationConfigurationTypeDef(TypedDict):
+    executionTimeoutSeconds: NotRequired[int]
+    generationContext: NotRequired[QueryGenerationContextUnionTypeDef]
 
 
 class WebDataSourceConfigurationTypeDef(TypedDict):
@@ -2786,6 +2973,20 @@ class SharePointCrawlerConfigurationTypeDef(TypedDict):
     filterConfiguration: NotRequired[CrawlFilterConfigurationUnionTypeDef]
 
 
+KnowledgeBaseConfigurationOutputTypeDef = TypedDict(
+    "KnowledgeBaseConfigurationOutputTypeDef",
+    {
+        "type": KnowledgeBaseTypeType,
+        "kendraKnowledgeBaseConfiguration": NotRequired[KendraKnowledgeBaseConfigurationTypeDef],
+        "sqlKnowledgeBaseConfiguration": NotRequired[SqlKnowledgeBaseConfigurationOutputTypeDef],
+        "vectorKnowledgeBaseConfiguration": NotRequired[
+            VectorKnowledgeBaseConfigurationOutputTypeDef
+        ],
+    },
+)
+QueryGenerationConfigurationUnionTypeDef = Union[
+    QueryGenerationConfigurationTypeDef, QueryGenerationConfigurationOutputTypeDef
+]
 WebDataSourceConfigurationUnionTypeDef = Union[
     WebDataSourceConfigurationTypeDef, WebDataSourceConfigurationOutputTypeDef
 ]
@@ -2841,6 +3042,26 @@ SharePointCrawlerConfigurationUnionTypeDef = Union[
 ]
 
 
+class KnowledgeBaseTypeDef(TypedDict):
+    createdAt: datetime
+    knowledgeBaseArn: str
+    knowledgeBaseConfiguration: KnowledgeBaseConfigurationOutputTypeDef
+    knowledgeBaseId: str
+    name: str
+    roleArn: str
+    status: KnowledgeBaseStatusType
+    updatedAt: datetime
+    description: NotRequired[str]
+    failureReasons: NotRequired[List[str]]
+    storageConfiguration: NotRequired[StorageConfigurationTypeDef]
+
+
+class RedshiftConfigurationTypeDef(TypedDict):
+    queryEngineConfiguration: RedshiftQueryEngineConfigurationTypeDef
+    storageConfigurations: Sequence[RedshiftQueryEngineStorageConfigurationUnionTypeDef]
+    queryGenerationConfiguration: NotRequired[QueryGenerationConfigurationUnionTypeDef]
+
+
 class PromptFlowNodeSourceConfigurationOutputTypeDef(TypedDict):
     inline: NotRequired[PromptFlowNodeInlineConfigurationOutputTypeDef]
     resource: NotRequired[PromptFlowNodeResourceConfigurationTypeDef]
@@ -2928,6 +3149,26 @@ class SharePointDataSourceConfigurationTypeDef(TypedDict):
     crawlerConfiguration: NotRequired[SharePointCrawlerConfigurationUnionTypeDef]
 
 
+class CreateKnowledgeBaseResponseTypeDef(TypedDict):
+    knowledgeBase: KnowledgeBaseTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class GetKnowledgeBaseResponseTypeDef(TypedDict):
+    knowledgeBase: KnowledgeBaseTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class UpdateKnowledgeBaseResponseTypeDef(TypedDict):
+    knowledgeBase: KnowledgeBaseTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+RedshiftConfigurationUnionTypeDef = Union[
+    RedshiftConfigurationTypeDef, RedshiftConfigurationOutputTypeDef
+]
+
+
 class PromptFlowNodeConfigurationOutputTypeDef(TypedDict):
     sourceConfiguration: PromptFlowNodeSourceConfigurationOutputTypeDef
     guardrailConfiguration: NotRequired[GuardrailConfigurationTypeDef]
@@ -2949,6 +3190,13 @@ SalesforceDataSourceConfigurationUnionTypeDef = Union[
 SharePointDataSourceConfigurationUnionTypeDef = Union[
     SharePointDataSourceConfigurationTypeDef, SharePointDataSourceConfigurationOutputTypeDef
 ]
+SqlKnowledgeBaseConfigurationTypeDef = TypedDict(
+    "SqlKnowledgeBaseConfigurationTypeDef",
+    {
+        "type": Literal["REDSHIFT"],
+        "redshiftConfiguration": NotRequired[RedshiftConfigurationUnionTypeDef],
+    },
+)
 FlowNodeConfigurationOutputTypeDef = TypedDict(
     "FlowNodeConfigurationOutputTypeDef",
     {
@@ -2980,6 +3228,9 @@ DataSourceConfigurationTypeDef = TypedDict(
         "webConfiguration": NotRequired[WebDataSourceConfigurationUnionTypeDef],
     },
 )
+SqlKnowledgeBaseConfigurationUnionTypeDef = Union[
+    SqlKnowledgeBaseConfigurationTypeDef, SqlKnowledgeBaseConfigurationOutputTypeDef
+]
 FlowNodeExtraOutputTypeDef = TypedDict(
     "FlowNodeExtraOutputTypeDef",
     {
@@ -3019,6 +3270,19 @@ class UpdateDataSourceRequestRequestTypeDef(TypedDict):
     vectorIngestionConfiguration: NotRequired[VectorIngestionConfigurationTypeDef]
 
 
+KnowledgeBaseConfigurationTypeDef = TypedDict(
+    "KnowledgeBaseConfigurationTypeDef",
+    {
+        "type": KnowledgeBaseTypeType,
+        "kendraKnowledgeBaseConfiguration": NotRequired[KendraKnowledgeBaseConfigurationTypeDef],
+        "sqlKnowledgeBaseConfiguration": NotRequired[SqlKnowledgeBaseConfigurationUnionTypeDef],
+        "vectorKnowledgeBaseConfiguration": NotRequired[
+            VectorKnowledgeBaseConfigurationUnionTypeDef
+        ],
+    },
+)
+
+
 class FlowDefinitionOutputTypeDef(TypedDict):
     connections: NotRequired[List[FlowConnectionTypeDef]]
     nodes: NotRequired[List[FlowNodeExtraOutputTypeDef]]
@@ -3027,6 +3291,27 @@ class FlowDefinitionOutputTypeDef(TypedDict):
 PromptTemplateConfigurationUnionTypeDef = Union[
     PromptTemplateConfigurationTypeDef, PromptTemplateConfigurationOutputTypeDef
 ]
+
+
+class CreateKnowledgeBaseRequestRequestTypeDef(TypedDict):
+    knowledgeBaseConfiguration: KnowledgeBaseConfigurationTypeDef
+    name: str
+    roleArn: str
+    clientToken: NotRequired[str]
+    description: NotRequired[str]
+    storageConfiguration: NotRequired[StorageConfigurationTypeDef]
+    tags: NotRequired[Mapping[str, str]]
+
+
+class UpdateKnowledgeBaseRequestRequestTypeDef(TypedDict):
+    knowledgeBaseConfiguration: KnowledgeBaseConfigurationTypeDef
+    knowledgeBaseId: str
+    name: str
+    roleArn: str
+    description: NotRequired[str]
+    storageConfiguration: NotRequired[StorageConfigurationTypeDef]
+
+
 CreateFlowResponseTypeDef = TypedDict(
     "CreateFlowResponseTypeDef",
     {