We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent eebb601 commit 632fad4Copy full SHA for 632fad4
backend/apps/datasource/embedding/ds_embedding.py
@@ -53,7 +53,7 @@ def get_ds_embedding(session: SessionDep, current_user: CurrentUser, _ds_list, q
53
_list.sort(key=lambda x: x['cosine_similarity'], reverse=True)
54
# print(len(_list))
55
SQLBotLogUtil.info(json.dumps(
56
- [{"id": ele.get("id"), "name": ele.get("name"), "cosine_similarity": ele.get("cosine_similarity")}
+ [{"id": ele.get("id"), "name": ele.get("ds").name, "cosine_similarity": ele.get("cosine_similarity")}
57
for ele in _list]))
58
ds = _list[0].get('ds')
59
return {"id": ds.id, "name": ds.name, "description": ds.description}
0 commit comments