Enable Ollama Testcontainers in chat-models tests

ThomasVitale · Aug 24, 2024 · 375f172 · 375f172
1 parent 11b6a25
commit 375f172
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...els-ollama/src/test/java/com/thomasvitale/ai/spring/ChatModelsOllamaApplicationTests.java b/...els-ollama/src/test/java/com/thomasvitale/ai/spring/ChatModelsOllamaApplicationTests.java
@@ -5,13 +5,14 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.autoconfigure.web.reactive.AutoConfigureWebTestClient;
 import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.context.annotation.Import;
 import org.springframework.test.web.reactive.server.WebTestClient;
 
 import static org.assertj.core.api.Assertions.assertThat;
 
 @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
 @AutoConfigureWebTestClient(timeout = "60s")
-//@Import(TestChatModelsOllamaApplication.class)
+@Import(TestcontainersConfiguration.class)
 class ChatModelsOllamaApplicationTests {
 
     @Autowired