Skip to content

Commit f4682b3

Browse files
committed
feat: Vector retrieval matches tables
1 parent 0b0066f commit f4682b3

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backend/apps/datasource/crud/datasource.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from sqlmodel import select
99

1010
from apps.datasource.crud.permission import get_column_permission_fields, get_row_permission_filters, is_normal_user
11-
from apps.datasource.embedding.table_embedding import get_table_embedding, calc_table_embedding
11+
from apps.datasource.embedding.table_embedding import calc_table_embedding
1212
from apps.datasource.utils.utils import aes_decrypt
1313
from apps.db.constant import DB
1414
from apps.db.db import get_tables, get_fields, exec_sql, check_connection

backend/apps/system/crud/assistant.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from sqlmodel import Session, select
99
from starlette.middleware.cors import CORSMiddleware
1010

11-
from apps.datasource.embedding.table_embedding import get_table_embedding
11+
# from apps.datasource.embedding.table_embedding import get_table_embedding
1212
from apps.datasource.models.datasource import CoreDatasource, DatasourceConf
1313
from apps.system.models.system_model import AssistantModel
1414
from apps.system.schemas.auth import CacheName, CacheNamespace

0 commit comments

Comments
 (0)