Skip to content

Commit f3aefae

Browse files
committed
fix: build
1 parent eb05f66 commit f3aefae

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

backend/src/main/java/ch/xxx/aidoclibchat/usecase/service/CodeGenerationService.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ public String generateTest(String url, Optional<String> testUrlOpt) {
124124
LOGGER.info(response.getResult().getOutput().getText());
125125
}
126126
LOGGER.info("Prompt tokens: " + response.getMetadata().getUsage().getPromptTokens());
127-
LOGGER.info("Generation tokens: " + response.getMetadata().getUsage().getGenerationTokens());
127+
LOGGER.info("Generation tokens: " + response.getMetadata().getUsage().getCompletionTokens());
128128
LOGGER.info("Total tokens: " + response.getMetadata().getUsage().getTotalTokens());
129129
LOGGER.info("Time in seconds: {}", (Instant.now().toEpochMilli() - start.toEpochMilli()) / 1000.0);
130130
return response.getResult().getOutput().getText();

backend/src/main/java/ch/xxx/aidoclibchat/usecase/service/ImageService.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,8 @@
3131
import org.springframework.ai.chat.client.ChatClient.Builder;
3232
import org.springframework.ai.chat.messages.UserMessage;
3333
import org.springframework.ai.chat.prompt.Prompt;
34+
import org.springframework.ai.content.Media;
3435
import org.springframework.ai.document.Document;
35-
import org.springframework.ai.model.Media;
3636
import org.springframework.beans.factory.annotation.Value;
3737
import org.springframework.core.io.ByteArrayResource;
3838
import org.springframework.stereotype.Service;
@@ -119,8 +119,8 @@ private ResultData createAIResult(ImageQueryDto imageDto) {
119119
imageDto = this.resizeImage(imageDto);
120120
}
121121
var prompt = new Prompt(
122-
new UserMessage(imageDto.getQuery(), new Media(MimeType.valueOf(imageDto.getImageType().getMediaType()),
123-
new ByteArrayResource(imageDto.getImageContent()))));
122+
new UserMessage(imageDto.getQuery(), List.of(new Media(MimeType.valueOf(imageDto.getImageType().getMediaType()),
123+
new ByteArrayResource(imageDto.getImageContent())))));
124124

125125
var response = this.chatClient.prompt(prompt).call().chatResponse();
126126
var resultData = new ResultData(response.getResult().getOutput().getText(), imageDto);

0 commit comments

Comments
 (0)