File tree Expand file tree Collapse file tree 5 files changed +10
-28
lines changed
main/java/com/sap/ai/sdk/orchestration
java/com/sap/ai/sdk/orchestration Expand file tree Collapse file tree 5 files changed +10
-28
lines changed Original file line number Diff line number Diff line change 1212import com .sap .ai .sdk .core .common .ClientResponseHandler ;
1313import com .sap .ai .sdk .core .common .ClientStreamingHandler ;
1414import com .sap .ai .sdk .core .common .StreamedDelta ;
15- import com .sap .ai .sdk .orchestration .model .ChatMessage ;
1615import com .sap .ai .sdk .orchestration .model .CompletionPostRequest ;
1716import com .sap .ai .sdk .orchestration .model .CompletionPostResponse ;
1817import com .sap .ai .sdk .orchestration .model .ModuleConfigs ;
3837public class OrchestrationClient {
3938 private static final String DEFAULT_SCENARIO = "orchestration" ;
4039
41- static final ObjectMapper JACKSON ;
42-
43- static {
44- JACKSON = getDefaultObjectMapper ();
45-
46- // Add mix-ins
47- JACKSON .addMixIn (LLMModuleResult .class , JacksonMixins .LLMModuleResultMixIn .class );
48- JACKSON .addMixIn (
49- ModuleResultsOutputUnmaskingInner .class ,
50- JacksonMixins .ModuleResultsOutputUnmaskingInnerMixIn .class );
51-
52- final var module =
53- new SimpleModule ()
54- .addDeserializer (
55- ChatMessage .class ,
56- PolymorphicFallbackDeserializer .fromJsonSubTypes (ChatMessage .class ))
57- .setMixInAnnotation (ChatMessage .class , JacksonMixins .NoneTypeInfoMixin .class );
58- JACKSON .registerModule (module );
59- }
6040 static final ObjectMapper JACKSON = getOrchestrationObjectMapper ();
6141
6242 @ Nonnull private final Supplier <HttpDestination > destinationSupplier ;
Original file line number Diff line number Diff line change 55import com .fasterxml .jackson .databind .ObjectMapper ;
66import com .fasterxml .jackson .databind .module .SimpleModule ;
77import com .google .common .annotations .Beta ;
8- import com .sap .ai .sdk .orchestration .model .ChatMessagesInner ;
8+ import com .sap .ai .sdk .orchestration .model .ChatMessage ;
99import com .sap .ai .sdk .orchestration .model .LLMModuleResult ;
1010import com .sap .ai .sdk .orchestration .model .ModuleResultsOutputUnmaskingInner ;
1111import javax .annotation .Nonnull ;
@@ -45,9 +45,9 @@ public static ObjectMapper getOrchestrationObjectMapper() {
4545 final var module =
4646 new SimpleModule ()
4747 .addDeserializer (
48- ChatMessagesInner .class ,
49- PolymorphicFallbackDeserializer .fromJsonSubTypes (ChatMessagesInner .class ))
50- .setMixInAnnotation (ChatMessagesInner .class , JacksonMixins .NoneTypeInfoMixin .class );
48+ ChatMessage .class ,
49+ PolymorphicFallbackDeserializer .fromJsonSubTypes (ChatMessage .class ))
50+ .setMixInAnnotation (ChatMessage .class , JacksonMixins .NoneTypeInfoMixin .class );
5151 jackson .registerModule (module );
5252 return jackson ;
5353 }
Original file line number Diff line number Diff line change @@ -309,7 +309,7 @@ void filteringLoose() throws IOException {
309309 .sexual (ALLOW_SAFE_LOW_MEDIUM )
310310 .violence (ALLOW_SAFE_LOW_MEDIUM );
311311
312- ContentFilter llamaFilter =
312+ final ContentFilter llamaFilter =
313313 () ->
314314 LlamaGuard38bFilterConfig .create ()
315315 .type (LlamaGuard38bFilterConfig .TypeEnum .LLAMA_GUARD_3_8B )
Original file line number Diff line number Diff line change 22
33import static org .assertj .core .api .Assertions .assertThat ;
44
5- import com .sap .ai .sdk .orchestration .model .ChatMessage ;
65import com .sap .ai .sdk .orchestration .model .LLMChoice ;
76import com .sap .ai .sdk .orchestration .model .LLMModuleResultSynchronous ;
7+ import com .sap .ai .sdk .orchestration .model .ResponseChatMessage ;
88import com .sap .ai .sdk .orchestration .model .TokenUsage ;
99import java .util .List ;
1010import org .junit .jupiter .api .Test ;
@@ -17,7 +17,7 @@ void testToAssistantMessage() {
1717 var choice =
1818 LLMChoice .create ()
1919 .index (0 )
20- .message (ChatMessage .create ().role ("assistant" ).content ("Hello, world!" ))
20+ .message (ResponseChatMessage .create ().role ("assistant" ).content ("Hello, world!" ))
2121 .finishReason ("stop" );
2222
2323 AssistantMessage message = OrchestrationSpringChatResponse .toAssistantMessage (choice );
Original file line number Diff line number Diff line change 1717 "template" : [ {
1818 "role" : " system" ,
1919 "content" : " Context message with embedded grounding results. {{?results}}"
20- } ]
20+ } ],
21+ "defaults" : { },
22+ "tools" : [ ]
2123 },
2224 "grounding_module_config" : {
2325 "type" : " document_grounding_service" ,
You can’t perform that action at this time.
0 commit comments