File tree Expand file tree Collapse file tree 2 files changed +60
-0
lines changed
main/kotlin/org/springframework/ai/model/function
test/kotlin/org/springframework/ai/model/function Expand file tree Collapse file tree 2 files changed +60
-0
lines changed Original file line number Diff line number Diff line change 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)
Original file line number Diff line number Diff line change 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+ }
You can’t perform that action at this time.
0 commit comments