This commit is contained in:
Dmytro Liubarskyi 2026-07-17 13:43:28 +02:00
parent 855e06b43c
commit 3ea186f6e5
1 changed files with 4 additions and 2 deletions

View File

@ -33,7 +33,8 @@ class OllamaEmbeddingModelIT extends AbstractOllamaEmbeddingModelInfrastructure
assertThat(response.content().vector()).isNotEmpty();
assertThat(response.content().dimension()).isEqualTo(model.dimension());
assertThat(response.tokenUsage()).isNull();
assertThat(response.tokenUsage()).isNotNull();
assertThat(response.tokenUsage().inputTokenCount()).isPositive();
assertThat(response.finishReason()).isNull();
}
@ -51,7 +52,8 @@ class OllamaEmbeddingModelIT extends AbstractOllamaEmbeddingModelInfrastructure
assertThat(response.content().get(0).dimension()).isEqualTo(model.dimension());
assertThat(response.content().get(1).dimension()).isEqualTo(model.dimension());
assertThat(response.tokenUsage()).isNull();
assertThat(response.tokenUsage()).isNotNull();
assertThat(response.tokenUsage().inputTokenCount()).isPositive();
assertThat(response.finishReason()).isNull();
}