Skip to content

Commit 286d6b1

Browse files
Unit test
1 parent 20a106a commit 286d6b1

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

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

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353
import com.sap.ai.sdk.orchestration.model.LLMModuleResult;
5454
import com.sap.ai.sdk.orchestration.model.LLMModuleResultSynchronous;
5555
import com.sap.ai.sdk.orchestration.model.LlamaGuard38b;
56-
import com.sap.ai.sdk.orchestration.model.LlamaGuard38bFilterConfig;
56+
import com.sap.ai.sdk.orchestration.model.LlamaGuardFilter;
5757
import com.sap.ai.sdk.orchestration.model.ModuleConfigs;
5858
import com.sap.ai.sdk.orchestration.model.MultiChatMessage;
5959
import com.sap.ai.sdk.orchestration.model.OrchestrationConfig;
@@ -309,11 +309,7 @@ void filteringLoose() throws IOException {
309309
.sexual(ALLOW_SAFE_LOW_MEDIUM)
310310
.violence(ALLOW_SAFE_LOW_MEDIUM);
311311

312-
final ContentFilter llamaFilter =
313-
() ->
314-
LlamaGuard38bFilterConfig.create()
315-
.type(LlamaGuard38bFilterConfig.TypeEnum.LLAMA_GUARD_3_8B)
316-
.config(LlamaGuard38b.create().selfHarm(true));
312+
final var llamaFilter = new LlamaGuardFilter().config(LlamaGuard38b.create().selfHarm(true));
317313

318314
client.chatCompletion(
319315
prompt,

0 commit comments

Comments
 (0)