query schema update

robinji0 · robinji0 · commit 454adcc064c2 · 2023-10-04T14:55:50.000+08:00
diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/ChatController.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/ChatController.java
@@ -528,7 +528,7 @@ public String queryDatabaseSchema(ChatQueryRequest queryRequest) throws IOExcept
         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())) {
diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/http/GatewayClientService.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/http/GatewayClientService.java
@@ -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);
 }
diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/http/request/TableSchemaRequest.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/http/request/TableSchemaRequest.java
@@ -19,4 +19,6 @@ public class TableSchemaRequest {
     private List<List<BigDecimal>> contentVector;
 
     private List<String> sentenceList;
+
+    private Boolean deleteBeforeInsert;
 }

Original file line number	Diff line number	Diff line change
`@@ -87,8 +87,9 @@ public interface GatewayClientService {`
`87`	`87`	`* save table schema vector`
`88`	`88`	`*`
`89`	`89`	`* @param searchVectors`
	`90`	`+ * @param datasourceId`
`90`	`91`	`* @return`
`91`	`92`	`*/`
`92`	`93`	`@Get("/api/milvus/schema/search")`
`93`		`- DataResult<TableSchemaResponse> schemaVectorSearch(List<List<BigDecimal>> searchVectors);`
	`94`	`+ DataResult<TableSchemaResponse> schemaVectorSearch(List<List<BigDecimal>> searchVectors, Long datasourceId);`
`94`	`95`	`}`
Original file line number	Diff line number	Diff line change
`@@ -19,4 +19,6 @@ public class TableSchemaRequest {`
`19`	`19`	`private List<List<BigDecimal>> contentVector;`
`20`	`20`
`21`	`21`	`private List<String> sentenceList;`
	`22`	`+`
	`23`	`+ private Boolean deleteBeforeInsert;`
`22`	`24`	`}`