Merge pull request #37 from EnvisionMindCa/codex/edit-pdf-files-using-terminal-tool

README.md

@@ -60,3 +60,11 @@ python -m bot.discord_bot
 
 Any attachments sent to the bot are uploaded to the VM and the bot replies with
 their paths so they can be used in later messages.
+
+## VM Configuration
+
+The Linux VM used for tool execution runs inside a Docker container. By default
+it pulls the image defined by the ``VM_IMAGE`` environment variable, falling
+back to ``python:3.11-slim``. This base image includes Python and ``pip`` so
+packages can be installed immediately. The container has network access enabled
+which allows fetching additional dependencies as needed.

src/config.py

@@ -9,6 +9,7 @@ OLLAMA_HOST: Final[str] = os.getenv("OLLAMA_HOST", "http://localhost:11434")
 MAX_TOOL_CALL_DEPTH: Final[int] = 5
 NUM_CTX: Final[int] = int(os.getenv("OLLAMA_NUM_CTX", "32000"))
 UPLOAD_DIR: Final[str] = os.getenv("UPLOAD_DIR", str(Path.cwd() / "uploads"))
+VM_IMAGE: Final[str] = os.getenv("VM_IMAGE", "python:3.11-slim")
 
 SYSTEM_PROMPT: Final[str] = (
     "You are Starlette, a professional AI assistant with advanced tool orchestration. "

src/vm.py

@@ -8,7 +8,7 @@ from pathlib import Path
 
 from threading import Lock
 
-from .config import UPLOAD_DIR
+from .config import UPLOAD_DIR, VM_IMAGE
 
 from .log import get_logger
 
@@ -16,10 +16,14 @@ _LOG = get_logger(__name__)
 
 
 class LinuxVM:
-    """Manage a lightweight Linux VM using Docker."""
+    """Manage a lightweight Docker-based VM.
+
+    The default image provides Python and pip so packages can be installed
+    immediately. A custom image can be supplied via ``VM_IMAGE``.
+    """
 
     def __init__(
-        self, image: str = "ubuntu:latest", host_dir: str = UPLOAD_DIR
+        self, image: str = VM_IMAGE, host_dir: str = UPLOAD_DIR
     ) -> None:
         self._image = image
         self._name = f"chat-vm-{uuid.uuid4().hex[:8]}"