FilesExpand file tree

 main

/

schema.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

27 lines (20 loc) · 974 Bytes

 main

/

schema.py

Copy path

Top

File metadata and controls

• Code
• Blame

27 lines (20 loc) · 974 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

from typing import Optional

from packaging import version

# TODO: this code is shared with replicate's backend. Maybe we should put it in the Cog Python package as the source of truth?

def version_has_no_array_type(cog_version: str) -> Optional[bool]:

"""Iterators have x-cog-array-type=iterator in the schema from 0.3.9 onward"""

try:

return version.parse(cog_version) < version.parse("0.3.9")

except version.InvalidVersion:

return None

def make_schema_backwards_compatible(

schema: dict,

cog_version: str,

) -> dict:

"""A place to add backwards compatibility logic for our openapi schema"""

# If the top-level output is an array, assume it is an iterator in old versions which didn't have an array type

if version_has_no_array_type(cog_version):

output = schema["components"]["schemas"]["Output"]

if output.get("type") == "array":

output["x-cog-array-type"] = "iterator"

return schema