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

add knowledge support

Browse Source
This commit is contained in:
robin
2023-10-04 11:34:45 +08:00
parent 9b16b3c35e
commit eed1111fc9
3 changed files with 5 additions and 4 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

@ -454,7 +454,7 @@ public class ChatController {
* @return
*/
private String buildPrompt(ChatQueryRequest queryRequest) {
if (PromptType.QUESTION_ANSWERING.getCode().equals(queryRequest.getPromptType())) {
if (PromptType.TEXT_GENERATION.getCode().equals(queryRequest.getPromptType())) {
return queryRequest.getMessage();
}

3
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/KnowledgeController.java
View File

@ -6,6 +6,7 @@ import ai.chat2db.server.tools.common.exception.ParamBusinessException;
import ai.chat2db.server.web.api.aspect.ConnectionInfoAspect;
import ai.chat2db.server.web.api.controller.ai.DocParser.AbstractParser;
import ai.chat2db.server.web.api.controller.ai.DocParser.PdfParse;
import ai.chat2db.server.web.api.controller.ai.enums.PromptType;
import ai.chat2db.server.web.api.controller.ai.fastchat.embeddings.FastChatEmbeddingResponse;
import ai.chat2db.server.web.api.controller.ai.request.ChatQueryRequest;
import ai.chat2db.server.web.api.http.GatewayClientService;
@ -102,7 +103,7 @@ public class KnowledgeController extends ChatController {
// search embedding
DataResult<KnowledgeResponse> result = gatewayClientService.knowledgeVectorSearch(contentVector);
queryRequest.setPromptType(PromptType.TEXT_GENERATION.getCode());
String prompt = queryRequest.getMessage();
if (CollectionUtils.isNotEmpty(result.getData().getKnowledgeList())) {
List<String> contents = new ArrayList<>();

4
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/enums/PromptType.java
View File

@ -35,9 +35,9 @@ public enum PromptType implements BaseEnum<String> {
SQL_2_SQL("进行SQL转换"),
/**
* knowledge qa
* text generation
*/
QUESTION_ANSWERING("问答"),
TEXT_GENERATION("文本生成"),
;
final String description;