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

ensure primitive types can be used as root of schema

Browse source
This commit is contained in:
Evan Jones 2023-07-27 21:24:05 -04:00
parent 5ef33fbd5c
commit 766ec56642
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
examples/json-schema-to-grammar.py
View file

@ -90,7 +90,10 @@ class SchemaConverter:
else: else:
assert schema_type in PRIMITIVE_RULES, f'Unrecognized schema: {schema}' assert schema_type in PRIMITIVE_RULES, f'Unrecognized schema: {schema}'
return self._add_rule(schema_type, PRIMITIVE_RULES[schema_type]) return self._add_rule(
'root' if rule_name == 'root' else schema_type,
PRIMITIVE_RULES[schema_type]
)
def format_grammar(self): def format_grammar(self):
return '\n'.join((f'{name} ::= {rule}' for name, rule in self._rules.items())) return '\n'.join((f'{name} ::= {rule}' for name, rule in self._rules.items()))