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

embedding query

Browse Source
This commit is contained in:
robin
2023-10-17 13:05:46 +08:00
parent f792ec99b4
commit b466a50a9d
3 changed files with 23 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/ChatController.java
View File

@ -542,20 +542,27 @@ public class ChatController {
TableSchemaRequest tableSchemaRequest = new TableSchemaRequest(); TableSchemaRequest tableSchemaRequest = new TableSchemaRequest();
tableSchemaRequest.setSchemaVector(contentVector); tableSchemaRequest.setSchemaVector(contentVector);
tableSchemaRequest.setDataSourceId(queryRequest.getDataSourceId()); tableSchemaRequest.setDataSourceId(queryRequest.getDataSourceId());
String databaseName = StringUtils.isNotBlank(queryRequest.getDatabaseName()) ? queryRequest.getDatabaseName() : queryRequest.getSchemaName(); tableSchemaRequest.setDatabaseName(queryRequest.getDatabaseName());
if (Objects.isNull(databaseName)) { tableSchemaRequest.setDataSourceSchema(queryRequest.getSchemaName());
databaseName = ""; ConfigService configService = ApplicationContextUtil.getBean(ConfigService.class);
Config keyConfig = configService.find(Chat2dbAIClient.CHAT2DB_OPENAI_KEY).getData();
if (Objects.isNull(keyConfig) || StringUtils.isBlank(keyConfig.getContent())) {
return "";
} }
tableSchemaRequest.setDatabaseName(databaseName); tableSchemaRequest.setApiKey(keyConfig.getContent());
DataResult<TableSchemaResponse> result = gatewayClientService.schemaVectorSearch(tableSchemaRequest); try {
DataResult<TableSchemaResponse> result = gatewayClientService.schemaVectorSearch(tableSchemaRequest);
List<String> schemas = Lists.newArrayList(); List<String> schemas = Lists.newArrayList();
if (CollectionUtils.isNotEmpty(result.getData().getTableSchemas())) { if (Objects.nonNull(result.getData()) && CollectionUtils.isNotEmpty(result.getData().getTableSchemas())) {
for(TableSchema data: result.getData().getTableSchemas()){ for(TableSchema data: result.getData().getTableSchemas()){
schemas.add(data.getTableSchema()); schemas.add(data.getTableSchema());
}
} }
return JSON.toJSONString(schemas);
} catch (Exception exception) {
log.error("query table error, do nothing");
return "";
} }
return JSON.toJSONString(schemas);
} }
/** /**

8
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/http/GatewayClientService.java
View File

@ -79,8 +79,8 @@ public interface GatewayClientService {
* @param searchVectors * @param searchVectors
* @return * @return
*/ */
@Get("/api/client/milvus/knowledge/search") @Post(url = "/api/client/milvus/knowledge/search", contentType = "application/json")
DataResult<KnowledgeResponse> knowledgeVectorSearch(@Query KnowledgeRequest searchVectors); DataResult<KnowledgeResponse> knowledgeVectorSearch(@Body KnowledgeRequest searchVectors);
/** /**
* save table schema vector * save table schema vector
@ -88,8 +88,8 @@ public interface GatewayClientService {
* @param request * @param request
* @return * @return
*/ */
@Get("/api/client/milvus/schema/search") @Post(url = "/api/client/milvus/schema/search", contentType = "application/json")
DataResult<TableSchemaResponse> schemaVectorSearch(@Query TableSchemaRequest request); DataResult<TableSchemaResponse> schemaVectorSearch(@Body TableSchemaRequest request);
/** /**
* check in white list * check in white list

2
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/http/request/TableSchemaRequest.java
View File

@ -26,5 +26,5 @@ public class TableSchemaRequest {
private List<String> schemaList; private List<String> schemaList;
private Boolean deleteBeforeInsert; private Boolean deleteBeforeInsert = false;
} }