opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-09-22 12:35:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

embedding sql

Browse Source
This commit is contained in:
robin
2023-10-17 10:57:42 +08:00
parent 8ebfe94c2e
commit ae893fefa0
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/EmbeddingController.java
View File

@ -157,10 +157,6 @@ public class EmbeddingController extends ChatController {
if (StringUtils.isBlank(vectorParam.getDatabase()) && StringUtils.isBlank(vectorParam.getSchema())) {
return;
}
DataResult<Boolean> result = tableService.checkTableVector(vectorParam);
if (result.getData()) {
return;
}
ConfigService configService = ApplicationContextUtil.getBean(ConfigService.class);
Config config = configService.find(RestAIClient.AI_SQL_SOURCE).getData();
@ -178,6 +174,12 @@ public class EmbeddingController extends ChatController {
String apiKey = keyConfig.getContent();
request.setApikey(apiKey);
vectorParam.setApiKey(apiKey);
DataResult<Boolean> result = tableService.checkTableVector(vectorParam);
if (result.getData()) {
return;
}
// check if in white list
boolean res = gatewayClientService.checkInWhite(new WhiteListRequest(apiKey, WhiteListTypeEnum.VECTOR.getCode())).getData();
if (!res) {