Typing fix again

Author: I8dNLo

fastembed/common/model_management.py

@@ -295,7 +295,11 @@ def decompress_to_cache(cls, targz_path: str, cache_dir: str) -> str:
 
     @classmethod
     def retrieve_model_gcs(
-        cls, model_name: str, source_url: str, cache_dir: str, local_files_only: bool = False
+        cls,
+        model_name: str,
+        source_url: Optional[str],
+        cache_dir: str,
+        local_files_only: bool = False,
     ) -> Path:
         fast_model_name = f"fast-{model_name.split('/')[-1]}"
         cache_tmp_dir = Path(cache_dir) / "tmp"

fastembed/late_interaction/colbert.py

@@ -185,7 +185,7 @@ def __init__(
     def load_onnx_model(self) -> None:
         self._load_onnx_model(
             model_dir=self._model_dir,
-            model_file=self.model_description["model_file"],
+            model_file=self.model_description.model_file,
             threads=self.threads,
             providers=self.providers,
             cuda=self.cuda,

fastembed/late_interaction_multimodal/colpali.py

@@ -119,7 +119,7 @@ def list_supported_models(cls) -> list[ModelDescription]:
     def load_onnx_model(self) -> None:
         self._load_onnx_model(
             model_dir=self._model_dir,
-            model_file=self.model_description["model_file"],
+            model_file=self.model_description.model_file,
             threads=self.threads,
             providers=self.providers,
             cuda=self.cuda,
@@ -140,7 +140,7 @@ def _post_process_onnx_image_output(
             Iterable[NumpyArray]: Post-processed output as NumPy arrays.
         """
         return output.model_output.reshape(
-            output.model_output.shape[0], -1, self.model_description["dim"]
+            output.model_output.shape[0], -1, self.model_description.dim
         ).astype(np.float32)
 
     def _post_process_onnx_text_output(