Skip to content

Ollama

Testcontainers module for Ollama .

Ollama's usage examples

You can start an Ollama container instance from any Java application by using:

OllamaContainer ollama = new OllamaContainer("ollama/ollama:0.1.26")

Pulling the model

Testcontainers allows executing commands in the container. So, pulling the model is as simple as:

ollama.execInContainer("ollama", "pull", "all-minilm");

Create a new Image

In order to create a new image that contains the model, you can use the following code:

ollama.commitToImage(newImageName);

And use the new image along with Image name Substitution

OllamaContainer ollama = new OllamaContainer(
    DockerImageName.parse(newImageName)
        .asCompatibleSubstituteFor("ollama/ollama")
)

Adding this module to your project dependencies

Add the following dependency to your pom.xml/build.gradle file:

testImplementation "org.testcontainers:ollama:1.20.3"
<dependency>
    <groupId>org.testcontainers</groupId>
    <artifactId>ollama</artifactId>
    <version>1.20.3</version>
    <scope>test</scope>
</dependency>