From 396ff5b4ceae4bc6b6f097bec6b58e4265215928 Mon Sep 17 00:00:00 2001
From: wangbing <wangbing@wbsite.xyz>
Date: Thu, 6 Mar 2025 22:26:46 +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

---
 src/main/java/xyz/wbsite/ai/Main.java | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/main/java/xyz/wbsite/ai/Main.java b/src/main/java/xyz/wbsite/ai/Main.java
index bd3ca9d..15af1fb 100644
--- a/src/main/java/xyz/wbsite/ai/Main.java
+++ b/src/main/java/xyz/wbsite/ai/Main.java
@@ -29,11 +29,11 @@ public class Main {
     public static void main(String[] args) {
 //        testSimpleChat(args);
 //        testStreamChat(args);
-        testTool(args);
+//        testTool(args);
+        testRagChat(args);
     }
 
     public static void testSimpleChat(String[] args) {
-
         ChatLanguageModel model = OllamaChatModel.builder()
                 .baseUrl("http://36.138.207.178:11434")
                 .modelName("deepseek-r1:32B")
@@ -51,7 +51,24 @@ public class Main {
                 .build();
         ChatResponse chatResponse = model.chat(build);
         System.out.println(chatResponse.aiMessage().text());
+    }
+
+    public static void testRagChat(String[] args) {
+        ChatLanguageModel model = OllamaChatModel.builder()
+                .baseUrl("http://36.138.207.178:11434")
+                .modelName("deepseek-r1:32B")
+                .build();
 
+        ChatRequest build = ChatRequest.builder()
+                .messages(new ChatMessage[]{
+                        SystemMessage.from("你需要使用文档内容对用户提出的问题进行回复，这点非常重要。\n" +
+                                "当用户提出的问题无法根据文档内容进行回复时，回复不知道即可。请明确说明，你的答案是否是从文档中获取的。\n" +
+                                "文档内容如下:小王和小吴是情侣关系，他们两个年龄一样都是20岁，小王有一辆大众牌小轿车，小吴有一辆雅迪电动车。"),
+                        UserMessage.from("小王的车是什么牌子")
+                })
+                .build();
+        ChatResponse chatResponse = model.chat(build);
+        System.out.println(chatResponse.aiMessage().text());
     }
 
     public static void testStreamChat(String[] args) {