add .deep_clone()

guidance-ai · Oct 3, 2024 · d4e35a8 · d4e35a8
1 parent a3a94b4
commit d4e35a8
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/parser/src/constraint.rs b/parser/src/constraint.rs
@@ -65,6 +65,12 @@ impl Constraint {
         }
     }
 
+    pub fn deep_clone(&self) -> Self {
+        let mut copy = self.clone();
+        copy.parser = self.parser.deep_clone();
+        copy
+    }
+
     fn save_progress_and_result(&mut self, res: StepResult) {
         self.last_res = res;
         if self.log_json_progress {

diff --git a/parser/src/earley/parser.rs b/parser/src/earley/parser.rs
@@ -1855,4 +1855,11 @@ impl Parser {
         let mut shared = self.shared.lock().unwrap();
         self.state.model_variables(&mut shared)
     }
+
+    pub fn deep_clone(&self) -> Self {
+        let mut copy = self.clone();
+        let shared = self.shared.lock().unwrap();
+        copy.shared = Arc::new(Mutex::new(shared.clone()));
+        copy
+    }
 }
diff --git a/parser/src/tokenparser.rs b/parser/src/tokenparser.rs
@@ -109,6 +109,13 @@ impl TokenParser {
         })
     }
 
+    // regular .clone() uses a shared lexer state
+    pub fn deep_clone(&self) -> Self {
+        let mut copy = self.clone();
+        copy.parser = self.parser.deep_clone();
+        copy
+    }
+
     pub fn stop_reason(&self) -> StopReason {
         self.stop_reason
     }