Q i`dZddlmZmZddlmZmZGddeZGddZdS) z/SVS-VAMANA compression configuration utilities.)LiteralOptional) BaseModelFieldceZdZUdZdZeded<dZee ed<dZ ee ed<e ddZ ee ed <dZee ed <dZee ed <dZee ed <dS) SVSConfigaSVS-VAMANA configuration model. Attributes: algorithm: Always "svs-vamana" datatype: Vector datatype (float16, float32) compression: Compression type (LVQ4, LeanVec4x8, etc.) reduce: Reduced dimensionality (only for LeanVec) graph_max_degree: Max edges per node construction_window_size: Build-time candidates search_window_size: Query-time candidates svs-vamana algorithmNdatatype compressionz)Reduced dimensionality (only for LeanVec))default descriptionreducegraph_max_degreeconstruction_window_sizesearch_window_size)__name__ __module__ __qualname____doc__r r__annotations__r rstrr rrintrrrC:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\redisvl/utils/compression.pyrrs  (4Iw|$333"Hhsm"""!%K#%%%!E"MFHSM'+hsm***.2hsm222(, ,,,,,rrc eZdZdZdZdddddddZe dd ed ed d e e de fdZ e dde d ede ede fdZd S)CompressionAdvisoraHelper to recommend compression settings based on vector characteristics. This class provides utilities to: - Recommend optimal SVS-VAMANA configurations based on vector dimensions and priorities - Estimate memory savings from compression and dimensionality reduction Examples: >>> # Get recommendations for high-dimensional vectors >>> config = CompressionAdvisor.recommend(dims=1536, priority="balanced") >>> config.compression 'LeanVec4x8' >>> config.reduce 768 >>> # Estimate memory savings >>> savings = CompressionAdvisor.estimate_memory_savings( ... compression="LeanVec4x8", ... dims=1536, ... reduce=768 ... ) >>> savings 81.2 i )LVQ4LVQ4x4LVQ4x8LVQ8 LeanVec4x8 LeanVec8x8balancedNdimspriority)speedmemoryr)r returnc |dkrtd||tjkri|pd}|dkrtd|ddd|d zd S|d kr'td|dddt d |dzd Std|ddd|d zd S|pd}|dkrtd|ddd dS|d krtd|ddd dStd|ddd dS)aRecommend compression settings based on dimensions and priorities. Args: dims: Vector dimensionality (must be > 0) priority: Optimization priority: - "memory": Maximize memory savings - "speed": Optimize for query speed - "balanced": Balance between memory and speed datatype: Override datatype (default: float16 for high-dim, float32 for low-dim) Returns: dict: Complete SVS-VAMANA configuration including: - algorithm: "svs-vamana" - datatype: Recommended datatype - compression: Compression type - reduce: Dimensionality reduction (for LeanVec only) - graph_max_degree: Graph connectivity - construction_window_size: Build-time candidates - search_window_size: Query-time candidates Raises: ValueError: If dims <= 0 Examples: >>> # High-dimensional embeddings (e.g., OpenAI ada-002) >>> config = CompressionAdvisor.recommend(dims=1536, priority="memory") >>> config.compression 'LeanVec4x8' >>> config.reduce 768 >>> # Lower-dimensional embeddings >>> config = CompressionAdvisor.recommend(dims=384, priority="speed") >>> config.compression 'LVQ4x8' rzdims must be positive, got float16r-r @i,r')r r rrr rrr,r(float32r#)r r rrrr r%r$) ValueErrorrHIGH_DIM_THRESHOLDrmax)r*r+r base_datatypes r recommendzCompressionAdvisor.recommendFsT 199A4AABB B %8 8 8$1 M8## **%'-0 ,19')W$$ **%'-0 ,sDAI..')!**%'-0 ,19')%1 M8## **%'-0') & W$$ **%'-0') ( !**%'-0') ( rr rcd}tj||}|r|n|}||z}||z}d||z z dz}t|dS)aEstimate memory savings percentage from compression. Calculates the percentage of memory saved compared to uncompressed float32 vectors. Args: compression: Compression type (e.g., "LVQ4", "LeanVec4x8") dims: Original vector dimensionality reduce: Reduced dimensionality (for LeanVec compression) Returns: float: Memory savings percentage (0-100) Examples: >>> # LeanVec with dimensionality reduction >>> CompressionAdvisor.estimate_memory_savings( ... compression="LeanVec4x8", ... dims=1536, ... reduce=768 ... ) 81.2 >>> # LVQ without dimensionality reduction >>> CompressionAdvisor.estimate_memory_savings( ... compression="LVQ4", ... dims=384 ... ) 87.5 d)rCOMPRESSION_BITSgetround) r r*r base_bitscompression_bitseffective_dims original_sizecompressed_sizesavingss restimate_memory_savingsz*CompressionAdvisor.estimate_memory_savingssuB .>BB    $*3ty( (+;;66#=Wa   r)r)N)N)rrrrr:rB staticmethodrrrrrr=floatrKrrrrr s2 rPs55$$$$$$$$%%%%%%%%----- ---0F!F!F!F!F!F!F!F!F!F!r