Support both x86_64 and arm

Author: jakubno

codegen.Dockerfile

@@ -15,10 +15,16 @@ FROM python:3.9
 # Set working directory
 WORKDIR /workspace
 
-ENV PROTOC_ZIP=protoc-29.3-linux-aarch_64.zip
-
-RUN curl -OL https://github.com/protocolbuffers/protobuf/releases/download/v29.3/$PROTOC_ZIP
-RUN unzip -o $PROTOC_ZIP -d /usr/local bin/protoc
+ENV PROTOC_VERSION=29.3
+RUN ARCH=$(uname -m) && \
+    case "$ARCH" in \
+        x86_64) PROTOC_ARCH="x86_64" ;; \
+        arm64|aarch64) PROTOC_ARCH="aarch_64" ;; \
+        *) echo "Unsupported architecture: $ARCH" && exit 1 ;; \
+    esac && \
+    curl -LO https://github.com/protocolbuffers/protobuf/releases/download/v${PROTOC_VERSION}/protoc-${PROTOC_VERSION}-linux-${PROTOC_ARCH}.zip && \
+    unzip -o protoc-${PROTOC_VERSION}-linux-${PROTOC_ARCH}.zip -d /usr/local && \
+    rm protoc-${PROTOC_VERSION}-linux-${PROTOC_ARCH}.zip
 
 # Copy installed Go deps from previous build step
 COPY --from=0 /go /go