Skip to content

Commit dc537f4

Browse files
committed
Fix compilation and test code
1 parent 8267ca8 commit dc537f4

File tree

5 files changed

+12
-17
lines changed

5 files changed

+12
-17
lines changed

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import com.sap.ai.sdk.foundationmodels.openai.generated.model.ChatCompletionRequestMessageContentPartText;
77
import com.sap.ai.sdk.foundationmodels.openai.generated.model.ChatCompletionRequestSystemMessage;
88
import com.sap.ai.sdk.foundationmodels.openai.generated.model.ChatCompletionRequestSystemMessageContent;
9-
import com.sap.ai.sdk.foundationmodels.openai.generated.model.ChatCompletionRequestSystemMessageContentPart;
109
import java.util.LinkedList;
1110
import java.util.List;
1211
import javax.annotation.Nonnull;
@@ -74,7 +73,7 @@ ChatCompletionRequestSystemMessage createChatCompletionRequestMessage()
7473
.content(ChatCompletionRequestSystemMessageContent.create(textItem.text()));
7574
}
7675

77-
final var contentList = new LinkedList<ChatCompletionRequestSystemMessageContentPart>();
76+
final var contentList = new LinkedList<ChatCompletionRequestMessageContentPartText>();
7877
for (final OpenAiContentItem item : itemList) {
7978
if (item instanceof OpenAiTextItem textItem) {
8079
contentList.add(

foundation-models/openai/src/test/java/com/sap/ai/sdk/foundationmodels/openai/OpenAIMessageTest.java

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -171,18 +171,16 @@ void systemMessageToDto() {
171171
.isEqualTo(ChatCompletionRequestSystemMessage.RoleEnum.SYSTEM);
172172
var values =
173173
((ChatCompletionRequestSystemMessageContent
174-
.InnerChatCompletionRequestSystemMessageContentParts)
174+
.InnerChatCompletionRequestMessageContentPartTexts)
175175
requestMessageWithText.getContent())
176176
.values();
177177
assertThat(values).hasSize(2);
178-
assertThat(((ChatCompletionRequestMessageContentPartText) values.get(0)).getType())
178+
assertThat(values.get(0).getType())
179179
.isEqualTo(ChatCompletionRequestMessageContentPartText.TypeEnum.TEXT);
180-
assertThat(((ChatCompletionRequestMessageContentPartText) values.get(0)).getText())
181-
.isEqualTo(validText);
182-
assertThat(((ChatCompletionRequestMessageContentPartText) values.get(1)).getType())
180+
assertThat(values.get(0).getText()).isEqualTo(validText);
181+
assertThat(values.get(1).getType())
183182
.isEqualTo(ChatCompletionRequestMessageContentPartText.TypeEnum.TEXT);
184-
assertThat(((ChatCompletionRequestMessageContentPartText) values.get(1)).getText())
185-
.isEqualTo("Additional text");
183+
assertThat(values.get(1).getText()).isEqualTo("Additional text");
186184
}
187185

188186
@Test

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
import com.sap.ai.sdk.orchestration.model.DPIConfigMaskGroundingInput;
99
import com.sap.ai.sdk.orchestration.model.DPIEntities;
1010
import com.sap.ai.sdk.orchestration.model.DPIEntityConfig;
11-
import com.sap.ai.sdk.orchestration.model.MaskingProviderConfig;
1211
import java.util.ArrayList;
1312
import java.util.Arrays;
1413
import java.util.List;
@@ -95,7 +94,7 @@ public DpiMasking withAllowList(@Nonnull final List<String> allowList) {
9594

9695
@Nonnull
9796
@Override
98-
public MaskingProviderConfig createConfig() {
97+
public DPIConfig createConfig() {
9998
val entitiesDTO = entities.stream().map(it -> DPIEntityConfig.create().type(it)).toList();
10099
return DPIConfig.create()
101100
.type(SAP_DATA_PRIVACY_INTEGRATION)

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
package com.sap.ai.sdk.orchestration;
22

3-
import com.sap.ai.sdk.orchestration.model.MaskingProviderConfig;
3+
import com.sap.ai.sdk.orchestration.model.DPIConfig;
44
import javax.annotation.Nonnull;
55

66
/**
@@ -17,5 +17,5 @@ public interface MaskingProvider {
1717
* @return the masking configuration
1818
*/
1919
@Nonnull
20-
MaskingProviderConfig createConfig();
20+
DPIConfig createConfig();
2121
}

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

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,10 @@
44
import com.sap.ai.sdk.orchestration.model.FilteringModuleConfig;
55
import com.sap.ai.sdk.orchestration.model.GroundingModuleConfig;
66
import com.sap.ai.sdk.orchestration.model.InputFilteringConfig;
7-
import com.sap.ai.sdk.orchestration.model.InputTranslationModuleConfig;
87
import com.sap.ai.sdk.orchestration.model.LLMModuleConfig;
98
import com.sap.ai.sdk.orchestration.model.MaskingModuleConfig;
109
import com.sap.ai.sdk.orchestration.model.OutputFilteringConfig;
11-
import com.sap.ai.sdk.orchestration.model.OutputTranslationModuleConfig;
10+
import com.sap.ai.sdk.orchestration.model.SAPDocumentTranslation;
1211
import com.sap.ai.sdk.orchestration.model.TemplatingModuleConfig;
1312
import java.util.ArrayList;
1413
import java.util.Arrays;
@@ -96,9 +95,9 @@ public class OrchestrationModuleConfig {
9695
*/
9796
@Nullable GroundingModuleConfig groundingConfig;
9897

99-
@Nullable InputTranslationModuleConfig inputTranslationConfig;
98+
@Nullable SAPDocumentTranslation inputTranslationConfig;
10099

101-
@Nullable OutputTranslationModuleConfig outputTranslationConfig;
100+
@Nullable SAPDocumentTranslation outputTranslationConfig;
102101

103102
/**
104103
* Creates a new configuration with the given LLM configuration.

0 commit comments

Comments
 (0)