klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Vector retrieval matches tables

Browse Source
This commit is contained in:
junjun
2025-10-09 14:17:38 +08:00
parent c54902e15c
commit c1364302ae
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/datasource/crud/table.py
+2 -2
View File
@@ -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()