diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 624da5ec..888d2ded 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -2,7 +2,7 @@
 // README at: https://github.com/devcontainers/templates/tree/main/src/docker-existing-dockerfile
 {
     "name": "ten_agent_dev",
-    "image": "ghcr.io/ten-framework/ten_agent_build:0.2.2",
+    "image": "ghcr.io/ten-framework/ten_agent_build:0.2.3",
     "customizations": {
         "vscode": {
             "extensions": [
diff --git a/Dockerfile b/Dockerfile
index 5777ae64..d6777ef7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ghcr.io/ten-framework/ten_agent_build:0.2.2 AS builder
+FROM ghcr.io/ten-framework/ten_agent_build:0.2.3 AS builder
 
 ARG SESSION_CONTROL_CONF=session_control.conf
 
@@ -22,7 +22,7 @@ RUN apt-get clean && apt-get update && apt-get install -y --no-install-recommend
     python3-venv \
     python3-pip \
     python3-dev \
-    jq \
+    jq vim \
     ca-certificates \
     && apt-get clean && rm -rf /var/lib/apt/lists/* && rm -rf /tmp/*
 
diff --git a/docker-compose.yml b/docker-compose.yml
index 7e705c82..cd4168bb 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,6 +1,6 @@
 services:
   ten_agent_dev:
-    image: ghcr.io/ten-framework/ten_agent_build:0.2.2
+    image: ghcr.io/ten-framework/ten_agent_build:0.2.3
     container_name: ten_agent_dev
     platform: linux/amd64
     tty: true