feat: 修复生成预加载计划时反序列化失败 TencentBlueKing#2626

yaoxuwan · Sep 29, 2024 · 827c00b · 827c00b
1 parent 61f2460
commit 827c00b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ain/kotlin/com/tencent/bkrepo/job/batch/task/cache/preload/ai/milvus/MilvusVectorStore.kt b/...ain/kotlin/com/tencent/bkrepo/job/batch/task/cache/preload/ai/milvus/MilvusVectorStore.kt
@@ -108,7 +108,7 @@ class MilvusVectorStore(
         val results = ArrayList<Document>()
         val respSearch = milvusClient.search(req)
         for (vector in respSearch) {
-            val similarity = getResultSimilarity((vector[DISTANCE_FIELD_NAME] as Double).toFloat())
+            val similarity = getResultSimilarity(vector[DISTANCE_FIELD_NAME].toString().toFloat())
             if (similarity < request.similarityThreshold) {
                 continue
             }