Spring/spring-ai-examples
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Replace FunctionCallback to ToolCallback

Browse Source
This commit is contained in:
Ilayaperumal Gopinathan
2025-04-29 20:22:19 +01:00
parent 40e1b43d53
commit c0cc26b6c3
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
kotlin/kotlin-function-callback/src/main/kotlin/com/example/kotlin_function_callback/KotlinFunctionCallbackApplication.kt
View File

@@ -1,7 +1,8 @@
package com.example.kotlin_function_callback
import org.springframework.ai.chat.client.ChatClient
import org.springframework.ai.model.function.FunctionCallback
import org.springframework.ai.tool.ToolCallback
import org.springframework.ai.tool.function.FunctionToolCallback
import org.springframework.boot.CommandLineRunner
import org.springframework.boot.autoconfigure.SpringBootApplication
import org.springframework.boot.runApplication
@@ -18,7 +19,7 @@ class KotlinFunctionCallbackApplication {
val chatClient = chatClientBuilder.build();
val response = chatClient
.prompt("What are the weather conditions in San Francisco, Tokyo, and Paris? Find the temperature in Celsius for each of the three locations.")
.functions("WeatherInfo")
.tools("WeatherInfo")
.call().chatResponse();
println("Response: $response")
@@ -33,9 +34,8 @@ class KotlinFunctionCallbackApplication {
class Config {
@Bean
fun weatherFunctionInfo(currentWeather: (WeatherRequest) -> WeatherResponse): FunctionCallback {
return FunctionCallback.builder()
.function("WeatherInfo", currentWeather)
fun weatherFunctionInfo(currentWeather: (WeatherRequest) -> WeatherResponse): ToolCallback {
return FunctionToolCallback.builder("WeatherInfo", currentWeather)
.description(
"Find the weather conditions, forecasts, and temperatures for a location, like a city or state."
)