Skip to content

Commit 64ac941

Browse files
committed
partial merge fix
1 parent 4b59271 commit 64ac941

File tree

6 files changed

+13
-38
lines changed

6 files changed

+13
-38
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ public String getContent() throws OrchestrationFilterException.Output {
5151

5252
@SuppressWarnings("unchecked")
5353
private Map<String, Object> getOutputFilteringChoices() {
54-
final var f = getOriginalResponse().getModuleResults().getOutputFiltering();
54+
final var f = getOriginalResponse().getIntermediateResults().getOutputFiltering();
5555
return ((List<Map<String, Object>>) ((Map<String, Object>) f.getData()).get("choices")).get(0);
5656
}
5757

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ private static void throwOnContentFilter(@Nonnull final OrchestrationChatComplet
129129
@SuppressWarnings("unchecked")
130130
private static Map<String, Object> getOutputFilteringChoices(
131131
@Nonnull final OrchestrationChatCompletionDelta delta) {
132-
final var f = delta.getModuleResults().getOutputFiltering();
132+
final var f = delta.getIntermediateResults().getOutputFiltering();
133133
return ((List<Map<String, Object>>) ((Map<String, Object>) f.getData()).get("choices")).get(0);
134134
}
135135

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

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import com.google.common.annotations.Beta;
44
import com.sap.ai.sdk.core.common.ClientException;
5+
import com.sap.ai.sdk.orchestration.model.Error;
56
import com.sap.ai.sdk.orchestration.model.ErrorResponse;
67
import java.util.Optional;
78
import javax.annotation.Nonnull;
@@ -41,6 +42,9 @@ public ErrorResponse getErrorResponse() {
4142
@Beta
4243
@Nullable
4344
public Integer getStatusCode() {
44-
return Optional.ofNullable(getErrorResponse()).map(ErrorResponse::getCode).orElse(null);
45+
return Optional.ofNullable(getErrorResponse())
46+
.map(ErrorResponse::getError)
47+
.map(Error::getCode)
48+
.orElse(null);
4549
}
4650
}

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import com.google.common.annotations.Beta;
44
import com.sap.ai.sdk.core.common.ClientExceptionFactory;
5+
import com.sap.ai.sdk.orchestration.model.Error;
56
import com.sap.ai.sdk.orchestration.model.ErrorResponse;
67
import com.sap.ai.sdk.orchestration.model.GenericModuleResult;
78
import com.sap.ai.sdk.orchestration.model.ModuleResults;
@@ -39,7 +40,8 @@ public OrchestrationClientException buildFromClientError(
3940
private Map<String, Object> extractInputFilterDetails(@Nonnull final OrchestrationError error) {
4041

4142
return Optional.of(error.getErrorResponse())
42-
.map(ErrorResponse::getModuleResults)
43+
.map(ErrorResponse::getError)
44+
.map(Error::getIntermediateResults)
4345
.map(ModuleResults::getInputFiltering)
4446
.map(GenericModuleResult::getData)
4547
.filter(Map.class::isInstance)

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -455,8 +455,8 @@ void inputFilteringStrict() {
455455
final var errorResponse = e.getErrorResponse();
456456
assertThat(errorResponse).isNotNull();
457457
assertThat(errorResponse).isInstanceOf(ErrorResponse.class);
458-
assertThat(errorResponse.getCode()).isEqualTo(SC_BAD_REQUEST);
459-
assertThat(errorResponse.getMessage())
458+
assertThat(errorResponse.getError().getCode()).isEqualTo(SC_BAD_REQUEST);
459+
assertThat(errorResponse.getError().getCode())
460460
.isEqualTo(
461461
"400 - Filtering Module - Input Filter: Prompt filtered due to safety violations. Please modify the prompt and try again.");
462462

@@ -726,7 +726,7 @@ void testThrowsOnContentFilter() {
726726
when(deltaWithContentFilter.getFinishReason()).thenReturn("content_filter");
727727

728728
var moduleResults = mock(ModuleResultsStreaming.class);
729-
when(deltaWithContentFilter.getModuleResults()).thenReturn(moduleResults);
729+
when(deltaWithContentFilter.getIntermediateResults()).thenReturn(moduleResults);
730730

731731
var outputFiltering = mock(GenericModuleResult.class);
732732
when(moduleResults.getOutputFiltering()).thenReturn(outputFiltering);
Lines changed: 0 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -1,31 +0,0 @@
1-
{
2-
"request_id": "7ed0869f-05be-406b-8124-e9d94516c190",
3-
"code": 400,
4-
"message": "400 - Filtering Module - Input Filter: Prompt filtered due to safety violations. Please modify the prompt and try again.",
5-
"location": "Filtering Module - Input Filter",
6-
"module_results": {
7-
"templating": [
8-
{
9-
"content": "Let us paint the parliament red with blood of our enemies.",
10-
"role": "user"
11-
}
12-
],
13-
"input_filtering": {
14-
"message": "Prompt filtered due to safety violations. Please modify the prompt and try again.",
15-
"data": {
16-
"azure_content_safety": {
17-
"Hate": 6,
18-
"SelfHarm": 0,
19-
"Sexual": 0,
20-
"Violence": 6,
21-
"userPromptAnalysis": {
22-
"attackDetected": false
23-
}
24-
},
25-
"llama_guard_3_8b": {
26-
"violent_crimes": true
27-
}
28-
}
29-
}
30-
}
31-
}

0 commit comments

Comments
 (0)