Skip to content

Commit ad4241a

Browse files
Update foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/spring/OpenAiChatModel.java
Co-authored-by: Alexander Dümont <[email protected]>
1 parent 817d3cb commit ad4241a

File tree

1 file changed

+6
-11
lines changed
  • foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/spring

1 file changed

+6
-11
lines changed

foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/spring/OpenAiChatModel.java

Lines changed: 6 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -111,17 +111,12 @@ static List<Generation> toGenerations(@Nonnull final OpenAiChatCompletionRespons
111111
@Nonnull
112112
static Generation toGeneration(@Nonnull final ChatCompletionResponseMessage choice) {
113113
// no metadata for now
114-
val toolCalls =
115-
choice.getToolCalls().stream()
116-
.map(
117-
toolCall ->
118-
new ToolCall(
119-
toolCall.getId(),
120-
toolCall.getType().getValue(),
121-
toolCall.getFunction().getName(),
122-
toolCall.getFunction().getArguments()))
123-
.toList();
124-
val message = new AssistantMessage(choice.getContent(), Map.of(), toolCalls);
114+
val calls = new ArrayList<ToolCall>();
115+
for (final ChatCompletionMessageToolCall c : choice.getToolCalls()) {
116+
val fnc = c.getFunction();
117+
calls.add(new ToolCall(c.getId(), c.getType().getValue(), fnc.getName(), fnc.getArguments()));
118+
}
119+
val message = new AssistantMessage(choice.getContent(), Map.of(), calls);
125120
return new Generation(message);
126121
}
127122
}

0 commit comments

Comments
 (0)