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

Fix where previous Assistant Messages were not being added to the parts list.

Browse Source 
Results in: io.grpc.StatusRuntimeException: INVALID_ARGUMENT: Unable to submit request because it must include at least one parts field, which describes the prompt input. Learn more: https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/gemini
This commit is contained in:
Peter Dolukhanov
2024-09-05 12:27:02 -07:00
committed by Christian Tzolov
parent 2a62235f73
commit 26c5674acb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -431,7 +431,7 @@ public class VertexAiGeminiChatModel extends AbstractToolCallSupport implements
else if (message instanceof AssistantMessage assistantMessage) {
List<Part> parts = new ArrayList<>();
if (StringUtils.hasText(assistantMessage.getContent())) {
List.of(Part.newBuilder().setText(assistantMessage.getContent()).build());
parts.add(Part.newBuilder().setText(assistantMessage.getContent()).build());
}
if (!CollectionUtils.isEmpty(assistantMessage.getToolCalls())) {
parts.addAll(assistantMessage.getToolCalls()