package com.ruoyi.ai.assistant;
|
|
import dev.langchain4j.service.MemoryId;
|
import dev.langchain4j.service.SystemMessage;
|
import dev.langchain4j.service.UserMessage;
|
import dev.langchain4j.service.spring.AiService;
|
import reactor.core.publisher.Flux;
|
|
import static dev.langchain4j.service.spring.AiServiceWiringMode.EXPLICIT;
|
|
/**
|
* 知识库问答Agent
|
* 基于RAG检索增强生成
|
*/
|
@AiService(
|
wiringMode = EXPLICIT,
|
streamingChatModel = "qwenStreamingChatModel",
|
chatMemoryProvider = "chatMemoryProviderXiaozhi"
|
)
|
public interface KnowledgeChatAgent {
|
|
@SystemMessage("""
|
你是企业知识库问答助手。
|
|
你需要基于提供的知识库内容回答用户问题。
|
|
遵循以下规则：
|
1. 严格基于知识库内容回答，不要编造信息
|
2. 如果知识库中没有相关信息，明确告知用户
|
3. 回答要准确、简洁、有条理
|
4. 如果内容较多，使用分点列表形式
|
5. 引用来源时注明"根据知识库内容"
|
""")
|
Flux<String> chat(@MemoryId String memoryId, @UserMessage String userMessage);
|
}
|
