File tree Expand file tree Collapse file tree 1 file changed +13
-1
lines changed Expand file tree Collapse file tree 1 file changed +13
-1
lines changed Original file line number Diff line number Diff line change 14
14
* See the License for the specific language governing permissions and
15
15
* limitations under the License.
16
16
*/
17
- import { ImagenModelParams , ModelParams , VertexAIErrorCode } from './types' ;
17
+ import {
18
+ ImagenModelParams ,
19
+ InferenceMode ,
20
+ ModelParams ,
21
+ VertexAIErrorCode
22
+ } from './types' ;
18
23
import { VertexAIError } from './errors' ;
19
24
import { ImagenModel , getGenerativeModel , getImagenModel } from './api' ;
20
25
import { expect } from 'chai' ;
@@ -112,6 +117,13 @@ describe('Top level API', () => {
112
117
) ;
113
118
}
114
119
} ) ;
120
+ it ( 'getGenerativeModel with HybridParams sets the model' , ( ) => {
121
+ const genModel = getGenerativeModel ( fakeVertexAI , {
122
+ mode : InferenceMode . ONLY_ON_CLOUD ,
123
+ onCloudParams : { model : 'my-model' }
124
+ } ) ;
125
+ expect ( genModel . model ) . to . equal ( 'publishers/google/models/my-model' ) ;
126
+ } ) ;
115
127
it ( 'getImagenModel throws if no apiKey is provided' , ( ) => {
116
128
const fakeVertexNoApiKey = {
117
129
...fakeVertexAI ,
You can’t perform that action at this time.
0 commit comments