add StopReason::is_ok()

guidance-ai · Oct 4, 2024 · a99d152 · a99d152
1 parent 6a4380c
commit a99d152
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/parser/src/api.rs b/parser/src/api.rs
@@ -303,6 +303,16 @@ impl StopReason {
             .unwrap()
             .to_string()
     }
+
+    pub fn is_ok(&self) -> bool {
+        matches!(
+            self,
+            StopReason::NotStopped
+                | StopReason::EndOfSentence
+                | StopReason::NoExtension
+                | StopReason::NoExtensionBias
+        )
+    }
 }
 
 #[derive(Clone, Serialize, Deserialize, Debug)]