small bug with prev commit

Author: atharvas

src/datasmith/agents/context_synthesis.py

@@ -9,14 +9,14 @@
 from datetime import datetime, timezone
 from pathlib import Path
 
-import docker
 import dspy
-from docker.errors import APIError, ImageNotFound, NotFound
 
+import docker
 from datasmith.agents.tool_executor import ContainerToolExecutor
 from datasmith.docker.context import BuildResult, ContextRegistry, DockerContext
 from datasmith.docker.orchestrator import gen_run_labels
 from datasmith.docker.validation import Task
+from docker.errors import APIError, ImageNotFound, NotFound
 
 logger = logging.getLogger(__name__)
 
@@ -503,7 +503,7 @@ def agent_build_and_validate(  # noqa: C901
             timeout_s=args.build_timeout,
             tail_chars=args.tail_chars,
             probe=True,
-            pull=True,
+            pull=False,
             force=False,  # don't rebuild if already present
             run_labels=run_labels,
         )

src/datasmith/docker/context.py

@@ -16,10 +16,9 @@
 from typing import Any, ClassVar
 
 import docker
-from docker.errors import APIError, DockerException, ImageNotFound
-
 from datasmith.execution.utils import _get_commit_info
 from datasmith.logging_config import get_logger
+from docker.errors import APIError, DockerException, ImageNotFound
 
 logger = get_logger("docker.context")
 
@@ -226,6 +225,7 @@ def build_container(
         if not image_exists:
             cache_from = None
             if base_image := os.environ.get("DOCKER_CACHE_FROM", None):
+                logger.info("Using DOCKER_CACHE_FROM='%s' for build cache.", base_image)
                 build_args = {**build_args, "BASE_IMAGE": base_image}
                 cache_from = [base_image]
 
@@ -305,6 +305,7 @@ def build_container_streaming(  # noqa: C901
 
             cache_from = None
             if base_image := os.environ.get("DOCKER_CACHE_FROM", None):
+                logger.info("Using DOCKER_CACHE_FROM='%s' for build cache.", base_image)
                 build_args = {**build_args, "BASE_IMAGE": base_image}
                 cache_from = [base_image]