From 7cbfb9bc3c6c7eb99b63a8710e35ce665f8e6195 Mon Sep 17 00:00:00 2001
From: wangbing <wangbing@wbsite.xyz>
Date: Fri, 18 Apr 2025 16:15:06 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E5=A4=87=E4=BB=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../wbsite/ai/Easy_RAG_Augmentor_Example.java | 58 +++++++++++++++++++
 src/main/java/xyz/wbsite/ai/Helper.java       |  2 +
 2 files changed, 60 insertions(+)
 create mode 100644 src/main/java/xyz/wbsite/ai/Easy_RAG_Augmentor_Example.java

diff --git a/src/main/java/xyz/wbsite/ai/Easy_RAG_Augmentor_Example.java b/src/main/java/xyz/wbsite/ai/Easy_RAG_Augmentor_Example.java
new file mode 100644
index 0000000..55560ff
--- /dev/null
+++ b/src/main/java/xyz/wbsite/ai/Easy_RAG_Augmentor_Example.java
@@ -0,0 +1,58 @@
+package xyz.wbsite.ai;
+
+import cn.hutool.core.collection.CollUtil;
+import dev.langchain4j.data.document.Document;
+import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.memory.chat.MessageWindowChatMemory;
+import dev.langchain4j.model.input.PromptTemplate;
+import dev.langchain4j.rag.DefaultRetrievalAugmentor;
+import dev.langchain4j.rag.content.injector.DefaultContentInjector;
+import dev.langchain4j.rag.content.retriever.EmbeddingStoreContentRetriever;
+import dev.langchain4j.service.AiServices;
+import dev.langchain4j.service.SystemMessage;
+import dev.langchain4j.store.embedding.EmbeddingStoreIngestor;
+import dev.langchain4j.store.embedding.inmemory.InMemoryEmbeddingStore;
+
+import java.util.List;
+
+/**
+ * Easy-RAG 自定义注入提示词
+ */
+public class Easy_RAG_Augmentor_Example {
+
+    public static void main(String[] args) {
+        // 加载文档
+        List<Document> documents = CollUtil.newArrayList(Document.from("小明+小芳=小强"));
+
+        // 创建一个内存存储器，用于存储文档和其嵌入
+        InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
+
+        // 使用easy-rag，可以最快捷的方式使用rag
+        EmbeddingStoreIngestor.ingest(documents, embeddingStore);
+
+
+        PromptTemplate promptTemplate = PromptTemplate.from("{{userMessage}} \n\n 使用以下信息回答：\n{{contents}}");
+
+        Assistant assistant = AiServices.builder(Assistant.class)
+                .chatLanguageModel(Helper.getChatModel())
+                .chatMemory(MessageWindowChatMemory.withMaxMessages(10))
+//                .contentRetriever(EmbeddingStoreContentRetriever.from(embeddingStore))
+                .retrievalAugmentor(DefaultRetrievalAugmentor.builder()
+                        .contentRetriever(EmbeddingStoreContentRetriever.from(embeddingStore))
+                        .contentInjector(DefaultContentInjector.builder().promptTemplate(promptTemplate).build())
+                        .build()
+                )
+                .build();
+
+        String chat = assistant.chat("小明+小芳等于什么");
+        System.out.println(chat);
+    }
+
+    /**
+     * 助手
+     */
+    interface Assistant {
+
+        String chat(String userMessage);
+    }
+}
\ No newline at end of file
diff --git a/src/main/java/xyz/wbsite/ai/Helper.java b/src/main/java/xyz/wbsite/ai/Helper.java
index 56a6773..ecdff8b 100644
--- a/src/main/java/xyz/wbsite/ai/Helper.java
+++ b/src/main/java/xyz/wbsite/ai/Helper.java
@@ -6,6 +6,7 @@ import dev.langchain4j.model.openai.OpenAiChatModel;
 import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
 import dev.langchain4j.model.openai.OpenAiStreamingChatModel;
 
+import java.time.Duration;
 import java.util.List;
 
 public class Helper {
@@ -21,6 +22,7 @@ public class Helper {
     private static  OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()
             .baseUrl("http://192.168.88.106:11434/v1")
             .apiKey("1")
+            .timeout(Duration.ofSeconds(120))
             .modelName("deepseek-r1:1.5b")
             .logRequests(true)
             .logResponses(true)