opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-01 18:53:35 +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 * @return
*/ */
private String buildPrompt(ChatQueryRequest queryRequest) { private String buildPrompt(ChatQueryRequest queryRequest) {
if (PromptType.QUESTION_ANSWERING.getCode().equals(queryRequest.getPromptType())) { if (PromptType.TEXT_GENERATION.getCode().equals(queryRequest.getPromptType())) {
return queryRequest.getMessage(); 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.aspect.ConnectionInfoAspect;
import ai.chat2db.server.web.api.controller.ai.DocParser.AbstractParser; 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.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.fastchat.embeddings.FastChatEmbeddingResponse;
import ai.chat2db.server.web.api.controller.ai.request.ChatQueryRequest; import ai.chat2db.server.web.api.controller.ai.request.ChatQueryRequest;
import ai.chat2db.server.web.api.http.GatewayClientService; import ai.chat2db.server.web.api.http.GatewayClientService;
@ -102,7 +103,7 @@ public class KnowledgeController extends ChatController {
// search embedding // search embedding
DataResult<KnowledgeResponse> result = gatewayClientService.knowledgeVectorSearch(contentVector); DataResult<KnowledgeResponse> result = gatewayClientService.knowledgeVectorSearch(contentVector);
queryRequest.setPromptType(PromptType.TEXT_GENERATION.getCode());
String prompt = queryRequest.getMessage(); String prompt = queryRequest.getMessage();
if (CollectionUtils.isNotEmpty(result.getData().getKnowledgeList())) { if (CollectionUtils.isNotEmpty(result.getData().getKnowledgeList())) {
List<String> contents = new ArrayList<>(); 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转换"), SQL_2_SQL("进行SQL转换"),
/** /**
* knowledge qa * text generation
*/ */
QUESTION_ANSWERING("问答"), TEXT_GENERATION("文本生成"),
; ;
final String description; final String description;