feat(choice): escape strings

dottxt-ai · Nov 27, 2024 · 33df55a · 33df55a
1 parent 5e78c7b
commit 33df55a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/outlines/generate/choice.py b/outlines/generate/choice.py
@@ -1,4 +1,5 @@
 import json as pyjson
+import re
 from enum import Enum
 from functools import singledispatch
 from typing import Callable, List, Union
@@ -19,6 +20,7 @@ def choice(
     if isinstance(choices, type(Enum)):
         regex_str = build_regex_from_schema(pyjson.dumps(get_schema_from_enum(choices)))
     else:
+        choices = [re.escape(choice) for choice in choices]  # type: ignore
         regex_str = r"(" + r"|".join(choices) + r")"
 
     generator = regex(model, regex_str, sampler)