opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-30 19:22:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

query gateway update

Browse Source
This commit is contained in:
robin
2023-10-16 19:24:46 +08:00
parent 99d8dd6c16
commit 8ebfe94c2e
2 changed files with 6 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/chat2db/client/Chat2DBAIStreamClient.java
View File

@ -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;

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

@ -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