miss messages

InternLM · Sep 24, 2024 · 81530ee · 81530ee
1 parent e66731a
commit 81530ee
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/lmdeploy/messages.py b/lmdeploy/messages.py
@@ -308,6 +308,31 @@ def __lt__(self, other):
             return self.value < other.value
         return NotImplemented
 
+    def __le__(self, other):
+        if self.__class__ is other.__class__:
+            return self.value <= other.value
+        return NotImplemented
+
+    def __eq__(self, other):
+        if self.__class__ is other.__class__:
+            return self.value == other.value
+        return NotImplemented
+
+    def __ne__(self, other):
+        if self.__class__ is other.__class__:
+            return self.value != other.value
+        return NotImplemented
+
+    def __gt__(self, other):
+        if self.__class__ is other.__class__:
+            return self.value > other.value
+        return NotImplemented
+
+    def __ge__(self, other):
+        if self.__class__ is other.__class__:
+            return self.value >= other.value
+        return NotImplemented
+
 
 @dataclass
 class Response: