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

types_boto3_bedrock_agent/type_defs.pyi

@@ -16,7 +16,9 @@ Copyright 2024 Vlad Emelianov
 
 import sys
 from datetime import datetime
-from typing import Any, Dict, List, Mapping, Sequence, Union
+from typing import IO, Any, Dict, List, Mapping, Sequence, Union
+
+from botocore.response import StreamingBody
 
 from .literals import (
     ActionGroupSignatureType,
@@ -25,11 +27,14 @@ from .literals import (
     AgentStatusType,
     ChunkingStrategyType,
     ConfluenceAuthTypeType,
+    ContentDataSourceTypeType,
     ConversationRoleType,
     CreationModeType,
+    CustomSourceTypeType,
     DataDeletionPolicyType,
     DataSourceStatusType,
     DataSourceTypeType,
+    DocumentStatusType,
     EmbeddingDataTypeType,
     FlowConnectionTypeType,
     FlowNodeIODataTypeType,
@@ -39,9 +44,12 @@ from .literals import (
     FlowValidationTypeType,
     IngestionJobSortByAttributeType,
     IngestionJobStatusType,
+    InlineContentTypeType,
     KnowledgeBaseStateType,
     KnowledgeBaseStatusType,
     KnowledgeBaseStorageTypeType,
+    MetadataSourceTypeType,
+    MetadataValueTypeType,
     OrchestrationTypeType,
     PromptStateType,
     PromptTemplateTypeType,
@@ -77,6 +85,8 @@ __all__ = (
     "AssociateAgentKnowledgeBaseResponseTypeDef",
     "BedrockEmbeddingModelConfigurationTypeDef",
     "BedrockFoundationModelConfigurationTypeDef",
+    "BlobTypeDef",
+    "ByteContentDocTypeDef",
     "ChatPromptTemplateConfigurationOutputTypeDef",
     "ChatPromptTemplateConfigurationTypeDef",
     "ChatPromptTemplateConfigurationUnionTypeDef",
@@ -117,7 +127,10 @@ __all__ = (
     "CreatePromptResponseTypeDef",
     "CreatePromptVersionRequestRequestTypeDef",
     "CreatePromptVersionResponseTypeDef",
+    "CustomContentTypeDef",
+    "CustomDocumentIdentifierTypeDef",
     "CustomOrchestrationTypeDef",
+    "CustomS3LocationTypeDef",
     "CustomTransformationConfigurationOutputTypeDef",
     "CustomTransformationConfigurationTypeDef",
     "CustomTransformationConfigurationUnionTypeDef",
@@ -141,11 +154,16 @@ __all__ = (
     "DeleteFlowResponseTypeDef",
     "DeleteFlowVersionRequestRequestTypeDef",
     "DeleteFlowVersionResponseTypeDef",
+    "DeleteKnowledgeBaseDocumentsRequestRequestTypeDef",
+    "DeleteKnowledgeBaseDocumentsResponseTypeDef",
     "DeleteKnowledgeBaseRequestRequestTypeDef",
     "DeleteKnowledgeBaseResponseTypeDef",
     "DeletePromptRequestRequestTypeDef",
     "DeletePromptResponseTypeDef",
     "DisassociateAgentKnowledgeBaseRequestRequestTypeDef",
+    "DocumentContentTypeDef",
+    "DocumentIdentifierTypeDef",
+    "DocumentMetadataTypeDef",
     "DuplicateConditionExpressionFlowValidationDetailsTypeDef",
     "DuplicateConnectionsFlowValidationDetailsTypeDef",
     "EmbeddingModelConfigurationTypeDef",
@@ -196,6 +214,8 @@ __all__ = (
     "GetFlowVersionResponseTypeDef",
     "GetIngestionJobRequestRequestTypeDef",
     "GetIngestionJobResponseTypeDef",
+    "GetKnowledgeBaseDocumentsRequestRequestTypeDef",
+    "GetKnowledgeBaseDocumentsResponseTypeDef",
     "GetKnowledgeBaseRequestRequestTypeDef",
     "GetKnowledgeBaseResponseTypeDef",
     "GetPromptRequestRequestTypeDef",
@@ -209,13 +229,18 @@ __all__ = (
     "InferenceConfigurationOutputTypeDef",
     "InferenceConfigurationTypeDef",
     "InferenceConfigurationUnionTypeDef",
+    "IngestKnowledgeBaseDocumentsRequestRequestTypeDef",
+    "IngestKnowledgeBaseDocumentsResponseTypeDef",
     "IngestionJobFilterTypeDef",
     "IngestionJobSortByTypeDef",
     "IngestionJobStatisticsTypeDef",
     "IngestionJobSummaryTypeDef",
     "IngestionJobTypeDef",
+    "InlineContentTypeDef",
     "IntermediateStorageTypeDef",
     "KnowledgeBaseConfigurationTypeDef",
+    "KnowledgeBaseDocumentDetailTypeDef",
+    "KnowledgeBaseDocumentTypeDef",
     "KnowledgeBaseFlowNodeConfigurationTypeDef",
     "KnowledgeBaseSummaryTypeDef",
     "KnowledgeBaseTypeDef",
@@ -251,6 +276,9 @@ __all__ = (
     "ListIngestionJobsRequestListIngestionJobsPaginateTypeDef",
     "ListIngestionJobsRequestRequestTypeDef",
     "ListIngestionJobsResponseTypeDef",
+    "ListKnowledgeBaseDocumentsRequestListKnowledgeBaseDocumentsPaginateTypeDef",
+    "ListKnowledgeBaseDocumentsRequestRequestTypeDef",
+    "ListKnowledgeBaseDocumentsResponseTypeDef",
     "ListKnowledgeBasesRequestListKnowledgeBasesPaginateTypeDef",
     "ListKnowledgeBasesRequestRequestTypeDef",
     "ListKnowledgeBasesResponseTypeDef",
@@ -266,6 +294,8 @@ __all__ = (
     "MessageOutputTypeDef",
     "MessageTypeDef",
     "MessageUnionTypeDef",
+    "MetadataAttributeTypeDef",
+    "MetadataAttributeValueTypeDef",
     "MismatchedNodeInputTypeFlowValidationDetailsTypeDef",
     "MismatchedNodeOutputTypeFlowValidationDetailsTypeDef",
     "MissingConnectionConfigurationFlowValidationDetailsTypeDef",
@@ -335,6 +365,7 @@ __all__ = (
     "RetrievalFlowNodeConfigurationTypeDef",
     "RetrievalFlowNodeS3ConfigurationTypeDef",
     "RetrievalFlowNodeServiceConfigurationTypeDef",
+    "S3ContentTypeDef",
     "S3DataSourceConfigurationOutputTypeDef",
     "S3DataSourceConfigurationTypeDef",
     "S3DataSourceConfigurationUnionTypeDef",
@@ -370,6 +401,7 @@ __all__ = (
     "StorageFlowNodeServiceConfigurationTypeDef",
     "SystemContentBlockTypeDef",
     "TagResourceRequestRequestTypeDef",
+    "TextContentDocTypeDef",
     "TextPromptTemplateConfigurationOutputTypeDef",
     "TextPromptTemplateConfigurationTypeDef",
     "TextPromptTemplateConfigurationUnionTypeDef",
@@ -508,6 +540,8 @@ class BedrockEmbeddingModelConfigurationTypeDef(TypedDict):
 class ParsingPromptTypeDef(TypedDict):
     parsingPromptText: str
 
+BlobTypeDef = Union[str, bytes, IO[Any], StreamingBody]
+
 class PromptInputVariableTypeDef(TypedDict):
     name: NotRequired[str]
 
@@ -557,6 +591,17 @@ class CreatePromptVersionRequestRequestTypeDef(TypedDict):
     description: NotRequired[str]
     tags: NotRequired[Mapping[str, str]]
 
+CustomDocumentIdentifierTypeDef = TypedDict(
+    "CustomDocumentIdentifierTypeDef",
+    {
+        "id": str,
+    },
+)
+
+class CustomS3LocationTypeDef(TypedDict):
+    uri: str
+    bucketOwnerAccountId: NotRequired[str]
+
 OrchestrationExecutorTypeDef = TypedDict(
     "OrchestrationExecutorTypeDef",
     {
@@ -628,6 +673,9 @@ class DisassociateAgentKnowledgeBaseRequestRequestTypeDef(TypedDict):
     agentVersion: str
     knowledgeBaseId: str
 
+class S3LocationTypeDef(TypedDict):
+    uri: str
+
 class DuplicateConditionExpressionFlowValidationDetailsTypeDef(TypedDict):
     expression: str
     node: str
@@ -872,8 +920,8 @@ class IngestionJobStatisticsTypeDef(TypedDict):
     numberOfModifiedDocumentsIndexed: NotRequired[int]
     numberOfNewDocumentsIndexed: NotRequired[int]
 
-class S3LocationTypeDef(TypedDict):
-    uri: str
+class TextContentDocTypeDef(TypedDict):
+    data: str
 
 class KnowledgeBaseSummaryTypeDef(TypedDict):
     knowledgeBaseId: str
@@ -932,6 +980,12 @@ class ListFlowsRequestRequestTypeDef(TypedDict):
     maxResults: NotRequired[int]
     nextToken: NotRequired[str]
 
+class ListKnowledgeBaseDocumentsRequestRequestTypeDef(TypedDict):
+    dataSourceId: str
+    knowledgeBaseId: str
+    maxResults: NotRequired[int]
+    nextToken: NotRequired[str]
+
 class ListKnowledgeBasesRequestRequestTypeDef(TypedDict):
     maxResults: NotRequired[int]
     nextToken: NotRequired[str]
@@ -957,6 +1011,17 @@ PromptSummaryTypeDef = TypedDict(
 class ListTagsForResourceRequestRequestTypeDef(TypedDict):
     resourceArn: str
 
+MetadataAttributeValueTypeDef = TypedDict(
+    "MetadataAttributeValueTypeDef",
+    {
+        "type": MetadataValueTypeType,
+        "booleanValue": NotRequired[bool],
+        "numberValue": NotRequired[float],
+        "stringListValue": NotRequired[Sequence[str]],
+        "stringValue": NotRequired[str],
+    },
+)
+
 class MongoDbAtlasFieldMappingTypeDef(TypedDict):
     metadataField: str
     textField: str
@@ -1262,6 +1327,10 @@ class BedrockFoundationModelConfigurationTypeDef(TypedDict):
     modelArn: str
     parsingPrompt: NotRequired[ParsingPromptTypeDef]
 
+class ByteContentDocTypeDef(TypedDict):
+    data: BlobTypeDef
+    mimeType: str
+
 class TextPromptTemplateConfigurationOutputTypeDef(TypedDict):
     text: str
     inputVariables: NotRequired[List[PromptInputVariableTypeDef]]
@@ -1364,6 +1433,17 @@ class ListDataSourcesResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
     nextToken: NotRequired[str]
 
+class DocumentIdentifierTypeDef(TypedDict):
+    dataSourceType: ContentDataSourceTypeType
+    custom: NotRequired[CustomDocumentIdentifierTypeDef]
+    s3: NotRequired[S3LocationTypeDef]
+
+class IntermediateStorageTypeDef(TypedDict):
+    s3Location: S3LocationTypeDef
+
+class S3ContentTypeDef(TypedDict):
+    s3Location: S3LocationTypeDef
+
 class FlowConnectionConfigurationTypeDef(TypedDict):
     conditional: NotRequired[FlowConditionalConnectionConfigurationTypeDef]
     data: NotRequired[FlowDataConnectionConfigurationTypeDef]
@@ -1484,9 +1564,6 @@ class IngestionJobTypeDef(TypedDict):
     failureReasons: NotRequired[List[str]]
     statistics: NotRequired[IngestionJobStatisticsTypeDef]
 
-class IntermediateStorageTypeDef(TypedDict):
-    s3Location: S3LocationTypeDef
-
 class ListKnowledgeBasesResponseTypeDef(TypedDict):
     knowledgeBaseSummaries: List[KnowledgeBaseSummaryTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -1535,6 +1612,11 @@ class ListIngestionJobsRequestListIngestionJobsPaginateTypeDef(TypedDict):
     sortBy: NotRequired[IngestionJobSortByTypeDef]
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
+class ListKnowledgeBaseDocumentsRequestListKnowledgeBaseDocumentsPaginateTypeDef(TypedDict):
+    dataSourceId: str
+    knowledgeBaseId: str
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
 class ListKnowledgeBasesRequestListKnowledgeBasesPaginateTypeDef(TypedDict):
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
@@ -1547,6 +1629,10 @@ class ListPromptsResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
     nextToken: NotRequired[str]
 
+class MetadataAttributeTypeDef(TypedDict):
+    key: str
+    value: MetadataAttributeValueTypeDef
+
 class MongoDbAtlasConfigurationTypeDef(TypedDict):
     collectionName: str
     credentialsSecretArn: str
@@ -1692,6 +1778,14 @@ class ParsingConfigurationTypeDef(TypedDict):
     parsingStrategy: Literal["BEDROCK_FOUNDATION_MODEL"]
     bedrockFoundationModelConfiguration: NotRequired[BedrockFoundationModelConfigurationTypeDef]
 
+InlineContentTypeDef = TypedDict(
+    "InlineContentTypeDef",
+    {
+        "type": InlineContentTypeType,
+        "byteContent": NotRequired[ByteContentDocTypeDef],
+        "textContent": NotRequired[TextContentDocTypeDef],
+    },
+)
 TextPromptTemplateConfigurationUnionTypeDef = Union[
     TextPromptTemplateConfigurationTypeDef, TextPromptTemplateConfigurationOutputTypeDef
 ]
@@ -1705,6 +1799,25 @@ class ListFlowAliasesResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
     nextToken: NotRequired[str]
 
+class DeleteKnowledgeBaseDocumentsRequestRequestTypeDef(TypedDict):
+    dataSourceId: str
+    documentIdentifiers: Sequence[DocumentIdentifierTypeDef]
+    knowledgeBaseId: str
+    clientToken: NotRequired[str]
+
+class GetKnowledgeBaseDocumentsRequestRequestTypeDef(TypedDict):
+    dataSourceId: str
+    documentIdentifiers: Sequence[DocumentIdentifierTypeDef]
+    knowledgeBaseId: str
+
+class KnowledgeBaseDocumentDetailTypeDef(TypedDict):
+    dataSourceId: str
+    identifier: DocumentIdentifierTypeDef
+    knowledgeBaseId: str
+    status: DocumentStatusType
+    statusReason: NotRequired[str]
+    updatedAt: NotRequired[datetime]
+
 FlowConnectionTypeDef = TypedDict(
     "FlowConnectionTypeDef",
     {
@@ -1769,6 +1882,14 @@ class StopIngestionJobResponseTypeDef(TypedDict):
     ingestionJob: IngestionJobTypeDef
     ResponseMetadata: ResponseMetadataTypeDef
 
+DocumentMetadataTypeDef = TypedDict(
+    "DocumentMetadataTypeDef",
+    {
+        "type": MetadataSourceTypeType,
+        "inlineAttributes": NotRequired[Sequence[MetadataAttributeTypeDef]],
+        "s3Location": NotRequired[CustomS3LocationTypeDef],
+    },
+)
 CrawlFilterConfigurationOutputTypeDef = TypedDict(
     "CrawlFilterConfigurationOutputTypeDef",
     {
@@ -1843,6 +1964,29 @@ KnowledgeBaseConfigurationTypeDef = TypedDict(
     },
 )
 
+class CustomContentTypeDef(TypedDict):
+    customDocumentIdentifier: CustomDocumentIdentifierTypeDef
+    sourceType: CustomSourceTypeType
+    inlineContent: NotRequired[InlineContentTypeDef]
+    s3Location: NotRequired[CustomS3LocationTypeDef]
+
+class DeleteKnowledgeBaseDocumentsResponseTypeDef(TypedDict):
+    documentDetails: List[KnowledgeBaseDocumentDetailTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+
+class GetKnowledgeBaseDocumentsResponseTypeDef(TypedDict):
+    documentDetails: List[KnowledgeBaseDocumentDetailTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+
+class IngestKnowledgeBaseDocumentsResponseTypeDef(TypedDict):
+    documentDetails: List[KnowledgeBaseDocumentDetailTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+
+class ListKnowledgeBaseDocumentsResponseTypeDef(TypedDict):
+    documentDetails: List[KnowledgeBaseDocumentDetailTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    nextToken: NotRequired[str]
+
 class ValidateFlowDefinitionResponseTypeDef(TypedDict):
     validations: List[FlowValidationTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -1986,6 +2130,11 @@ class UpdateKnowledgeBaseRequestRequestTypeDef(TypedDict):
     storageConfiguration: StorageConfigurationTypeDef
     description: NotRequired[str]
 
+class DocumentContentTypeDef(TypedDict):
+    dataSourceType: ContentDataSourceTypeType
+    custom: NotRequired[CustomContentTypeDef]
+    s3: NotRequired[S3ContentTypeDef]
+
 class CreateAgentActionGroupResponseTypeDef(TypedDict):
     agentActionGroup: AgentActionGroupTypeDef
     ResponseMetadata: ResponseMetadataTypeDef
@@ -2099,6 +2248,10 @@ class UpdateKnowledgeBaseResponseTypeDef(TypedDict):
     knowledgeBase: KnowledgeBaseTypeDef
     ResponseMetadata: ResponseMetadataTypeDef
 
+class KnowledgeBaseDocumentTypeDef(TypedDict):
+    content: DocumentContentTypeDef
+    metadata: NotRequired[DocumentMetadataTypeDef]
+
 class CreateAgentRequestRequestTypeDef(TypedDict):
     agentName: str
     agentResourceRoleArn: NotRequired[str]
@@ -2160,6 +2313,12 @@ class VectorIngestionConfigurationTypeDef(TypedDict):
     customTransformationConfiguration: NotRequired[CustomTransformationConfigurationUnionTypeDef]
     parsingConfiguration: NotRequired[ParsingConfigurationTypeDef]
 
+class IngestKnowledgeBaseDocumentsRequestRequestTypeDef(TypedDict):
+    dataSourceId: str
+    documents: Sequence[KnowledgeBaseDocumentTypeDef]
+    knowledgeBaseId: str
+    clientToken: NotRequired[str]
+
 class DataSourceTypeDef(TypedDict):
     createdAt: datetime
     dataSourceConfiguration: DataSourceConfigurationOutputTypeDef

types_boto3_bedrock_agent/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_agent_runtime/__init__.py

@@ -9,6 +9,7 @@ Usage::
         AgentsforBedrockRuntimeClient,
         Client,
         GetAgentMemoryPaginator,
+        RerankPaginator,
         RetrievePaginator,
     )
 
@@ -16,6 +17,7 @@ Usage::
     client: AgentsforBedrockRuntimeClient = session.client("bedrock-agent-runtime")
 
     get_agent_memory_paginator: GetAgentMemoryPaginator = client.get_paginator("get_agent_memory")
+    rerank_paginator: RerankPaginator = client.get_paginator("rerank")
     retrieve_paginator: RetrievePaginator = client.get_paginator("retrieve")
     ```
 
@@ -23,7 +25,7 @@ Copyright 2024 Vlad Emelianov
 """
 
 from .client import AgentsforBedrockRuntimeClient
-from .paginator import GetAgentMemoryPaginator, RetrievePaginator
+from .paginator import GetAgentMemoryPaginator, RerankPaginator, RetrievePaginator
 
 Client = AgentsforBedrockRuntimeClient
 
@@ -32,5 +34,6 @@ __all__ = (
     "AgentsforBedrockRuntimeClient",
     "Client",
     "GetAgentMemoryPaginator",
+    "RerankPaginator",
     "RetrievePaginator",
 )

types_boto3_bedrock_agent_runtime/__init__.pyi

@@ -9,6 +9,7 @@ Usage::
         AgentsforBedrockRuntimeClient,
         Client,
         GetAgentMemoryPaginator,
+        RerankPaginator,
         RetrievePaginator,
     )
 
@@ -16,6 +17,7 @@ Usage::
     client: AgentsforBedrockRuntimeClient = session.client("bedrock-agent-runtime")
 
     get_agent_memory_paginator: GetAgentMemoryPaginator = client.get_paginator("get_agent_memory")
+    rerank_paginator: RerankPaginator = client.get_paginator("rerank")
     retrieve_paginator: RetrievePaginator = client.get_paginator("retrieve")
     ```
 
@@ -23,7 +25,7 @@ Copyright 2024 Vlad Emelianov
 """
 
 from .client import AgentsforBedrockRuntimeClient
-from .paginator import GetAgentMemoryPaginator, RetrievePaginator
+from .paginator import GetAgentMemoryPaginator, RerankPaginator, RetrievePaginator
 
 Client = AgentsforBedrockRuntimeClient
 
@@ -31,5 +33,6 @@ __all__ = (
     "AgentsforBedrockRuntimeClient",
     "Client",
     "GetAgentMemoryPaginator",
+    "RerankPaginator",
     "RetrievePaginator",
 )

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.71\n"
-        "Version:         1.35.71\n"
+        "Type annotations for boto3 AgentsforBedrockRuntime 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_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.71")
+    print("1.35.72")
 
 
 def main() -> None:

types_boto3_bedrock_agent_runtime/client.py

@@ -21,7 +21,7 @@ from typing import Any, Dict, Mapping, Type, overload
 
 from botocore.client import BaseClient, ClientMeta
 
-from .paginator import GetAgentMemoryPaginator, RetrievePaginator
+from .paginator import GetAgentMemoryPaginator, RerankPaginator, RetrievePaginator
 from .type_defs import (
     DeleteAgentMemoryRequestRequestTypeDef,
     GetAgentMemoryRequestRequestTypeDef,
@@ -34,8 +34,12 @@ from .type_defs import (
     InvokeInlineAgentResponseTypeDef,
     OptimizePromptRequestRequestTypeDef,
     OptimizePromptResponseTypeDef,
+    RerankRequestRequestTypeDef,
+    RerankResponseTypeDef,
     RetrieveAndGenerateRequestRequestTypeDef,
     RetrieveAndGenerateResponseTypeDef,
+    RetrieveAndGenerateStreamRequestRequestTypeDef,
+    RetrieveAndGenerateStreamResponseTypeDef,
     RetrieveRequestRequestTypeDef,
     RetrieveResponseTypeDef,
 )
@@ -174,6 +178,14 @@ class AgentsforBedrockRuntimeClient(BaseClient):
         [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#optimize_prompt)
         """
 
+    def rerank(self, **kwargs: Unpack[RerankRequestRequestTypeDef]) -> RerankResponseTypeDef:
+        """
+        Reranks the relevance of sources based on queries.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent-runtime/client/rerank.html)
+        [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#rerank)
+        """
+
     def retrieve(self, **kwargs: Unpack[RetrieveRequestRequestTypeDef]) -> RetrieveResponseTypeDef:
         """
         Queries a knowledge base and retrieves information from it.
@@ -195,6 +207,17 @@ class AgentsforBedrockRuntimeClient(BaseClient):
         [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#retrieve_and_generate)
         """
 
+    def retrieve_and_generate_stream(
+        self, **kwargs: Unpack[RetrieveAndGenerateStreamRequestRequestTypeDef]
+    ) -> RetrieveAndGenerateStreamResponseTypeDef:
+        """
+        Queries a knowledge base and generates responses based on the retrieved
+        results, with output in streaming format.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent-runtime/client/retrieve_and_generate_stream.html)
+        [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#retrieve_and_generate_stream)
+        """
+
     @overload
     def get_paginator(self, operation_name: Literal["get_agent_memory"]) -> GetAgentMemoryPaginator:
         """
@@ -204,6 +227,15 @@ class AgentsforBedrockRuntimeClient(BaseClient):
         [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#get_paginator)
         """
 
+    @overload
+    def get_paginator(self, operation_name: Literal["rerank"]) -> RerankPaginator:
+        """
+        Create a paginator for an operation.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent-runtime/client/get_paginator.html)
+        [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#get_paginator)
+        """
+
     @overload
     def get_paginator(self, operation_name: Literal["retrieve"]) -> RetrievePaginator:
         """

types_boto3_bedrock_agent_runtime/client.pyi

@@ -21,7 +21,7 @@ from typing import Any, Dict, Mapping, Type, overload
 
 from botocore.client import BaseClient, ClientMeta
 
-from .paginator import GetAgentMemoryPaginator, RetrievePaginator
+from .paginator import GetAgentMemoryPaginator, RerankPaginator, RetrievePaginator
 from .type_defs import (
     DeleteAgentMemoryRequestRequestTypeDef,
     GetAgentMemoryRequestRequestTypeDef,
@@ -34,8 +34,12 @@ from .type_defs import (
     InvokeInlineAgentResponseTypeDef,
     OptimizePromptRequestRequestTypeDef,
     OptimizePromptResponseTypeDef,
+    RerankRequestRequestTypeDef,
+    RerankResponseTypeDef,
     RetrieveAndGenerateRequestRequestTypeDef,
     RetrieveAndGenerateResponseTypeDef,
+    RetrieveAndGenerateStreamRequestRequestTypeDef,
+    RetrieveAndGenerateStreamResponseTypeDef,
     RetrieveRequestRequestTypeDef,
     RetrieveResponseTypeDef,
 )
@@ -170,6 +174,14 @@ class AgentsforBedrockRuntimeClient(BaseClient):
         [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#optimize_prompt)
         """
 
+    def rerank(self, **kwargs: Unpack[RerankRequestRequestTypeDef]) -> RerankResponseTypeDef:
+        """
+        Reranks the relevance of sources based on queries.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent-runtime/client/rerank.html)
+        [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#rerank)
+        """
+
     def retrieve(self, **kwargs: Unpack[RetrieveRequestRequestTypeDef]) -> RetrieveResponseTypeDef:
         """
         Queries a knowledge base and retrieves information from it.
@@ -191,6 +203,17 @@ class AgentsforBedrockRuntimeClient(BaseClient):
         [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#retrieve_and_generate)
         """
 
+    def retrieve_and_generate_stream(
+        self, **kwargs: Unpack[RetrieveAndGenerateStreamRequestRequestTypeDef]
+    ) -> RetrieveAndGenerateStreamResponseTypeDef:
+        """
+        Queries a knowledge base and generates responses based on the retrieved
+        results, with output in streaming format.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent-runtime/client/retrieve_and_generate_stream.html)
+        [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#retrieve_and_generate_stream)
+        """
+
     @overload
     def get_paginator(self, operation_name: Literal["get_agent_memory"]) -> GetAgentMemoryPaginator:
         """
@@ -200,6 +223,15 @@ class AgentsforBedrockRuntimeClient(BaseClient):
         [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#get_paginator)
         """
 
+    @overload
+    def get_paginator(self, operation_name: Literal["rerank"]) -> RerankPaginator:
+        """
+        Create a paginator for an operation.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent-runtime/client/get_paginator.html)
+        [Show types-boto3-full documentation](https://youtype.github.io/types_boto3_docs/types_boto3_bedrock_agent_runtime/client/#get_paginator)
+        """
+
     @overload
     def get_paginator(self, operation_name: Literal["retrieve"]) -> RetrievePaginator:
         """