mirror of
https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git
synced 2026-11-08 09:17:06 +08:00
feat: Vector retrieval matches tables
This commit is contained in:
@@ -377,9 +377,10 @@ def get_table_schema(session: SessionDep, current_user: CurrentUser, ds: CoreDat
|
||||
schema_table += ",\n".join(field_list)
|
||||
schema_table += '\n]\n'
|
||||
tables.append(schema_table)
|
||||
# do table embedding
|
||||
if embedding:
|
||||
tables = get_table_embedding(session, current_user, tables, question)
|
||||
|
||||
# do table embedding
|
||||
if embedding:
|
||||
tables = get_table_embedding(session, current_user, tables, question)
|
||||
|
||||
# todo 外键
|
||||
for s in tables:
|
||||
|
||||
Reference in New Issue
Block a user