From 8e8dd984d807dac30a752ae3025925883b11ebc8 Mon Sep 17 00:00:00 2001 From: Nick Hill Date: Mon, 27 Mar 2023 00:25:15 -0700 Subject: [PATCH] feat(server): Add mypy-protobuf (#141) Generates .pyi files for protobuf stubs which provide strong typing information. Very helpful for IDE auto-completion, etc. --- server/Makefile | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/server/Makefile b/server/Makefile index 69ef9bc5..d2a8cf7a 100644 --- a/server/Makefile +++ b/server/Makefile @@ -3,9 +3,10 @@ flash_att_commit := 4d87e4d875077ad9efd25030efa4ab0ba92c19e1 gen-server: # Compile protos - pip install grpcio-tools==1.51.1 --no-cache-dir + pip install grpcio-tools==1.51.1 mypy-protobuf==3.4.0 'types-protobuf>=3.20.4' --no-cache-dir mkdir text_generation_server/pb || true - python -m grpc_tools.protoc -I../proto --python_out=text_generation_server/pb --grpc_python_out=text_generation_server/pb ../proto/generate.proto + python -m grpc_tools.protoc -I../proto --python_out=text_generation_server/pb \ + --grpc_python_out=text_generation_server/pb --mypy_out=text_generation_server/pb ../proto/generate.proto find text_generation_server/pb/ -type f -name "*.py" -print0 -exec sed -i -e 's/^\(import.*pb2\)/from . \1/g' {} \; touch text_generation_server/pb/__init__.py