Skip to content

Commit 632fad4

Browse files
committed
feat: Vector retrieval matches datasource
1 parent eebb601 commit 632fad4

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
@@ -53,7 +53,7 @@ def get_ds_embedding(session: SessionDep, current_user: CurrentUser, _ds_list, q
5353
_list.sort(key=lambda x: x['cosine_similarity'], reverse=True)
5454
# print(len(_list))
5555
SQLBotLogUtil.info(json.dumps(
56-
[{"id": ele.get("id"), "name": ele.get("name"), "cosine_similarity": ele.get("cosine_similarity")}
56+
[{"id": ele.get("id"), "name": ele.get("ds").name, "cosine_similarity": ele.get("cosine_similarity")}
5757
for ele in _list]))
5858
ds = _list[0].get('ds')
5959
return {"id": ds.id, "name": ds.name, "description": ds.description}

0 commit comments

Comments
 (0)