Skip to content

Commit 3b430a5

Browse files
sobychackoilayaperumalg
authored andcommitted
Fix Kotlin test deprecation warning
1 parent d858327 commit 3b430a5

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed
Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@ import org.springframework.ai.autoconfigure.ollama.OllamaAutoConfiguration
2626
import org.springframework.ai.chat.messages.UserMessage
2727
import org.springframework.ai.chat.prompt.Prompt
2828
import org.springframework.ai.model.function.FunctionCallback
29-
import org.springframework.ai.model.function.FunctionCallbackWrapper
3029
import org.springframework.ai.model.function.FunctionCallingOptions
3130
import org.springframework.ai.ollama.OllamaChatModel
3231
import org.springframework.ai.ollama.api.OllamaModel
@@ -36,7 +35,7 @@ import org.springframework.boot.test.context.runner.ApplicationContextRunner
3635
import org.springframework.context.annotation.Bean
3736
import org.springframework.context.annotation.Configuration
3837

39-
class FunctionCallbackWrapperKotlinIT : BaseOllamaIT() {
38+
class FunctionCallbackKotlinIT : BaseOllamaIT() {
4039

4140
companion object {
4241

@@ -49,7 +48,7 @@ class FunctionCallbackWrapperKotlinIT : BaseOllamaIT() {
4948
}
5049
}
5150

52-
private val logger = LoggerFactory.getLogger(FunctionCallbackWrapperKotlinIT::class.java)
51+
private val logger = LoggerFactory.getLogger(FunctionCallbackKotlinIT::class.java)
5352

5453
private val contextRunner = ApplicationContextRunner()
5554
.withPropertyValues(
@@ -106,13 +105,13 @@ class FunctionCallbackWrapperKotlinIT : BaseOllamaIT() {
106105

107106
@Bean
108107
open fun weatherFunctionInfo(): FunctionCallback {
109-
110-
return FunctionCallbackWrapper.builder(MockKotlinWeatherService())
111-
.withName("WeatherInfo")
112-
.withInputType(KotlinRequest::class.java)
113-
.withDescription(
114-
"Find the weather conditions, forecasts, and temperatures for a location, like a city or state.")
115-
.build();
108+
return FunctionCallback.builder()
109+
.description(
110+
"Find the weather conditions, forecasts, and temperatures for a location, like a city or state."
111+
)
112+
.function("WeatherInfo", MockKotlinWeatherService())
113+
.inputType(KotlinRequest::class.java)
114+
.build()
116115
}
117116

118117
}

0 commit comments

Comments
 (0)