Skip to content

Commit 88f552a

Browse files
author
I750911
committed
detection of output filter as finish reason
1 parent 15d2ce9 commit 88f552a

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

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

Lines changed: 7 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.orchestration.OrchestrationChatResponse;
5+
import com.sap.ai.sdk.orchestration.OrchestrationClientException;
56
import com.sap.ai.sdk.orchestration.model.LLMChoice;
67
import com.sap.ai.sdk.orchestration.model.LLMModuleResultSynchronous;
78
import com.sap.ai.sdk.orchestration.model.TokenUsage;
@@ -48,7 +49,12 @@ static List<Generation> toGenerations(@Nonnull final LLMModuleResultSynchronous
4849
}
4950

5051
@Nonnull
51-
static Generation toGeneration(@Nonnull final LLMChoice choice) {
52+
static Generation toGeneration(@Nonnull final LLMChoice choice)
53+
throws OrchestrationClientException {
54+
if ("content_filter".equals(choice.getFinishReason())) {
55+
throw new OrchestrationClientException("Content filter filtered the output.");
56+
}
57+
5258
val metadata = ChatGenerationMetadata.builder().finishReason(choice.getFinishReason());
5359
metadata.metadata("index", choice.getIndex());
5460
if (!choice.getLogprobs().isEmpty()) {

0 commit comments

Comments
 (0)