Fix minor issues related to total tokens calculation

This commit is contained in:
Carl-Robert Linnupuu 2023-11-15 00:44:13 +02:00
parent 346218b512
commit 318dd4286a
24 changed files with 424 additions and 397 deletions

View file

@ -49,7 +49,11 @@ public class StandardChatToolWindowTabPanelTest extends IntegrationTest {
panel.sendMessage(message);
await().atMost(5, SECONDS).until(() -> !panel.isStreaming());
await().atMost(5, SECONDS)
.until(() -> {
var messages = conversation.getMessages();
return !messages.isEmpty() && "Hello!".equals(messages.get(0).getResponse());
});
var encodingManager = EncodingManager.getInstance();
assertThat(panel.getTokenDetails()).extracting(
"systemPromptTokens",