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

query schema update

Browse Source
This commit is contained in:
robin
2023-10-04 14:55:50 +08:00
parent 4d95eae76b
commit f9d52b08a2
3 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -528,7 +528,7 @@ public class ChatController {
contentVector.add(response.getData().get(0).getEmbedding());
// search embedding
DataResult<TableSchemaResponse> result = gatewayClientService.schemaVectorSearch(contentVector);
DataResult<TableSchemaResponse> result = gatewayClientService.schemaVectorSearch(contentVector, queryRequest.getDataSourceId());
List<String> schemas = Lists.newArrayList();
if (CollectionUtils.isNotEmpty(result.getData().getTableSchemas())) {

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

@ -87,8 +87,9 @@ public interface GatewayClientService {
* save table schema vector
*
* @param searchVectors
* @param datasourceId
* @return
*/
@Get("/api/milvus/schema/search")
DataResult<TableSchemaResponse> schemaVectorSearch(List<List<BigDecimal>> searchVectors);
DataResult<TableSchemaResponse> schemaVectorSearch(List<List<BigDecimal>> searchVectors, Long datasourceId);
}

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

@ -19,4 +19,6 @@ public class TableSchemaRequest {
private List<List<BigDecimal>> contentVector;
private List<String> sentenceList;
private Boolean deleteBeforeInsert;
}