update

lvhan028 · Sep 24, 2024 · 468446a · 468446a
1 parent 579ec09
commit 468446a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/lmdeploy/messages.py b/lmdeploy/messages.py
@@ -303,6 +303,11 @@ class ResponseType(enum.Enum):
     INPUT_LENGTH_ERROR = enum.auto()
     INTERNAL_ENGINE_ERROR = enum.auto()
 
+    def __lt__(self, other):
+        if self.__class__ is other.__class__:
+            return self.value < other.value
+        return NotImplemented
+
 
 @dataclass
 class Response: