fix proto import in docker

services/aggregator/Dockerfile

@@ -8,7 +8,7 @@ WORKDIR /app
 RUN apt-get update && apt-get install -y --no-install-recommends \
     curl \
     && curl -fsSL https://github.com/grpc-ecosystem/grpc-health-probe/releases/download/v0.4.24/grpc_health_probe-linux-amd64 \
-       -o /bin/grpc_health_probe \
+    -o /bin/grpc_health_probe \
     && chmod +x /bin/grpc_health_probe \
     && rm -rf /var/lib/apt/lists/*
 
@@ -22,7 +22,8 @@ RUN python -m grpc_tools.protoc \
     -I/app/proto \
     --python_out=/app/shared \
     --grpc_python_out=/app/shared \
-    /app/proto/metrics.proto
+    /app/proto/metrics.proto \
+    && sed -i 's/^import metrics_pb2/from shared import metrics_pb2/' /app/shared/metrics_pb2_grpc.py
 
 COPY services/aggregator /app/services/aggregator
 

services/alerts/Dockerfile

@@ -18,7 +18,8 @@ RUN python -m grpc_tools.protoc \
     -I/app/proto \
     --python_out=/app/shared \
     --grpc_python_out=/app/shared \
-    /app/proto/metrics.proto
+    /app/proto/metrics.proto \
+    && sed -i 's/^import metrics_pb2/from shared import metrics_pb2/' /app/shared/metrics_pb2_grpc.py
 
 COPY services/alerts /app/services/alerts
 

services/collector/Dockerfile

@@ -26,7 +26,8 @@ RUN python -m grpc_tools.protoc \
     -I/app/proto \
     --python_out=/app/shared \
     --grpc_python_out=/app/shared \
-    /app/proto/metrics.proto
+    /app/proto/metrics.proto \
+    && sed -i 's/^import metrics_pb2/from shared import metrics_pb2/' /app/shared/metrics_pb2_grpc.py
 
 # Copy service code
 COPY services/collector /app/services/collector

services/gateway/Dockerfile

@@ -18,7 +18,8 @@ RUN python -m grpc_tools.protoc \
     -I/app/proto \
     --python_out=/app/shared \
     --grpc_python_out=/app/shared \
-    /app/proto/metrics.proto
+    /app/proto/metrics.proto \
+    && sed -i 's/^import metrics_pb2/from shared import metrics_pb2/' /app/shared/metrics_pb2_grpc.py
 
 COPY services/gateway /app/services/gateway
 COPY services/aggregator/__init__.py /app/services/aggregator/__init__.py