more changes

Author: korikuzma

src/anyvlm/config.py

@@ -23,6 +23,7 @@ class Settings(BaseSettings):
 
     env: ServiceEnvironment = ServiceEnvironment.LOCAL
     anyvar_uri: str = "http://localhost:8000"
+    storage_uri: str = "postgresql://postgres@localhost:5432/anyvlm"
 
 
 @cache

src/anyvlm/main.py

@@ -3,7 +3,6 @@
 import logging
 from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
-from os import environ
 from urllib.parse import urlparse
 
 from anyvar.anyvar import create_storage, create_translator
@@ -20,7 +19,6 @@
     ServiceOrganization,
     ServiceType,
 )
-from anyvlm.storage import DEFAULT_STORAGE_URI
 from anyvlm.storage.base_storage import Storage
 from anyvlm.utils.types import (
     EndpointTag,
@@ -68,7 +66,7 @@ def create_anyvlm_storage(uri: str | None = None) -> Storage:
     :return: AnyVLM storage instance
     """
     if not uri:
-        uri = environ.get("ANYVLM_STORAGE_URI", DEFAULT_STORAGE_URI)
+        uri = get_config().storage_uri
 
     parsed_uri = urlparse(uri)
     if parsed_uri.scheme == "postgresql":

src/anyvlm/storage/__init__.py

@@ -2,6 +2,4 @@
 
 from .base_storage import Storage
 
-DEFAULT_STORAGE_URI = "postgresql://postgres@localhost:5432/anyvlm"
-
-__all__ = ["DEFAULT_STORAGE_URI", "Storage"]
+__all__ = ["Storage"]