mirror of
https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git
synced 2026-11-08 01:06:59 +08:00
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
from concurrent.futures import ThreadPoolExecutor
|
|
from typing import List
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from common.core.config import settings
|
|
|
|
executor = ThreadPoolExecutor(max_workers=200)
|
|
|
|
engine = create_engine(str(settings.SQLALCHEMY_DATABASE_URI))
|
|
session_maker = sessionmaker(bind=engine)
|
|
session = session_maker()
|
|
|
|
|
|
def run_save_terminology_embeddings(ids: List[int]):
|
|
from apps.terminology.curd.terminology import save_embeddings
|
|
executor.submit(save_embeddings, session, ids)
|
|
|
|
|
|
def fill_empty_terminology_embeddings():
|
|
from apps.terminology.curd.terminology import run_fill_empty_embeddings
|
|
executor.submit(run_fill_empty_embeddings, session)
|
|
|
|
|
|
def run_save_data_training_embeddings(ids: List[int]):
|
|
from apps.data_training.curd.data_training import save_embeddings
|
|
executor.submit(save_embeddings, session, ids)
|
|
|
|
|
|
def fill_empty_data_training_embeddings():
|
|
from apps.data_training.curd.data_training import run_fill_empty_embeddings
|
|
executor.submit(run_fill_empty_embeddings, session)
|
|
|
|
|
|
def run_save_table_embeddings(ids: List[int]):
|
|
from apps.datasource.crud.table import save_table_embedding
|
|
executor.submit(save_table_embedding, session, ids)
|
|
|
|
|
|
def fill_empty_table_embeddings():
|
|
from apps.datasource.crud.table import run_fill_empty_table_embedding
|
|
executor.submit(run_fill_empty_table_embedding, session)
|