File tree Expand file tree Collapse file tree
packages/media-generation-core/src Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -55,4 +55,27 @@ describe("media-generation catalog", () => {
5555 } ) ,
5656 ) . toEqual ( [ "video-default" , "video-pro" ] ) ;
5757 } ) ;
58+
59+ it ( "marks a trimmed default model as the catalog default" , ( ) => {
60+ expect (
61+ synthesizeMediaGenerationCatalogEntries ( {
62+ kind : "video_generation" ,
63+ provider : {
64+ id : "example" ,
65+ defaultModel : " video-default " ,
66+ models : [ "video-default" ] ,
67+ capabilities : { } ,
68+ } ,
69+ } ) ,
70+ ) . toEqual ( [
71+ {
72+ kind : "video_generation" ,
73+ provider : "example" ,
74+ model : "video-default" ,
75+ source : "static" ,
76+ default : true ,
77+ capabilities : { } ,
78+ } ,
79+ ] ) ;
80+ } ) ;
5881} ) ;
Original file line number Diff line number Diff line change @@ -51,6 +51,7 @@ export function synthesizeMediaGenerationCatalogEntries<TCapabilities>(params: {
5151 provider : MediaGenerationCatalogProvider < TCapabilities > ;
5252 modes ?: readonly string [ ] ;
5353} ) : Array < MediaGenerationCatalogEntry < TCapabilities > > {
54+ const defaultModel = uniqueTrimmedStrings ( [ params . provider . defaultModel ] ) [ 0 ] ;
5455 return uniqueModels ( params . provider ) . map ( ( model ) => {
5556 const entry : MediaGenerationCatalogEntry < TCapabilities > = {
5657 kind : params . kind ,
@@ -62,7 +63,7 @@ export function synthesizeMediaGenerationCatalogEntries<TCapabilities>(params: {
6263 if ( params . provider . label ) {
6364 entry . label = params . provider . label ;
6465 }
65- if ( model === params . provider . defaultModel ) {
66+ if ( model === defaultModel ) {
6667 entry . default = true ;
6768 }
6869 if ( params . modes ) {
You can’t perform that action at this time.
0 commit comments