From c1364302ae1fa5fafb6c7f4864b9af014d85f8ef Mon Sep 17 00:00:00 2001 From: junjun Date: Thu, 9 Oct 2025 14:17:38 +0800 Subject: [PATCH] feat: Vector retrieval matches tables --- backend/apps/datasource/crud/table.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/apps/datasource/crud/table.py b/backend/apps/datasource/crud/table.py index 220d5e3b..95a34414 100644 --- a/backend/apps/datasource/crud/table.py +++ b/backend/apps/datasource/crud/table.py @@ -80,9 +80,9 @@ def save_table_embedding(session: Session, ids: List[int]): schema_table += ",\n".join(field_list) schema_table += '\n]\n' # table_schema.append(schema_table) - emb = model.embed_query(schema_table) + emb = json.dumps(model.embed_query(schema_table)) - stmt = update(CoreTable).where(and_(CoreTable.id == id)).values(embedding=json.dumps(emb)) + stmt = update(CoreTable).where(and_(CoreTable.id == id)).values(embedding=emb) session.execute(stmt) session.commit()