diff --git a/src/dynavec/embeddings/bedrock.py b/src/dynavec/embeddings/bedrock.py index 523fdc5..d1c624f 100644 --- a/src/dynavec/embeddings/bedrock.py +++ b/src/dynavec/embeddings/bedrock.py @@ -9,7 +9,7 @@ import base64 import json from pathlib import Path -from typing import BinaryIO +from typing import Any, BinaryIO from .base import Embedder, Vector @@ -39,7 +39,7 @@ def __init__( model_id: str = "amazon.titan-embed-text-v2:0", region: str | None = None, dimension: int | None = None, - boto_session=None, + boto_session: Any | None = None, ) -> None: import boto3 # local import keeps base import cheap @@ -99,7 +99,7 @@ def __init__( self, dimension: int = 1024, region: str | None = None, - boto_session=None, + boto_session: Any | None = None, ) -> None: import boto3 @@ -140,7 +140,7 @@ def _invoke( if text is None and image_base64 is None: raise ValueError("At least one of 'text' or 'image' must be provided.") - body: dict = { + body: dict[str , Any] = { "embeddingConfig": { "outputEmbeddingLength": self.dimension, } diff --git a/src/dynavec/eval/chart.py b/src/dynavec/eval/chart.py index 8772436..32b6140 100644 --- a/src/dynavec/eval/chart.py +++ b/src/dynavec/eval/chart.py @@ -229,3 +229,5 @@ def plot_retrieval_metrics( plt.close(fig) return output_path + +