fix: review fixes

aidial_adapter_vertexai/llm/imagen_chat_completion_adapter.py

@@ -61,7 +61,7 @@ def get_image_type(image: PIL_Image.Image) -> str:
             case "PNG":
                 return "image/png"
             case _:
-                raise ValueError(f"Unknown image mode: {image.mode}")
+                raise ValueError(f"Unknown image format: {image.format}")
 
     @override
     async def chat(

aidial_adapter_vertexai/utils/env.py

@@ -1,15 +1,10 @@
 import os
-from typing import Optional
 
 
-def get_env(name: str, err_msg: Optional[str] = None) -> str:
+def get_env(name: str) -> str:
     if name in os.environ:
         val = os.environ.get(name)
         if val is not None:
             return val
 
-    raise Exception(err_msg or f"{name} env variable is not set")
-
-
-def get_env_bool(name: str, default: bool = False) -> bool:
-    return os.getenv(name, str(default)).lower() == "true"
+    raise Exception(f"{name} env variable is not set")

client/chat/sdk.py

@@ -151,12 +151,6 @@ async def create(
         model = ImageGenerationModel.from_pretrained(deployment.value)
         return cls(model)
 
-    @staticmethod
-    def save_file(data: bytes, ext: str) -> Path:
-        path: Path = SDKImagenChat.get_filename(ext)
-        path.write_bytes(data)
-        return path
-
     @staticmethod
     def get_filename(ext) -> Path:
         dir = get_project_root() / "~images"