diff --git a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java index b347d551871..30f030c3737 100644 --- a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java +++ b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java @@ -92,6 +92,7 @@ * @author Chris Turchin * @author Mark Pollack * @author Soby Chacko + * @author Jihoon Kim * @since 0.8.1 */ public class VertexAiGeminiChatModel extends AbstractToolCallSupport implements ChatModel, DisposableBean { @@ -304,7 +305,7 @@ public ChatResponse call(Prompt prompt) { List generations = generateContentResponse.getCandidatesList() .stream() - .map(this::responseCandiateToGeneration) + .map(this::responseCandidateToGeneration) .flatMap(List::stream) .toList(); @@ -352,7 +353,7 @@ public Flux stream(Prompt prompt) { List generations = response.getCandidatesList() .stream() - .map(this::responseCandiateToGeneration) + .map(this::responseCandidateToGeneration) .flatMap(List::stream) .toList(); @@ -380,21 +381,20 @@ public Flux stream(Prompt prompt) { }); } - protected List responseCandiateToGeneration(Candidate candidate) { + protected List responseCandidateToGeneration(Candidate candidate) { // TODO - The candidateIndex (e.g. choice must be asigned to the generation). int candidateIndex = candidate.getIndex(); - FinishReason candidateFinishReasonn = candidate.getFinishReason(); + FinishReason candidateFinishReason = candidate.getFinishReason(); Map messageMetadata = Map.of("candidateIndex", candidateIndex, "finishReason", - candidateFinishReasonn); + candidateFinishReason); - ChatGenerationMetadata chatGenerationMetadata = ChatGenerationMetadata.from(candidateFinishReasonn.name(), - null); + ChatGenerationMetadata chatGenerationMetadata = ChatGenerationMetadata.from(candidateFinishReason.name(), null); - boolean isFunctinCall = candidate.getContent().getPartsList().stream().allMatch(Part::hasFunctionCall); + boolean isFunctionCall = candidate.getContent().getPartsList().stream().allMatch(Part::hasFunctionCall); - if (isFunctinCall) { + if (isFunctionCall) { List assistantToolCalls = candidate.getContent() .getPartsList() .stream()