baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix: json_schema_to_gbnf should take string dump of json schema as input

Browse source
This commit is contained in:
Andrei Betlen 2023-11-10 02:50:17 -05:00
parent e7962d2c73
commit faeae181b1
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
llama_cpp/llama_grammar.py
View file

@ -1484,7 +1484,7 @@ class SchemaConverter:
def json_schema_to_gbnf(schema: str, prop_order: Optional[List[str]] = None):
prop_order = prop_order or []
schema = json.load(schema)
schema = json.loads(schema)
prop_order = {name: idx for idx, name in enumerate(prop_order)}
converter = SchemaConverter(prop_order)
converter.visit(schema, "")