added pydantic settings, config loading from .env or env var

Author: gittb

.env.example

@@ -0,0 +1,2 @@
+INFERADMIN_HF_TOKEN='your_hf_token'
+INFERADMIN_MODEL_STORAGE_PATH='/your/path/to/store/models'

pyproject.toml

@@ -12,6 +12,7 @@ dependencies = [
     "aiodocker>=0.24.0",
     "fastapi>=0.115.12",
     "pydantic>=2.11.2",
+    "pydantic-settings>=2.8.1",
     "uvicorn>=0.34.0",
 ]
 

src/inferadmin/config/loader.py

@@ -0,0 +1,11 @@
+from .models import InferAdminConfig
+
+class Config:
+    async def load(self):
+        self.config = InferAdminConfig()
+        print(self.get_config())
+
+    def get_config(self) -> InferAdminConfig:
+        return self.config
+    
+config_manager = Config()

src/inferadmin/config/models.py

@@ -0,0 +1,8 @@
+# from pydantic import BaseModel
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+class InferAdminConfig(BaseSettings):
+    model_config = SettingsConfigDict(env_file='.env', env_prefix='INFERADMIN_', extra='ignore')
+
+    model_storage_path: str
+    hf_token: str

src/inferadmin/lifespan.py

@@ -0,0 +1,8 @@
+from contextlib import asynccontextmanager
+from fastapi import FastAPI
+from inferadmin.config.loader import config_manager
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    await config_manager.load()
+    yield

src/inferadmin/main.py

@@ -1,10 +1,12 @@
 from fastapi import FastAPI
 from inferadmin.routes import router
 from .openapi_tags import tags_metadata
+from .lifespan import lifespan
 
 app = FastAPI(
     title="InferAdmin",
     description="A lightweight management interface for local LLM infrastructure.",
-    tags_metadata=tags_metadata
+    tags_metadata=tags_metadata,
+    lifespan=lifespan
 )
 app.include_router(router=router)