1
1
import * as fs from "fs" ;
2
2
import { expect , test , vi } from "vitest" ;
3
3
import type { Mock } from "vitest" ;
4
- import Gemini from "../src/index.ts " ;
5
- import { Command , QueryResponseMap } from "../src/types.ts " ;
4
+ import Gemini from "../src/index" ;
5
+ import { Command , QueryResponseMap } from "../src/types" ;
6
6
7
7
const API_KEY = "demo-key" ;
8
8
@@ -98,7 +98,7 @@ const countResponse: QueryResponseMap[Command.Count] = {
98
98
test ( "Gemini.ask()" , async ( ) => {
99
99
global . fetch = vi . fn ( ) as Mock ;
100
100
( fetch as Mock ) . mockReturnValueOnce (
101
- createFetchResponse ( generateContentResponse ) ,
101
+ createFetchResponse ( generateContentResponse )
102
102
) ;
103
103
104
104
const gemini = new Gemini ( API_KEY ) ;
@@ -108,7 +108,7 @@ test("Gemini.ask()", async () => {
108
108
test ( "Gemini.ask() with Previous Messages" , async ( ) => {
109
109
global . fetch = vi . fn ( ) as Mock ;
110
110
( fetch as Mock ) . mockReturnValueOnce (
111
- createFetchResponse ( generateContentResponse ) ,
111
+ createFetchResponse ( generateContentResponse )
112
112
) ;
113
113
114
114
const gemini = new Gemini ( API_KEY ) ;
@@ -119,14 +119,14 @@ test("Gemini.ask() with Previous Messages", async () => {
119
119
format : Gemini . JSON ,
120
120
data : [ fileBuffer ] ,
121
121
messages : [ [ "Hi" , "Sup?" ] ] ,
122
- } ) ,
122
+ } )
123
123
) . toBe ( generateContentResponse ) ;
124
124
} ) ;
125
125
126
126
test ( "Gemini.ask() with Data" , async ( ) => {
127
127
global . fetch = vi . fn ( ) as Mock ;
128
128
( fetch as Mock ) . mockReturnValueOnce (
129
- createFetchResponse ( generateContentResponse ) ,
129
+ createFetchResponse ( generateContentResponse )
130
130
) ;
131
131
132
132
const gemini = new Gemini ( API_KEY ) ;
@@ -136,29 +136,29 @@ test("Gemini.ask() with Data", async () => {
136
136
await gemini . ask ( "What does this show?" , {
137
137
format : Gemini . JSON ,
138
138
data : [ fileBuffer ] ,
139
- } ) ,
139
+ } )
140
140
) . toBe ( generateContentResponse ) ;
141
141
} ) ;
142
142
143
143
test ( "Gemini.ask() with JSON Response" , async ( ) => {
144
144
global . fetch = vi . fn ( ) as Mock ;
145
145
( fetch as Mock ) . mockReturnValueOnce (
146
- createFetchResponse ( generateContentResponse ) ,
146
+ createFetchResponse ( generateContentResponse )
147
147
) ;
148
148
149
149
const gemini = new Gemini ( API_KEY ) ;
150
150
151
151
expect (
152
152
await gemini . ask ( "Hello!" , {
153
153
format : Gemini . JSON ,
154
- } ) ,
154
+ } )
155
155
) . toBe ( generateContentResponse ) ;
156
156
} ) ;
157
157
158
158
test ( "Fetch Polyfill" , async ( ) => {
159
159
const fetchPolyfill = vi . fn ( ) ;
160
160
fetchPolyfill . mockReturnValueOnce (
161
- createFetchResponse ( generateContentResponse ) ,
161
+ createFetchResponse ( generateContentResponse )
162
162
) ;
163
163
164
164
const gemini = new Gemini ( API_KEY , {
@@ -188,7 +188,7 @@ test("Gemini.count()", async () => {
188
188
test ( "Gemini.createChat()" , async ( ) => {
189
189
global . fetch = vi . fn ( ) as Mock ;
190
190
( fetch as Mock ) . mockReturnValueOnce (
191
- createFetchResponse ( generateContentResponse ) ,
191
+ createFetchResponse ( generateContentResponse )
192
192
) ;
193
193
194
194
const gemini = new Gemini ( API_KEY ) ;
@@ -219,7 +219,7 @@ test("Gemini.createChat()", async () => {
219
219
test ( "Gemini.createChat() with JSON" , async ( ) => {
220
220
global . fetch = vi . fn ( ) as Mock ;
221
221
( fetch as Mock ) . mockReturnValueOnce (
222
- createFetchResponse ( generateContentResponse ) ,
222
+ createFetchResponse ( generateContentResponse )
223
223
) ;
224
224
225
225
const gemini = new Gemini ( API_KEY ) ;
@@ -228,6 +228,6 @@ test("Gemini.createChat() with JSON", async () => {
228
228
expect (
229
229
await chat . ask ( "Hello!" , {
230
230
format : Gemini . JSON ,
231
- } ) ,
231
+ } )
232
232
) . toBe ( generateContentResponse ) ;
233
233
} ) ;
0 commit comments