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

types_boto3_bedrock_agent/type_defs.pyi

@@ -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,
@@ -90,6 +97,7 @@ __all__ = (
     "AssociateAgentCollaboratorResponseTypeDef",
     "AssociateAgentKnowledgeBaseRequestRequestTypeDef",
     "AssociateAgentKnowledgeBaseResponseTypeDef",
+    "BedrockDataAutomationConfigurationTypeDef",
     "BedrockEmbeddingModelConfigurationTypeDef",
     "BedrockFoundationModelConfigurationTypeDef",
     "BlobTypeDef",
@@ -134,6 +142,7 @@ __all__ = (
     "CreatePromptResponseTypeDef",
     "CreatePromptVersionRequestRequestTypeDef",
     "CreatePromptVersionResponseTypeDef",
+    "CuratedQueryTypeDef",
     "CustomContentTypeDef",
     "CustomDocumentIdentifierTypeDef",
     "CustomOrchestrationTypeDef",
@@ -248,6 +257,8 @@ __all__ = (
     "IngestionJobTypeDef",
     "InlineContentTypeDef",
     "IntermediateStorageTypeDef",
+    "KendraKnowledgeBaseConfigurationTypeDef",
+    "KnowledgeBaseConfigurationOutputTypeDef",
     "KnowledgeBaseConfigurationTypeDef",
     "KnowledgeBaseDocumentDetailTypeDef",
     "KnowledgeBaseDocumentTypeDef",
@@ -370,10 +381,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",
@@ -404,6 +440,9 @@ __all__ = (
     "SharePointSourceConfigurationTypeDef",
     "SharePointSourceConfigurationUnionTypeDef",
     "SpecificToolChoiceTypeDef",
+    "SqlKnowledgeBaseConfigurationOutputTypeDef",
+    "SqlKnowledgeBaseConfigurationTypeDef",
+    "SqlKnowledgeBaseConfigurationUnionTypeDef",
     "StartIngestionJobRequestRequestTypeDef",
     "StartIngestionJobResponseTypeDef",
     "StopIngestionJobRequestRequestTypeDef",
@@ -412,6 +451,10 @@ __all__ = (
     "StorageFlowNodeConfigurationTypeDef",
     "StorageFlowNodeS3ConfigurationTypeDef",
     "StorageFlowNodeServiceConfigurationTypeDef",
+    "SupplementalDataStorageConfigurationOutputTypeDef",
+    "SupplementalDataStorageConfigurationTypeDef",
+    "SupplementalDataStorageConfigurationUnionTypeDef",
+    "SupplementalDataStorageLocationTypeDef",
     "SystemContentBlockTypeDef",
     "TagResourceRequestRequestTypeDef",
     "TextContentDocTypeDef",
@@ -472,7 +515,9 @@ __all__ = (
     "ValidateFlowDefinitionResponseTypeDef",
     "VectorIngestionConfigurationOutputTypeDef",
     "VectorIngestionConfigurationTypeDef",
+    "VectorKnowledgeBaseConfigurationOutputTypeDef",
     "VectorKnowledgeBaseConfigurationTypeDef",
+    "VectorKnowledgeBaseConfigurationUnionTypeDef",
     "WebCrawlerConfigurationOutputTypeDef",
     "WebCrawlerConfigurationTypeDef",
     "WebCrawlerConfigurationUnionTypeDef",
@@ -551,6 +596,9 @@ class AssociateAgentKnowledgeBaseRequestRequestTypeDef(TypedDict):
     knowledgeBaseId: str
     knowledgeBaseState: NotRequired[KnowledgeBaseStateType]
 
+class BedrockDataAutomationConfigurationTypeDef(TypedDict):
+    parsingModality: NotRequired[Literal["MULTIMODAL"]]
+
 class BedrockEmbeddingModelConfigurationTypeDef(TypedDict):
     dimensions: NotRequired[int]
     embeddingDataType: NotRequired[EmbeddingDataTypeType]
@@ -609,6 +657,10 @@ class CreatePromptVersionRequestRequestTypeDef(TypedDict):
     description: NotRequired[str]
     tags: NotRequired[Mapping[str, str]]
 
+class CuratedQueryTypeDef(TypedDict):
+    naturalLanguage: str
+    sql: str
+
 CustomDocumentIdentifierTypeDef = TypedDict(
     "CustomDocumentIdentifierTypeDef",
     {
@@ -951,6 +1003,9 @@ class IngestionJobStatisticsTypeDef(TypedDict):
 class TextContentDocTypeDef(TypedDict):
     data: str
 
+class KendraKnowledgeBaseConfigurationTypeDef(TypedDict):
+    kendraIndexArn: str
+
 class KnowledgeBaseSummaryTypeDef(TypedDict):
     knowledgeBaseId: str
     name: str
@@ -1108,6 +1163,11 @@ class PromptModelInferenceConfigurationTypeDef(TypedDict):
     temperature: NotRequired[float]
     topP: NotRequired[float]
 
+class QueryGenerationColumnTypeDef(TypedDict):
+    description: NotRequired[str]
+    inclusion: NotRequired[IncludeExcludeType]
+    name: NotRequired[str]
+
 class RdsFieldMappingTypeDef(TypedDict):
     metadataField: str
     primaryKeyField: str
@@ -1119,6 +1179,32 @@ class RedisEnterpriseCloudFieldMappingTypeDef(TypedDict):
     textField: str
     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
 
@@ -1400,6 +1486,7 @@ class EmbeddingModelConfigurationTypeDef(TypedDict):
 
 class BedrockFoundationModelConfigurationTypeDef(TypedDict):
     modelArn: str
+    parsingModality: NotRequired[Literal["MULTIMODAL"]]
     parsingPrompt: NotRequired[ParsingPromptTypeDef]
 
 class ByteContentDocTypeDef(TypedDict):
@@ -1519,6 +1606,14 @@ class IntermediateStorageTypeDef(TypedDict):
 class S3ContentTypeDef(TypedDict):
     s3Location: S3LocationTypeDef
 
+SupplementalDataStorageLocationTypeDef = TypedDict(
+    "SupplementalDataStorageLocationTypeDef",
+    {
+        "type": Literal["S3"],
+        "s3Location": NotRequired[S3LocationTypeDef],
+    },
+)
+
 class FlowConnectionConfigurationTypeDef(TypedDict):
     conditional: NotRequired[FlowConditionalConnectionConfigurationTypeDef]
     data: NotRequired[FlowDataConnectionConfigurationTypeDef]
@@ -1751,6 +1846,18 @@ PromptModelInferenceConfigurationUnionTypeDef = Union[
     PromptModelInferenceConfigurationTypeDef, PromptModelInferenceConfigurationOutputTypeDef
 ]
 
+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
@@ -1764,6 +1871,31 @@ class RedisEnterpriseCloudConfigurationTypeDef(TypedDict):
     fieldMapping: RedisEnterpriseCloudFieldMappingTypeDef
     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]
 
@@ -1868,12 +2000,9 @@ class ListAgentVersionsResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
     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]
 
 InlineContentTypeDef = TypedDict(
@@ -1916,6 +2045,12 @@ class KnowledgeBaseDocumentDetailTypeDef(TypedDict):
     statusReason: NotRequired[str]
     updatedAt: NotRequired[datetime]
 
+class SupplementalDataStorageConfigurationOutputTypeDef(TypedDict):
+    storageLocations: List[SupplementalDataStorageLocationTypeDef]
+
+class SupplementalDataStorageConfigurationTypeDef(TypedDict):
+    storageLocations: Sequence[SupplementalDataStorageLocationTypeDef]
+
 FlowConnectionTypeDef = TypedDict(
     "FlowConnectionTypeDef",
     {
@@ -2003,6 +2138,13 @@ class PatternObjectFilterConfigurationTypeDef(TypedDict):
 class PromptInferenceConfigurationTypeDef(TypedDict):
     text: NotRequired[PromptModelInferenceConfigurationUnionTypeDef]
 
+class QueryGenerationContextOutputTypeDef(TypedDict):
+    curatedQueries: NotRequired[List[CuratedQueryTypeDef]]
+    tables: NotRequired[List[QueryGenerationTableOutputTypeDef]]
+
+QueryGenerationTableUnionTypeDef = Union[
+    QueryGenerationTableTypeDef, QueryGenerationTableOutputTypeDef
+]
 StorageConfigurationTypeDef = TypedDict(
     "StorageConfigurationTypeDef",
     {
@@ -2014,6 +2156,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):
     serviceConfiguration: RetrievalFlowNodeServiceConfigurationTypeDef
@@ -2055,14 +2217,6 @@ class UpdateAgentAliasResponseTypeDef(TypedDict):
     agentAlias: AgentAliasTypeDef
     ResponseMetadata: ResponseMetadataTypeDef
 
-KnowledgeBaseConfigurationTypeDef = TypedDict(
-    "KnowledgeBaseConfigurationTypeDef",
-    {
-        "type": Literal["VECTOR"],
-        "vectorKnowledgeBaseConfiguration": NotRequired[VectorKnowledgeBaseConfigurationTypeDef],
-    },
-)
-
 class CustomContentTypeDef(TypedDict):
     customDocumentIdentifier: CustomDocumentIdentifierTypeDef
     sourceType: CustomSourceTypeType
@@ -2086,6 +2240,17 @@ class ListKnowledgeBaseDocumentsResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
     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
@@ -2180,6 +2345,19 @@ PromptInferenceConfigurationUnionTypeDef = Union[
     PromptInferenceConfigurationTypeDef, PromptInferenceConfigurationOutputTypeDef
 ]
 
+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]
@@ -2201,41 +2379,18 @@ class CustomTransformationConfigurationTypeDef(TypedDict):
     intermediateStorage: IntermediateStorageTypeDef
     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
@@ -2315,6 +2470,15 @@ CrawlFilterConfigurationTypeDef = TypedDict(
         "patternObjectFilter": NotRequired[PatternObjectFilterConfigurationUnionTypeDef],
     },
 )
+
+class RedshiftConfigurationOutputTypeDef(TypedDict):
+    queryEngineConfiguration: RedshiftQueryEngineConfigurationTypeDef
+    storageConfigurations: List[RedshiftQueryEngineStorageConfigurationOutputTypeDef]
+    queryGenerationConfiguration: NotRequired[QueryGenerationConfigurationOutputTypeDef]
+
+QueryGenerationContextUnionTypeDef = Union[
+    QueryGenerationContextTypeDef, QueryGenerationContextOutputTypeDef
+]
 WebSourceConfigurationUnionTypeDef = Union[
     WebSourceConfigurationTypeDef, WebSourceConfigurationOutputTypeDef
 ]
@@ -2337,22 +2501,14 @@ CustomTransformationConfigurationUnionTypeDef = Union[
     CustomTransformationConfigurationTypeDef, CustomTransformationConfigurationOutputTypeDef
 ]
 
-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]
@@ -2400,6 +2556,17 @@ 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):
     sourceConfiguration: WebSourceConfigurationUnionTypeDef
@@ -2445,6 +2612,20 @@ class SalesforceCrawlerConfigurationTypeDef(TypedDict):
 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
 ]
@@ -2492,6 +2673,24 @@ SharePointCrawlerConfigurationUnionTypeDef = Union[
     SharePointCrawlerConfigurationTypeDef, SharePointCrawlerConfigurationOutputTypeDef
 ]
 
+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]
@@ -2574,6 +2773,22 @@ class SharePointDataSourceConfigurationTypeDef(TypedDict):
     sourceConfiguration: SharePointSourceConfigurationUnionTypeDef
     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]
@@ -2593,6 +2808,13 @@ SalesforceDataSourceConfigurationUnionTypeDef = Union[
 SharePointDataSourceConfigurationUnionTypeDef = Union[
     SharePointDataSourceConfigurationTypeDef, SharePointDataSourceConfigurationOutputTypeDef
 ]
+SqlKnowledgeBaseConfigurationTypeDef = TypedDict(
+    "SqlKnowledgeBaseConfigurationTypeDef",
+    {
+        "type": Literal["REDSHIFT"],
+        "redshiftConfiguration": NotRequired[RedshiftConfigurationUnionTypeDef],
+    },
+)
 FlowNodeConfigurationOutputTypeDef = TypedDict(
     "FlowNodeConfigurationOutputTypeDef",
     {
@@ -2624,6 +2846,9 @@ DataSourceConfigurationTypeDef = TypedDict(
         "webConfiguration": NotRequired[WebDataSourceConfigurationUnionTypeDef],
     },
 )
+SqlKnowledgeBaseConfigurationUnionTypeDef = Union[
+    SqlKnowledgeBaseConfigurationTypeDef, SqlKnowledgeBaseConfigurationOutputTypeDef
+]
 FlowNodeExtraOutputTypeDef = TypedDict(
     "FlowNodeExtraOutputTypeDef",
     {
@@ -2659,6 +2884,18 @@ class UpdateDataSourceRequestRequestTypeDef(TypedDict):
     serverSideEncryptionConfiguration: NotRequired[ServerSideEncryptionConfigurationTypeDef]
     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]]
@@ -2666,6 +2903,24 @@ 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",
     {

types_boto3_bedrock_agent/version.py

@@ -4,4 +4,4 @@ Source of truth for version.
 Copyright 2024 Vlad Emelianov
 """
 
-__version__ = "1.35.74"
+__version__ = "1.35.75"

types_boto3_bedrock_agent_runtime/__main__.py

@@ -12,8 +12,8 @@ def print_info() -> None:
     Print package info to stdout.
     """
     print(
-        "Type annotations for boto3 AgentsforBedrockRuntime 1.35.74\n"
-        "Version:         1.35.74\n"
+        "Type annotations for boto3 AgentsforBedrockRuntime 1.35.75\n"
+        "Version:         1.35.75\n"
         "Builder version: 8.5.0\n"
         "Docs:            https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime//\n"
         "Boto3 docs:      https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent-runtime.html#agentsforbedrockruntime\n"
@@ -26,7 +26,7 @@ def print_version() -> None:
     """
     Print package version to stdout.
     """
-    print("1.35.74")
+    print("1.35.75")
 
 
 def main() -> None:

types_boto3_bedrock_agent_runtime/client.py

@@ -24,6 +24,8 @@ from botocore.client import BaseClient, ClientMeta
 from .paginator import GetAgentMemoryPaginator, RerankPaginator, RetrievePaginator
 from .type_defs import (
     DeleteAgentMemoryRequestRequestTypeDef,
+    GenerateQueryRequestRequestTypeDef,
+    GenerateQueryResponseTypeDef,
     GetAgentMemoryRequestRequestTypeDef,
     GetAgentMemoryResponseTypeDef,
     InvokeAgentRequestRequestTypeDef,
@@ -125,6 +127,16 @@ class AgentsforBedrockRuntimeClient(BaseClient):
         [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#delete_agent_memory)
         """
 
+    def generate_query(
+        self, **kwargs: Unpack[GenerateQueryRequestRequestTypeDef]
+    ) -> GenerateQueryResponseTypeDef:
+        """
+        Generates an SQL query from a natural language query.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent-runtime/client/generate_query.html)
+        [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#generate_query)
+        """
+
     def get_agent_memory(
         self, **kwargs: Unpack[GetAgentMemoryRequestRequestTypeDef]
     ) -> GetAgentMemoryResponseTypeDef: