j 4UdZddlmZddlmZddlmZddlmZedZ ed Gd d Z d e d dde ddde dddfDZ de d<d d d d d d dddd Z de d<d dddZde d<d$d"Zd#S)%zLModel registry: public model ids exposed on /v1/models and alias resolution.) annotations) dataclass)Literal)InvalidRequest)chatimagevideoT)frozenc@eZdZUded<ded<dZded<dZded<d S) ModelSpecstrid ModelKindkind descriptionmuseowned_byN)__name__ __module__ __qualname____annotations__rr0C:\Projects\Muse2API\src\muse2api\core\models.pyr r sF GGGOOOKHrr ci|] }|j| Sr)r).0ms r r s,    D!   rz muse-chatrzDefault muse.ai assistantz muse-imager zText-to-image / image editingz muse-videor z*Text-to-video / first-frame image-to-videozdict[str, ModelSpec]MODELS) zgpt-4oz gpt-4o-minizgpt-4.1zgpt-5zclaude-sonnet-4z deepseek-chatzdall-e-3z gpt-image-1sorazdict[str, str]ALIASESzdict[ModelKind, str]DEFAULT_BY_KINDname str | Nonerrreturnc(|stt|St||}t|}|tt|S|j|krt d|d|jd||S)zResolve a client-supplied model name to a ``ModelSpec`` of the expected kind. Unknown names fall back to the default model of that kind; a known model of the wrong kind is rejected so that e.g. ``muse-video`` is never silently used for chat. Nzmodel 'z' is a z model, expected )r!r$r#getrr)r%rkeyspecs r resolve_modelr,3s -od+,, ++dD ! !C ::c??D |od+,, yDVtVVDIVVPTVVWWW KrN)r%r&rrr'r )__doc__ __future__r dataclassesrtypingrerrorsrrr r!rr#r$r,rrrr2sRRR""""""!!!!!!###### , -  $    +v'BCC ,)HII ,)UVV    "           ))r