feat: Vector retrieval matches tables

2026-11-08 01:06:59 +08:00 · 2025-10-10 13:16:54 +08:00
parent b1dda8b454
commit 2022b7ac34
1 changed files with 1 additions and 1 deletions
@@ -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))