feat: Vector retrieval matches datasource

2026-11-08 01:06:59 +08:00 · 2025-09-19 15:19:03 +08:00
parent eebb601a6d
commit 632fad4dec
1 changed files with 1 additions and 1 deletions
@@ -53,7 +53,7 @@ def get_ds_embedding(session: SessionDep, current_user: CurrentUser, _ds_list, q
            _list.sort(key=lambda x: x['cosine_similarity'], reverse=True)
            # print(len(_list))
            SQLBotLogUtil.info(json.dumps(
-                [{"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")}
                 for ele in _list]))
            ds = _list[0].get('ds')
            return {"id": ds.id, "name": ds.name, "description": ds.description}