gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-04-24 20:04:42 +08:00
parent f6129857a1
commit 02e9055f9d
1 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
lightrag/utils.py
View File

@@ -576,21 +576,27 @@ async def get_best_cached_response(
# Check if cache data is valid # Check if cache data is valid
if cache_data["embedding"] is None: if cache_data["embedding"] is None:
continue continue
try: try:
# Safely convert cached embedding # Safely convert cached embedding
cached_quantized = np.frombuffer( cached_quantized = np.frombuffer(
bytes.fromhex(cache_data["embedding"]), dtype=np.uint8 bytes.fromhex(cache_data["embedding"]), dtype=np.uint8
).reshape(cache_data["embedding_shape"]) ).reshape(cache_data["embedding_shape"])
# Ensure min_val and max_val are valid float values # Ensure min_val and max_val are valid float values
embedding_min = cache_data.get("embedding_min") embedding_min = cache_data.get("embedding_min")
embedding_max = cache_data.get("embedding_max") embedding_max = cache_data.get("embedding_max")
if embedding_min is None or embedding_max is None or embedding_min >= embedding_max: if (
logger.warning(f"Invalid embedding min/max values: min={embedding_min}, max={embedding_max}") embedding_min is None
or embedding_max is None
or embedding_min >= embedding_max
):
logger.warning(
f"Invalid embedding min/max values: min={embedding_min}, max={embedding_max}"
)
continue continue
cached_embedding = dequantize_embedding( cached_embedding = dequantize_embedding(
cached_quantized, cached_quantized,
embedding_min, embedding_min,
@@ -701,7 +707,7 @@ def dequantize_embedding(
if min_val == max_val: if min_val == max_val:
# handle constant vector # handle constant vector
return np.full_like(quantized, min_val, dtype=np.float32) return np.full_like(quantized, min_val, dtype=np.float32)
scale = (max_val - min_val) / (2**bits - 1) scale = (max_val - min_val) / (2**bits - 1)
return (quantized * scale + min_val).astype(np.float32) return (quantized * scale + min_val).astype(np.float32)