mirror of
https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git
synced 2026-11-08 17:26:55 +08:00
feat: Vector retrieval matches tables
This commit is contained in:
@@ -81,7 +81,7 @@ def save_table_embedding(session: Session, ids: List[int]):
|
||||
# table_schema.append(schema_table)
|
||||
emb = json.dumps(model.embed_query(schema_table))
|
||||
|
||||
stmt = update(CoreTable).where(and_(CoreTable.id == id)).values(embedding=emb)
|
||||
stmt = update(CoreTable).where(and_(CoreTable.id == _id)).values(embedding=emb)
|
||||
session.execute(stmt)
|
||||
session.commit()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user