Skip to content

Commit 9832e60

Browse files
committed
Clean up
- Rename toChat.. to createChat.. - extract variables
1 parent cdac75a commit 9832e60

File tree

8 files changed

+18
-17
lines changed

8 files changed

+18
-17
lines changed

orchestration/src/main/java/com/sap/ai/sdk/orchestration/AssistantMessage.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ public class AssistantMessage implements Message {
2121
* @return the corresponding {@code ChatMessage} object.
2222
*/
2323
@Nonnull
24-
public ChatMessage toChatMessage() {
24+
public ChatMessage createChatMessage() {
2525
return ChatMessage.create().role(ROLE).content(content);
2626
}
2727
}

orchestration/src/main/java/com/sap/ai/sdk/orchestration/ConfigToRequestTransformer.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,8 @@ static CompletionPostRequest toCompletionPostRequest(
3131
.orchestrationConfig(
3232
OrchestrationConfig.create().moduleConfigurations(toModuleConfigs(configCopy)))
3333
.inputParams(prompt.getTemplateParameters())
34-
.messagesHistory(prompt.getMessagesHistory().stream().map(Message::toChatMessage).toList());
34+
.messagesHistory(
35+
prompt.getMessagesHistory().stream().map(Message::createChatMessage).toList());
3536
}
3637

3738
@Nonnull
@@ -46,7 +47,8 @@ static TemplatingModuleConfig toTemplateModuleConfig(
4647
*/
4748
val messages = template instanceof Template t ? t.getTemplate() : List.<ChatMessage>of();
4849
val messagesWithPrompt = new ArrayList<>(messages);
49-
messagesWithPrompt.addAll(prompt.getMessages().stream().map(Message::toChatMessage).toList());
50+
messagesWithPrompt.addAll(
51+
prompt.getMessages().stream().map(Message::createChatMessage).toList());
5052
if (messagesWithPrompt.isEmpty()) {
5153
throw new IllegalStateException(
5254
"A prompt is required. Pass at least one message or configure a template with messages or a template reference.");

orchestration/src/main/java/com/sap/ai/sdk/orchestration/Message.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,5 +12,5 @@ public sealed interface Message permits UserMessage, AssistantMessage, SystemMes
1212
* @return the corresponding {@code ChatMessage} object.
1313
*/
1414
@Nonnull
15-
ChatMessage toChatMessage();
15+
ChatMessage createChatMessage();
1616
}

orchestration/src/main/java/com/sap/ai/sdk/orchestration/OrchestrationClient.java

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -169,10 +169,9 @@ public OrchestrationChatResponse executeRequestFromJsonModuleConfig(
169169
}
170170

171171
final ObjectNode requestJson = JACKSON.createObjectNode();
172-
requestJson.set(
173-
"messages_history",
174-
JACKSON.valueToTree(
175-
prompt.getMessagesHistory().stream().map(Message::toChatMessage).toList()));
172+
var chatMessageHistory =
173+
prompt.getMessagesHistory().stream().map(Message::createChatMessage).toList();
174+
requestJson.set("messages_history", JACKSON.valueToTree(chatMessageHistory));
176175
requestJson.set("input_params", JACKSON.valueToTree(prompt.getTemplateParameters()));
177176

178177
final JsonNode moduleConfigJson;

orchestration/src/main/java/com/sap/ai/sdk/orchestration/SystemMessage.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ public class SystemMessage implements Message {
2121
* @return the corresponding {@code ChatMessage} object.
2222
*/
2323
@Nonnull
24-
public ChatMessage toChatMessage() {
24+
public ChatMessage createChatMessage() {
2525
return ChatMessage.create().role(ROLE).content(content);
2626
}
2727
}

orchestration/src/main/java/com/sap/ai/sdk/orchestration/UserMessage.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ public class UserMessage implements Message {
2121
* @return the corresponding {@code ChatMessage} object.
2222
*/
2323
@Nonnull
24-
public ChatMessage toChatMessage() {
24+
public ChatMessage createChatMessage() {
2525
return ChatMessage.create().role(ROLE).content(content);
2626
}
2727
}

orchestration/src/test/java/com/sap/ai/sdk/orchestration/ConfigToRequestTransformerTest.java

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ void testEmptyTemplateConfig() {
3535

3636
var expected =
3737
Template.create()
38-
.template(List.of(systemMessage.toChatMessage(), userMessage.toChatMessage()));
38+
.template(List.of(systemMessage.createChatMessage(), userMessage.createChatMessage()));
3939

4040
var prompt = new OrchestrationPrompt(systemMessage, userMessage);
4141
var actual =
@@ -60,14 +60,14 @@ void testMergingTemplateConfig() {
6060
Template.create()
6161
.template(
6262
List.of(
63-
systemMessage.toChatMessage(),
64-
userMessage.toChatMessage(),
65-
userMessage2.toChatMessage()));
63+
systemMessage.createChatMessage(),
64+
userMessage.createChatMessage(),
65+
userMessage2.createChatMessage()));
6666

6767
var prompt = new OrchestrationPrompt(userMessage2);
6868
var templateConfig =
6969
Template.create()
70-
.template(List.of(systemMessage.toChatMessage(), userMessage.toChatMessage()));
70+
.template(List.of(systemMessage.createChatMessage(), userMessage.createChatMessage()));
7171
var actual = ConfigToRequestTransformer.toTemplateModuleConfig(prompt, templateConfig);
7272

7373
assertThat(actual).isEqualTo(expected);
@@ -82,6 +82,6 @@ void testMessagesHistory() {
8282
ConfigToRequestTransformer.toCompletionPostRequest(
8383
prompt, new OrchestrationModuleConfig().withLlmConfig(CUSTOM_GPT_35));
8484

85-
assertThat(actual.getMessagesHistory()).containsExactly(systemMessage.toChatMessage());
85+
assertThat(actual.getMessagesHistory()).containsExactly(systemMessage.createChatMessage());
8686
}
8787
}

sample-code/spring-app/src/main/java/com/sap/ai/sdk/app/controllers/OrchestrationController.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ public OrchestrationChatResponse completion() {
5555
public OrchestrationChatResponse template() {
5656
final var template =
5757
new UserMessage("Reply with 'Orchestration Service is working!' in {{?language}}");
58-
final var templatingConfig = Template.create().template(List.of(template.toChatMessage()));
58+
final var templatingConfig = Template.create().template(List.of(template.createChatMessage()));
5959
final var configWithTemplate = config.withTemplateConfig(templatingConfig);
6060

6161
final var inputParams = Map.of("language", "German");

0 commit comments

Comments
 (0)