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

types_boto3_bedrock_agent_runtime/client.pyi

@@ -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,
@@ -121,6 +123,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:

types_boto3_bedrock_agent_runtime/literals.py

@@ -36,6 +36,7 @@ __all__ = (
     "FileSourceTypeType",
     "FileUseCaseType",
     "FlowCompletionReasonType",
+    "GeneratedQueryTypeType",
     "GetAgentMemoryPaginatorName",
     "GuadrailActionType",
     "GuardrailActionType",
@@ -48,6 +49,7 @@ __all__ = (
     "GuardrailTopicPolicyActionType",
     "GuardrailTopicTypeType",
     "GuardrailWordPolicyActionType",
+    "InputQueryTypeType",
     "InvocationTypeType",
     "MemoryTypeType",
     "NodeTypeType",
@@ -56,6 +58,7 @@ __all__ = (
     "PayloadTypeType",
     "PromptStateType",
     "PromptTypeType",
+    "QueryTransformationModeType",
     "QueryTransformationTypeType",
     "RequireConfirmationType",
     "RerankDocumentTypeType",
@@ -66,12 +69,15 @@ __all__ = (
     "RerankingMetadataSelectionModeType",
     "ResourceServiceName",
     "ResponseStateType",
+    "RetrievalResultContentColumnTypeType",
+    "RetrievalResultContentTypeType",
     "RetrievalResultLocationTypeType",
     "RetrieveAndGenerateTypeType",
     "RetrievePaginatorName",
     "SearchTypeType",
     "ServiceName",
     "SourceType",
+    "TextToSqlConfigurationTypeType",
     "TypeType",
     "VectorSearchRerankingConfigurationTypeType",
 )
@@ -89,6 +95,7 @@ ExternalSourceTypeType = Literal["BYTE_CONTENT", "S3"]
 FileSourceTypeType = Literal["BYTE_CONTENT", "S3"]
 FileUseCaseType = Literal["CHAT", "CODE_INTERPRETER"]
 FlowCompletionReasonType = Literal["SUCCESS"]
+GeneratedQueryTypeType = Literal["REDSHIFT_SQL"]
 GetAgentMemoryPaginatorName = Literal["get_agent_memory"]
 GuadrailActionType = Literal["INTERVENED", "NONE"]
 GuardrailActionType = Literal["INTERVENED", "NONE"]
@@ -135,6 +142,7 @@ GuardrailSensitiveInformationPolicyActionType = Literal["ANONYMIZED", "BLOCKED"]
 GuardrailTopicPolicyActionType = Literal["BLOCKED"]
 GuardrailTopicTypeType = Literal["DENY"]
 GuardrailWordPolicyActionType = Literal["BLOCKED"]
+InputQueryTypeType = Literal["TEXT"]
 InvocationTypeType = Literal[
     "ACTION_GROUP",
     "ACTION_GROUP_CODE_INTERPRETER",
@@ -162,6 +170,7 @@ PromptTypeType = Literal[
     "PRE_PROCESSING",
     "ROUTING_CLASSIFIER",
 ]
+QueryTransformationModeType = Literal["TEXT_TO_SQL"]
 QueryTransformationTypeType = Literal["QUERY_DECOMPOSITION"]
 RequireConfirmationType = Literal["DISABLED", "ENABLED"]
 RerankDocumentTypeType = Literal["JSON", "TEXT"]
@@ -171,13 +180,18 @@ RerankSourceTypeType = Literal["INLINE"]
 RerankingConfigurationTypeType = Literal["BEDROCK_RERANKING_MODEL"]
 RerankingMetadataSelectionModeType = Literal["ALL", "SELECTIVE"]
 ResponseStateType = Literal["FAILURE", "REPROMPT"]
+RetrievalResultContentColumnTypeType = Literal[
+    "BLOB", "BOOLEAN", "DOUBLE", "LONG", "NULL", "STRING"
+]
+RetrievalResultContentTypeType = Literal["IMAGE", "ROW", "TEXT"]
 RetrievalResultLocationTypeType = Literal[
-    "CONFLUENCE", "CUSTOM", "S3", "SALESFORCE", "SHAREPOINT", "WEB"
+    "CONFLUENCE", "CUSTOM", "KENDRA", "S3", "SALESFORCE", "SHAREPOINT", "SQL", "WEB"
 ]
 RetrieveAndGenerateTypeType = Literal["EXTERNAL_SOURCES", "KNOWLEDGE_BASE"]
 RetrievePaginatorName = Literal["retrieve"]
 SearchTypeType = Literal["HYBRID", "SEMANTIC"]
 SourceType = Literal["ACTION_GROUP", "KNOWLEDGE_BASE", "PARSER"]
+TextToSqlConfigurationTypeType = Literal["KNOWLEDGE_BASE"]
 TypeType = Literal[
     "ACTION_GROUP", "AGENT_COLLABORATOR", "ASK_USER", "FINISH", "KNOWLEDGE_BASE", "REPROMPT"
 ]
@@ -224,6 +238,8 @@ ServiceName = Literal[
     "bedrock",
     "bedrock-agent",
     "bedrock-agent-runtime",
+    "bedrock-data-automation",
+    "bedrock-data-automation-runtime",
     "bedrock-runtime",
     "billing",
     "billingconductor",

types_boto3_bedrock_agent_runtime/literals.pyi

@@ -35,6 +35,7 @@ __all__ = (
     "FileSourceTypeType",
     "FileUseCaseType",
     "FlowCompletionReasonType",
+    "GeneratedQueryTypeType",
     "GetAgentMemoryPaginatorName",
     "GuadrailActionType",
     "GuardrailActionType",
@@ -47,6 +48,7 @@ __all__ = (
     "GuardrailTopicPolicyActionType",
     "GuardrailTopicTypeType",
     "GuardrailWordPolicyActionType",
+    "InputQueryTypeType",
     "InvocationTypeType",
     "MemoryTypeType",
     "NodeTypeType",
@@ -55,6 +57,7 @@ __all__ = (
     "PayloadTypeType",
     "PromptStateType",
     "PromptTypeType",
+    "QueryTransformationModeType",
     "QueryTransformationTypeType",
     "RequireConfirmationType",
     "RerankDocumentTypeType",
@@ -65,12 +68,15 @@ __all__ = (
     "RerankingMetadataSelectionModeType",
     "ResourceServiceName",
     "ResponseStateType",
+    "RetrievalResultContentColumnTypeType",
+    "RetrievalResultContentTypeType",
     "RetrievalResultLocationTypeType",
     "RetrieveAndGenerateTypeType",
     "RetrievePaginatorName",
     "SearchTypeType",
     "ServiceName",
     "SourceType",
+    "TextToSqlConfigurationTypeType",
     "TypeType",
     "VectorSearchRerankingConfigurationTypeType",
 )
@@ -87,6 +93,7 @@ ExternalSourceTypeType = Literal["BYTE_CONTENT", "S3"]
 FileSourceTypeType = Literal["BYTE_CONTENT", "S3"]
 FileUseCaseType = Literal["CHAT", "CODE_INTERPRETER"]
 FlowCompletionReasonType = Literal["SUCCESS"]
+GeneratedQueryTypeType = Literal["REDSHIFT_SQL"]
 GetAgentMemoryPaginatorName = Literal["get_agent_memory"]
 GuadrailActionType = Literal["INTERVENED", "NONE"]
 GuardrailActionType = Literal["INTERVENED", "NONE"]
@@ -133,6 +140,7 @@ GuardrailSensitiveInformationPolicyActionType = Literal["ANONYMIZED", "BLOCKED"]
 GuardrailTopicPolicyActionType = Literal["BLOCKED"]
 GuardrailTopicTypeType = Literal["DENY"]
 GuardrailWordPolicyActionType = Literal["BLOCKED"]
+InputQueryTypeType = Literal["TEXT"]
 InvocationTypeType = Literal[
     "ACTION_GROUP",
     "ACTION_GROUP_CODE_INTERPRETER",
@@ -160,6 +168,7 @@ PromptTypeType = Literal[
     "PRE_PROCESSING",
     "ROUTING_CLASSIFIER",
 ]
+QueryTransformationModeType = Literal["TEXT_TO_SQL"]
 QueryTransformationTypeType = Literal["QUERY_DECOMPOSITION"]
 RequireConfirmationType = Literal["DISABLED", "ENABLED"]
 RerankDocumentTypeType = Literal["JSON", "TEXT"]
@@ -169,13 +178,18 @@ RerankSourceTypeType = Literal["INLINE"]
 RerankingConfigurationTypeType = Literal["BEDROCK_RERANKING_MODEL"]
 RerankingMetadataSelectionModeType = Literal["ALL", "SELECTIVE"]
 ResponseStateType = Literal["FAILURE", "REPROMPT"]
+RetrievalResultContentColumnTypeType = Literal[
+    "BLOB", "BOOLEAN", "DOUBLE", "LONG", "NULL", "STRING"
+]
+RetrievalResultContentTypeType = Literal["IMAGE", "ROW", "TEXT"]
 RetrievalResultLocationTypeType = Literal[
-    "CONFLUENCE", "CUSTOM", "S3", "SALESFORCE", "SHAREPOINT", "WEB"
+    "CONFLUENCE", "CUSTOM", "KENDRA", "S3", "SALESFORCE", "SHAREPOINT", "SQL", "WEB"
 ]
 RetrieveAndGenerateTypeType = Literal["EXTERNAL_SOURCES", "KNOWLEDGE_BASE"]
 RetrievePaginatorName = Literal["retrieve"]
 SearchTypeType = Literal["HYBRID", "SEMANTIC"]
 SourceType = Literal["ACTION_GROUP", "KNOWLEDGE_BASE", "PARSER"]
+TextToSqlConfigurationTypeType = Literal["KNOWLEDGE_BASE"]
 TypeType = Literal[
     "ACTION_GROUP", "AGENT_COLLABORATOR", "ASK_USER", "FINISH", "KNOWLEDGE_BASE", "REPROMPT"
 ]
@@ -222,6 +236,8 @@ ServiceName = Literal[
     "bedrock",
     "bedrock-agent",
     "bedrock-agent-runtime",
+    "bedrock-data-automation",
+    "bedrock-data-automation-runtime",
     "bedrock-runtime",
     "billing",
     "billingconductor",

types_boto3_bedrock_agent_runtime/type_defs.py

@@ -48,6 +48,8 @@ from .literals import (
     RerankDocumentTypeType,
     RerankingMetadataSelectionModeType,
     ResponseStateType,
+    RetrievalResultContentColumnTypeType,
+    RetrievalResultContentTypeType,
     RetrievalResultLocationTypeType,
     RetrieveAndGenerateTypeType,
     SearchTypeType,
@@ -131,6 +133,9 @@ __all__ = (
     "FunctionResultTypeDef",
     "FunctionResultUnionTypeDef",
     "FunctionSchemaTypeDef",
+    "GenerateQueryRequestRequestTypeDef",
+    "GenerateQueryResponseTypeDef",
+    "GeneratedQueryTypeDef",
     "GeneratedResponsePartTypeDef",
     "GenerationConfigurationTypeDef",
     "GetAgentMemoryRequestGetAgentMemoryPaginateTypeDef",
@@ -218,6 +223,7 @@ __all__ = (
     "PromptOverrideConfigurationTypeDef",
     "PromptTemplateTypeDef",
     "PropertyParametersTypeDef",
+    "QueryGenerationInputTypeDef",
     "QueryTransformationConfigurationTypeDef",
     "RationaleTypeDef",
     "RawResponseTypeDef",
@@ -241,12 +247,15 @@ __all__ = (
     "RetrievalFilterPaginatorTypeDef",
     "RetrievalFilterTypeDef",
     "RetrievalResultConfluenceLocationTypeDef",
+    "RetrievalResultContentColumnTypeDef",
     "RetrievalResultContentTypeDef",
     "RetrievalResultCustomDocumentLocationTypeDef",
+    "RetrievalResultKendraDocumentLocationTypeDef",
     "RetrievalResultLocationTypeDef",
     "RetrievalResultS3LocationTypeDef",
     "RetrievalResultSalesforceLocationTypeDef",
     "RetrievalResultSharePointLocationTypeDef",
+    "RetrievalResultSqlLocationTypeDef",
     "RetrievalResultWebLocationTypeDef",
     "RetrieveAndGenerateConfigurationTypeDef",
     "RetrieveAndGenerateInputTypeDef",
@@ -276,9 +285,12 @@ __all__ = (
     "TextInferenceConfigTypeDef",
     "TextPromptTypeDef",
     "TextResponsePartTypeDef",
+    "TextToSqlConfigurationTypeDef",
+    "TextToSqlKnowledgeBaseConfigurationTypeDef",
     "ThrottlingExceptionTypeDef",
     "TracePartTypeDef",
     "TraceTypeDef",
+    "TransformationConfigurationTypeDef",
     "UsageTypeDef",
     "ValidationExceptionTypeDef",
     "VectorSearchBedrockRerankingConfigurationTypeDef",
@@ -492,6 +504,28 @@ FunctionParameterTypeDef = TypedDict(
         "value": NotRequired[str],
     },
 )
+QueryGenerationInputTypeDef = TypedDict(
+    "QueryGenerationInputTypeDef",
+    {
+        "text": str,
+        "type": Literal["TEXT"],
+    },
+)
+GeneratedQueryTypeDef = TypedDict(
+    "GeneratedQueryTypeDef",
+    {
+        "sql": NotRequired[str],
+        "type": NotRequired[Literal["REDSHIFT_SQL"]],
+    },
+)
+
+
+class ResponseMetadataTypeDef(TypedDict):
+    RequestId: str
+    HTTPStatusCode: int
+    HTTPHeaders: Dict[str, str]
+    RetryAttempts: int
+    HostId: NotRequired[str]
 
 
 class PaginatorConfigTypeDef(TypedDict):
@@ -509,14 +543,6 @@ class GetAgentMemoryRequestRequestTypeDef(TypedDict):
     nextToken: NotRequired[str]
 
 
-class ResponseMetadataTypeDef(TypedDict):
-    RequestId: str
-    HTTPStatusCode: int
-    HTTPHeaders: Dict[str, str]
-    RetryAttempts: int
-    HostId: NotRequired[str]
-
-
 class GuardrailConfigurationWithArnTypeDef(TypedDict):
     guardrailIdentifier: str
     guardrailVersion: str
@@ -625,10 +651,6 @@ class KnowledgeBaseQueryTypeDef(TypedDict):
     text: str
 
 
-class RetrievalResultContentTypeDef(TypedDict):
-    text: str
-
-
 class MemorySessionSummaryTypeDef(TypedDict):
     memoryId: NotRequired[str]
     sessionExpiryTime: NotRequired[datetime]
@@ -681,6 +703,14 @@ class RetrievalResultConfluenceLocationTypeDef(TypedDict):
     url: NotRequired[str]
 
 
+RetrievalResultContentColumnTypeDef = TypedDict(
+    "RetrievalResultContentColumnTypeDef",
+    {
+        "columnName": NotRequired[str],
+        "columnValue": NotRequired[str],
+        "type": NotRequired[RetrievalResultContentColumnTypeType],
+    },
+)
 RetrievalResultCustomDocumentLocationTypeDef = TypedDict(
     "RetrievalResultCustomDocumentLocationTypeDef",
     {
@@ -689,6 +719,10 @@ RetrievalResultCustomDocumentLocationTypeDef = TypedDict(
 )
 
 
+class RetrievalResultKendraDocumentLocationTypeDef(TypedDict):
+    uri: NotRequired[str]
+
+
 class RetrievalResultS3LocationTypeDef(TypedDict):
     uri: NotRequired[str]
 
@@ -701,6 +735,10 @@ class RetrievalResultSharePointLocationTypeDef(TypedDict):
     url: NotRequired[str]
 
 
+class RetrievalResultSqlLocationTypeDef(TypedDict):
+    query: NotRequired[str]
+
+
 class RetrievalResultWebLocationTypeDef(TypedDict):
     url: NotRequired[str]
 
@@ -726,6 +764,10 @@ class SpanTypeDef(TypedDict):
     start: NotRequired[int]
 
 
+class TextToSqlKnowledgeBaseConfigurationTypeDef(TypedDict):
+    knowledgeBaseArn: str
+
+
 class VectorSearchBedrockRerankingModelConfigurationTypeDef(TypedDict):
     modelArn: str
     additionalModelRequestFields: NotRequired[Mapping[str, Mapping[str, Any]]]
@@ -905,6 +947,11 @@ class FunctionInvocationInputTypeDef(TypedDict):
     parameters: NotRequired[List[FunctionParameterTypeDef]]
 
 
+class GenerateQueryResponseTypeDef(TypedDict):
+    queries: List[GeneratedQueryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class GetAgentMemoryRequestGetAgentMemoryPaginateTypeDef(TypedDict):
     agentAliasId: str
     agentId: str
@@ -997,15 +1044,26 @@ RerankQueryTypeDef = TypedDict(
         "type": Literal["TEXT"],
     },
 )
+RetrievalResultContentTypeDef = TypedDict(
+    "RetrievalResultContentTypeDef",
+    {
+        "byteContent": NotRequired[str],
+        "row": NotRequired[List[RetrievalResultContentColumnTypeDef]],
+        "text": NotRequired[str],
+        "type": NotRequired[RetrievalResultContentTypeType],
+    },
+)
 RetrievalResultLocationTypeDef = TypedDict(
     "RetrievalResultLocationTypeDef",
     {
         "type": RetrievalResultLocationTypeType,
         "confluenceLocation": NotRequired[RetrievalResultConfluenceLocationTypeDef],
         "customDocumentLocation": NotRequired[RetrievalResultCustomDocumentLocationTypeDef],
+        "kendraDocumentLocation": NotRequired[RetrievalResultKendraDocumentLocationTypeDef],
         "s3Location": NotRequired[RetrievalResultS3LocationTypeDef],
         "salesforceLocation": NotRequired[RetrievalResultSalesforceLocationTypeDef],
         "sharePointLocation": NotRequired[RetrievalResultSharePointLocationTypeDef],
+        "sqlLocation": NotRequired[RetrievalResultSqlLocationTypeDef],
         "webLocation": NotRequired[RetrievalResultWebLocationTypeDef],
     },
 )
@@ -1016,6 +1074,15 @@ class TextResponsePartTypeDef(TypedDict):
     text: NotRequired[str]
 
 
+TextToSqlConfigurationTypeDef = TypedDict(
+    "TextToSqlConfigurationTypeDef",
+    {
+        "type": Literal["KNOWLEDGE_BASE"],
+        "knowledgeBaseConfiguration": NotRequired[TextToSqlKnowledgeBaseConfigurationTypeDef],
+    },
+)
+
+
 class ApiRequestBodyTypeDef(TypedDict):
     content: NotRequired[Dict[str, PropertyParametersTypeDef]]
 
@@ -1201,6 +1268,11 @@ class GeneratedResponsePartTypeDef(TypedDict):
     textResponsePart: NotRequired[TextResponsePartTypeDef]
 
 
+class TransformationConfigurationTypeDef(TypedDict):
+    mode: Literal["TEXT_TO_SQL"]
+    textToSqlConfiguration: NotRequired[TextToSqlConfigurationTypeDef]
+
+
 class ApiInvocationInputTypeDef(TypedDict):
     actionGroup: str
     actionInvocationType: NotRequired[ActionInvocationTypeType]
@@ -1319,6 +1391,11 @@ class CitationTypeDef(TypedDict):
     retrievedReferences: NotRequired[List[RetrievedReferenceTypeDef]]
 
 
+class GenerateQueryRequestRequestTypeDef(TypedDict):
+    queryGenerationInput: QueryGenerationInputTypeDef
+    transformationConfiguration: TransformationConfigurationTypeDef
+
+
 class InvocationInputMemberTypeDef(TypedDict):
     apiInvocationInput: NotRequired[ApiInvocationInputTypeDef]
     functionInvocationInput: NotRequired[FunctionInvocationInputTypeDef]

types_boto3_bedrock_agent_runtime/type_defs.pyi

@@ -48,6 +48,8 @@ from .literals import (
     RerankDocumentTypeType,
     RerankingMetadataSelectionModeType,
     ResponseStateType,
+    RetrievalResultContentColumnTypeType,
+    RetrievalResultContentTypeType,
     RetrievalResultLocationTypeType,
     RetrieveAndGenerateTypeType,
     SearchTypeType,
@@ -130,6 +132,9 @@ __all__ = (
     "FunctionResultTypeDef",
     "FunctionResultUnionTypeDef",
     "FunctionSchemaTypeDef",
+    "GenerateQueryRequestRequestTypeDef",
+    "GenerateQueryResponseTypeDef",
+    "GeneratedQueryTypeDef",
     "GeneratedResponsePartTypeDef",
     "GenerationConfigurationTypeDef",
     "GetAgentMemoryRequestGetAgentMemoryPaginateTypeDef",
@@ -217,6 +222,7 @@ __all__ = (
     "PromptOverrideConfigurationTypeDef",
     "PromptTemplateTypeDef",
     "PropertyParametersTypeDef",
+    "QueryGenerationInputTypeDef",
     "QueryTransformationConfigurationTypeDef",
     "RationaleTypeDef",
     "RawResponseTypeDef",
@@ -240,12 +246,15 @@ __all__ = (
     "RetrievalFilterPaginatorTypeDef",
     "RetrievalFilterTypeDef",
     "RetrievalResultConfluenceLocationTypeDef",
+    "RetrievalResultContentColumnTypeDef",
     "RetrievalResultContentTypeDef",
     "RetrievalResultCustomDocumentLocationTypeDef",
+    "RetrievalResultKendraDocumentLocationTypeDef",
     "RetrievalResultLocationTypeDef",
     "RetrievalResultS3LocationTypeDef",
     "RetrievalResultSalesforceLocationTypeDef",
     "RetrievalResultSharePointLocationTypeDef",
+    "RetrievalResultSqlLocationTypeDef",
     "RetrievalResultWebLocationTypeDef",
     "RetrieveAndGenerateConfigurationTypeDef",
     "RetrieveAndGenerateInputTypeDef",
@@ -275,9 +284,12 @@ __all__ = (
     "TextInferenceConfigTypeDef",
     "TextPromptTypeDef",
     "TextResponsePartTypeDef",
+    "TextToSqlConfigurationTypeDef",
+    "TextToSqlKnowledgeBaseConfigurationTypeDef",
     "ThrottlingExceptionTypeDef",
     "TracePartTypeDef",
     "TraceTypeDef",
+    "TransformationConfigurationTypeDef",
     "UsageTypeDef",
     "ValidationExceptionTypeDef",
     "VectorSearchBedrockRerankingConfigurationTypeDef",
@@ -452,6 +464,27 @@ FunctionParameterTypeDef = TypedDict(
         "value": NotRequired[str],
     },
 )
+QueryGenerationInputTypeDef = TypedDict(
+    "QueryGenerationInputTypeDef",
+    {
+        "text": str,
+        "type": Literal["TEXT"],
+    },
+)
+GeneratedQueryTypeDef = TypedDict(
+    "GeneratedQueryTypeDef",
+    {
+        "sql": NotRequired[str],
+        "type": NotRequired[Literal["REDSHIFT_SQL"]],
+    },
+)
+
+class ResponseMetadataTypeDef(TypedDict):
+    RequestId: str
+    HTTPStatusCode: int
+    HTTPHeaders: Dict[str, str]
+    RetryAttempts: int
+    HostId: NotRequired[str]
 
 class PaginatorConfigTypeDef(TypedDict):
     MaxItems: NotRequired[int]
@@ -466,13 +499,6 @@ class GetAgentMemoryRequestRequestTypeDef(TypedDict):
     maxItems: NotRequired[int]
     nextToken: NotRequired[str]
 
-class ResponseMetadataTypeDef(TypedDict):
-    RequestId: str
-    HTTPStatusCode: int
-    HTTPHeaders: Dict[str, str]
-    RetryAttempts: int
-    HostId: NotRequired[str]
-
 class GuardrailConfigurationWithArnTypeDef(TypedDict):
     guardrailIdentifier: str
     guardrailVersion: str
@@ -567,9 +593,6 @@ class StreamingConfigurationsTypeDef(TypedDict):
 class KnowledgeBaseQueryTypeDef(TypedDict):
     text: str
 
-class RetrievalResultContentTypeDef(TypedDict):
-    text: str
-
 class MemorySessionSummaryTypeDef(TypedDict):
     memoryId: NotRequired[str]
     sessionExpiryTime: NotRequired[datetime]
@@ -612,6 +635,14 @@ class RerankTextDocumentTypeDef(TypedDict):
 class RetrievalResultConfluenceLocationTypeDef(TypedDict):
     url: NotRequired[str]
 
+RetrievalResultContentColumnTypeDef = TypedDict(
+    "RetrievalResultContentColumnTypeDef",
+    {
+        "columnName": NotRequired[str],
+        "columnValue": NotRequired[str],
+        "type": NotRequired[RetrievalResultContentColumnTypeType],
+    },
+)
 RetrievalResultCustomDocumentLocationTypeDef = TypedDict(
     "RetrievalResultCustomDocumentLocationTypeDef",
     {
@@ -619,6 +650,9 @@ RetrievalResultCustomDocumentLocationTypeDef = TypedDict(
     },
 )
 
+class RetrievalResultKendraDocumentLocationTypeDef(TypedDict):
+    uri: NotRequired[str]
+
 class RetrievalResultS3LocationTypeDef(TypedDict):
     uri: NotRequired[str]
 
@@ -628,6 +662,9 @@ class RetrievalResultSalesforceLocationTypeDef(TypedDict):
 class RetrievalResultSharePointLocationTypeDef(TypedDict):
     url: NotRequired[str]
 
+class RetrievalResultSqlLocationTypeDef(TypedDict):
+    query: NotRequired[str]
+
 class RetrievalResultWebLocationTypeDef(TypedDict):
     url: NotRequired[str]
 
@@ -647,6 +684,9 @@ class SpanTypeDef(TypedDict):
     end: NotRequired[int]
     start: NotRequired[int]
 
+class TextToSqlKnowledgeBaseConfigurationTypeDef(TypedDict):
+    knowledgeBaseArn: str
+
 class VectorSearchBedrockRerankingModelConfigurationTypeDef(TypedDict):
     modelArn: str
     additionalModelRequestFields: NotRequired[Mapping[str, Mapping[str, Any]]]
@@ -802,6 +842,10 @@ class FunctionInvocationInputTypeDef(TypedDict):
     function: NotRequired[str]
     parameters: NotRequired[List[FunctionParameterTypeDef]]
 
+class GenerateQueryResponseTypeDef(TypedDict):
+    queries: List[GeneratedQueryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+
 class GetAgentMemoryRequestGetAgentMemoryPaginateTypeDef(TypedDict):
     agentAliasId: str
     agentId: str
@@ -882,15 +926,26 @@ RerankQueryTypeDef = TypedDict(
         "type": Literal["TEXT"],
     },
 )
+RetrievalResultContentTypeDef = TypedDict(
+    "RetrievalResultContentTypeDef",
+    {
+        "byteContent": NotRequired[str],
+        "row": NotRequired[List[RetrievalResultContentColumnTypeDef]],
+        "text": NotRequired[str],
+        "type": NotRequired[RetrievalResultContentTypeType],
+    },
+)
 RetrievalResultLocationTypeDef = TypedDict(
     "RetrievalResultLocationTypeDef",
     {
         "type": RetrievalResultLocationTypeType,
         "confluenceLocation": NotRequired[RetrievalResultConfluenceLocationTypeDef],
         "customDocumentLocation": NotRequired[RetrievalResultCustomDocumentLocationTypeDef],
+        "kendraDocumentLocation": NotRequired[RetrievalResultKendraDocumentLocationTypeDef],
         "s3Location": NotRequired[RetrievalResultS3LocationTypeDef],
         "salesforceLocation": NotRequired[RetrievalResultSalesforceLocationTypeDef],
         "sharePointLocation": NotRequired[RetrievalResultSharePointLocationTypeDef],
+        "sqlLocation": NotRequired[RetrievalResultSqlLocationTypeDef],
         "webLocation": NotRequired[RetrievalResultWebLocationTypeDef],
     },
 )
@@ -899,6 +954,14 @@ class TextResponsePartTypeDef(TypedDict):
     span: NotRequired[SpanTypeDef]
     text: NotRequired[str]
 
+TextToSqlConfigurationTypeDef = TypedDict(
+    "TextToSqlConfigurationTypeDef",
+    {
+        "type": Literal["KNOWLEDGE_BASE"],
+        "knowledgeBaseConfiguration": NotRequired[TextToSqlKnowledgeBaseConfigurationTypeDef],
+    },
+)
+
 class ApiRequestBodyTypeDef(TypedDict):
     content: NotRequired[Dict[str, PropertyParametersTypeDef]]
 
@@ -1054,6 +1117,10 @@ class RetrievedReferenceTypeDef(TypedDict):
 class GeneratedResponsePartTypeDef(TypedDict):
     textResponsePart: NotRequired[TextResponsePartTypeDef]
 
+class TransformationConfigurationTypeDef(TypedDict):
+    mode: Literal["TEXT_TO_SQL"]
+    textToSqlConfiguration: NotRequired[TextToSqlConfigurationTypeDef]
+
 class ApiInvocationInputTypeDef(TypedDict):
     actionGroup: str
     actionInvocationType: NotRequired[ActionInvocationTypeType]
@@ -1154,6 +1221,10 @@ class CitationTypeDef(TypedDict):
     generatedResponsePart: NotRequired[GeneratedResponsePartTypeDef]
     retrievedReferences: NotRequired[List[RetrievedReferenceTypeDef]]
 
+class GenerateQueryRequestRequestTypeDef(TypedDict):
+    queryGenerationInput: QueryGenerationInputTypeDef
+    transformationConfiguration: TransformationConfigurationTypeDef
+
 class InvocationInputMemberTypeDef(TypedDict):
     apiInvocationInput: NotRequired[ApiInvocationInputTypeDef]
     functionInvocationInput: NotRequired[FunctionInvocationInputTypeDef]

types_boto3_bedrock_agent_runtime/version.py

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