From dee4d2705bc133afe168fe31c8e3bccf8ea7b5f3 Mon Sep 17 00:00:00 2001
From: linarkou <abzaltdinov@gmail.com>
Date: Mon, 23 Dec 2024 13:31:50 +0300
Subject: [PATCH] fix generic type in explicit type casting

---
 .../ai/model/function/DefaultFunctionCallbackBuilder.java   | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/spring-ai-core/src/main/java/org/springframework/ai/model/function/DefaultFunctionCallbackBuilder.java b/spring-ai-core/src/main/java/org/springframework/ai/model/function/DefaultFunctionCallbackBuilder.java
index fe66bacd0..584501a34 100644
--- a/spring-ai-core/src/main/java/org/springframework/ai/model/function/DefaultFunctionCallbackBuilder.java
+++ b/spring-ai-core/src/main/java/org/springframework/ai/model/function/DefaultFunctionCallbackBuilder.java
@@ -144,9 +144,9 @@ public class DefaultFunctionCallbackBuilder implements FunctionCallback.Builder
 			BiFunction<I, ToolContext, O> finalBiFunction = (this.biFunction != null) ? this.biFunction
 					: (request, context) -> this.function.apply(request);
 
-			return new FunctionInvokingFunctionCallback(this.name, this.getDescriptionExt(), this.getInputTypeSchema(),
-					this.inputType, (Function<I, String>) this.getResponseConverter(), this.getObjectMapper(),
-					finalBiFunction);
+			return new FunctionInvokingFunctionCallback<>(this.name, this.getDescriptionExt(),
+					this.getInputTypeSchema(), this.inputType, (Function<O, String>) this.getResponseConverter(),
+					this.getObjectMapper(), finalBiFunction);
 		}
 
 		private String getDescriptionExt() {