Skip to content

Commit 1fd1223

Browse files
fix tool-use-tests build_regex_from_schema import path
1 parent 8f12c49 commit 1fd1223

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tool_use/test_tool_choice_required.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def _compile_and_check(tools: list[ChatCompletionToolsParam], sample_output,
7171
assert isinstance(schema, dict)
7272

7373
# use build_regex_from_schema used in JSONLogitsProcessor to create Guide
74-
from outlines_core.fsm.json_schema import build_regex_from_schema
74+
from outlines_core.json_schema import build_regex_from_schema
7575
regex = build_regex_from_schema(json.dumps(schema))
7676
compiled = re.compile(regex)
7777
matches = compiled.fullmatch(json.dumps(sample_output)) is not None

0 commit comments

Comments
 (0)