Skip to content

Commit b625b56

Browse files
committed
refactor: print log
1 parent 879f753 commit b625b56

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

backend/apps/datasource/embedding/table_embedding.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# Author: Junjun
22
# Date: 2025/9/23
33
import json
4+
import time
45
import traceback
56

67
from apps.ai_model.embedding import EmbeddingModelCache
@@ -20,7 +21,10 @@ def get_table_embedding(session: SessionDep, current_user: CurrentUser, tables:
2021
text = [s.get('schema_table') for s in _list]
2122

2223
model = EmbeddingModelCache.get_model()
24+
start_time = time.time()
2325
results = model.embed_documents(text)
26+
end_time = time.time()
27+
SQLBotLogUtil.info(str(end_time - start_time))
2428

2529
q_embedding = model.embed_query(question)
2630
for index in range(len(results)):

0 commit comments

Comments
 (0)