fix: prefer direct path to pb files in script

neptune-ai · Dec 13, 2024 · f2d5bd3 · f2d5bd3
1 parent 43b7273
commit f2d5bd3
Showing 1 changed file with 15 additions and 15 deletions.
diff --git a/scripts/generate_proto.sh b/scripts/generate_proto.sh
@@ -4,37 +4,37 @@ set -euxo pipefail
 
 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
 SRC_DIR="${SCRIPT_DIR}/../src"
-NEPTUNE_PROJECT_DIR="${1:-}"
+PROTOBUF_SRC_DIR="${1:-}"
 
-if [ -z "${NEPTUNE_PROJECT_DIR}" ]; then
-    echo "Usage: $0 <neptune_project_dir>"
+if [ -z "${PROTBUF_SRC_DIR}" ]; then
+    echo "Usage: $0 <protobuf_src_dir>"
     exit 1
 fi
 
 python -m grpc_tools.protoc \
   --python_out="${SRC_DIR}/neptune_api/proto" \
   --mypy_out="${SRC_DIR}/neptune_api/proto" \
-  --proto_path="${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto" \
-  "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto/google_rpc/"*".proto" \
-  "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto/neptune_pb/ingest/v1/"*".proto" \
-  "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto/neptune_pb/ingest/v1/pub/"*".proto"
+  --proto_path="${PROTBUF_SRC_DIR}" \
+  "${PROTBUF_SRC_DIR}/google_rpc/"*".proto" \
+  "${PROTBUF_SRC_DIR}/neptune_pb/ingest/v1/"*".proto" \
+  "${PROTBUF_SRC_DIR}/neptune_pb/ingest/v1/pub/"*".proto"
 
 python -m grpc_tools.protoc \
   --python_out="${SRC_DIR}/neptune_retrieval_api/proto" \
   --mypy_out="${SRC_DIR}/neptune_retrieval_api/proto" \
-  --proto_path="${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto" \
-  "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto/neptune_pb/api/v1/model/"*".proto"
+  --proto_path="${PROTBUF_SRC_DIR}" \
+  "${PROTBUF_SRC_DIR}/neptune_pb/api/v1/model/"*".proto"
 
 protol --create-package --in-place \
   --python-out "${SRC_DIR}/neptune_api/proto" \
   protoc \
-  --proto-path "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto" \
-  "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto/google_rpc/"*".proto" \
-  "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto/neptune_pb/ingest/v1/"*".proto" \
-  "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto/neptune_pb/ingest/v1/pub/"*".proto"
+  --proto-path "${PROTBUF_SRC_DIR}" \
+  "${PROTBUF_SRC_DIR}/google_rpc/"*".proto" \
+  "${PROTBUF_SRC_DIR}/neptune_pb/ingest/v1/"*".proto" \
+  "${PROTBUF_SRC_DIR}/neptune_pb/ingest/v1/pub/"*".proto"
 
 protol --create-package --in-place \
   --python-out "${SRC_DIR}/neptune_retrieval_api/proto" \
   protoc \
-  --proto-path "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto" \
-  "${NEPTUNE_PROJECT_DIR}/libs/models/leaderboard-proto/src/main/proto/neptune_pb/api/v1/model/"*".proto"
+  --proto-path "${PROTBUF_SRC_DIR}" \
+  "${PROTBUF_SRC_DIR}/neptune_pb/api/v1/model/"*".proto"