diff --git a/src/main/java/com/hanserwei/snailsai/config/ChatClientConfiguration.java b/src/main/java/com/hanserwei/snailsai/config/ChatClientConfiguration.java
index 7aeed1f..3493856 100644
--- a/src/main/java/com/hanserwei/snailsai/config/ChatClientConfiguration.java
+++ b/src/main/java/com/hanserwei/snailsai/config/ChatClientConfiguration.java
@@ -3,6 +3,7 @@ package com.hanserwei.snailsai.config;
 import com.alibaba.cloud.ai.dashscope.chat.DashScopeChatModel;
 import com.alibaba.cloud.ai.memory.redis.BaseRedisChatMemoryRepository;
 import com.alibaba.cloud.ai.memory.redis.LettuceRedisChatMemoryRepository;
+import com.hanserwei.snailsai.tools.QueryTool;
 import jakarta.annotation.Resource;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.chat.client.advisor.PromptChatMemoryAdvisor;
@@ -28,6 +29,8 @@ public class ChatClientConfiguration {
 
     @Resource
     private DashScopeChatModel dashScopeChatModel;
+    @Resource
+    private QueryTool queryTool;
 
     @Bean
     public BaseRedisChatMemoryRepository redisChatMemoryRepository() {
@@ -51,6 +54,7 @@ public class ChatClientConfiguration {
     @Bean
     public ChatClient dashScopeChatClient(ChatMemory chatMemory) {
         return ChatClient.builder(dashScopeChatModel)
+                .defaultTools(queryTool)
                 .defaultAdvisors(PromptChatMemoryAdvisor.builder(chatMemory).build(), new SimpleLoggerAdvisor())
                 .build();
     }
diff --git a/src/main/java/com/hanserwei/snailsai/config/RedisConfig.java b/src/main/java/com/hanserwei/snailsai/config/RedisConfig.java
new file mode 100644
index 0000000..b75f668
--- /dev/null
+++ b/src/main/java/com/hanserwei/snailsai/config/RedisConfig.java
@@ -0,0 +1,7 @@
+package com.hanserwei.snailsai.config;
+
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class RedisConfig {
+}
diff --git a/src/main/java/com/hanserwei/snailsai/controller/DashScopeController.java b/src/main/java/com/hanserwei/snailsai/controller/DashScopeController.java
index 667fc28..1c5df91 100644
--- a/src/main/java/com/hanserwei/snailsai/controller/DashScopeController.java
+++ b/src/main/java/com/hanserwei/snailsai/controller/DashScopeController.java
@@ -1,7 +1,7 @@
 package com.hanserwei.snailsai.controller;
 
 import com.hanserwei.snailsai.model.AIResponse;
-import com.hanserwei.snailsai.model.ChatMessageDTO;
+import com.hanserwei.snailsai.dto.ChatMessageDTO;
 import jakarta.annotation.Resource;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.ai.chat.client.ChatClient;
diff --git a/src/main/java/com/hanserwei/snailsai/controller/TestDataController.java b/src/main/java/com/hanserwei/snailsai/controller/TestDataController.java
new file mode 100644
index 0000000..47dd755
--- /dev/null
+++ b/src/main/java/com/hanserwei/snailsai/controller/TestDataController.java
@@ -0,0 +1,29 @@
+package com.hanserwei.snailsai.controller;
+
+import com.hanserwei.snailsai.entity.UserEntity;
+import com.hanserwei.snailsai.service.UserService;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.List;
+
+@RestController
+public class TestDataController {
+
+    private final UserService userService;
+
+    public TestDataController(UserService userService) {
+        this.userService = userService;
+    }
+
+    /**
+     * POST /api/test/generate-users?count=100
+     * 插入假数据用于分页测试
+     */
+    @PostMapping("/api/test/generate-users")
+    public String generateTestData(@RequestParam(defaultValue = "100") int count) {
+        List<UserEntity> insertedUsers = userService.insertDummyUsers(count);
+        return String.format("成功插入了 %d 条假数据！", insertedUsers.size());
+    }
+}
\ No newline at end of file
diff --git a/src/main/java/com/hanserwei/snailsai/model/ChatMessageDTO.java b/src/main/java/com/hanserwei/snailsai/dto/ChatMessageDTO.java
similarity index 87%
rename from src/main/java/com/hanserwei/snailsai/model/ChatMessageDTO.java
rename to src/main/java/com/hanserwei/snailsai/dto/ChatMessageDTO.java
index 923e838..11b1cf8 100644
--- a/src/main/java/com/hanserwei/snailsai/model/ChatMessageDTO.java
+++ b/src/main/java/com/hanserwei/snailsai/dto/ChatMessageDTO.java
@@ -1,4 +1,4 @@
-package com.hanserwei.snailsai.model;
+package com.hanserwei.snailsai.dto;
 
 import lombok.AllArgsConstructor;
 import lombok.Builder;
diff --git a/src/main/java/com/hanserwei/snailsai/entity/UserEntity.java b/src/main/java/com/hanserwei/snailsai/entity/UserEntity.java
index 28a20eb..247405f 100644
--- a/src/main/java/com/hanserwei/snailsai/entity/UserEntity.java
+++ b/src/main/java/com/hanserwei/snailsai/entity/UserEntity.java
@@ -7,6 +7,7 @@ import lombok.NoArgsConstructor;
 
 import java.io.Serial;
 import java.io.Serializable;
+import java.time.LocalDateTime;
 
 
 @Entity
@@ -45,9 +46,9 @@ public class UserEntity implements Serializable {
 
     // 5. 补充审计字段 (推荐)
     @Column(name = "create_time", nullable = false)
-    private java.time.LocalDateTime createTime;
+    private LocalDateTime createTime;
 
     @Column(name = "update_time")
-    private java.time.LocalDateTime updateTime;
+    private LocalDateTime updateTime;
 
 }
diff --git a/src/main/java/com/hanserwei/snailsai/repository/UserRepository.java b/src/main/java/com/hanserwei/snailsai/repository/UserRepository.java
new file mode 100644
index 0000000..7d83d2f
--- /dev/null
+++ b/src/main/java/com/hanserwei/snailsai/repository/UserRepository.java
@@ -0,0 +1,21 @@
+package com.hanserwei.snailsai.repository;
+
+import com.hanserwei.snailsai.entity.UserEntity;
+import org.jetbrains.annotations.NotNull;
+import org.springframework.ai.tool.annotation.Tool;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+import java.util.List;
+
+@Repository
+public interface UserRepository extends JpaRepository<UserEntity, Long> {
+
+    @NotNull List<UserEntity> findAll();
+
+    @NotNull List<UserEntity> findAllByIdIn(List<Long> ids);
+
+    @NotNull List<UserEntity> findAllByUsernameContaining(String name);
+
+
+}
\ No newline at end of file
diff --git a/src/main/java/com/hanserwei/snailsai/service/UserService.java b/src/main/java/com/hanserwei/snailsai/service/UserService.java
new file mode 100644
index 0000000..44012da
--- /dev/null
+++ b/src/main/java/com/hanserwei/snailsai/service/UserService.java
@@ -0,0 +1,69 @@
+package com.hanserwei.snailsai.service;
+
+import com.hanserwei.snailsai.entity.UserEntity;
+import com.hanserwei.snailsai.repository.UserRepository;
+import jakarta.transaction.Transactional;
+import org.springframework.stereotype.Service;
+
+import java.time.LocalDateTime;
+import java.util.ArrayList;
+import java.util.List;
+
+@Service
+public class UserService {
+
+    private final UserRepository userRepository;
+
+    // 构造器注入
+    public UserService(UserRepository userRepository) {
+        this.userRepository = userRepository;
+    }
+
+    /**
+     * 批量生成并插入指定数量的假用户数据
+     * @param count 要插入的用户数量
+     * @return 插入成功的用户列表
+     */
+    @Transactional // 确保整个批量操作在一个事务中完成
+    public List<UserEntity> insertDummyUsers(int count) {
+        if (count <= 0) {
+            return List.of(); // 返回空列表
+        }
+
+        List<UserEntity> dummyUsers = new ArrayList<>(count);
+        LocalDateTime now = LocalDateTime.now();
+        
+        // 确保用户名是唯一的
+        // 我们可以先获取当前数据库中用户数量，作为生成唯一用户名的起始点
+        long startId = userRepository.count(); 
+
+        for (int i = 1; i <= count; i++) {
+            
+            // 使用 startId + i 来保证生成的用户名在多次运行时尽可能不重复
+            long userIndex = startId + i; 
+
+            // 注意：密码通常应该是加密后的，这里为了演示使用明文
+            UserEntity user = new UserEntity(
+                null, // ID 设为 null，让 JPA 自动生成
+                "testUser_" + userIndex, // 确保用户名唯一
+                "123456", // 模拟一个加密后的密码，或者使用一个测试用的明文，例如 "password"
+                (long) (i % 3) + 1, // 随机分配一个 hobbyId (例如 1, 2, 3)
+                now.plusSeconds(i), // 模拟创建时间略微递增
+                null // updateTime 初始为 null
+            );
+            dummyUsers.add(user);
+        }
+
+        // 使用 JpaRepository 的 saveAll 方法进行批量插入，效率比单个 save 要高
+        return userRepository.saveAll(dummyUsers);
+    }
+
+    public List<UserEntity> findAll() {
+        return userRepository.findAll();
+    }
+
+    public List<UserEntity> findAllByIdIn(List<Long> ids) {
+        return userRepository.findAllByIdIn(ids);
+    }
+
+}
\ No newline at end of file
diff --git a/src/main/java/com/hanserwei/snailsai/tools/QueryTool.java b/src/main/java/com/hanserwei/snailsai/tools/QueryTool.java
new file mode 100644
index 0000000..f52773a
--- /dev/null
+++ b/src/main/java/com/hanserwei/snailsai/tools/QueryTool.java
@@ -0,0 +1,27 @@
+package com.hanserwei.snailsai.tools;
+
+import com.hanserwei.snailsai.entity.UserEntity;
+import com.hanserwei.snailsai.service.UserService;
+import jakarta.annotation.Resource;
+import org.springframework.ai.tool.annotation.Tool;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+
+@Component
+public class QueryTool {
+
+    @Resource
+    private UserService userService;
+
+    @Tool(name = "findAll", description = "查询所有用户")
+    public List<UserEntity> findAll() {
+        return userService.findAll();
+    }
+
+    @Tool(name = "findAllByIdIn", description = "根据id列表查询用户")
+    public List<UserEntity> findAllByIdIn(List<Long> ids) {
+        return userService.findAllByIdIn(ids);
+    }
+
+}