Add lazy import to nixl.py

container/deps/vllm/vllm_v0.7.2-triton-kv-disagg-patch.patch

@@ -618,11 +618,9 @@ index 00000000..bc962726
 +import msgspec
 +import time
 +import uuid
-+from nixl_wrapper import nixl_wrapper as NixlWrapper
 +
 +logger = init_logger(__name__)
 +
-+
 +def nixl_wrapper_init_patch(self, agent_name, nixl_config):
 +    logger.info("Initializing patched NixlWrapper")
 +    import nixl_bindings as nixl
@@ -646,9 +644,17 @@ index 00000000..bc962726
 +                        "WRITE_NOTIF": nixl.NIXL_WR_NOTIF,
 +                        "READ_NOTIF":  nixl.NIXL_RD_NOTIF}
 +
-+    print("Initializied NIXL agent:", agent_name)
++    logger.info("Initializied NIXL agent: %s", agent_name)
++
 +
-+NixlWrapper.__init__ = nixl_wrapper_init_patch
++# Lazy import nixl_wrapper to avoid loading nixl_bindings if nixl is not used
++try:
++    from nixl_wrapper import nixl_wrapper as NixlWrapper # type: ignore
++    logger.info("NIXL is available")
++    NixlWrapper.__init__ = nixl_wrapper_init_patch
++except ImportError:
++    logger.warning("NIXL is not available")
++    NixlWrapper = None
 +
 +
 +
@@ -665,6 +671,10 @@ index 00000000..bc962726
 +class TritonNixlConnector:
 +    def __init__(self, vllm_config: VllmConfig, engine_id: str, rank: int):
 +        self.vllm_config = vllm_config
++        if NixlWrapper is None:
++            logger.error("NIXL is not available")
++            raise RuntimeError("NIXL is not available")
++        logger.info("Initializing NIXL wrapper")
 +        self.nixl_wrapper = NixlWrapper(str(uuid.uuid4()), None)
 +
 +        self.num_layers = None

examples/python_rs/llm/vllm/__init__.py

-# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-# SPDX-License-Identifier: Apache-2.0
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.