diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/chat2db/client/Chat2DBAIStreamClient.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/chat2db/client/Chat2DBAIStreamClient.java index cf6ff6f3..b2b5d377 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/chat2db/client/Chat2DBAIStreamClient.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/chat2db/client/Chat2DBAIStreamClient.java @@ -246,25 +246,9 @@ public class Chat2DBAIStreamClient { FastChatEmbeddingResponse chatEmbeddingResponse = null; Response response = this.okHttpClient.newCall(request).execute(); - StringBuilder body = new StringBuilder(); if (response.isSuccessful()) { - ResponseBody responseBody = response.body(); - if (responseBody != null) { - // 获取响应体的输入流 - java.io.InputStream inputStream = responseBody.byteStream(); - java.io.BufferedReader reader = new java.io.BufferedReader(new java.io.InputStreamReader(inputStream)); - - String line; - while ((line = reader.readLine()) != null) { - // 在这里处理每行响应内容 - body.append(line); - } - - // 关闭流 - reader.close(); - inputStream.close(); - } - chatEmbeddingResponse = mapper.readValue(body.toString(), FastChatEmbeddingResponse.class); + String body = response.body().string(); + chatEmbeddingResponse = mapper.readValue(body, FastChatEmbeddingResponse.class); } log.info("finish invoking chat embedding"); return chatEmbeddingResponse; 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 index e13c2391..07512229 100644 --- 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 @@ -61,8 +61,8 @@ public interface GatewayClientService { * @param request * @return */ - @Post("/api/client/milvus/knowledge/save") - ActionResult knowledgeVectorSave(@Query KnowledgeRequest request); + @Post(url = "/api/client/milvus/knowledge/save", contentType = "application/json") + ActionResult knowledgeVectorSave(@Body KnowledgeRequest request); /** * save table schema vector @@ -70,8 +70,8 @@ public interface GatewayClientService { * @param request * @return */ - @Post("/api/client/milvus/schema/save") - ActionResult schemaVectorSave(@Query TableSchemaRequest request); + @Post(url = "/api/client/milvus/schema/save", contentType = "application/json") + ActionResult schemaVectorSave(@Body TableSchemaRequest request); /** * save knowledge vector