module 6 - ollama

Author: shelajev

module-6/pom.xml

@@ -0,0 +1,64 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+    <groupId>org.testcontainers</groupId>
+    <artifactId>testcontainers-workshop-6</artifactId>
+    <version>0.0.1-SNAPSHOT</version>
+    <name>testcontainers-workshop-6</name>
+    <description>testcontainers-workshop-6</description>
+    <properties>
+        <java.version>17</java.version>
+        <testcontainers.version>1.19.8</testcontainers.version>
+    </properties>
+    <dependencies>
+        <dependency>
+            <groupId>org.junit.jupiter</groupId>
+            <artifactId>junit-jupiter</artifactId>
+            <version>5.10.2</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.testcontainers</groupId>
+            <artifactId>ollama</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.assertj</groupId>
+            <artifactId>assertj-core</artifactId>
+            <version>3.25.3</version>
+        </dependency>
+        <dependency>
+            <groupId>io.rest-assured</groupId>
+            <artifactId>rest-assured</artifactId>
+            <version>5.4.0</version>
+        </dependency>
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-classic</artifactId>
+            <version>1.3.14</version>
+        </dependency>
+    </dependencies>
+
+    <dependencyManagement>
+        <dependencies>
+            <dependency>
+                <groupId>org.testcontainers</groupId>
+                <artifactId>testcontainers-bom</artifactId>
+                <version>${testcontainers.version}</version>
+                <scope>import</scope>
+                <type>pom</type>
+            </dependency>
+        </dependencies>
+    </dependencyManagement>
+
+    <build>
+        <plugins>
+            <plugin>
+                <artifactId>maven-surefire-plugin</artifactId>
+                <version>3.2.5</version>
+            </plugin>
+        </plugins>
+    </build>
+
+</project>

module-6/src/test/java/org/testcontainers/ollama/OllamaContainerTest.java

@@ -0,0 +1,57 @@
+package org.testcontainers.ollama;
+
+import org.junit.Test;
+import org.testcontainers.utility.Base58;
+import org.testcontainers.utility.DockerImageName;
+
+import java.io.IOException;
+
+import static io.restassured.RestAssured.given;
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class OllamaContainerTest {
+
+    @Test
+    public void withDefaultConfig() {
+        try (
+            OllamaContainer ollama = new OllamaContainer("ollama/ollama:0.1.26")
+
+        ) {
+            ollama.start();
+
+            String version = given().baseUri(ollama.getEndpoint()).get("/api/version").jsonPath().get("version");
+            assertThat(version).isEqualTo("0.1.26");
+        }
+    }
+
+    @Test
+    public void downloadModelAndCommitToImage() throws IOException, InterruptedException {
+        String newImageName = "tc-ollama-allminilm-" + Base58.randomString(4).toLowerCase();
+        try (OllamaContainer ollama = new OllamaContainer("ollama/ollama:0.1.26")) {
+            ollama.start();
+            ollama.execInContainer("ollama", "pull", "all-minilm");
+
+            String modelName = given()
+                .baseUri(ollama.getEndpoint())
+                .get("/api/tags")
+                .jsonPath()
+                .getString("models[0].name");
+            assertThat(modelName).contains("all-minilm");
+            ollama.commitToImage(newImageName);
+        }
+        try (
+            OllamaContainer ollama = new OllamaContainer(
+                DockerImageName.parse(newImageName)
+                    .asCompatibleSubstituteFor("ollama/ollama")
+            )
+        ) {
+            ollama.start();
+            String modelName = given()
+                .baseUri(ollama.getEndpoint())
+                .get("/api/tags")
+                .jsonPath()
+                .getString("models[0].name");
+            assertThat(modelName).contains("all-minilm");
+        }
+    }
+}