fix: linting in model.py

caretech-owl · Dec 1, 2024 · 12b9cf5 · 12b9cf5
1 parent 507da9e
commit 12b9cf5
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/gerd/models/model.py b/gerd/models/model.py
@@ -143,14 +143,20 @@ def validate_field(cls, data: Any) -> Any:  # noqa: ANN401
                 and f"{env_name}_URL" in os.environ
                 and f"{env_name}_TYPE" in os.environ
             ):
+                # Check if the endpoint type is valid
+                # Since MyPy does not correctly infer the type of `os.environ`
+                # we need to use `type: ignore`
+                if os.environ[f"{env_name}_TYPE"] not in ["llama.cpp", "openai"]:
+                    msg = f"Unknown endpoint type: {os.environ[f'{env_name}_TYPE']}"
+                    raise ValueError(msg)
                 setattr(
                     data,
                     field,
                     ModelEndpoint(
                         url=os.environ[f"{env_name}_URL"],
-                        type=os.environ[f"{env_name}_TYPE"],
+                        type=os.environ[f"{env_name}_TYPE"],  # type: ignore[arg-type]
                         key=(
-                            SecretStr(os.environ.get(f"{env_name}_KEY"))
+                            SecretStr(os.environ.get(f"{env_name}_KEY") or "")
                             if f"{env_name}_KEY" in os.environ
                             else None
                         ),