First pass at a fix for guidance-ai#863

hudson-ai · Jun 2, 2024 · f17e434 · f17e434
1 parent c009b1a
commit f17e434
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/guidance/_parser.py b/guidance/_parser.py
@@ -594,10 +594,14 @@ def _record_captures_from_root(self, initial_item, data, log_prob_data):
 
     def _compute_parse_tree(self, initial_pos, initial_item, reversed_state_sets):
         stack = [(initial_pos, initial_item)]
+        seen = set()
 
         while stack:
             pos, item = stack.pop()
-
+            if (pos, item) in seen:
+                # Skip items we have already processed
+                continue
+            seen.add((pos, item))
             # compute the children for this item
             assert self._compute_children(pos, item, reversed_state_sets)