From e3c0f3f94aa814a7f7b94cd0523b4674567564d5 Mon Sep 17 00:00:00 2001 From: junjun Date: Fri, 10 Oct 2025 13:16:03 +0800 Subject: [PATCH] feat: Vector retrieval matches tables --- backend/apps/datasource/embedding/ds_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/apps/datasource/embedding/ds_embedding.py b/backend/apps/datasource/embedding/ds_embedding.py index 34ee7c7b1..49fa40208 100644 --- a/backend/apps/datasource/embedding/ds_embedding.py +++ b/backend/apps/datasource/embedding/ds_embedding.py @@ -71,7 +71,7 @@ def get_ds_embedding(session: SessionDep, current_user: CurrentUser, _ds_list, o for index in range(len(results)): item = results[index] if item: - _list[index]['cosine_similarity'] = cosine_similarity(q_embedding, item) + _list[index]['cosine_similarity'] = cosine_similarity(q_embedding, json.loads(item)) _list.sort(key=lambda x: x['cosine_similarity'], reverse=True) # print(len(_list))