o ϖi @sddlZddlmZmZmZmZmZer ddlmZddl m Z GdddeZ dedd de fd eed ee ee ee ffd d Z dS) N) TYPE_CHECKING NamedTupleOptionalTupleUnionModelVersionc@sHeZdZUdZeed<eed<dZeeed<ededdfdd Z dS) ModelVersionIdentifierzX A reference to a model version in the format owner/name or owner/name:version. ownernameNversionrefreturncCs>td|}|std|d||d|d|dS)zY Split a reference in the format owner/name:version into its components. z6^(?P[^/]+)/(?P[^/:]+)(:(?P.+))?$z$Invalid reference to model version: z%. Expected format: owner/name:versionr r r)rematch ValueErrorgroup)clsrrrH/home/ubuntu/.local/lib/python3.10/site-packages/replicate/identifier.pyparses  zModelVersionIdentifier.parse) __name__ __module__ __qualname____doc__str__annotations__rr classmethodrrrrrr s r rrr rcCsddlm}ddlm}d}d\}}}t||r |j|j}}n#t||r+|}|j}nt|tr6|\}}}n t|t rCt |\}}}||||fS)Nrrr )NNN) replicate.modelrreplicate.versionr isinstancer r idr rr)rrr rr r version_idrrr_resolve!s         r%)rtypingrrrrrr rr!r r rr%rrrrs