chore: import upstream snapshot with attribution

python/ray/serve/tests/unit/test_llm_imports.py

@@ -0,0 +1,37 @@
+import sys
+
+import pytest
+
+# skip the test if vllm is installed
+HAS_VLLM = False
+try:
+    import vllm  # noqa: F401
+
+    HAS_VLLM = True
+except ImportError:
+    pass
+
+
+@pytest.mark.skipif(not HAS_VLLM, reason="vllm is not installed")
+def test_serve_llm_import_does_not_error():
+    # expected ImportError because of missing
+    # dependencies without ray[llm] dependencies
+    with pytest.raises(ImportError):
+        import ray.serve.llm  # noqa: F401
+    with pytest.raises(ImportError):
+        from ray.serve.llm import (
+            LLMConfig,  # noqa: F401
+        )
+    with pytest.raises(ImportError):
+        from ray.serve.llm.deployment import (
+            LLMServer,  # noqa: F401
+        )
+    with pytest.raises(ImportError):
+        from ray.serve.llm import (
+            build_llm_deployment,  # noqa: F401
+            build_openai_app,  # noqa: F401
+        )
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(["-v", __file__]))