INNER CODE UNIT · Python

model_store_from_env

operatorai/modelstore · modelstore/utils/cli.py:95

def model_store_from_env() -> ModelStore:
    """Builds a modelstore instance from environment variables."""
    if "MODEL_STORE_STORAGE" not in os.environ:
        failure("❌  No value for MODEL_STORE_STORAGE set in os.environ")
        sys.exit(1)

    storage_name = os.environ["MODEL_STORE_STORAGE"]
    if storage_name not in STORAGE_TYPES:
        failure(f"❌  Unknown storage name in MODEL_STORE_STORAGE: {storage_name}")
        sys.exit(1)

    storage_type = STORAGE_TYPES[storage_name]
    assert_environ_exists(storage_name, storage_type.BUILD_FROM_ENVIRONMENT)
    info(f"Using model store with storage={storage_name}")
    return MODEL_STORE_TYPES[storage_name]()

View source record →