@@ -20,7 +20,6 @@ import { FunctionCallingMode, VertexAI } from '../public-types';
20
20
import * as request from '../requests/request' ;
21
21
import { match , restore , stub } from 'sinon' ;
22
22
import { getMockResponse } from '../../test-utils/mock-response' ;
23
- import { Task } from '../requests/request' ;
24
23
import sinonChai from 'sinon-chai' ;
25
24
26
25
use ( sinonChai ) ;
@@ -81,7 +80,7 @@ describe('GenerativeModel', () => {
81
80
await genModel . generateContent ( 'hello' ) ;
82
81
expect ( makeRequestStub ) . to . be . calledWith (
83
82
'publishers/google/models/my-model' ,
84
- Task . GENERATE_CONTENT ,
83
+ request . Task . GENERATE_CONTENT ,
85
84
match . any ,
86
85
false ,
87
86
match ( ( value : string ) => {
@@ -121,7 +120,7 @@ describe('GenerativeModel', () => {
121
120
} ) ;
122
121
expect ( makeRequestStub ) . to . be . calledWith (
123
122
'publishers/google/models/my-model' ,
124
- Task . GENERATE_CONTENT ,
123
+ request . Task . GENERATE_CONTENT ,
125
124
match . any ,
126
125
false ,
127
126
match ( ( value : string ) => {
@@ -156,7 +155,7 @@ describe('GenerativeModel', () => {
156
155
await genModel . startChat ( ) . sendMessage ( 'hello' ) ;
157
156
expect ( makeRequestStub ) . to . be . calledWith (
158
157
'publishers/google/models/my-model' ,
159
- Task . GENERATE_CONTENT ,
158
+ request . Task . GENERATE_CONTENT ,
160
159
match . any ,
161
160
false ,
162
161
match ( ( value : string ) => {
@@ -199,7 +198,7 @@ describe('GenerativeModel', () => {
199
198
. sendMessage ( 'hello' ) ;
200
199
expect ( makeRequestStub ) . to . be . calledWith (
201
200
'publishers/google/models/my-model' ,
202
- Task . GENERATE_CONTENT ,
201
+ request . Task . GENERATE_CONTENT ,
203
202
match . any ,
204
203
false ,
205
204
match ( ( value : string ) => {
0 commit comments