Nicer standalone engine inferface (#4480)

c614dbdf · Yinghai Lu · GitHub · 927ca935 · c614dbdf
Unverified Commit c614dbdf authored Mar 17, 2025 by Yinghai Lu Committed by GitHub Mar 17, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

python/sglang/srt/entrypoints/engine.py python/sglang/srt/entrypoints/engine.py +7 -0

No files found.
--- a/python/sglang/srt/entrypoints/engine.py
+++ b/python/sglang/srt/entrypoints/engine.py
@@ -249,6 +249,13 @@ class Engine:
        """Shutdown the engine"""
        kill_process_tree(os.getpid(), include_parent=False)

+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.shutdown()
+        return False
+
    def start_profile(self):
        loop = asyncio.get_event_loop()
        loop.run_until_complete(self.tokenizer_manager.start_profile())