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 13:00:33 +08:00
parent 090df35b48
commit cc076244f1
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/datasource/crud/table.py
+4 -2
View File
@@ -4,6 +4,7 @@ import traceback
from typing import List
from sqlalchemy import and_, select, update
from sqlalchemy.orm.session import Session
from apps.ai_model.embedding import EmbeddingModelCache
from common.core.config import settings
@@ -30,10 +31,11 @@ def update_table(session: SessionDep, item: CoreTable):
session.commit()
def run_fill_empty_table_embedding(session: SessionDep):
def run_fill_empty_table_embedding(session: Session):
if not settings.EMBEDDING_ENABLED:
return
SQLBotLogUtil.info('get tables')
stmt = select(CoreTable.id).where(and_(CoreTable.embedding.is_(None)))
results = session.execute(stmt).scalars().all()
SQLBotLogUtil.info(json.dumps(results))
@@ -41,7 +43,7 @@ def run_fill_empty_table_embedding(session: SessionDep):
save_table_embedding(session, results)
def save_table_embedding(session: SessionDep, ids: List[int]):
def save_table_embedding(session: Session, ids: List[int]):
if not settings.EMBEDDING_ENABLED:
return