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

types_boto3_bedrock_agent_runtime/type_defs.pyi

@@ -24,6 +24,7 @@ from botocore.response import StreamingBody
 from .literals import (
     ActionGroupSignatureType,
     ActionInvocationTypeType,
+    AttributeTypeType,
     ConfirmationStateType,
     CreationModeType,
     ExecutionTypeType,
@@ -42,6 +43,8 @@ from .literals import (
     PromptStateType,
     PromptTypeType,
     RequireConfirmationType,
+    RerankDocumentTypeType,
+    RerankingMetadataSelectionModeType,
     ResponseStateType,
     RetrievalResultLocationTypeType,
     RetrieveAndGenerateTypeType,
@@ -69,9 +72,12 @@ __all__ = (
     "ApiResultTypeDef",
     "AttributionTypeDef",
     "BadGatewayExceptionTypeDef",
+    "BedrockRerankingConfigurationTypeDef",
+    "BedrockRerankingModelConfigurationTypeDef",
     "BlobTypeDef",
     "ByteContentDocTypeDef",
     "ByteContentFileTypeDef",
+    "CitationEventTypeDef",
     "CitationTypeDef",
     "CodeInterpreterInvocationInputTypeDef",
     "CodeInterpreterInvocationOutputTypeDef",
@@ -85,6 +91,7 @@ __all__ = (
     "ExternalSourcesGenerationConfigurationTypeDef",
     "ExternalSourcesRetrieveAndGenerateConfigurationTypeDef",
     "FailureTraceTypeDef",
+    "FieldForRerankingTypeDef",
     "FilePartTypeDef",
     "FileSourceTypeDef",
     "FilterAttributeTypeDef",
@@ -121,6 +128,7 @@ __all__ = (
     "GuardrailContentFilterTypeDef",
     "GuardrailContentPolicyAssessmentTypeDef",
     "GuardrailCustomWordTypeDef",
+    "GuardrailEventTypeDef",
     "GuardrailManagedWordTypeDef",
     "GuardrailPiiEntityFilterTypeDef",
     "GuardrailRegexFilterTypeDef",
@@ -129,6 +137,7 @@ __all__ = (
     "GuardrailTopicTypeDef",
     "GuardrailTraceTypeDef",
     "GuardrailWordPolicyAssessmentTypeDef",
+    "ImplicitFilterConfigurationTypeDef",
     "InferenceConfigTypeDef",
     "InferenceConfigurationOutputTypeDef",
     "InferenceConfigurationTypeDef",
@@ -164,6 +173,8 @@ __all__ = (
     "KnowledgeBaseVectorSearchConfigurationTypeDef",
     "MemorySessionSummaryTypeDef",
     "MemoryTypeDef",
+    "MetadataAttributeSchemaTypeDef",
+    "MetadataConfigurationForRerankingTypeDef",
     "MetadataTypeDef",
     "ModelInvocationInputTypeDef",
     "ObservationTypeDef",
@@ -195,6 +206,18 @@ __all__ = (
     "RawResponseTypeDef",
     "RepromptResponseTypeDef",
     "RequestBodyTypeDef",
+    "RerankDocumentOutputTypeDef",
+    "RerankDocumentTypeDef",
+    "RerankDocumentUnionTypeDef",
+    "RerankQueryTypeDef",
+    "RerankRequestRequestTypeDef",
+    "RerankRequestRerankPaginateTypeDef",
+    "RerankResponseTypeDef",
+    "RerankResultTypeDef",
+    "RerankSourceTypeDef",
+    "RerankTextDocumentTypeDef",
+    "RerankingConfigurationTypeDef",
+    "RerankingMetadataSelectiveModeConfigurationTypeDef",
     "ResourceNotFoundExceptionTypeDef",
     "ResponseMetadataTypeDef",
     "ResponseStreamTypeDef",
@@ -202,6 +225,7 @@ __all__ = (
     "RetrievalFilterTypeDef",
     "RetrievalResultConfluenceLocationTypeDef",
     "RetrievalResultContentTypeDef",
+    "RetrievalResultCustomDocumentLocationTypeDef",
     "RetrievalResultLocationTypeDef",
     "RetrievalResultS3LocationTypeDef",
     "RetrievalResultSalesforceLocationTypeDef",
@@ -209,10 +233,14 @@ __all__ = (
     "RetrievalResultWebLocationTypeDef",
     "RetrieveAndGenerateConfigurationTypeDef",
     "RetrieveAndGenerateInputTypeDef",
+    "RetrieveAndGenerateOutputEventTypeDef",
     "RetrieveAndGenerateOutputTypeDef",
     "RetrieveAndGenerateRequestRequestTypeDef",
     "RetrieveAndGenerateResponseTypeDef",
     "RetrieveAndGenerateSessionConfigurationTypeDef",
+    "RetrieveAndGenerateStreamRequestRequestTypeDef",
+    "RetrieveAndGenerateStreamResponseOutputTypeDef",
+    "RetrieveAndGenerateStreamResponseTypeDef",
     "RetrieveRequestRequestTypeDef",
     "RetrieveRequestRetrievePaginateTypeDef",
     "RetrieveResponseTypeDef",
@@ -233,6 +261,9 @@ __all__ = (
     "TraceTypeDef",
     "UsageTypeDef",
     "ValidationExceptionTypeDef",
+    "VectorSearchBedrockRerankingConfigurationTypeDef",
+    "VectorSearchBedrockRerankingModelConfigurationTypeDef",
+    "VectorSearchRerankingConfigurationTypeDef",
 )
 
 class S3IdentifierTypeDef(TypedDict):
@@ -280,6 +311,10 @@ class BadGatewayExceptionTypeDef(TypedDict):
     message: NotRequired[str]
     resourceName: NotRequired[str]
 
+class BedrockRerankingModelConfigurationTypeDef(TypedDict):
+    modelArn: str
+    additionalModelRequestFields: NotRequired[Mapping[str, Mapping[str, Any]]]
+
 BlobTypeDef = Union[str, bytes, IO[Any], StreamingBody]
 
 class CodeInterpreterInvocationInputTypeDef(TypedDict):
@@ -321,6 +356,9 @@ class FailureTraceTypeDef(TypedDict):
     failureReason: NotRequired[str]
     traceId: NotRequired[str]
 
+class FieldForRerankingTypeDef(TypedDict):
+    fieldName: str
+
 OutputFileTypeDef = TypedDict(
     "OutputFileTypeDef",
     {
@@ -427,6 +465,9 @@ class GuardrailCustomWordTypeDef(TypedDict):
     action: NotRequired[Literal["BLOCKED"]]
     match: NotRequired[str]
 
+class GuardrailEventTypeDef(TypedDict):
+    action: NotRequired[GuadrailActionType]
+
 GuardrailManagedWordTypeDef = TypedDict(
     "GuardrailManagedWordTypeDef",
     {
@@ -458,6 +499,14 @@ GuardrailTopicTypeDef = TypedDict(
         "type": NotRequired[Literal["DENY"]],
     },
 )
+MetadataAttributeSchemaTypeDef = TypedDict(
+    "MetadataAttributeSchemaTypeDef",
+    {
+        "description": str,
+        "key": str,
+        "type": AttributeTypeType,
+    },
+)
 
 class TextInferenceConfigTypeDef(TypedDict):
     maxTokens: NotRequired[int]
@@ -532,9 +581,19 @@ class PreProcessingParsedResponseTypeDef(TypedDict):
     isValid: NotRequired[bool]
     rationale: NotRequired[str]
 
+class RerankTextDocumentTypeDef(TypedDict):
+    text: NotRequired[str]
+
 class RetrievalResultConfluenceLocationTypeDef(TypedDict):
     url: NotRequired[str]
 
+RetrievalResultCustomDocumentLocationTypeDef = TypedDict(
+    "RetrievalResultCustomDocumentLocationTypeDef",
+    {
+        "id": NotRequired[str],
+    },
+)
+
 class RetrievalResultS3LocationTypeDef(TypedDict):
     uri: NotRequired[str]
 
@@ -550,6 +609,9 @@ class RetrievalResultWebLocationTypeDef(TypedDict):
 class RetrieveAndGenerateInputTypeDef(TypedDict):
     text: str
 
+class RetrieveAndGenerateOutputEventTypeDef(TypedDict):
+    text: str
+
 class RetrieveAndGenerateOutputTypeDef(TypedDict):
     text: str
 
@@ -560,6 +622,10 @@ class SpanTypeDef(TypedDict):
     end: NotRequired[int]
     start: NotRequired[int]
 
+class VectorSearchBedrockRerankingModelConfigurationTypeDef(TypedDict):
+    modelArn: str
+    additionalModelRequestFields: NotRequired[Mapping[str, Mapping[str, Any]]]
+
 class APISchemaTypeDef(TypedDict):
     payload: NotRequired[str]
     s3: NotRequired[S3IdentifierTypeDef]
@@ -586,6 +652,10 @@ class FunctionResultTypeDef(TypedDict):
     responseBody: NotRequired[Mapping[str, ContentBodyTypeDef]]
     responseState: NotRequired[ResponseStateType]
 
+class BedrockRerankingConfigurationTypeDef(TypedDict):
+    modelConfiguration: BedrockRerankingModelConfigurationTypeDef
+    numberOfResults: NotRequired[int]
+
 class ByteContentDocTypeDef(TypedDict):
     contentType: str
     data: BlobTypeDef
@@ -599,6 +669,10 @@ class CustomOrchestrationTraceTypeDef(TypedDict):
     event: NotRequired[CustomOrchestrationTraceEventTypeDef]
     traceId: NotRequired[str]
 
+class RerankingMetadataSelectiveModeConfigurationTypeDef(TypedDict):
+    fieldsToExclude: NotRequired[Sequence[FieldForRerankingTypeDef]]
+    fieldsToInclude: NotRequired[Sequence[FieldForRerankingTypeDef]]
+
 class FilePartTypeDef(TypedDict):
     files: NotRequired[List[OutputFileTypeDef]]
 
@@ -698,6 +772,10 @@ class GuardrailSensitiveInformationPolicyAssessmentTypeDef(TypedDict):
 class GuardrailTopicPolicyAssessmentTypeDef(TypedDict):
     topics: NotRequired[List[GuardrailTopicTypeDef]]
 
+class ImplicitFilterConfigurationTypeDef(TypedDict):
+    metadataAttributes: Sequence[MetadataAttributeSchemaTypeDef]
+    modelArn: str
+
 class InferenceConfigTypeDef(TypedDict):
     textInferenceConfig: NotRequired[TextInferenceConfigTypeDef]
 
@@ -729,11 +807,35 @@ class MemoryTypeDef(TypedDict):
 class MetadataTypeDef(TypedDict):
     usage: NotRequired[UsageTypeDef]
 
+RerankDocumentOutputTypeDef = TypedDict(
+    "RerankDocumentOutputTypeDef",
+    {
+        "type": RerankDocumentTypeType,
+        "jsonDocument": NotRequired[Dict[str, Any]],
+        "textDocument": NotRequired[RerankTextDocumentTypeDef],
+    },
+)
+RerankDocumentTypeDef = TypedDict(
+    "RerankDocumentTypeDef",
+    {
+        "type": RerankDocumentTypeType,
+        "jsonDocument": NotRequired[Mapping[str, Any]],
+        "textDocument": NotRequired[RerankTextDocumentTypeDef],
+    },
+)
+RerankQueryTypeDef = TypedDict(
+    "RerankQueryTypeDef",
+    {
+        "textQuery": RerankTextDocumentTypeDef,
+        "type": Literal["TEXT"],
+    },
+)
 RetrievalResultLocationTypeDef = TypedDict(
     "RetrievalResultLocationTypeDef",
     {
         "type": RetrievalResultLocationTypeType,
         "confluenceLocation": NotRequired[RetrievalResultConfluenceLocationTypeDef],
+        "customDocumentLocation": NotRequired[RetrievalResultCustomDocumentLocationTypeDef],
         "s3Location": NotRequired[RetrievalResultS3LocationTypeDef],
         "salesforceLocation": NotRequired[RetrievalResultSalesforceLocationTypeDef],
         "sharePointLocation": NotRequired[RetrievalResultSharePointLocationTypeDef],
@@ -762,6 +864,14 @@ class InvocationResultMemberTypeDef(TypedDict):
     apiResult: NotRequired[ApiResultTypeDef]
     functionResult: NotRequired[FunctionResultTypeDef]
 
+RerankingConfigurationTypeDef = TypedDict(
+    "RerankingConfigurationTypeDef",
+    {
+        "bedrockRerankingConfiguration": BedrockRerankingConfigurationTypeDef,
+        "type": Literal["BEDROCK_RERANKING_MODEL"],
+    },
+)
+
 class ExternalSourceTypeDef(TypedDict):
     sourceType: ExternalSourceTypeType
     byteContent: NotRequired[ByteContentDocTypeDef]
@@ -772,22 +882,9 @@ class FileSourceTypeDef(TypedDict):
     byteContent: NotRequired[ByteContentFileTypeDef]
     s3Location: NotRequired[S3ObjectFileTypeDef]
 
-KnowledgeBaseVectorSearchConfigurationPaginatorTypeDef = TypedDict(
-    "KnowledgeBaseVectorSearchConfigurationPaginatorTypeDef",
-    {
-        "filter": NotRequired[RetrievalFilterPaginatorTypeDef],
-        "numberOfResults": NotRequired[int],
-        "overrideSearchType": NotRequired[SearchTypeType],
-    },
-)
-KnowledgeBaseVectorSearchConfigurationTypeDef = TypedDict(
-    "KnowledgeBaseVectorSearchConfigurationTypeDef",
-    {
-        "filter": NotRequired[RetrievalFilterTypeDef],
-        "numberOfResults": NotRequired[int],
-        "overrideSearchType": NotRequired[SearchTypeType],
-    },
-)
+class MetadataConfigurationForRerankingTypeDef(TypedDict):
+    selectionMode: RerankingMetadataSelectionModeType
+    selectiveModeConfiguration: NotRequired[RerankingMetadataSelectiveModeConfigurationTypeDef]
 
 class InvokeFlowRequestRequestTypeDef(TypedDict):
     flowAliasIdentifier: str
@@ -873,6 +970,13 @@ class PreProcessingModelInvocationOutputTypeDef(TypedDict):
     rawResponse: NotRequired[RawResponseTypeDef]
     traceId: NotRequired[str]
 
+class RerankResultTypeDef(TypedDict):
+    index: int
+    relevanceScore: float
+    document: NotRequired[RerankDocumentOutputTypeDef]
+
+RerankDocumentUnionTypeDef = Union[RerankDocumentTypeDef, RerankDocumentOutputTypeDef]
+
 class KnowledgeBaseRetrievalResultTypeDef(TypedDict):
     content: RetrievalResultContentTypeDef
     location: NotRequired[RetrievalResultLocationTypeDef]
@@ -907,11 +1011,10 @@ class InputFileTypeDef(TypedDict):
     source: FileSourceTypeDef
     useCase: FileUseCaseType
 
-class KnowledgeBaseRetrievalConfigurationPaginatorTypeDef(TypedDict):
-    vectorSearchConfiguration: KnowledgeBaseVectorSearchConfigurationPaginatorTypeDef
-
-class KnowledgeBaseRetrievalConfigurationTypeDef(TypedDict):
-    vectorSearchConfiguration: KnowledgeBaseVectorSearchConfigurationTypeDef
+class VectorSearchBedrockRerankingConfigurationTypeDef(TypedDict):
+    modelConfiguration: VectorSearchBedrockRerankingModelConfigurationTypeDef
+    metadataConfiguration: NotRequired[MetadataConfigurationForRerankingTypeDef]
+    numberOfRerankedResults: NotRequired[int]
 
 class FlowTraceTypeDef(TypedDict):
     conditionNodeResultTrace: NotRequired[FlowTraceConditionNodeResultEventTypeDef]
@@ -959,7 +1062,21 @@ class PreProcessingTraceTypeDef(TypedDict):
     modelInvocationInput: NotRequired[ModelInvocationInputTypeDef]
     modelInvocationOutput: NotRequired[PreProcessingModelInvocationOutputTypeDef]
 
+class RerankResponseTypeDef(TypedDict):
+    results: List[RerankResultTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    nextToken: NotRequired[str]
+
+RerankSourceTypeDef = TypedDict(
+    "RerankSourceTypeDef",
+    {
+        "inlineDocumentSource": RerankDocumentUnionTypeDef,
+        "type": Literal["INLINE"],
+    },
+)
+
 class RetrieveResponseTypeDef(TypedDict):
+    guardrailAction: GuadrailActionType
     retrievalResults: List[KnowledgeBaseRetrievalResultTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
     nextToken: NotRequired[str]
@@ -982,33 +1099,15 @@ class InlineSessionStateTypeDef(TypedDict):
     returnControlInvocationResults: NotRequired[Sequence[InvocationResultMemberTypeDef]]
     sessionAttributes: NotRequired[Mapping[str, str]]
 
-class RetrieveRequestRetrievePaginateTypeDef(TypedDict):
-    knowledgeBaseId: str
-    retrievalQuery: KnowledgeBaseQueryTypeDef
-    retrievalConfiguration: NotRequired[KnowledgeBaseRetrievalConfigurationPaginatorTypeDef]
-    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
-
-class KnowledgeBaseConfigurationTypeDef(TypedDict):
-    knowledgeBaseId: str
-    retrievalConfiguration: KnowledgeBaseRetrievalConfigurationTypeDef
-
-class KnowledgeBaseRetrieveAndGenerateConfigurationTypeDef(TypedDict):
-    knowledgeBaseId: str
-    modelArn: str
-    generationConfiguration: NotRequired[GenerationConfigurationTypeDef]
-    orchestrationConfiguration: NotRequired[OrchestrationConfigurationTypeDef]
-    retrievalConfiguration: NotRequired[KnowledgeBaseRetrievalConfigurationTypeDef]
-
-class KnowledgeBaseTypeDef(TypedDict):
-    description: str
-    knowledgeBaseId: str
-    retrievalConfiguration: NotRequired[KnowledgeBaseRetrievalConfigurationTypeDef]
-
-class RetrieveRequestRequestTypeDef(TypedDict):
-    knowledgeBaseId: str
-    retrievalQuery: KnowledgeBaseQueryTypeDef
-    nextToken: NotRequired[str]
-    retrievalConfiguration: NotRequired[KnowledgeBaseRetrievalConfigurationTypeDef]
+VectorSearchRerankingConfigurationTypeDef = TypedDict(
+    "VectorSearchRerankingConfigurationTypeDef",
+    {
+        "type": Literal["BEDROCK_RERANKING_MODEL"],
+        "bedrockRerankingConfiguration": NotRequired[
+            VectorSearchBedrockRerankingConfigurationTypeDef
+        ],
+    },
+)
 
 class FlowTraceEventTypeDef(TypedDict):
     trace: FlowTraceTypeDef
@@ -1017,6 +1116,18 @@ class OptimizePromptResponseTypeDef(TypedDict):
     optimizedPrompt: "EventStream[OptimizedPromptStreamTypeDef]"
     ResponseMetadata: ResponseMetadataTypeDef
 
+class RerankRequestRequestTypeDef(TypedDict):
+    queries: Sequence[RerankQueryTypeDef]
+    rerankingConfiguration: RerankingConfigurationTypeDef
+    sources: Sequence[RerankSourceTypeDef]
+    nextToken: NotRequired[str]
+
+class RerankRequestRerankPaginateTypeDef(TypedDict):
+    queries: Sequence[RerankQueryTypeDef]
+    rerankingConfiguration: RerankingConfigurationTypeDef
+    sources: Sequence[RerankSourceTypeDef]
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
 ObservationTypeDef = TypedDict(
     "ObservationTypeDef",
     {
@@ -1033,6 +1144,9 @@ ObservationTypeDef = TypedDict(
 class AttributionTypeDef(TypedDict):
     citations: NotRequired[List[CitationTypeDef]]
 
+class CitationEventTypeDef(TypedDict):
+    citation: NotRequired[CitationTypeDef]
+
 class RetrieveAndGenerateResponseTypeDef(TypedDict):
     citations: List[CitationTypeDef]
     guardrailAction: GuadrailActionType
@@ -1048,41 +1162,26 @@ class ReturnControlPayloadTypeDef(TypedDict):
     invocationId: NotRequired[str]
     invocationInputs: NotRequired[List[InvocationInputMemberTypeDef]]
 
-class SessionStateTypeDef(TypedDict):
-    files: NotRequired[Sequence[InputFileTypeDef]]
-    invocationId: NotRequired[str]
-    knowledgeBaseConfigurations: NotRequired[Sequence[KnowledgeBaseConfigurationTypeDef]]
-    promptSessionAttributes: NotRequired[Mapping[str, str]]
-    returnControlInvocationResults: NotRequired[Sequence[InvocationResultMemberTypeDef]]
-    sessionAttributes: NotRequired[Mapping[str, str]]
-
-RetrieveAndGenerateConfigurationTypeDef = TypedDict(
-    "RetrieveAndGenerateConfigurationTypeDef",
+KnowledgeBaseVectorSearchConfigurationPaginatorTypeDef = TypedDict(
+    "KnowledgeBaseVectorSearchConfigurationPaginatorTypeDef",
     {
-        "type": RetrieveAndGenerateTypeType,
-        "externalSourcesConfiguration": NotRequired[
-            ExternalSourcesRetrieveAndGenerateConfigurationTypeDef
-        ],
-        "knowledgeBaseConfiguration": NotRequired[
-            KnowledgeBaseRetrieveAndGenerateConfigurationTypeDef
-        ],
+        "filter": NotRequired[RetrievalFilterPaginatorTypeDef],
+        "implicitFilterConfiguration": NotRequired[ImplicitFilterConfigurationTypeDef],
+        "numberOfResults": NotRequired[int],
+        "overrideSearchType": NotRequired[SearchTypeType],
+        "rerankingConfiguration": NotRequired[VectorSearchRerankingConfigurationTypeDef],
+    },
+)
+KnowledgeBaseVectorSearchConfigurationTypeDef = TypedDict(
+    "KnowledgeBaseVectorSearchConfigurationTypeDef",
+    {
+        "filter": NotRequired[RetrievalFilterTypeDef],
+        "implicitFilterConfiguration": NotRequired[ImplicitFilterConfigurationTypeDef],
+        "numberOfResults": NotRequired[int],
+        "overrideSearchType": NotRequired[SearchTypeType],
+        "rerankingConfiguration": NotRequired[VectorSearchRerankingConfigurationTypeDef],
     },
 )
-
-class InvokeInlineAgentRequestRequestTypeDef(TypedDict):
-    foundationModel: str
-    instruction: str
-    sessionId: str
-    actionGroups: NotRequired[Sequence[AgentActionGroupTypeDef]]
-    customerEncryptionKeyArn: NotRequired[str]
-    enableTrace: NotRequired[bool]
-    endSession: NotRequired[bool]
-    guardrailConfiguration: NotRequired[GuardrailConfigurationWithArnTypeDef]
-    idleSessionTTLInSeconds: NotRequired[int]
-    inlineSessionState: NotRequired[InlineSessionStateTypeDef]
-    inputText: NotRequired[str]
-    knowledgeBases: NotRequired[Sequence[KnowledgeBaseTypeDef]]
-    promptOverrideConfiguration: NotRequired[PromptOverrideConfigurationTypeDef]
 
 class FlowResponseStreamTypeDef(TypedDict):
     accessDeniedException: NotRequired[AccessDeniedExceptionTypeDef]
@@ -1120,26 +1219,25 @@ PayloadPartTypeDef = TypedDict(
     },
 )
 
-class InvokeAgentRequestRequestTypeDef(TypedDict):
-    agentAliasId: str
-    agentId: str
-    sessionId: str
-    enableTrace: NotRequired[bool]
-    endSession: NotRequired[bool]
-    inputText: NotRequired[str]
-    memoryId: NotRequired[str]
-    sessionState: NotRequired[SessionStateTypeDef]
-    streamingConfigurations: NotRequired[StreamingConfigurationsTypeDef]
+class RetrieveAndGenerateStreamResponseOutputTypeDef(TypedDict):
+    accessDeniedException: NotRequired[AccessDeniedExceptionTypeDef]
+    badGatewayException: NotRequired[BadGatewayExceptionTypeDef]
+    citation: NotRequired[CitationEventTypeDef]
+    conflictException: NotRequired[ConflictExceptionTypeDef]
+    dependencyFailedException: NotRequired[DependencyFailedExceptionTypeDef]
+    guardrail: NotRequired[GuardrailEventTypeDef]
+    internalServerException: NotRequired[InternalServerExceptionTypeDef]
+    output: NotRequired[RetrieveAndGenerateOutputEventTypeDef]
+    resourceNotFoundException: NotRequired[ResourceNotFoundExceptionTypeDef]
+    serviceQuotaExceededException: NotRequired[ServiceQuotaExceededExceptionTypeDef]
+    throttlingException: NotRequired[ThrottlingExceptionTypeDef]
+    validationException: NotRequired[ValidationExceptionTypeDef]
 
-RetrieveAndGenerateRequestRequestTypeDef = TypedDict(
-    "RetrieveAndGenerateRequestRequestTypeDef",
-    {
-        "input": RetrieveAndGenerateInputTypeDef,
-        "retrieveAndGenerateConfiguration": NotRequired[RetrieveAndGenerateConfigurationTypeDef],
-        "sessionConfiguration": NotRequired[RetrieveAndGenerateSessionConfigurationTypeDef],
-        "sessionId": NotRequired[str],
-    },
-)
+class KnowledgeBaseRetrievalConfigurationPaginatorTypeDef(TypedDict):
+    vectorSearchConfiguration: KnowledgeBaseVectorSearchConfigurationPaginatorTypeDef
+
+class KnowledgeBaseRetrievalConfigurationTypeDef(TypedDict):
+    vectorSearchConfiguration: KnowledgeBaseVectorSearchConfigurationTypeDef
 
 class InvokeFlowResponseTypeDef(TypedDict):
     responseStream: "EventStream[FlowResponseStreamTypeDef]"
@@ -1153,6 +1251,41 @@ class TraceTypeDef(TypedDict):
     postProcessingTrace: NotRequired[PostProcessingTraceTypeDef]
     preProcessingTrace: NotRequired[PreProcessingTraceTypeDef]
 
+class RetrieveAndGenerateStreamResponseTypeDef(TypedDict):
+    sessionId: str
+    stream: "EventStream[RetrieveAndGenerateStreamResponseOutputTypeDef]"
+    ResponseMetadata: ResponseMetadataTypeDef
+
+class RetrieveRequestRetrievePaginateTypeDef(TypedDict):
+    knowledgeBaseId: str
+    retrievalQuery: KnowledgeBaseQueryTypeDef
+    guardrailConfiguration: NotRequired[GuardrailConfigurationTypeDef]
+    retrievalConfiguration: NotRequired[KnowledgeBaseRetrievalConfigurationPaginatorTypeDef]
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+class KnowledgeBaseConfigurationTypeDef(TypedDict):
+    knowledgeBaseId: str
+    retrievalConfiguration: KnowledgeBaseRetrievalConfigurationTypeDef
+
+class KnowledgeBaseRetrieveAndGenerateConfigurationTypeDef(TypedDict):
+    knowledgeBaseId: str
+    modelArn: str
+    generationConfiguration: NotRequired[GenerationConfigurationTypeDef]
+    orchestrationConfiguration: NotRequired[OrchestrationConfigurationTypeDef]
+    retrievalConfiguration: NotRequired[KnowledgeBaseRetrievalConfigurationTypeDef]
+
+class KnowledgeBaseTypeDef(TypedDict):
+    description: str
+    knowledgeBaseId: str
+    retrievalConfiguration: NotRequired[KnowledgeBaseRetrievalConfigurationTypeDef]
+
+class RetrieveRequestRequestTypeDef(TypedDict):
+    knowledgeBaseId: str
+    retrievalQuery: KnowledgeBaseQueryTypeDef
+    guardrailConfiguration: NotRequired[GuardrailConfigurationTypeDef]
+    nextToken: NotRequired[str]
+    retrievalConfiguration: NotRequired[KnowledgeBaseRetrievalConfigurationTypeDef]
+
 class InlineAgentTracePartTypeDef(TypedDict):
     sessionId: NotRequired[str]
     trace: NotRequired[TraceTypeDef]
@@ -1164,6 +1297,42 @@ class TracePartTypeDef(TypedDict):
     sessionId: NotRequired[str]
     trace: NotRequired[TraceTypeDef]
 
+class SessionStateTypeDef(TypedDict):
+    files: NotRequired[Sequence[InputFileTypeDef]]
+    invocationId: NotRequired[str]
+    knowledgeBaseConfigurations: NotRequired[Sequence[KnowledgeBaseConfigurationTypeDef]]
+    promptSessionAttributes: NotRequired[Mapping[str, str]]
+    returnControlInvocationResults: NotRequired[Sequence[InvocationResultMemberTypeDef]]
+    sessionAttributes: NotRequired[Mapping[str, str]]
+
+RetrieveAndGenerateConfigurationTypeDef = TypedDict(
+    "RetrieveAndGenerateConfigurationTypeDef",
+    {
+        "type": RetrieveAndGenerateTypeType,
+        "externalSourcesConfiguration": NotRequired[
+            ExternalSourcesRetrieveAndGenerateConfigurationTypeDef
+        ],
+        "knowledgeBaseConfiguration": NotRequired[
+            KnowledgeBaseRetrieveAndGenerateConfigurationTypeDef
+        ],
+    },
+)
+
+class InvokeInlineAgentRequestRequestTypeDef(TypedDict):
+    foundationModel: str
+    instruction: str
+    sessionId: str
+    actionGroups: NotRequired[Sequence[AgentActionGroupTypeDef]]
+    customerEncryptionKeyArn: NotRequired[str]
+    enableTrace: NotRequired[bool]
+    endSession: NotRequired[bool]
+    guardrailConfiguration: NotRequired[GuardrailConfigurationWithArnTypeDef]
+    idleSessionTTLInSeconds: NotRequired[int]
+    inlineSessionState: NotRequired[InlineSessionStateTypeDef]
+    inputText: NotRequired[str]
+    knowledgeBases: NotRequired[Sequence[KnowledgeBaseTypeDef]]
+    promptOverrideConfiguration: NotRequired[PromptOverrideConfigurationTypeDef]
+
 class InlineAgentResponseStreamTypeDef(TypedDict):
     accessDeniedException: NotRequired[AccessDeniedExceptionTypeDef]
     badGatewayException: NotRequired[BadGatewayExceptionTypeDef]
@@ -1194,6 +1363,36 @@ class ResponseStreamTypeDef(TypedDict):
     trace: NotRequired[TracePartTypeDef]
     validationException: NotRequired[ValidationExceptionTypeDef]
 
+class InvokeAgentRequestRequestTypeDef(TypedDict):
+    agentAliasId: str
+    agentId: str
+    sessionId: str
+    enableTrace: NotRequired[bool]
+    endSession: NotRequired[bool]
+    inputText: NotRequired[str]
+    memoryId: NotRequired[str]
+    sessionState: NotRequired[SessionStateTypeDef]
+    streamingConfigurations: NotRequired[StreamingConfigurationsTypeDef]
+
+RetrieveAndGenerateRequestRequestTypeDef = TypedDict(
+    "RetrieveAndGenerateRequestRequestTypeDef",
+    {
+        "input": RetrieveAndGenerateInputTypeDef,
+        "retrieveAndGenerateConfiguration": NotRequired[RetrieveAndGenerateConfigurationTypeDef],
+        "sessionConfiguration": NotRequired[RetrieveAndGenerateSessionConfigurationTypeDef],
+        "sessionId": NotRequired[str],
+    },
+)
+RetrieveAndGenerateStreamRequestRequestTypeDef = TypedDict(
+    "RetrieveAndGenerateStreamRequestRequestTypeDef",
+    {
+        "input": RetrieveAndGenerateInputTypeDef,
+        "retrieveAndGenerateConfiguration": NotRequired[RetrieveAndGenerateConfigurationTypeDef],
+        "sessionConfiguration": NotRequired[RetrieveAndGenerateSessionConfigurationTypeDef],
+        "sessionId": NotRequired[str],
+    },
+)
+
 class InvokeInlineAgentResponseTypeDef(TypedDict):
     completion: "EventStream[InlineAgentResponseStreamTypeDef]"
     contentType: str

types_boto3_bedrock_agent_runtime/version.py

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

types_boto3_bedrock_runtime/__main__.py

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