File tree Expand file tree Collapse file tree 2 files changed +5
-8
lines changed
orchestration/src/main/java/com/sap/ai/sdk/orchestration
sample-code/spring-app/src/test/java/com/sap/ai/sdk/app/controllers Expand file tree Collapse file tree 2 files changed +5
-8
lines changed Original file line number Diff line number Diff line change 33import com .sap .ai .sdk .orchestration .client .model .LLMModuleConfig ;
44import java .util .Map ;
55import javax .annotation .Nonnull ;
6- import javax .annotation .Nullable ;
7-
8- import com .sap .ai .sdk .orchestration .client .model .LLMModuleResult ;
96import lombok .AccessLevel ;
107import lombok .Getter ;
118import lombok .RequiredArgsConstructor ;
@@ -138,7 +135,8 @@ public class OrchestrationAiModel {
138135 */
139136 @ Nonnull
140137 public OrchestrationAiModel modelVersion (@ Nonnull final String version ) {
141- return new OrchestrationAiModel (new LLMModuleConfig ()
138+ return new OrchestrationAiModel (
139+ new LLMModuleConfig ()
142140 .modelVersion (version )
143141 .modelParams (config .getModelParams ())
144142 .modelName (config .getModelName ()));
Original file line number Diff line number Diff line change @@ -37,8 +37,8 @@ void testTemplate() {
3737 assertThat (controller .config .getLlmConfig ()).isNotNull ();
3838 final var model = controller .config .getLlmConfig ().getModelName ();
3939
40- final var response = controller .template ();
41- final var result = response .getOriginalResponse ();
40+ final var response = controller .template ();
41+ final var result = response .getOriginalResponse ();
4242
4343 assertThat (result .getRequestId ()).isNotEmpty ();
4444 assertThat (result .getModuleResults ().getTemplating ().get (0 ).getContent ())
@@ -62,8 +62,7 @@ void testTemplate() {
6262 var orchestrationResult = ((LLMModuleResultSynchronous ) result .getOrchestrationResult ());
6363 assertThat (orchestrationResult .getObject ()).isEqualTo ("chat.completion" );
6464 assertThat (orchestrationResult .getCreated ()).isGreaterThan (1 );
65- assertThat (orchestrationResult .getModel ())
66- .isEqualTo (model );
65+ assertThat (orchestrationResult .getModel ()).isEqualTo (model );
6766 choices = orchestrationResult .getChoices ();
6867 assertThat (choices .get (0 ).getIndex ()).isZero ();
6968 assertThat (choices .get (0 ).getMessage ().getContent ()).isNotEmpty ();
You can’t perform that action at this time.
0 commit comments