ai-model-vault 4.6.1

DEPRECATED - renamed to `ironvault`. This crate is no longer maintained; 4.6.1 is identical to 4.6.0 apart from this notice.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

"""
Compression utilities for efficient model storage.
"""

import gzip
import lzma
import zlib
from enum import Enum
from typing import Protocol


class CompressionLevel(Enum):
    """Compression level options."""
    NONE = 0
    FAST = 1
    BALANCED = 6
    MAXIMUM = 9


class Compressor(Protocol):
    """Protocol for compression implementations."""
    
    def compress(self, data: bytes, level: int) -> bytes:
        """Compress data."""
        ...
    
    def decompress(self, data: bytes) -> bytes:
        """Decompress data."""
        ...


class GzipCompressor:
    """Gzip compression (fast, good compression ratio)."""
    
    def compress(self, data: bytes, level: int = 6) -> bytes:
        """Compress using gzip."""
        return gzip.compress(data, compresslevel=level)
    
    def decompress(self, data: bytes) -> bytes:
        """Decompress gzip data."""
        return gzip.decompress(data)


class LZMACompressor:
    """LZMA compression (slower, better compression for large models)."""
    
    def compress(self, data: bytes, level: int = 6) -> bytes:
        """Compress using LZMA."""
        return lzma.compress(data, preset=level)
    
    def decompress(self, data: bytes) -> bytes:
        """Decompress LZMA data."""
        return lzma.decompress(data)


class ZlibCompressor:
    """Zlib compression (balanced speed/ratio)."""
    
    def compress(self, data: bytes, level: int = 6) -> bytes:
        """Compress using zlib."""
        return zlib.compress(data, level=level)
    
    def decompress(self, data: bytes) -> bytes:
        """Decompress zlib data."""
        return zlib.decompress(data)


def get_compressor(algorithm: str = "gzip") -> Compressor:
    """
    Get compressor implementation.
    
    Args:
        algorithm: Compression algorithm ('gzip', 'lzma', 'zlib')
    
    Returns:
        Compressor instance
    """
    compressors = {
        "gzip": GzipCompressor(),
        "lzma": LZMACompressor(),
        "zlib": ZlibCompressor(),
    }
    
    if algorithm not in compressors:
        raise ValueError(f"Unknown compression algorithm: {algorithm}")
    
    return compressors[algorithm]