We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ac49f57 commit 26f9f21Copy full SHA for 26f9f21
outlines/generate/json.py
@@ -71,6 +71,12 @@ def json(
71
regex_str = build_regex_from_schema(schema, whitespace_pattern)
72
generator = regex(model, regex_str, sampler)
73
generator.format_sequence = lambda x: pyjson.loads(x)
74
+ elif isinstance(schema_object, dict) and schema_object.get("type") == "function":
75
+ # Handle OpenAI function call format
76
+ schema = pyjson.dumps(schema_object["function"]["parameters"])
77
+ regex_str = build_regex_from_schema(schema, whitespace_pattern)
78
+ generator = regex(model, regex_str, sampler)
79
+ generator.format_sequence = lambda x: pyjson.loads(x)
80
else:
81
raise ValueError(
82
f"Cannot parse schema {schema_object}. The schema must be either "
0 commit comments