diff --git a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
index f6bcbae8cd..a4a336885e 100644
--- a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
+++ b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
@@ -10,6 +10,7 @@ RUN apt update && apt install -y -V ca-certificates lsb-release wget && \
     apt install -y \
     jq \
     libarrow-dev && \
+    npm install -g yarn && \
     apt remove -y lsb-release wget && \
     apt-get clean && rm -rf /var/cache/apt/lists
 
diff --git a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
index e9b9f8d78c..e41806d22e 100644
--- a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
+++ b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
@@ -3,7 +3,7 @@ FROM python:3.11-slim-bullseye
 RUN pip install --no-cache-dir pip --upgrade
 RUN pip install --no-cache-dir pip-tools
 
-RUN apt update && apt install -y -V ca-certificates lsb-release wget make git && \
+RUN apt update && apt install -y -V ca-certificates lsb-release wget make git npm && \
     wget https://apache.jfrog.io/artifactory/arrow/$(lsb_release --id --short | tr 'A-Z' 'a-z')/apache-arrow-apt-source-latest-$(lsb_release --codename --short).deb && \
     apt install -y -V ./apache-arrow-apt-source-latest-$(lsb_release --codename --short).deb && apt update && \
     apt install -y \
@@ -13,6 +13,10 @@ RUN apt update && apt install -y -V ca-certificates lsb-release wget make git &&
     apt-get clean && rm -rf /var/cache/apt/lists
 
 COPY . .
+WORKDIR /ui
+RUN yarn install && yarn build:lib
+
+WORKDIR /
 RUN make install-python build-ui
 
 # modify permissions to support running with a random uid