1

spring-projects · Nov 13, 2024 · 58ee212 · 58ee212
1 parent b4e0a45
commit 58ee212
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/spring-ai-core/src/main/java/org/springframework/ai/chat/model/AbstractToolCallSupport.java b/spring-ai-core/src/main/java/org/springframework/ai/chat/model/AbstractToolCallSupport.java
@@ -27,6 +27,7 @@
 import org.springframework.ai.chat.messages.AssistantMessage;
 import org.springframework.ai.chat.messages.Message;
 import org.springframework.ai.chat.messages.ToolResponseMessage;
+import org.springframework.ai.chat.metadata.ChatGenerationMetadata;
 import org.springframework.ai.chat.prompt.Prompt;
 import org.springframework.ai.model.function.FunctionCallback;
 import org.springframework.ai.model.function.FunctionCallbackContext;
@@ -240,10 +241,11 @@ protected boolean isToolCall(ChatResponse chatResponse, Set<String> toolCallFini
 	protected boolean isToolCall(Generation generation, Set<String> toolCallFinishReasons) {
 		var finishReason = (generation.getMetadata().getFinishReason() != null)
 				? generation.getMetadata().getFinishReason() : "";
-		return generation.getOutput().hasToolCalls() && toolCallFinishReasons.stream()
+		ChatGenerationMetadata metadata = generation.getMetadata();
+		return generation.getOutput().hasToolCalls() && (toolCallFinishReasons.stream()
 			.map(s -> s.toLowerCase())
 			.toList()
-			.contains(finishReason.toLowerCase());
+			.contains(finishReason.toLowerCase()) || metadata != null);
 	}
 
 	/**