Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix typos in VertexAiGeminiChatModel class

Browse Source 
Signed-off-by: jitokim <pigberger70@gmail.com>
This commit is contained in:
jitokim
2024-11-22 02:28:12 +09:00
committed by Ilayaperumal Gopinathan
parent 357efd0b0b
commit 7feb13b63d
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java
View File

@@ -92,6 +92,7 @@ import org.springframework.util.StringUtils;
* @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 class VertexAiGeminiChatModel extends AbstractToolCallSupport implements
List<Generation> generations = generateContentResponse.getCandidatesList()
.stream()
.map(this::responseCandiateToGeneration)
.map(this::responseCandidateToGeneration)
.flatMap(List::stream)
.toList();
@@ -352,7 +353,7 @@ public class VertexAiGeminiChatModel extends AbstractToolCallSupport implements
List<Generation> generations = response.getCandidatesList()
.stream()
.map(this::responseCandiateToGeneration)
.map(this::responseCandidateToGeneration)
.flatMap(List::stream)
.toList();
@@ -380,21 +381,20 @@ public class VertexAiGeminiChatModel extends AbstractToolCallSupport implements
});
}
protected List<Generation> responseCandiateToGeneration(Candidate candidate) {
protected List<Generation> 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<String, Object> 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<AssistantMessage.ToolCall> assistantToolCalls = candidate.getContent()
.getPartsList()
.stream()