Skip to content

Commit 162d480

Browse files
committed
feat: Vector retrieval matches tables
1 parent 56c0359 commit 162d480

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

backend/apps/datasource/crud/table.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,21 +42,21 @@ def update_table(session: SessionDep, item: CoreTable):
4242

4343
def run_fill_empty_table_embedding(session: Session):
4444
try:
45-
if not settings.EMBEDDING_ENABLED:
45+
if not settings.TABLE_EMBEDDING_ENABLED:
4646
return
4747

4848
SQLBotLogUtil.info('get tables')
4949
stmt = select(CoreTable.id).where(and_(CoreTable.embedding.is_(None)))
5050
results = session.execute(stmt).scalars().all()
51-
SQLBotLogUtil.info('result:' + str(len(results)))
51+
SQLBotLogUtil.info('result: ' + str(len(results)))
5252

5353
save_table_embedding(session, results)
5454
except Exception:
5555
traceback.print_exc()
5656

5757

5858
def save_table_embedding(session: Session, ids: List[int]):
59-
if not settings.EMBEDDING_ENABLED:
59+
if not settings.TABLE_EMBEDDING_ENABLED:
6060
return
6161

6262
if not ids or len(ids) == 0:
@@ -99,7 +99,7 @@ def save_table_embedding(session: Session, ids: List[int]):
9999
session.commit()
100100

101101
end_time = time.time()
102-
SQLBotLogUtil.info('table embedding finished in:' + str(end_time - start_time) + 'seconds')
102+
SQLBotLogUtil.info('table embedding finished in: ' + str(end_time - start_time) + ' seconds')
103103
except Exception:
104104
traceback.print_exc()
105105

0 commit comments

Comments
 (0)