Fix json processor

dottxt-ai · Feb 13, 2024 · a7e14e0 · a7e14e0
1 parent c5998f8
commit a7e14e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/outlines/generate/processors.py b/outlines/generate/processors.py
@@ -7,7 +7,7 @@
 from numpy.typing import NDArray
 
 from outlines.fsm.fsm import CFGFSM, FSM, FSMState, RegexFSM
-from outlines.fsm.json_schema import build_regex_from_object
+from outlines.fsm.json_schema import build_regex_from_schema
 from outlines.models.tokenizer import Tokenizer
 
 
@@ -80,7 +80,7 @@ def __init__(self, schema: Union[str, dict], tokenizer: Tokenizer):
         # TODO: Why is this needed? We are using regexes
         if isinstance(schema, dict):
             schema = json.dumps(schema)
-        regex_string = build_regex_from_object(schema)
+        regex_string = build_regex_from_schema(schema)
         super().__init__(regex_string, tokenizer)