🤖 自动格式化代码 [skip ci]

This commit is contained in:
github-actions[bot]
2025-07-11 05:19:35 +00:00
parent 5f0a0c0e3a
commit 1bff478fcc
12 changed files with 15 additions and 13 deletions

View File

@@ -59,7 +59,7 @@ EMBEDDING_SIM_THRESHOLD = 0.99
def cosine_similarity(a, b):
# 计算余弦相似度
dot = sum(x * y for x, y in zip(a, b))
dot = sum(x * y for x, y in zip(a, b, strict=False))
norm_a = math.sqrt(sum(x * x for x in a))
norm_b = math.sqrt(sum(x * x for x in b))
if norm_a == 0 or norm_b == 0:
@@ -285,7 +285,7 @@ class EmbeddingStore:
distances = list(distances.flatten())
result = [
(self.idx2hash[str(int(idx))], float(sim))
for (idx, sim) in zip(indices, distances)
for (idx, sim) in zip(indices, distances, strict=False)
if idx in range(len(self.idx2hash))
]