Skip to content

Commit 17e4752

Browse files
SalazareoSalazareo
authored
fix: mistral ai tests from renames (#3091)
1 parent b948eaa commit 17e4752

2 files changed

Lines changed: 30 additions & 30 deletions

File tree

‎src/backend/drivers/ai-chat/providers/mistral/MistralAiProvider.integration.test.ts‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
/**
2121
* Integration test for the Mistral provider.
2222
*
23-
* Uses `mistral-small-2506` (provider default, cheapest tier). Skipped
23+
* Uses `mistral-small-2603` (provider default, cheapest tier). Skipped
2424
* when `PUTER_TEST_AI_MISTRAL_API_KEY` is unset.
2525
*/
2626

@@ -47,7 +47,7 @@ describe.skipIf(skipUnlessEnv(ENV_VAR))(
4747

4848
const result = await withTestActor(() =>
4949
provider.complete({
50-
model: 'mistral-small-2506',
50+
model: 'mistral-small-2603',
5151
messages: [
5252
{ role: 'user', content: 'Say hi in one word.' },
5353
],

‎src/backend/drivers/ai-chat/providers/mistral/MistralAiProvider.test.ts‎

Lines changed: 28 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -149,9 +149,9 @@ describe('MistralAIProvider construction', () => {
149149
// ── Model catalog ───────────────────────────────────────────────────
150150

151151
describe('MistralAIProvider model catalog', () => {
152-
it('returns mistral-small-2506 as the default', () => {
152+
it('returns the configured small model as the default', () => {
153153
const { provider } = makeProvider();
154-
expect(provider.getDefaultModel()).toBe('mistral-small-2506');
154+
expect(provider.getDefaultModel()).toBe('mistral-small-2603');
155155
});
156156

157157
it('exposes the static MISTRAL_MODELS list verbatim from models()', async () => {
@@ -171,7 +171,7 @@ describe('MistralAIProvider model catalog', () => {
171171
}
172172
// Sanity: a known alias is present alongside its canonical id.
173173
expect(ids).toContain('mistral-small-latest');
174-
expect(ids).toContain('mistral-small-2506');
174+
expect(ids).toContain('mistral-small-2603');
175175
});
176176
});
177177

@@ -194,15 +194,15 @@ describe('MistralAIProvider.complete request shape', () => {
194194

195195
await withTestActor(() =>
196196
provider.complete({
197-
model: 'mistral-small-2506',
197+
model: 'mistral-small-2603',
198198
messages: [{ role: 'user', content: 'hello' }],
199199
max_tokens: 256,
200200
temperature: 0.4,
201201
}),
202202
);
203203

204204
const [args] = completeMock.mock.calls[0]!;
205-
expect(args.model).toBe('mistral-small-2506');
205+
expect(args.model).toBe('mistral-small-2603');
206206
expect(args.messages).toEqual([{ role: 'user', content: 'hello' }]);
207207
// Mistral's SDK uses maxTokens (camelCase). The provider should
208208
// adapt our snake_case input.
@@ -216,7 +216,7 @@ describe('MistralAIProvider.complete request shape', () => {
216216

217217
await withTestActor(() =>
218218
provider.complete({
219-
model: 'mistral-small-2506',
219+
model: 'mistral-small-2603',
220220
messages: [{ role: 'user', content: 'hi' }],
221221
}),
222222
);
@@ -240,7 +240,7 @@ describe('MistralAIProvider.complete request shape', () => {
240240
];
241241
await withTestActor(() =>
242242
provider.complete({
243-
model: 'mistral-small-2506',
243+
model: 'mistral-small-2603',
244244
messages: [{ role: 'user', content: 'hi' }],
245245
tools,
246246
}),
@@ -257,7 +257,7 @@ describe('MistralAIProvider.complete request shape', () => {
257257

258258
await withTestActor(() =>
259259
provider.complete({
260-
model: 'mistral-small-2506',
260+
model: 'mistral-small-2603',
261261
messages: [
262262
{
263263
role: 'assistant',
@@ -308,7 +308,7 @@ describe('MistralAIProvider.complete request shape', () => {
308308
completeMock.mockResolvedValueOnce(baseCompletion);
309309
await withTestActor(() =>
310310
provider.complete({
311-
model: 'mistral-small-2506',
311+
model: 'mistral-small-2603',
312312
messages: [{ role: 'user', content: 'hi' }],
313313
stream: false,
314314
}),
@@ -320,7 +320,7 @@ describe('MistralAIProvider.complete request shape', () => {
320320
streamMock.mockReturnValueOnce(asAsyncIterable([]));
321321
await withTestActor(() =>
322322
provider.complete({
323-
model: 'mistral-small-2506',
323+
model: 'mistral-small-2603',
324324
messages: [{ role: 'user', content: 'hi' }],
325325
stream: true,
326326
}),
@@ -370,17 +370,17 @@ describe('MistralAIProvider model resolution', () => {
370370

371371
await withTestActor(() =>
372372
provider.complete({
373-
// `mistral-small-latest` is an alias of `mistral-small-2506`.
373+
// `mistral-small-latest` is an alias of `mistral-small-2603`.
374374
model: 'mistral-small-latest',
375375
messages: [{ role: 'user', content: 'hi' }],
376376
}),
377377
);
378378

379-
expect(completeMock.mock.calls[0]![0].model).toBe('mistral-small-2506');
379+
expect(completeMock.mock.calls[0]![0].model).toBe('mistral-small-2603');
380380
expect(recordSpy).toHaveBeenCalledWith(
381381
expect.any(Object),
382382
expect.anything(),
383-
'mistral:mistral-small-2506',
383+
'mistral:mistral-small-2603',
384384
expect.any(Object),
385385
);
386386
});
@@ -396,11 +396,11 @@ describe('MistralAIProvider model resolution', () => {
396396
}),
397397
);
398398

399-
expect(completeMock.mock.calls[0]![0].model).toBe('mistral-small-2506');
399+
expect(completeMock.mock.calls[0]![0].model).toBe('mistral-small-2603');
400400
expect(recordSpy).toHaveBeenCalledWith(
401401
expect.any(Object),
402402
expect.anything(),
403-
'mistral:mistral-small-2506',
403+
'mistral:mistral-small-2603',
404404
expect.any(Object),
405405
);
406406
});
@@ -424,7 +424,7 @@ describe('MistralAIProvider.complete non-stream output', () => {
424424

425425
const result = await withTestActor(() =>
426426
provider.complete({
427-
model: 'mistral-small-2506',
427+
model: 'mistral-small-2603',
428428
messages: [{ role: 'user', content: 'hi' }],
429429
}),
430430
);
@@ -450,12 +450,12 @@ describe('MistralAIProvider.complete non-stream output', () => {
450450
cached_tokens: 0,
451451
});
452452
expect(actor).toBe(SYSTEM_ACTOR);
453-
expect(prefix).toBe('mistral:mistral-small-2506');
454-
// mistral-small-2506 costs: prompt=10, completion=30. cached_tokens
453+
expect(prefix).toBe('mistral:mistral-small-2603');
454+
// mistral-small-2603 costs: prompt=15, completion=60. cached_tokens
455455
// is undefined in the model row → multiplied by 0 → NaN-safe 0.
456456
expect(overrides).toMatchObject({
457-
prompt_tokens: 100 * 10,
458-
completion_tokens: 50 * 30,
457+
prompt_tokens: 100 * 15,
458+
completion_tokens: 50 * 60,
459459
});
460460
});
461461

@@ -486,7 +486,7 @@ describe('MistralAIProvider.complete non-stream output', () => {
486486

487487
const result = (await withTestActor(() =>
488488
provider.complete({
489-
model: 'mistral-small-2506',
489+
model: 'mistral-small-2603',
490490
messages: [{ role: 'user', content: 'do a tool call' }],
491491
tools: [
492492
{
@@ -535,7 +535,7 @@ describe('MistralAIProvider.complete streaming', () => {
535535

536536
const result = await withTestActor(() =>
537537
provider.complete({
538-
model: 'mistral-small-2506',
538+
model: 'mistral-small-2603',
539539
messages: [{ role: 'user', content: 'say hi' }],
540540
stream: true,
541541
}),
@@ -560,14 +560,14 @@ describe('MistralAIProvider.complete streaming', () => {
560560
cached_tokens: 0,
561561
});
562562

563-
// mistral-small-2506: prompt=10, completion=30.
563+
// mistral-small-2603: prompt=15, completion=60.
564564
expect(recordSpy).toHaveBeenCalledTimes(1);
565565
const [, , prefix, overrides] =
566566
recordSpy.mock.calls[0]!;
567-
expect(prefix).toBe('mistral:mistral-small-2506');
567+
expect(prefix).toBe('mistral:mistral-small-2603');
568568
expect(overrides).toMatchObject({
569-
prompt_tokens: 4 * 10,
570-
completion_tokens: 2 * 30,
569+
prompt_tokens: 4 * 15,
570+
completion_tokens: 2 * 60,
571571
});
572572
});
573573

@@ -626,7 +626,7 @@ describe('MistralAIProvider.complete streaming', () => {
626626

627627
const result = await withTestActor(() =>
628628
provider.complete({
629-
model: 'mistral-small-2506',
629+
model: 'mistral-small-2603',
630630
messages: [{ role: 'user', content: 'do tool call' }],
631631
tools: [
632632
{
@@ -666,7 +666,7 @@ describe('MistralAIProvider.complete error mapping', () => {
666666
await expect(
667667
withTestActor(() =>
668668
provider.complete({
669-
model: 'mistral-small-2506',
669+
model: 'mistral-small-2603',
670670
messages: [{ role: 'user', content: 'boom' }],
671671
}),
672672
),

0 commit comments

Comments
 (0)