refactor: update OPEN_ROUTER_API_KEY to use pydantic settings

Author: onur-askui

src/askui/models/openrouter/handler.py

@@ -1,4 +1,3 @@
-import os
 from typing import Type
 
 from openai import OpenAI
@@ -10,17 +9,15 @@
 from askui.utils.image_utils import ImageSource
 
 from .prompts import PROMPT_QA
-from .settings import OpenRouterSettings
+from .settings import OpenRouterAuthSettings, OpenRouterSettings
 
 
 class OpenRouterGetModel(GetModel):
     def __init__(self, settings: OpenRouterSettings):
         self._settings = settings
 
-        _open_router_key = os.getenv("OPEN_ROUTER_API_KEY")
-        if _open_router_key is None:
-            error_msg = "OPEN_ROUTER_API_KEY is not set"
-            raise ValueError(error_msg)
+        _auth_settings = OpenRouterAuthSettings()
+        _open_router_key = _auth_settings.api_key.get_secret_value()
 
         self._client = OpenAI(
             api_key=_open_router_key,

src/askui/models/openrouter/settings.py

@@ -1,5 +1,13 @@
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, SecretStr
+from pydantic_settings import BaseSettings
 
 
 class OpenRouterSettings(BaseModel):
     model: str = Field(..., description="OpenRouter model name")
+
+
+class OpenRouterAuthSettings(BaseSettings):
+    api_key: SecretStr = Field(
+        min_length=1,
+        validation_alias="OPEN_ROUTER_API_KEY",
+    )