opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-30 03:03:13 +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.setSchemaVector(contentVector);
tableSchemaRequest.setDataSourceId(queryRequest.getDataSourceId());
String databaseName = StringUtils.isNotBlank(queryRequest.getDatabaseName()) ? queryRequest.getDatabaseName() : queryRequest.getSchemaName();
if (Objects.isNull(databaseName)) {
databaseName = "";
tableSchemaRequest.setDatabaseName(queryRequest.getDatabaseName());
tableSchemaRequest.setDataSourceSchema(queryRequest.getSchemaName());
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);
DataResult<TableSchemaResponse> result = gatewayClientService.schemaVectorSearch(tableSchemaRequest);
List<String> schemas = Lists.newArrayList();
if (CollectionUtils.isNotEmpty(result.getData().getTableSchemas())) {
for(TableSchema data: result.getData().getTableSchemas()){
schemas.add(data.getTableSchema());
tableSchemaRequest.setApiKey(keyConfig.getContent());
try {
DataResult<TableSchemaResponse> result = gatewayClientService.schemaVectorSearch(tableSchemaRequest);
List<String> schemas = Lists.newArrayList();
if (Objects.nonNull(result.getData()) && CollectionUtils.isNotEmpty(result.getData().getTableSchemas())) {
for(TableSchema data: result.getData().getTableSchemas()){
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
* @return
*/
@Get("/api/client/milvus/knowledge/search")
DataResult<KnowledgeResponse> knowledgeVectorSearch(@Query KnowledgeRequest searchVectors);
@Post(url = "/api/client/milvus/knowledge/search", contentType = "application/json")
DataResult<KnowledgeResponse> knowledgeVectorSearch(@Body KnowledgeRequest searchVectors);
/**
* save table schema vector
@ -88,8 +88,8 @@ public interface GatewayClientService {
* @param request
* @return
*/
@Get("/api/client/milvus/schema/search")
DataResult<TableSchemaResponse> schemaVectorSearch(@Query TableSchemaRequest request);
@Post(url = "/api/client/milvus/schema/search", contentType = "application/json")
DataResult<TableSchemaResponse> schemaVectorSearch(@Body TableSchemaRequest request);
/**
* 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 Boolean deleteBeforeInsert;
private Boolean deleteBeforeInsert = false;
}