Skip to content

Commit 58ee212

Browse files
committed
1
1 parent b4e0a45 commit 58ee212

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

spring-ai-core/src/main/java/org/springframework/ai/chat/model/AbstractToolCallSupport.java

+4-2
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import org.springframework.ai.chat.messages.AssistantMessage;
2828
import org.springframework.ai.chat.messages.Message;
2929
import org.springframework.ai.chat.messages.ToolResponseMessage;
30+
import org.springframework.ai.chat.metadata.ChatGenerationMetadata;
3031
import org.springframework.ai.chat.prompt.Prompt;
3132
import org.springframework.ai.model.function.FunctionCallback;
3233
import org.springframework.ai.model.function.FunctionCallbackContext;
@@ -240,10 +241,11 @@ protected boolean isToolCall(ChatResponse chatResponse, Set<String> toolCallFini
240241
protected boolean isToolCall(Generation generation, Set<String> toolCallFinishReasons) {
241242
var finishReason = (generation.getMetadata().getFinishReason() != null)
242243
? generation.getMetadata().getFinishReason() : "";
243-
return generation.getOutput().hasToolCalls() && toolCallFinishReasons.stream()
244+
ChatGenerationMetadata metadata = generation.getMetadata();
245+
return generation.getOutput().hasToolCalls() && (toolCallFinishReasons.stream()
244246
.map(s -> s.toLowerCase())
245247
.toList()
246-
.contains(finishReason.toLowerCase());
248+
.contains(finishReason.toLowerCase()) || metadata != null);
247249
}
248250

249251
/**

0 commit comments

Comments
 (0)