From 336ffcf2a11f550579b065db0801929eb99103ee Mon Sep 17 00:00:00 2001 From: junjun Date: Fri, 19 Sep 2025 14:49:04 +0800 Subject: [PATCH] feat: Vector retrieval matches datasource --- backend/apps/datasource/embedding/ds_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/apps/datasource/embedding/ds_embedding.py b/backend/apps/datasource/embedding/ds_embedding.py index 48ef07ef..afa1ec69 100644 --- a/backend/apps/datasource/embedding/ds_embedding.py +++ b/backend/apps/datasource/embedding/ds_embedding.py @@ -50,7 +50,7 @@ def get_ds_embedding(session: SessionDep, current_user: CurrentUser, _ds_list, q _list[index]['cosine_similarity'] = cosine_similarity(q_embedding, item) _list.sort(key=lambda x: x['cosine_similarity'], reverse=True) - print(json.dumps(_list)) + print(len(_list)) ds = _list[0].get('ds') return {"id": ds.id, "name": ds.name, "description": ds.description} except Exception: