Skip to content

Commit 945aabb

Browse files
committed
Fix formatting issues in BedrockTitanEmbeddingModel.java
1 parent 8da10e6 commit 945aabb

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

models/spring-ai-bedrock/src/test/java/org/springframework/ai/bedrock/titan/BedrockTitanEmbeddingModelIT.java

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,13 +40,18 @@
4040

4141
import static org.assertj.core.api.Assertions.assertThat;
4242

43+
import io.micrometer.observation.tck.TestObservationRegistry;
44+
4345
@SpringBootTest
4446
@RequiresAwsCredentials
4547
class BedrockTitanEmbeddingModelIT {
4648

4749
@Autowired
4850
private BedrockTitanEmbeddingModel embeddingModel;
4951

52+
@Autowired
53+
TestObservationRegistry observationRegistry;
54+
5055
@Test
5156
void singleEmbedding() {
5257
assertThat(this.embeddingModel).isNotNull();
@@ -82,8 +87,9 @@ public TitanEmbeddingBedrockApi titanEmbeddingApi() {
8287
}
8388

8489
@Bean
85-
public BedrockTitanEmbeddingModel titanEmbedding(TitanEmbeddingBedrockApi titanEmbeddingApi) {
86-
return new BedrockTitanEmbeddingModel(titanEmbeddingApi);
90+
public BedrockTitanEmbeddingModel titanEmbedding(TitanEmbeddingBedrockApi titanEmbeddingApi,
91+
TestObservationRegistry observationRegistry) {
92+
return new BedrockTitanEmbeddingModel(titanEmbeddingApi, observationRegistry);
8793
}
8894

8995
}

0 commit comments

Comments
 (0)