types-boto3-full 1.35.72__py3-none-any.whl → 1.35.75__py3-none-any.whl

types_boto3_bedrock_agent/type_defs.py

@@ -24,6 +24,7 @@ from .literals import (
     ActionGroupSignatureType,
     ActionGroupStateType,
     AgentAliasStatusType,
+    AgentCollaborationType,
     AgentStatusType,
     ChunkingStrategyType,
     ConfluenceAuthTypeType,
@@ -42,18 +43,26 @@ 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,
     TypeType,
@@ -75,6 +84,9 @@ __all__ = (
     "AgentAliasRoutingConfigurationListItemTypeDef",
     "AgentAliasSummaryTypeDef",
     "AgentAliasTypeDef",
+    "AgentCollaboratorSummaryTypeDef",
+    "AgentCollaboratorTypeDef",
+    "AgentDescriptorTypeDef",
     "AgentFlowNodeConfigurationTypeDef",
     "AgentKnowledgeBaseSummaryTypeDef",
     "AgentKnowledgeBaseTypeDef",
@@ -82,8 +94,11 @@ __all__ = (
     "AgentTypeDef",
     "AgentVersionSummaryTypeDef",
     "AgentVersionTypeDef",
+    "AssociateAgentCollaboratorRequestRequestTypeDef",
+    "AssociateAgentCollaboratorResponseTypeDef",
     "AssociateAgentKnowledgeBaseRequestRequestTypeDef",
     "AssociateAgentKnowledgeBaseResponseTypeDef",
+    "BedrockDataAutomationConfigurationTypeDef",
     "BedrockEmbeddingModelConfigurationTypeDef",
     "BedrockFoundationModelConfigurationTypeDef",
     "BlobTypeDef",
@@ -128,6 +143,7 @@ __all__ = (
     "CreatePromptResponseTypeDef",
     "CreatePromptVersionRequestRequestTypeDef",
     "CreatePromptVersionResponseTypeDef",
+    "CuratedQueryTypeDef",
     "CustomContentTypeDef",
     "CustomDocumentIdentifierTypeDef",
     "CustomOrchestrationTypeDef",
@@ -161,6 +177,7 @@ __all__ = (
     "DeleteKnowledgeBaseResponseTypeDef",
     "DeletePromptRequestRequestTypeDef",
     "DeletePromptResponseTypeDef",
+    "DisassociateAgentCollaboratorRequestRequestTypeDef",
     "DisassociateAgentKnowledgeBaseRequestRequestTypeDef",
     "DocumentContentTypeDef",
     "DocumentIdentifierTypeDef",
@@ -199,6 +216,8 @@ __all__ = (
     "GetAgentActionGroupResponseTypeDef",
     "GetAgentAliasRequestRequestTypeDef",
     "GetAgentAliasResponseTypeDef",
+    "GetAgentCollaboratorRequestRequestTypeDef",
+    "GetAgentCollaboratorResponseTypeDef",
     "GetAgentKnowledgeBaseRequestRequestTypeDef",
     "GetAgentKnowledgeBaseResponseTypeDef",
     "GetAgentRequestRequestTypeDef",
@@ -239,6 +258,8 @@ __all__ = (
     "IngestionJobTypeDef",
     "InlineContentTypeDef",
     "IntermediateStorageTypeDef",
+    "KendraKnowledgeBaseConfigurationTypeDef",
+    "KnowledgeBaseConfigurationOutputTypeDef",
     "KnowledgeBaseConfigurationTypeDef",
     "KnowledgeBaseDocumentDetailTypeDef",
     "KnowledgeBaseDocumentTypeDef",
@@ -253,6 +274,9 @@ __all__ = (
     "ListAgentAliasesRequestListAgentAliasesPaginateTypeDef",
     "ListAgentAliasesRequestRequestTypeDef",
     "ListAgentAliasesResponseTypeDef",
+    "ListAgentCollaboratorsRequestListAgentCollaboratorsPaginateTypeDef",
+    "ListAgentCollaboratorsRequestRequestTypeDef",
+    "ListAgentCollaboratorsResponseTypeDef",
     "ListAgentKnowledgeBasesRequestListAgentKnowledgeBasesPaginateTypeDef",
     "ListAgentKnowledgeBasesRequestRequestTypeDef",
     "ListAgentKnowledgeBasesResponseTypeDef",
@@ -358,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",
@@ -392,6 +441,9 @@ __all__ = (
     "SharePointSourceConfigurationTypeDef",
     "SharePointSourceConfigurationUnionTypeDef",
     "SpecificToolChoiceTypeDef",
+    "SqlKnowledgeBaseConfigurationOutputTypeDef",
+    "SqlKnowledgeBaseConfigurationTypeDef",
+    "SqlKnowledgeBaseConfigurationUnionTypeDef",
     "StartIngestionJobRequestRequestTypeDef",
     "StartIngestionJobResponseTypeDef",
     "StopIngestionJobRequestRequestTypeDef",
@@ -400,6 +452,10 @@ __all__ = (
     "StorageFlowNodeConfigurationTypeDef",
     "StorageFlowNodeS3ConfigurationTypeDef",
     "StorageFlowNodeServiceConfigurationTypeDef",
+    "SupplementalDataStorageConfigurationOutputTypeDef",
+    "SupplementalDataStorageConfigurationTypeDef",
+    "SupplementalDataStorageConfigurationUnionTypeDef",
+    "SupplementalDataStorageLocationTypeDef",
     "SystemContentBlockTypeDef",
     "TagResourceRequestRequestTypeDef",
     "TextContentDocTypeDef",
@@ -437,6 +493,8 @@ __all__ = (
     "UpdateAgentActionGroupResponseTypeDef",
     "UpdateAgentAliasRequestRequestTypeDef",
     "UpdateAgentAliasResponseTypeDef",
+    "UpdateAgentCollaboratorRequestRequestTypeDef",
+    "UpdateAgentCollaboratorResponseTypeDef",
     "UpdateAgentKnowledgeBaseRequestRequestTypeDef",
     "UpdateAgentKnowledgeBaseResponseTypeDef",
     "UpdateAgentRequestRequestTypeDef",
@@ -458,7 +516,9 @@ __all__ = (
     "ValidateFlowDefinitionResponseTypeDef",
     "VectorIngestionConfigurationOutputTypeDef",
     "VectorIngestionConfigurationTypeDef",
+    "VectorKnowledgeBaseConfigurationOutputTypeDef",
     "VectorKnowledgeBaseConfigurationTypeDef",
+    "VectorKnowledgeBaseConfigurationUnionTypeDef",
     "WebCrawlerConfigurationOutputTypeDef",
     "WebCrawlerConfigurationTypeDef",
     "WebCrawlerConfigurationUnionTypeDef",
@@ -499,6 +559,10 @@ class AgentAliasRoutingConfigurationListItemTypeDef(TypedDict):
     provisionedThroughput: NotRequired[str]
 
 
+class AgentDescriptorTypeDef(TypedDict):
+    aliasArn: NotRequired[str]
+
+
 class AgentFlowNodeConfigurationTypeDef(TypedDict):
     agentAliasArn: str
 
@@ -530,6 +594,14 @@ class MemoryConfigurationOutputTypeDef(TypedDict):
     storageDays: NotRequired[int]
 
 
+class ResponseMetadataTypeDef(TypedDict):
+    RequestId: str
+    HTTPStatusCode: int
+    HTTPHeaders: Dict[str, str]
+    RetryAttempts: int
+    HostId: NotRequired[str]
+
+
 class AssociateAgentKnowledgeBaseRequestRequestTypeDef(TypedDict):
     agentId: str
     agentVersion: str
@@ -538,12 +610,8 @@ class AssociateAgentKnowledgeBaseRequestRequestTypeDef(TypedDict):
     knowledgeBaseState: NotRequired[KnowledgeBaseStateType]
 
 
-class ResponseMetadataTypeDef(TypedDict):
-    RequestId: str
-    HTTPStatusCode: int
-    HTTPHeaders: Dict[str, str]
-    RetryAttempts: int
-    HostId: NotRequired[str]
+class BedrockDataAutomationConfigurationTypeDef(TypedDict):
+    parsingModality: NotRequired[Literal["MULTIMODAL"]]
 
 
 class BedrockEmbeddingModelConfigurationTypeDef(TypedDict):
@@ -619,6 +687,11 @@ class CreatePromptVersionRequestRequestTypeDef(TypedDict):
     tags: NotRequired[Mapping[str, str]]
 
 
+class CuratedQueryTypeDef(TypedDict):
+    naturalLanguage: str
+    sql: str
+
+
 CustomDocumentIdentifierTypeDef = TypedDict(
     "CustomDocumentIdentifierTypeDef",
     {
@@ -712,6 +785,12 @@ class DeletePromptRequestRequestTypeDef(TypedDict):
     promptVersion: NotRequired[str]
 
 
+class DisassociateAgentCollaboratorRequestRequestTypeDef(TypedDict):
+    agentId: str
+    agentVersion: str
+    collaboratorId: str
+
+
 class DisassociateAgentKnowledgeBaseRequestRequestTypeDef(TypedDict):
     agentId: str
     agentVersion: str
@@ -915,6 +994,12 @@ class GetAgentAliasRequestRequestTypeDef(TypedDict):
     agentId: str
 
 
+class GetAgentCollaboratorRequestRequestTypeDef(TypedDict):
+    agentId: str
+    agentVersion: str
+    collaboratorId: str
+
+
 class GetAgentKnowledgeBaseRequestRequestTypeDef(TypedDict):
     agentId: str
     agentVersion: str
@@ -1013,6 +1098,10 @@ class TextContentDocTypeDef(TypedDict):
     data: str
 
 
+class KendraKnowledgeBaseConfigurationTypeDef(TypedDict):
+    kendraIndexArn: str
+
+
 class KnowledgeBaseSummaryTypeDef(TypedDict):
     knowledgeBaseId: str
     name: str
@@ -1040,6 +1129,13 @@ class ListAgentAliasesRequestRequestTypeDef(TypedDict):
     nextToken: NotRequired[str]
 
 
+class ListAgentCollaboratorsRequestRequestTypeDef(TypedDict):
+    agentId: str
+    agentVersion: str
+    maxResults: NotRequired[int]
+    nextToken: NotRequired[str]
+
+
 class ListAgentKnowledgeBasesRequestRequestTypeDef(TypedDict):
     agentId: str
     agentVersion: str
@@ -1193,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
@@ -1206,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
 
@@ -1337,6 +1470,51 @@ class UpdateAgentAliasRequestRequestTypeDef(TypedDict):
     routingConfiguration: NotRequired[Sequence[AgentAliasRoutingConfigurationListItemTypeDef]]
 
 
+class AgentCollaboratorSummaryTypeDef(TypedDict):
+    agentDescriptor: AgentDescriptorTypeDef
+    agentId: str
+    agentVersion: str
+    collaborationInstruction: str
+    collaboratorId: str
+    collaboratorName: str
+    createdAt: datetime
+    lastUpdatedAt: datetime
+    relayConversationHistory: RelayConversationHistoryType
+
+
+class AgentCollaboratorTypeDef(TypedDict):
+    agentDescriptor: AgentDescriptorTypeDef
+    agentId: str
+    agentVersion: str
+    collaborationInstruction: str
+    collaboratorId: str
+    collaboratorName: str
+    createdAt: datetime
+    lastUpdatedAt: datetime
+    clientToken: NotRequired[str]
+    relayConversationHistory: NotRequired[RelayConversationHistoryType]
+
+
+class AssociateAgentCollaboratorRequestRequestTypeDef(TypedDict):
+    agentDescriptor: AgentDescriptorTypeDef
+    agentId: str
+    agentVersion: str
+    collaborationInstruction: str
+    collaboratorName: str
+    clientToken: NotRequired[str]
+    relayConversationHistory: NotRequired[RelayConversationHistoryType]
+
+
+class UpdateAgentCollaboratorRequestRequestTypeDef(TypedDict):
+    agentDescriptor: AgentDescriptorTypeDef
+    agentId: str
+    agentVersion: str
+    collaborationInstruction: str
+    collaboratorId: str
+    collaboratorName: str
+    relayConversationHistory: NotRequired[RelayConversationHistoryType]
+
+
 class AgentSummaryTypeDef(TypedDict):
     agentId: str
     agentName: str
@@ -1487,6 +1665,7 @@ class EmbeddingModelConfigurationTypeDef(TypedDict):
 
 class BedrockFoundationModelConfigurationTypeDef(TypedDict):
     modelArn: str
+    parsingModality: NotRequired[Literal["MULTIMODAL"]]
     parsingPrompt: NotRequired[ParsingPromptTypeDef]
 
 
@@ -1623,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]
@@ -1712,6 +1900,7 @@ class HierarchicalChunkingConfigurationTypeDef(TypedDict):
 
 class PromptConfigurationOutputTypeDef(TypedDict):
     basePromptTemplate: NotRequired[str]
+    foundationModel: NotRequired[str]
     inferenceConfiguration: NotRequired[InferenceConfigurationOutputTypeDef]
     parserMode: NotRequired[CreationModeType]
     promptCreationMode: NotRequired[CreationModeType]
@@ -1773,6 +1962,12 @@ class ListAgentAliasesRequestListAgentAliasesPaginateTypeDef(TypedDict):
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
 
+class ListAgentCollaboratorsRequestListAgentCollaboratorsPaginateTypeDef(TypedDict):
+    agentId: str
+    agentVersion: str
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+
 class ListAgentKnowledgeBasesRequestListAgentKnowledgeBasesPaginateTypeDef(TypedDict):
     agentId: str
     agentVersion: str
@@ -1886,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
@@ -1901,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]
 
@@ -1991,6 +2228,27 @@ class ListAgentAliasesResponseTypeDef(TypedDict):
     nextToken: NotRequired[str]
 
 
+class ListAgentCollaboratorsResponseTypeDef(TypedDict):
+    agentCollaboratorSummaries: List[AgentCollaboratorSummaryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    nextToken: NotRequired[str]
+
+
+class AssociateAgentCollaboratorResponseTypeDef(TypedDict):
+    agentCollaborator: AgentCollaboratorTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class GetAgentCollaboratorResponseTypeDef(TypedDict):
+    agentCollaborator: AgentCollaboratorTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class UpdateAgentCollaboratorResponseTypeDef(TypedDict):
+    agentCollaborator: AgentCollaboratorTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class ListAgentsResponseTypeDef(TypedDict):
     agentSummaries: List[AgentSummaryTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -2003,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]
 
 
@@ -2058,6 +2312,14 @@ class KnowledgeBaseDocumentDetailTypeDef(TypedDict):
     updatedAt: NotRequired[datetime]
 
 
+class SupplementalDataStorageConfigurationOutputTypeDef(TypedDict):
+    storageLocations: List[SupplementalDataStorageLocationTypeDef]
+
+
+class SupplementalDataStorageConfigurationTypeDef(TypedDict):
+    storageLocations: Sequence[SupplementalDataStorageLocationTypeDef]
+
+
 FlowConnectionTypeDef = TypedDict(
     "FlowConnectionTypeDef",
     {
@@ -2105,6 +2367,7 @@ class PromptOverrideConfigurationOutputTypeDef(TypedDict):
 
 class PromptConfigurationTypeDef(TypedDict):
     basePromptTemplate: NotRequired[str]
+    foundationModel: NotRequired[str]
     inferenceConfiguration: NotRequired[InferenceConfigurationUnionTypeDef]
     parserMode: NotRequired[CreationModeType]
     promptCreationMode: NotRequired[CreationModeType]
@@ -2158,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",
     {
@@ -2169,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):
@@ -2222,15 +2513,6 @@ class UpdateAgentAliasResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
-KnowledgeBaseConfigurationTypeDef = TypedDict(
-    "KnowledgeBaseConfigurationTypeDef",
-    {
-        "type": Literal["VECTOR"],
-        "vectorKnowledgeBaseConfiguration": NotRequired[VectorKnowledgeBaseConfigurationTypeDef],
-    },
-)
-
-
 class CustomContentTypeDef(TypedDict):
     customDocumentIdentifier: CustomDocumentIdentifierTypeDef
     sourceType: CustomSourceTypeType
@@ -2259,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
@@ -2301,6 +2596,7 @@ class AgentTypeDef(TypedDict):
     createdAt: datetime
     idleSessionTTLInSeconds: int
     updatedAt: datetime
+    agentCollaboration: NotRequired[AgentCollaborationType]
     clientToken: NotRequired[str]
     customOrchestration: NotRequired[CustomOrchestrationTypeDef]
     customerEncryptionKeyArn: NotRequired[str]
@@ -2326,6 +2622,7 @@ class AgentVersionTypeDef(TypedDict):
     idleSessionTTLInSeconds: int
     updatedAt: datetime
     version: str
+    agentCollaboration: NotRequired[AgentCollaborationType]
     customerEncryptionKeyArn: NotRequired[str]
     description: NotRequired[str]
     failureReasons: NotRequired[List[str]]
@@ -2362,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]
@@ -2389,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
@@ -2521,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
 ]
@@ -2548,28 +2847,19 @@ 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]
     agentResourceRoleArn: NotRequired[str]
     clientToken: NotRequired[str]
     customOrchestration: NotRequired[CustomOrchestrationTypeDef]
@@ -2590,6 +2880,7 @@ class UpdateAgentRequestRequestTypeDef(TypedDict):
     agentName: str
     agentResourceRoleArn: str
     foundationModel: str
+    agentCollaboration: NotRequired[AgentCollaborationType]
     customOrchestration: NotRequired[CustomOrchestrationTypeDef]
     customerEncryptionKeyArn: NotRequired[str]
     description: NotRequired[str]
@@ -2615,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):
@@ -2670,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
 ]
@@ -2725,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]
@@ -2812,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]
@@ -2833,6 +3190,13 @@ SalesforceDataSourceConfigurationUnionTypeDef = Union[
 SharePointDataSourceConfigurationUnionTypeDef = Union[
     SharePointDataSourceConfigurationTypeDef, SharePointDataSourceConfigurationOutputTypeDef
 ]
+SqlKnowledgeBaseConfigurationTypeDef = TypedDict(
+    "SqlKnowledgeBaseConfigurationTypeDef",
+    {
+        "type": Literal["REDSHIFT"],
+        "redshiftConfiguration": NotRequired[RedshiftConfigurationUnionTypeDef],
+    },
+)
 FlowNodeConfigurationOutputTypeDef = TypedDict(
     "FlowNodeConfigurationOutputTypeDef",
     {
@@ -2864,6 +3228,9 @@ DataSourceConfigurationTypeDef = TypedDict(
         "webConfiguration": NotRequired[WebDataSourceConfigurationUnionTypeDef],
     },
 )
+SqlKnowledgeBaseConfigurationUnionTypeDef = Union[
+    SqlKnowledgeBaseConfigurationTypeDef, SqlKnowledgeBaseConfigurationOutputTypeDef
+]
 FlowNodeExtraOutputTypeDef = TypedDict(
     "FlowNodeExtraOutputTypeDef",
     {
@@ -2903,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]]
@@ -2911,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",
     {