Skip to content

Commit 642507e

Browse files
committed
fix: updating unit tests
1 parent 3122b41 commit 642507e

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

src/api/providers/__tests__/gemini.spec.ts

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -173,13 +173,13 @@ describe("GeminiHandler", () => {
173173
expect(modelInfo.id).toBe("gemini-2.5-pro")
174174
})
175175

176-
it("should map gemini-1.5-pro-{variants} to gemini-2.0-pro-exp-02-05", () => {
176+
it("should map gemini-1.5-pro-{variants} to gemini-2.0-flash-001", () => {
177177
const legacyHandler = new GeminiHandler({
178178
apiModelId: "gemini-1.5-pro-002",
179179
geminiApiKey: "test-key",
180180
})
181181
const modelInfo = legacyHandler.getModel()
182-
expect(modelInfo.id).toBe("gemini-2.0-pro-exp-02-05")
182+
expect(modelInfo.id).toBe("gemini-2.0-flash-001")
183183
})
184184

185185
it("should map gemini-1.5-flash-{variants} to gemini-2.0-flash-001", () => {
@@ -200,13 +200,13 @@ describe("GeminiHandler", () => {
200200
expect(modelInfo.id).toBe("gemini-2.5-pro")
201201
})
202202

203-
it("should map gemini-exp-1206 to gemini-2.0-pro-exp-02-05", () => {
203+
it("should map gemini-exp-1206 to gemini-2.0-flash-001", () => {
204204
const legacyHandler = new GeminiHandler({
205205
apiModelId: "gemini-exp-1206",
206206
geminiApiKey: "test-key",
207207
})
208208
const modelInfo = legacyHandler.getModel()
209-
expect(modelInfo.id).toBe("gemini-2.0-pro-exp-02-05")
209+
expect(modelInfo.id).toBe("gemini-2.0-flash-001")
210210
})
211211

212212
it("should keep current models as-is", () => {

src/api/providers/__tests__/vertex.spec.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,14 +149,14 @@ describe("VertexHandler", () => {
149149
expect(modelInfo.id).toBe("gemini-2.5-pro")
150150
})
151151

152-
it("should map gemini-1.5-pro-{variants} to gemini-2.0-pro-exp-02-05", () => {
152+
it("should map gemini-1.5-pro-{variants} to gemini-2.0-flash-001", () => {
153153
const legacyHandler = new VertexHandler({
154154
apiModelId: "gemini-1.5-pro-002",
155155
vertexProjectId: "test-project",
156156
vertexRegion: "us-central1",
157157
})
158158
const modelInfo = legacyHandler.getModel()
159-
expect(modelInfo.id).toBe("gemini-2.0-pro-exp-02-05")
159+
expect(modelInfo.id).toBe("gemini-2.0-flash-001")
160160
})
161161

162162
it("should map gemini-1.5-flash-{variants} to gemini-2.0-flash-001", () => {

0 commit comments

Comments
 (0)