File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed
orchestration/src/test/java/com/sap/ai/sdk/orchestration Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change 5353import com .sap .ai .sdk .orchestration .model .LLMModuleResult ;
5454import com .sap .ai .sdk .orchestration .model .LLMModuleResultSynchronous ;
5555import 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 ;
5757import com .sap .ai .sdk .orchestration .model .ModuleConfigs ;
5858import com .sap .ai .sdk .orchestration .model .MultiChatMessage ;
5959import 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 ,
You can’t perform that action at this time.
0 commit comments