o ۾i @s<dZddlZddlmZddZedkreedSdS)a Compile vLLM protobuf definitions into Python code. This script uses grpcio-tools to generate *_pb2.py, *_pb2_grpc.py, and *_pb2.pyi (type stubs) files from the vllm_engine.proto definition. NOTE: Proto compilation happens automatically during package build (via setup.py). This script is provided for developers who want to regenerate protos manually, e.g., after modifying vllm_engine.proto. Usage: python vllm/grpc/compile_protos.py Requirements: pip install grpcio-tools N)Pathc Csttj}|jj}|d}|std|dStd|td|zzddlm}|dd |d |d |d |t|dg}|dkrd }|d|d|dfD]}|ru| }| dsu|d}| ||q[tdtd|dtd|dtd|ddWdStd||WSt ytdtdYdSt y} ztd| WYd} ~ dSd} ~ ww)zCompile protobuf definitions.zvllm_engine.protozError: Proto file not found at zCompiling protobuf: zOutput directory: r)protoczgrpc_tools.protocz --proto_path=z --python_out=z--grpc_python_out=z --pyi_out=zk# SPDX-License-Identifier: Apache-2.0 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project zvllm_engine_pb2.pyzvllm_engine_pb2_grpc.pyzvllm_engine_pb2.pyiz# SPDX-License-Identifierz# mypy: ignore-errors u$✓ Protobuf compilation successful!z Generated: z (type stubs)zError: protoc returned z!Error: grpcio-tools not installedz&Install with: pip install grpcio-toolszError during compilation: N)r__file__parentexistsprint grpc_toolsrmainstr read_text startswith write_text ImportError Exception) script_dirvllm_package_root proto_filerresult spdx_headergenerated_filecontentheadererL/home/ubuntu/.local/lib/python3.10/site-packages/vllm/grpc/compile_protos.pycompile_protoss`      r__main__)__doc__syspathlibrr__name__exitrrrrs D