Skip to content

Commit 2022b7a

Browse files
committed
feat: Vector retrieval matches tables
1 parent b1dda8b commit 2022b7a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backend/apps/datasource/embedding/ds_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def get_ds_embedding(session: SessionDep, current_user: CurrentUser, _ds_list, o
7171
for index in range(len(results)):
7272
item = results[index]
7373
if item:
74-
_list[index]['cosine_similarity'] = cosine_similarity(q_embedding, item)
74+
_list[index]['cosine_similarity'] = cosine_similarity(q_embedding, json.loads(item))
7575

7676
_list.sort(key=lambda x: x['cosine_similarity'], reverse=True)
7777
# print(len(_list))

0 commit comments

Comments
 (0)