Skip to content

Commit 8b15e88

Browse files
committed
Adding smoke test for new hybrid params
1 parent 96e7dd5 commit 8b15e88

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

packages/vertexai/src/api.test.ts

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,12 @@
1414
* See the License for the specific language governing permissions and
1515
* limitations under the License.
1616
*/
17-
import { ImagenModelParams, ModelParams, VertexAIErrorCode } from './types';
17+
import {
18+
ImagenModelParams,
19+
InferenceMode,
20+
ModelParams,
21+
VertexAIErrorCode
22+
} from './types';
1823
import { VertexAIError } from './errors';
1924
import { ImagenModel, getGenerativeModel, getImagenModel } from './api';
2025
import { expect } from 'chai';
@@ -112,6 +117,13 @@ describe('Top level API', () => {
112117
);
113118
}
114119
});
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+
});
115127
it('getImagenModel throws if no apiKey is provided', () => {
116128
const fakeVertexNoApiKey = {
117129
...fakeVertexAI,

0 commit comments

Comments
 (0)