Add docs

Author: eddumelendez

core/src/main/java/org/testcontainers/containers/DockerModelRunnerContainer.java

@@ -1,17 +1,8 @@
 package org.testcontainers.containers;
 
-import com.github.dockerjava.api.command.InspectContainerResponse;
 import org.testcontainers.containers.wait.strategy.Wait;
 import org.testcontainers.utility.DockerImageName;
 
-import java.io.BufferedReader;
-import java.io.IOException;
-import java.io.InputStreamReader;
-import java.io.OutputStream;
-import java.net.HttpURLConnection;
-import java.net.URL;
-import java.nio.charset.StandardCharsets;
-
 /**
  * Testcontainers proxy container for the Docker Model Runner service
  * provided by Docker Desktop.
@@ -22,9 +13,7 @@
  */
 public class DockerModelRunnerContainer extends SocatContainer {
 
-    public static final String MODEL_RUNNER_ENDPOINT = "model-runner.docker.internal";
-
-    private String model;
+    private static final String MODEL_RUNNER_ENDPOINT = "model-runner.docker.internal";
 
     public DockerModelRunnerContainer(String image) {
         this(DockerImageName.parse(image));
@@ -36,56 +25,11 @@ public DockerModelRunnerContainer(DockerImageName image) {
         waitingFor(Wait.forHttp("/").forResponsePredicate(res -> res.contains("The service is running")));
     }
 
-    @Override
-    protected void containerIsStarted(InspectContainerResponse containerInfo) {
-        pullModel();
-    }
-
-    private void pullModel() {
-        logger().info("Pulling model: {}. Please be patient, no progress bar yet!", this.model);
-        try {
-            String json = String.format("{\"from\":\"%s\"}", this.model);
-            String endpoint = "http://" + getHost() + ":" + getMappedPort(80) + "/models/create";
-
-            URL url = new URL(endpoint);
-            HttpURLConnection connection = (HttpURLConnection) url.openConnection();
-            connection.setRequestMethod("POST");
-            connection.setRequestProperty("Content-Type", "application/json");
-            connection.setDoOutput(true);
-
-            try (OutputStream os = connection.getOutputStream()) {
-                byte[] input = json.getBytes(StandardCharsets.UTF_8);
-                os.write(input, 0, input.length);
-            }
-
-            try (
-                BufferedReader br = new BufferedReader(
-                    new InputStreamReader(connection.getInputStream(), StandardCharsets.UTF_8)
-                )
-            ) {
-                StringBuilder response = new StringBuilder();
-                String responseLine;
-                while ((responseLine = br.readLine()) != null) {
-                    response.append(responseLine.trim());
-                }
-                logger().info(response.toString());
-            }
-        } catch (IOException e) {
-            throw new RuntimeException(e);
-        }
-        logger().info("Finished pulling model: {}", model);
-    }
-
     public String getBaseEndpoint() {
         return "http://" + getHost() + ":" + getMappedPort(80);
     }
 
     public String getOpenAIEndpoint() {
         return getBaseEndpoint() + "/engines";
     }
-
-    public DockerModelRunnerContainer withModel(String modelName) {
-        this.model = modelName;
-        return this;
-    }
 }

core/src/test/java/org/testcontainers/containers/DockerModelRunnerContainerTest.java

@@ -16,13 +16,23 @@ public void pullsModelAndExposesInference() {
         String modelName = "ai/smollm2:360M-Q4_K_M";
 
         try (
+            // container {
             DockerModelRunnerContainer dmr = new DockerModelRunnerContainer("alpine/socat:1.7.4.3-r0")
-                .withModel(modelName)
+            // }
         ) {
             dmr.start();
 
-            Response response = RestAssured.get(dmr.getBaseEndpoint() + "/models").thenReturn();
-            assertThat(response.body().jsonPath().getList("tags.flatten()")).contains(modelName);
+            // pullModel {
+            RestAssured
+                .given()
+                .body(String.format("{\"from\":\"%s\"}", modelName))
+                .post(dmr.getBaseEndpoint() + "/models/create")
+                .then()
+                .statusCode(200);
+            // }
+
+            Response modelResponse = RestAssured.get(dmr.getBaseEndpoint() + "/models").thenReturn();
+            assertThat(modelResponse.body().jsonPath().getList("tags.flatten()")).contains(modelName);
 
             Response openAiResponse = RestAssured.get(dmr.getOpenAIEndpoint() + "/v1/models").prettyPeek().thenReturn();
             assertThat(openAiResponse.body().jsonPath().getList("data.id")).contains(modelName);

docs/modules/docker_model_runner.md

@@ -0,0 +1,41 @@
+# Docker Model Runner
+
+This module helps connect to [Docker Model Runner](https://docs.docker.com/desktop/features/model-runner/)
+provided by Docker Desktop 4.40.0.
+
+## DockerModelRunner's usage examples
+
+You can start a Docker Model Runner proxy container instance from any Java application by using:
+
+<!--codeinclude-->
+[Create a DockerModelRunnerContainer](../../core/src/test/java/org/testcontainers/containers/DockerModelRunnerContainerTest.java) inside_block:container
+<!--/codeinclude-->
+
+### Pulling the model
+
+Pulling the model is as simple as:
+
+<!--codeinclude-->
+[Pull model](../../core/src/test/java/org/testcontainers/containers/DockerModelRunnerContainerTest.java) inside_block:pullModel
+<!--/codeinclude-->
+
+## Adding this module to your project dependencies
+
+*Docker Model Runner support is part of the core Testcontainers library.*
+
+Add the following dependency to your `pom.xml`/`build.gradle` file:
+
+=== "Gradle"
+    ```groovy
+    testImplementation "org.testcontainers:testcontainers:{{latest_version}}"
+    ```
+=== "Maven"
+    ```xml
+    <dependency>
+        <groupId>org.testcontainers</groupId>
+        <artifactId>testcontainers</artifactId>
+        <version>{{latest_version}}</version>
+        <scope>test</scope>
+    </dependency>
+    ```
+

mkdocs.yml

@@ -82,6 +82,7 @@ nav:
           - modules/chromadb.md
           - modules/consul.md
           - modules/docker_compose.md
+          - modules/docker_model_runner.md
           - modules/elasticsearch.md
           - modules/gcloud.md
           - modules/grafana.md