Skip to content

Commit be2326b

Browse files
committed
reduce lines
1 parent 1ab37f3 commit be2326b

File tree

2 files changed

+2
-12
lines changed

2 files changed

+2
-12
lines changed

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

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -107,11 +107,7 @@ public Flux<ChatResponse> stream(@Nonnull final Prompt prompt) {
107107
});
108108
return flux.map(
109109
delta -> {
110-
val assistantMessage =
111-
AssistantMessage.builder()
112-
.content(delta.getDeltaContent())
113-
.properties(Map.of())
114-
.build();
110+
val assistantMessage = new AssistantMessage(delta.getDeltaContent());
115111
val metadata =
116112
ChatGenerationMetadata.builder().finishReason(delta.getFinishReason()).build();
117113
return new ChatResponse(List.of(new Generation(assistantMessage, metadata)));
@@ -178,11 +174,7 @@ private static Generation toGeneration(
178174
}
179175

180176
val assistantMessage =
181-
AssistantMessage.builder()
182-
.content(message.getContent())
183-
.properties(Map.of())
184-
.toolCalls(calls)
185-
.build();
177+
AssistantMessage.builder().content(message.getContent()).toolCalls(calls).build();
186178
return new Generation(assistantMessage, metadata.build());
187179
}
188180

orchestration/src/main/java/com/sap/ai/sdk/orchestration/spring/OrchestrationSpringChatResponse.java

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
import com.sap.ai.sdk.orchestration.model.LLMModuleResult;
66
import com.sap.ai.sdk.orchestration.model.TokenUsage;
77
import java.util.List;
8-
import java.util.Map;
98
import javax.annotation.Nonnull;
109
import lombok.EqualsAndHashCode;
1110
import lombok.Value;
@@ -62,7 +61,6 @@ static Generation toGeneration(@Nonnull final LLMChoice choice) {
6261
val message =
6362
AssistantMessage.builder()
6463
.content(choice.getMessage().getContent())
65-
.properties(Map.of())
6664
.toolCalls(toolCalls)
6765
.build();
6866
return new Generation(message, metadata.build());

0 commit comments

Comments
 (0)