Skip to content

Commit d497510

Browse files
committed
chore: Add AIConfig keys to tests
1 parent a51dcdb commit d497510

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

packages/ai-providers/server-ai-openai/__tests__/OpenAIProvider.test.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -360,6 +360,7 @@ describe('OpenAIProvider', () => {
360360
describe('create', () => {
361361
it('creates OpenAIProvider with correct model and parameters', async () => {
362362
const mockAiConfig = {
363+
key: 'test-config',
363364
model: {
364365
name: 'gpt-4',
365366
parameters: {

packages/ai-providers/server-ai-vercel/__tests__/VercelProvider.test.ts

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,6 +330,7 @@ describe('VercelProvider', () => {
330330
describe('createVercelModel', () => {
331331
it('creates OpenAI model for openai provider', async () => {
332332
const mockAiConfig = {
333+
key: 'test-config',
333334
model: { name: 'gpt-4', parameters: {} },
334335
provider: { name: 'openai' },
335336
enabled: true,
@@ -348,6 +349,7 @@ describe('VercelProvider', () => {
348349

349350
it('throws error for unsupported provider', async () => {
350351
const mockAiConfig = {
352+
key: 'test-config',
351353
model: { name: 'test-model', parameters: {} },
352354
provider: { name: 'unsupported' },
353355
enabled: true,
@@ -364,6 +366,7 @@ describe('VercelProvider', () => {
364366
describe('create', () => {
365367
it('creates VercelProvider with correct model and parameters', async () => {
366368
const mockAiConfig = {
369+
key: 'test-config',
367370
model: {
368371
name: 'gpt-4',
369372
parameters: {
@@ -408,6 +411,7 @@ describe('VercelProvider', () => {
408411

409412
it('handles undefined model and messages', () => {
410413
const aiConfig = {
414+
key: 'test-config',
411415
enabled: true,
412416
};
413417

@@ -424,6 +428,7 @@ describe('VercelProvider', () => {
424428

425429
it('uses additional messages', () => {
426430
const aiConfig = {
431+
key: 'test-config',
427432
model: { name: 'test-ai-model' },
428433
enabled: true,
429434
};
@@ -441,6 +446,7 @@ describe('VercelProvider', () => {
441446

442447
it('combines config messages and additional messages', () => {
443448
const aiConfig = {
449+
key: 'test-config',
444450
model: { name: 'test-ai-model' },
445451
messages: mockMessages,
446452
enabled: true,
@@ -459,6 +465,7 @@ describe('VercelProvider', () => {
459465

460466
it('maps parameters correctly', () => {
461467
const aiConfig = {
468+
key: 'test-config',
462469
model: {
463470
name: 'test-ai-model',
464471
parameters: {
@@ -500,6 +507,7 @@ describe('VercelProvider', () => {
500507
};
501508

502509
const aiConfig = {
510+
key: 'test-config',
503511
model: { name: 'test-ai-model' },
504512
provider: { name: 'openai' },
505513
enabled: true,
@@ -514,6 +522,7 @@ describe('VercelProvider', () => {
514522

515523
it('throws error when model cannot be determined', () => {
516524
const aiConfig = {
525+
key: 'test-config',
517526
model: { name: 'test-ai-model' },
518527
provider: { name: 'unknown' },
519528
enabled: true,
@@ -530,7 +539,7 @@ describe('VercelProvider', () => {
530539

531540
it('throws error when function provider returns undefined', () => {
532541
const aiConfig = {
533-
model: { name: 'test-ai-model' },
542+
key: 'test-config',
534543
enabled: true,
535544
};
536545

0 commit comments

Comments
 (0)