此版本仍在开发中,尚未被视为稳定版本。最新的快照版本请使用 Spring AI 1.0.0-SNAPSHOT!spring-doc.cn

函数调用 API

在 AI 模型中集成函数支持,允许模型请求执行客户端函数,从而根据需要访问必要的信息或动态执行任务。spring-doc.cn

Spring AI 目前支持以下 AI 模型的函数调用:spring-doc.cn

函数调用

您可以使用 注册自定义 Java 函数,并让 AI 模型智能地选择输出包含参数的 JSON 对象,以调用一个或多个已注册的函数。 这允许您将 LLM 功能与外部工具和 API 连接起来。 AI 模型经过训练,可以检测何时应该调用函数,并使用符合函数签名的 JSON 进行响应。ChatClientspring-doc.cn

API 不直接调用该函数;相反,该模型会生成 JSON,您可以使用该 JSON 在代码中调用函数,并将结果返回给模型以完成对话。spring-doc.cn

Spring AI 提供了灵活且用户友好的方法来注册和调用自定义函数。 通常,自定义函数需要提供函数 、 和函数调用(作为 JSON 架构),以让模型知道函数需要哪些参数。这有助于模型了解何时调用函数。namedescriptionsignaturedescriptionspring-doc.cn

作为开发人员,您需要实现一个函数,该函数采用从 AI 模型发送的函数调用参数,并将结果返回给模型。您的函数可以反过来调用其他第三方服务来提供结果。spring-doc.cn

Spring AI 使这变得非常简单,只需定义一个返回 a 的定义,并在提示请求中动态调用或注册函数时提供 bean 名称作为选项即可。@Beanjava.util.FunctionChatClientspring-doc.cn

在后台, Spring 使用适当的适配器代码包装您的 POJO(函数),以便与 AI 模型进行交互,从而避免编写繁琐的样板代码。 底层基础架构的基础是 FunctionCallback.java 接口和配套的 Builder 实用程序类,以简化 Java 回调函数的实现和注册。spring-doc.cn

运作方式

假设我们希望 AI 模型使用它没有的信息进行响应,例如,给定位置的当前温度。spring-doc.cn

我们可以为 AI 模型提供有关我们自己的函数的元数据,它可以在处理您的提示时使用这些元数据来检索该信息。spring-doc.cn

例如,如果在处理提示期间,AI 模型确定它需要有关给定位置温度的其他信息,它将启动服务器端生成的请求/响应交互。 AI 模型不会返回最终响应消息,而是在特殊的 Tool Call 请求中返回,提供函数名称和参数(作为 JSON)。 客户端负责处理此消息并执行命名函数并返回响应 作为 Tool Response 消息返回给 AI 模型。spring-doc.cn

Spring AI 大大简化了您需要编写以支持函数调用的代码。 它为您代理函数调用对话。 您只需将函数定义作为 a 提供,然后在提示选项中提供函数的 bean 名称,或者直接在提示请求选项中将函数作为参数传递。@Beanspring-doc.cn

您还可以在提示符中引用多个函数 Bean 名称。spring-doc.cn

示例用例

让我们定义一个简单的用例,我们可以将其用作示例来解释函数调用的工作原理。 让我们创建一个聊天机器人,通过调用我们自己的函数来回答问题。 为了支持聊天机器人的响应,我们将注册自己的函数,该函数获取一个位置并返回该位置的当前天气。spring-doc.cn

当模型需要回答诸如 AI 模型之类的问题时,AI 模型将调用客户端,提供 location 值作为要传递给函数的参数。这种类似 RPC 的数据以 JSON 形式传递。"What’s the weather like in Boston?"spring-doc.cn

我们的函数调用一些基于 SaaS 的天气服务 API,并将天气响应返回给模型以完成对话。在此示例中,我们将使用一个名为 的简单实现,该实现对各个位置的温度进行硬编码。MockWeatherServicespring-doc.cn

以下类表示天气服务 API:MockWeatherServicespring-doc.cn

public class MockWeatherService implements Function<Request, Response> {

	public enum Unit { C, F }
	public record Request(String location, Unit unit) {}
	public record Response(double temp, Unit unit) {}

	public Response apply(Request request) {
		return new Response(30.0, Unit.C);
	}
}
class MockWeatherService : Function1<Request, Response> {
	override fun invoke(request: Request) = Response(30.0, Unit.C)
}

enum class Unit { C, F }
data class Request(val location: String, val unit: Unit) {}
data class Response(val temp: Double, val unit: Unit) {}

服务器端注册

函数作为 Bean

Spring AI 提供了多种在 Spring 上下文中将自定义函数注册为 bean 的方法。spring-doc.cn

我们首先介绍对 POJO 最友好的选项。spring-doc.cn

普通函数

在这种方法中,您可以在应用程序上下文中定义一个,就像定义任何其他 Spring 托管对象一样。@Beanspring-doc.cn

在内部, Spring AI 将创建一个 a 实例,该实例为通过 AI 模型调用它添加逻辑。 is used 函数名称的名称。ChatModelFunctionCallback@Beanspring-doc.cn

@Configuration
static class Config {

	@Bean
	@Description("Get the weather in location") // function description
	public Function<MockWeatherService.Request, MockWeatherService.Response> currentWeather() {
		return new MockWeatherService();
	}

}
@Configuration
class Config {

	@Bean
	@Description("Get the weather in location") // function description
	fun currentWeather(): (Request) -> Response = MockWeatherService()

}

注释是可选的,并提供函数描述,可帮助模型了解何时调用函数。这是一个重要的属性,可帮助 AI 模型确定要调用的客户端函数。@Descriptionspring-doc.cn

提供函数描述的另一个选项是在 :@JsonClassDescriptionMockWeatherService.Requestspring-doc.cn

@Configuration
static class Config {

	@Bean
	public Function<Request, Response> currentWeather() { // bean name as function name
		return new MockWeatherService();
	}
}

@JsonClassDescription("Get the weather in location") // function description
public record Request(String location, Unit unit) {}
@Configuration
class Config {

	@Bean
	fun currentWeather(): (Request) -> Response  { // bean name as function name
		return MockWeatherService()
	}
}

@JsonClassDescription("Get the weather in location") // function description
data class Request(val location: String, val unit: Unit)

最佳实践是使用信息对请求对象进行注释,以便为该函数生成的 JSON 架构尽可能具有描述性,以帮助 AI 模型选择要调用的正确函数。spring-doc.cn

函数回调

注册函数的另一种方法是创建如下所示的函数:FunctionCallbackspring-doc.cn

@Configuration
static class Config {

	@Bean
	public FunctionCallback weatherFunctionInfo() {

        return FunctionCallback.builder()
            .description("Get the weather in location") // (2) function description
            .function("CurrentWeather", new MockWeatherService()) // (1) function name and instance
            .inputType(MockWeatherService.Request.class) // (3) input type to build the JSON schema
            .build();
	}
}
import org.springframework.ai.model.function.withInputType

@Configuration
class Config {

	@Bean
	fun weatherFunctionInfo(): FunctionCallback {

        return FunctionCallback.builder()
            .description("Get the weather in location") // (2) function description
            .function("CurrentWeather", MockWeatherService()) // (1) function name and instance
            // (3) Required due to Kotlin SAM conversion being an opaque lambda
            .inputType<MockWeatherService.Request>()
            .build();
	}
}

它包装第三方函数,并将其注册为带有 . 它还提供了一个描述 (2) 和一个可选的响应转换器,用于将响应转换为模型预期的文本。MockWeatherServiceCurrentWeatherChatClientspring-doc.cn

默认情况下,响应转换器执行 Response 对象的 JSON 序列化。
根据类在内部解析函数调用签名。FunctionCallback.BuilderMockWeatherService.Request

按 Bean 名称启用函数

要让模型知道并调用您的函数,您需要在提示请求中启用它:CurrentWeatherspring-doc.cn

ChatClient chatClient = ...

ChatResponse response = this.chatClient.prompt("What's the weather like in San Francisco, Tokyo, and Paris?")
    .functions("CurrentWeather") // Enable the function
    .call().
    chatResponse();

logger.info("Response: {}", response);

上述用户问题将触发对函数的 3 次调用(每个城市 1 次),最终响应将如下所示:CurrentWeatherspring-doc.cn

Here is the current weather for the requested cities:
- San Francisco, CA: 30.0°C
- Tokyo, Japan: 10.0°C
- Paris, France: 15.0°C

客户端注册

除了自动配置之外,您还可以动态注册回调函数。 您可以使用函数调用或方法调用方法将函数注册到 or 请求中。ChatClientChatModelspring-doc.cn

默认情况下,客户端注册使您能够注册函数。spring-doc.cn

函数调用

ChatClient chatClient = ...

ChatResponse response = this.chatClient.prompt("What's the weather like in San Francisco, Tokyo, and Paris?")
    .functions(FunctionCallback.builder()
            .description("Get the weather in location") // (2) function description
            .function("currentWeather", (Request request) -> new Response(30.0, Unit.C)) // (1) function name and instance
            .inputType(MockWeatherService.Request.class) // (3) input type to build the JSON schema
            .build())
    .call()
    .chatResponse();
默认情况下,在此请求的持续时间内,动态功能处于启用状态。

这种方法允许根据用户输入动态选择要调用的不同函数。spring-doc.cn

FunctionCallbackInPromptIT.java 集成测试提供了一个完整的示例,说明如何向 注册函数并在提示请求中使用它。ChatClientspring-doc.cn

方法调用

它支持通过反射调用方法,同时自动处理 JSON 架构生成和参数转换。 它对于将 Java 方法作为可调用函数集成到 AI 模型交互中特别有用。MethodInvokingFunctionCallbackspring-doc.cn

实现接口并提供:MethodInvokingFunctionCallbackFunctionCallbackspring-doc.cn

您需要像这样创建:FunctionCallback.BuilderMethodInvokingFunctionCallbackspring-doc.cn

// Create using builder pattern
FunctionCallback methodInvokingCallback = FunctionCallback.builder()
    .description("Function calling description") // Hints the AI to know when to call this method
    .method("MethodName", Class<?>...argumentTypes) // The method to invoke and its argument types
    .targetObject(targetObject)       // Required instance methods for static methods use targetClass
    .build();

以下是一些使用示例:spring-doc.cn

public class WeatherService {
    public static String getWeather(String city, TemperatureUnit unit) {
        return "Temperature in " + city + ": 20" + unit;
    }
}

// Usage
FunctionCallback callback = FunctionCallback.builder()
    .description("Get weather information for a city")
    .method("getWeather", String.class, TemperatureUnit.class)
    .targetClass(WeatherService.class)
    .build();
public class DeviceController {
    public void setDeviceState(String deviceId, boolean state, ToolContext context) {
        Map<String, Object> contextData = context.getContext();
        // Implementation using context data
    }
}

// Usage
DeviceController controller = new DeviceController();

String response = ChatClient.create(chatModel).prompt()
    .user("Turn on the living room lights")
    .functions(FunctionCallback.builder()
        .description("Control device state")
        .method("setDeviceState", String.class,boolean.class,ToolContext.class)
        .targetObject(controller)
        .build())
    .toolContext(Map.of("location", "home"))
    .call()
    .content();

OpenAiChatClientMethodInvokingFunctionCallbackIT 集成测试提供了如何使用 FunctionCallback.Builder 创建方法调用 FunctionCallbacks 的额外示例。spring-doc.cn

工具上下文

Spring AI 现在支持通过工具上下文将额外的上下文信息传递给函数回调。 此功能允许您提供额外的用户提供的数据,这些数据可与 AI 模型传递的函数参数一起在函数执行中使用。spring-doc.cn

使用 Tool Context 进行函数调用

ToolContext 类提供了一种传递其他上下文信息的方法。spring-doc.cn

使用工具上下文

在函数调用的情况下,作为 .java.util.BiFunctionspring-doc.cn

对于方法调用,上下文信息作为 type 为 .ToolContextspring-doc.cn

函数调用

您可以在构建聊天选项时设置工具上下文,并将 BiFunction 用于回调:spring-doc.cn

BiFunction<MockWeatherService.Request, ToolContext, MockWeatherService.Response> weatherFunction =
    (request, toolContext) -> {
        String sessionId = (String) toolContext.getContext().get("sessionId");
        String userId = (String) toolContext.getContext().get("userId");

        // Use sessionId and userId in your function logic
        double temperature = 0;
        if (request.location().contains("Paris")) {
            temperature = 15;
        }
        else if (request.location().contains("Tokyo")) {
            temperature = 10;
        }
        else if (request.location().contains("San Francisco")) {
            temperature = 30;
        }

        return new MockWeatherService.Response(temperature, 15, 20, 2, 53, 45, MockWeatherService.Unit.C);
    };


ChatResponse response = chatClient.prompt("What's the weather like in San Francisco, Tokyo, and Paris?")
    .functions(FunctionCallback.builder()
        .description("Get the weather in location")
        .function("getCurrentWeather", this.weatherFunction)
        .inputType(MockWeatherService.Request.class)
        .build())
    .toolContext(Map.of("sessionId", "1234", "userId", "5678"))
    .call()
    .chatResponse();

在此示例中,被定义为将请求和工具上下文作为参数的 BiFunction。这允许您直接在函数 logic中访问上下文。weatherFunctionspring-doc.cn

这种方法允许您将特定于会话或用户的信息传递给您的函数,从而实现更多上下文和个性化的响应。spring-doc.cn

方法调用

public class DeviceController {
    public void setDeviceState(String deviceId, boolean state, ToolContext context) {
        Map<String, Object> contextData = context.getContext();
        // Implementation using context data
    }
}

// Usage
DeviceController controller = new DeviceController();

String response = ChatClient.create(chatModel).prompt()
    .user("Turn on the living room lights")
    .functions(FunctionCallback.builder()
        .description("Control device state")
        .method("setDeviceState", String.class,boolean.class,ToolContext.class)
        .targetObject(controller)
        .build())
    .toolContext(Map.of("location", "home"))
    .call()
    .content();