diff --git a/Dockerfile b/Dockerfile
index 1d08cf6a..f4a92a45 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,10 +18,10 @@ COPY package.json ./
 # Copy all package.json files from the monorepo
 # This ensures all workspace packages are correctly identified
 COPY apps/dbagent/package.json ./apps/dbagent/
-COPY packages/components/package.json ./packages/components/
-COPY packages/theme/package.json ./packages/theme/
-COPY configs/eslint-config/package.json ./configs/eslint-config/
-COPY configs/tsconfig/package.json ./configs/tsconfig/
+#COPY packages/components/package.json ./packages/components/
+#COPY packages/theme/package.json ./packages/theme/
+#COPY configs/eslint-config/package.json ./configs/eslint-config/
+#COPY configs/tsconfig/package.json ./configs/tsconfig/
 
 # Now copy the source code of all workspace packages
 COPY packages/ ./packages/
diff --git a/docker-compose.yml b/docker-compose.yml
index 364983a1..4c3ead73 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -12,7 +12,7 @@ services:
       - db_network
 
   xata-agent:
-    image: xataio/agent:0.2.2
+    image: xataio/agent:0.3.1
     environment:
       DATABASE_URL: postgresql://${POSTGRES_USER:-dbagent}:${POSTGRES_PASSWORD:-changeme}@postgres:5432/${POSTGRES_DB:-dbagent}
       NODE_ENV: production