Skip to content

Commit 01d3c97

Browse files
committed
Add withInputType Kotlin extension
This commit adds a Kotlin extension function for FunctionCallbackWrapper.Builder.withInputType allowing to specify withInputType<T>() instead of withInputType(T::class.java).
1 parent 0b4fcb4 commit 01d3c97

File tree

2 files changed

+60
-0
lines changed

2 files changed

+60
-0
lines changed
Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
/*
2+
* Copyright 2023-2024 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
package org.springframework.ai.model.function
18+
19+
/**
20+
* Extension for [FunctionCallbackWrapper.Builder.withInputType] providing a `withInputType<Foo>()`
21+
* variant.
22+
*
23+
* @author Sebastien Deleuze
24+
*/
25+
inline fun <reified T> FunctionCallbackWrapper.Builder<*, *>.withInputType() =
26+
withInputType(T::class.java)
Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
/*
2+
* Copyright 2023-2024 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
package org.springframework.ai.model.function
18+
19+
import io.mockk.every
20+
import io.mockk.mockk
21+
import io.mockk.verify
22+
import org.junit.jupiter.api.Test
23+
24+
class FunctionCallbackWrapperExtensionsTests {
25+
26+
private val builder = mockk<FunctionCallbackWrapper.Builder<WeatherRequest, WeatherResponse>>()
27+
28+
@Test
29+
fun withInputType() {
30+
every { builder.withInputType(any<Class<*>>()) } returns builder
31+
builder.withInputType<WeatherRequest>()
32+
verify { builder.withInputType(WeatherRequest::class.java) }
33+
}
34+
}

0 commit comments

Comments
 (0)