快捷導(dǎo)航

使用SpringAI整合Ollama實(shí)現(xiàn)工具鏈調(diào)用功能

更新時(shí)間：2026年05月20日 08:43:34 作者：霸道流氓氣質(zhì)

文章介紹了SpringAI的ToolCalling能力,允許大模型自主調(diào)用外部工具以完成多步驟任務(wù),具體而言,通過@Tool注解標(biāo)記方法,SpringAI自動解析生成JSONSchema,@Tools注解注冊多個(gè)工具,模型可自主選擇調(diào)用順序和組合,需要的朋友可以參考下

場景

Spring AI 提供了強(qiáng)大的 Tool Calling 能力，允許大模型在對話中自主決定調(diào)用哪些外部工具，

并自動組合它們的執(zhí)行順序。

這種機(jī)制讓智能體能夠完成復(fù)雜的多步驟任務(wù)，例如“查詢北京天氣并用英文回答”

——模型會先調(diào)用天氣工具獲取數(shù)據(jù)，再調(diào)用翻譯工具輸出英文。

核心概念

概念	說明
Tool Calling	大模型根據(jù)用戶問題，生成函數(shù)調(diào)用請求（含函數(shù)名和參數(shù)），Spring AI 攔截后執(zhí)行對應(yīng)的 Java 方法，將結(jié)果返回模型。
@Tool 注解	標(biāo)記在方法上，Spring AI 自動解析方法簽名和注釋，生成 JSON Schema 供模型理解。
多工具注冊	通過 `defaultTools(...)` 同時(shí)注冊多個(gè)工具，模型可自主選擇調(diào)用順序和組合。
鏈?zhǔn)秸{(diào)用	模型在一次響應(yīng)中先后調(diào)用多個(gè)工具，前一個(gè)的輸出作為后一個(gè)的輸入，形成“鏈”。

實(shí)現(xiàn)

pom.xml

    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>3.3.3</version> <!-- 降級為穩(wěn)定版，解決沖突 -->
    </parent>
    <groupId>com.example</groupId>
    <artifactId>spring-ai-ollama-tool-chain</artifactId>
    <version>1.0</version>
    <properties>
        <java.version>17</java.version>
        <spring-ai.version>1.1.2</spring-ai.version>
    </properties>
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <!-- Spring AI Ollama 核心 -->
        <dependency>
            <groupId>org.springframework.ai</groupId>
            <artifactId>spring-ai-starter-model-ollama</artifactId>
            <version>${spring-ai.version}</version>
        </dependency>
    </dependencies>

application.yml

?
server:
  port: 886
spring:
  ai:
    ollama:
      base-url: http://localhost:11434
      chat:
        model: qwen2.5:7b-instruct
        options:
          temperature: 0.7
          num-ctx: 4096                         # 上下文窗口大小
logging:
  level:
    org.springframework.ai.chat.client: DEBUG   # 查看工具調(diào)用詳情
?

天氣工具

package com.badao.ai.tools;

import org.springframework.ai.tool.annotation.Tool;
import org.springframework.ai.tool.annotation.ToolParam;
import org.springframework.stereotype.Component;

@Component
public class WeatherTool {

    @Tool(name = "get_weather", description = "查詢指定城市的實(shí)時(shí)天氣")
    public String getWeather(@ToolParam(description = "城市名稱") String city) {
        System.out.println("調(diào)用了天氣工具");
        // 模擬天氣數(shù)據(jù)
        return String.format("%s當(dāng)前天氣：晴，溫度22℃，濕度45%%。", city);
    }
}

翻譯工具

package com.badao.ai.tools;

import org.springframework.ai.tool.annotation.Tool;
import org.springframework.ai.tool.annotation.ToolParam;
import org.springframework.stereotype.Component;

@Component
public class TranslateTool {

    @Tool(name = "translate_to_english", description = "將中文文本翻譯成英文")
    public String translate(@ToolParam(description = "待翻譯的中文文本") String text) {
        System.out.println("調(diào)用了翻譯工具");
        // 模擬翻譯，實(shí)際可接入翻譯API
        return "Translated: " + text + " (This is the English version.)";
    }
}

工具注冊配置類

package com.badao.ai.config;

import com.badao.ai.tools.WeatherTool;
import com.badao.ai.tools.TranslateTool;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.model.ChatModel;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class ToolConfig {

    @Bean
    public ChatClient chatClient(ChatModel chatModel,
                                 WeatherTool weatherTool,
                                 TranslateTool translateTool) {
        return ChatClient.builder(chatModel)
                .defaultTools(weatherTool, translateTool)   // 注冊天氣和翻譯工具
                .build();
    }
}

Agent 服務(wù)層

package com.badao.ai.service;

import org.springframework.ai.chat.client.ChatClient;
import org.springframework.stereotype.Service;

@Service
public class AgentService {

    private final ChatClient chatClient;

    public AgentService(ChatClient chatClient) {
        this.chatClient = chatClient;
    }

    public String ask(String question) {
        return chatClient.prompt()
                .user(question+ "（請先用天氣工具，再用翻譯工具）")
                .call()
                .content();
    }
}

控制器

package com.badao.ai.controller;

import com.badao.ai.service.AgentService;
import org.springframework.web.bind.annotation.*;

@RestController
@RequestMapping("/api")
public class AgentController {

    private final AgentService agentService;

    public AgentController(AgentService agentService) {
        this.agentService = agentService;
    }

    @PostMapping("/agent")
    public String ask(@RequestBody String question) {
        return agentService.ask(question);
    }
}