Skip to content

Commit 82fbf7f

Browse files
committed
feat: Vector retrieval matches tables
1 parent 865ac8f commit 82fbf7f

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

backend/apps/datasource/crud/datasource.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -377,9 +377,10 @@ def get_table_schema(session: SessionDep, current_user: CurrentUser, ds: CoreDat
377377
schema_table += ",\n".join(field_list)
378378
schema_table += '\n]\n'
379379
tables.append(schema_table)
380-
# do table embedding
381-
if embedding:
382-
tables = get_table_embedding(session, current_user, tables, question)
380+
381+
# do table embedding
382+
if embedding:
383+
tables = get_table_embedding(session, current_user, tables, question)
383384

384385
# todo 外键
385386
for s in tables:

0 commit comments

Comments
 (0)