Skip to content

Commit ab72f6f

Browse files
committed
handle getting examples
1 parent 97d0d0f commit ab72f6f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

vetiver/utils.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,10 @@ def serialize_prototype(prototype):
6464

6565
serialized_schema = dict()
6666
for key, value in schema.items():
67-
example = value.get("example", None)
67+
example = value.get("example", None) or value.get("examples", None)
68+
# with pydantic's move from example to [examples], now we have to
69+
# make sure we're extracting the right value
70+
example = example[0] if isinstance(value, list) else example
6871
default = value.get("default", None)
6972
serialized_schema[key] = example if example is not None else default
7073

0 commit comments

Comments
 (0)