From b2e22a0069910151056d000cca4a5363df6a20ca Mon Sep 17 00:00:00 2001 From: Eric Deandrea Date: Tue, 10 Dec 2024 11:30:47 -0500 Subject: [PATCH 1/2] Adding constant and overridable method to OllamaContainer --- .../java/org/testcontainers/ollama/OllamaContainer.java | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/modules/ollama/src/main/java/org/testcontainers/ollama/OllamaContainer.java b/modules/ollama/src/main/java/org/testcontainers/ollama/OllamaContainer.java index 1c739b95a6b..9abb22b8200 100644 --- a/modules/ollama/src/main/java/org/testcontainers/ollama/OllamaContainer.java +++ b/modules/ollama/src/main/java/org/testcontainers/ollama/OllamaContainer.java @@ -23,6 +23,7 @@ public class OllamaContainer extends GenericContainer { private static final DockerImageName DOCKER_IMAGE_NAME = DockerImageName.parse("ollama/ollama"); + private static final int OLLAMA_PORT = 11434; public OllamaContainer(String image) { this(DockerImageName.parse(image)); @@ -49,7 +50,7 @@ public OllamaContainer(DockerImageName dockerImageName) { }); } } - withExposedPorts(11434); + withExposedPorts(OLLAMA_PORT); } /** @@ -74,7 +75,11 @@ public void commitToImage(String imageName) { } } + public int getPort() { + return getMappedPort(OLLAMA_PORT); + } + public String getEndpoint() { - return "http://" + getHost() + ":" + getMappedPort(11434); + return "http://" + getHost() + ":" + getPort(); } } From 8f8665e7e2c85f2bc014ce1c9e3d0606da441141 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Edd=C3=BA=20Mel=C3=A9ndez=20Gonzales?= Date: Tue, 10 Dec 2024 13:02:01 -0600 Subject: [PATCH 2/2] Update modules/ollama/src/main/java/org/testcontainers/ollama/OllamaContainer.java --- .../src/main/java/org/testcontainers/ollama/OllamaContainer.java | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/ollama/src/main/java/org/testcontainers/ollama/OllamaContainer.java b/modules/ollama/src/main/java/org/testcontainers/ollama/OllamaContainer.java index 9abb22b8200..db611a34a39 100644 --- a/modules/ollama/src/main/java/org/testcontainers/ollama/OllamaContainer.java +++ b/modules/ollama/src/main/java/org/testcontainers/ollama/OllamaContainer.java @@ -23,6 +23,7 @@ public class OllamaContainer extends GenericContainer { private static final DockerImageName DOCKER_IMAGE_NAME = DockerImageName.parse("ollama/ollama"); + private static final int OLLAMA_PORT = 11434; public OllamaContainer(String image) {