Fix finish reason (#2013)

78c1d644 · Lianmin Zheng · GitHub · 027e6524 · 78c1d644
Unverified Commit 78c1d644 authored Nov 11, 2024 by Lianmin Zheng Committed by GitHub Nov 11, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

python/sglang/srt/managers/schedule_batch.py python/sglang/srt/managers/schedule_batch.py +3 -1

No files found.
--- a/python/sglang/srt/managers/schedule_batch.py
+++ b/python/sglang/srt/managers/schedule_batch.py
@@ -107,12 +107,14 @@ class FINISH_LENGTH(BaseFinishReason):
 class FINISH_ABORT(BaseFinishReason):
-    def __init__(self):
+    def __init__(self, message="Unknown error"):
        super().__init__(is_error=True)
+        self.message = message
    def to_json(self):
        return {
            "type": "abort",
+            "message": self.message,
        }