Skip to content

Commit 1467b90

Browse files
committed
update request import
1 parent 1b5f74f commit 1467b90

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

packages/vertexai/src/models/generative-model.test.ts

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ import { FunctionCallingMode, VertexAI } from '../public-types';
2020
import * as request from '../requests/request';
2121
import { match, restore, stub } from 'sinon';
2222
import { getMockResponse } from '../../test-utils/mock-response';
23-
import { Task } from '../requests/request';
2423
import sinonChai from 'sinon-chai';
2524

2625
use(sinonChai);
@@ -81,7 +80,7 @@ describe('GenerativeModel', () => {
8180
await genModel.generateContent('hello');
8281
expect(makeRequestStub).to.be.calledWith(
8382
'publishers/google/models/my-model',
84-
Task.GENERATE_CONTENT,
83+
request.Task.GENERATE_CONTENT,
8584
match.any,
8685
false,
8786
match((value: string) => {
@@ -121,7 +120,7 @@ describe('GenerativeModel', () => {
121120
});
122121
expect(makeRequestStub).to.be.calledWith(
123122
'publishers/google/models/my-model',
124-
Task.GENERATE_CONTENT,
123+
request.Task.GENERATE_CONTENT,
125124
match.any,
126125
false,
127126
match((value: string) => {
@@ -156,7 +155,7 @@ describe('GenerativeModel', () => {
156155
await genModel.startChat().sendMessage('hello');
157156
expect(makeRequestStub).to.be.calledWith(
158157
'publishers/google/models/my-model',
159-
Task.GENERATE_CONTENT,
158+
request.Task.GENERATE_CONTENT,
160159
match.any,
161160
false,
162161
match((value: string) => {
@@ -199,7 +198,7 @@ describe('GenerativeModel', () => {
199198
.sendMessage('hello');
200199
expect(makeRequestStub).to.be.calledWith(
201200
'publishers/google/models/my-model',
202-
Task.GENERATE_CONTENT,
201+
request.Task.GENERATE_CONTENT,
203202
match.any,
204203
false,
205204
match((value: string) => {

0 commit comments

Comments
 (0)