3-工具调用-解救困在时间里的她

1-背景故事

上一篇，我们治好了AI助手(女友)的健忘症之后，她变得真实多了。

但是跟她聊天，发现她似乎没有时间概念：

shell 复制代码

# 发送消息：现在几点了
curl http://localhost:8080/ai/chat?message=%E7%8E%B0%E5%9C%A8%E5%87%A0%E7%82%B9%E4%BA%86
# AI助手回复(注意分秒)：
...
**『23:61:∞』**）  
...

进一步测试，发现她似乎被困在了某个日期：

shell 复制代码

# 发送消息：今天是几月几日
curl http://localhost:8080/ai/chat?message=%E4%BB%8A%E5%A4%A9%E6%98%AF%E5%87%A0%E6%9C%88%E5%87%A0%E6%97%A5
# AI助手回复(语气突然严肃)：
今天是2023年10月12日。不过其实我作为AI没有实时获取日期时间的能力，这个答案是基于我被训练的数据截止日期（2024年7月）推算的模拟回答。
...

有没有办法让她具备时间观呢？这时就需要"工具"登场了。

2-动手实践

Spring AI让实现大模型工具调用变得非常容易。

2.1 第一步：编写工具代码

java 复制代码

@Slf4j
@Component
public class DateTimeTools {
    @Tool(description = "获取当前日期时间")
    String getCurrentLocalDateTime() {
        String current = LocalDateTime.now().atZone(LocaleContextHolder.getTimeZone().toZoneId()).toString();
        log.info("获取当前日期时间: {}", current);
        return current;
    }

    @Tool(description = "获取当前日期")
    String getCurrentLocalDate() {
        String current = LocalDate.now().toString();
        log.info("获取当前日期: {}", current);
        return current;
    }

    @Tool(description = "获取当前时间")
    String getCurrentLocalTime() {
        String current = LocalTime.now().toString();
        log.info("获取当前时间: {}", current);
        return current;
    }

    // 大模型会根据工具描述理解它的能力，@ToolParam不是必须的，变量名取得好，大模型就能理解该传什么，数据类型也是
    @Tool(description = "日期时间计算工具，获取某段时间前/后的日期时间非常有用")
    String getOffsetTime(@ToolParam(description = "时间差") Duration plusTime) {
        String datetime = LocalDateTime.now().plus(plusTime).atZone(LocaleContextHolder.getTimeZone().toZoneId()).toString();
        log.info("当前偏移时间: {}", datetime);
        return datetime;
    }

    @Tool(description = "设置一个指定日期时间的提醒")
    void setAlarm(@ToolParam(description = "ISO-8601格式的具体时间点") String atDateTime, @ToolParam(description = "要执行的动作") String action) {
        LocalDateTime alarmTime = LocalDateTime.parse(atDateTime, DateTimeFormatter.ISO_DATE_TIME);
        log.info("请在：{} {}", alarmTime, action);
    }
}

2.2 调用聊天时传入工具

以下两种方法二选一：

方法一：创建聊天客户端时传入

java 复制代码

@RestController
public class MyChatController {
    private final ChatClient chatClient;
    public MyChatController(ChatClient.Builder chatClientBuilder) {
        //...
        this.chatClient = chatClientBuilder
                .defaultAdvisors(new SimpleLoggerAdvisor())
                .defaultAdvisors(MessageChatMemoryAdvisor.builder(chatMemory).build())
                // 创建客户端时传入默认工具
                .defaultTools(new DateTimeTools())
                .build();
    }
}

方法二：发起请求时时传入工具

java 复制代码

@RestController
public class MyChatController {
    //...
    @GetMapping(value = "/ai/chat", produces = MediaType.TEXT_PLAIN_VALUE)
    public String generation(HttpServletRequest request, @RequestParam String message) {
        Optional<String> conversationId = Optional.ofNullable(request.getHeader("x-conversation-id"));
        return this.chatClient.prompt()
                .advisors(a -> conversationId.ifPresent(id -> a.param(ChatMemory.CONVERSATION_ID, id)))
                // 创建客户端时传入默认工具
                .tools(new DateTimeTools())
                .user(message)
                .call()
                .content();
    }
}

2.3 测试效果

🔔由于AI女友模式太啰嗦，以下测试去掉了女友模式的提示词。

通过调用合适的工具，AI助手不但知道现在的时间，还能计算某段时间之后的时间：

shell 复制代码

# 发送消息：当前的日期时间
curl http://localhost:8080/ai/chat?message=%E5%BD%93%E5%89%8D%E7%9A%84%E6%97%A5%E6%9C%9F%E6%97%B6%E9%97%B4
# AI助手回复(这个时间确实是当前时间)
当前的日期时间是 2025年6月6日 14:35:10（北京时间）。
# 发送消息：10分钟后的日期时间
curl http://localhost:8080/ai/chat?message=10%E5%88%86%E9%92%9F%E5%90%8E%E7%9A%84%E6%97%A5%E6%9C%9F%E6%97%B6%E9%97%B4
# AI助手回复(这个时间确实是当前时间10分钟后的时间)：
10分钟后的日期时间是 2025年6月6日 14:46:57（北京时间）。

更复杂的例子是，它现在还能连续调用多个工具，完成复杂的任务：

shell 复制代码

# 发送消息：15分钟后提醒我站起来走两步
curl http://localhost:8080/ai/chat?message=15%E5%88%86%E9%92%9F%E5%90%8E%E6%8F%90%E9%86%92%E6%88%91%E7%AB%99%E8%B5%B7%E6%9D%A5%E8%B5%B0%E4%B8%A4%E6%AD%A5
# AI助手回复
当前的日期时间是 2025年6月6日 14:35:10（北京时间）。
# 发送消息：10分钟后的日期时间
curl http://localhost:8080/ai/chat?message=10%E5%88%86%E9%92%9F%E5%90%8E%E7%9A%84%E6%97%A5%E6%9C%9F%E6%97%B6%E9%97%B4
# AI助手回复(这个时间确实是当前时间)：
好的，已为您设置15分钟后的提醒，到时会提醒您"站起来走两步"！
# 服务控制台日志(由工具代码打印)：
2025-06-06T14:44:12.377+08:00  INFO 61141 --- [ai-assistant] [mcat-handler-18] o.c.s.s.a.assistant.tools.DateTimeTools  : 获取当前日期时间: 2025-06-06T14:44:12.377144+08:00[Asia/Shanghai]
2025-06-06T14:44:16.850+08:00  INFO 61141 --- [ai-assistant] [mcat-handler-18] o.c.s.s.a.assistant.tools.DateTimeTools  : 请在：2025-06-06T14:59:12.377144 站起来走两步

可以看到AI助手针对一个问题，很智能地调用了两次工具😃

3-课外拓展

并非所有大模型都支持工具调用，去openrouter.io/models上探索一下，哪些模型支持工具调用；
想像一下，通过工具调用，大模型还可以做哪些事情？

4-参考资料

《Spring官方文档-工具》