fix unit test

Author: jsonbailey

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

@@ -1,11 +1,12 @@
-import { generateObject, generateText } from 'ai';
+import { generateObject, generateText, jsonSchema } from 'ai';
 
 import { VercelProvider } from '../src/VercelProvider';
 
 // Mock Vercel AI SDK
 jest.mock('ai', () => ({
   generateText: jest.fn(),
   generateObject: jest.fn(),
+  jsonSchema: jest.fn((schema) => schema),
 }));
 
 describe('VercelProvider', () => {
@@ -15,6 +16,7 @@ describe('VercelProvider', () => {
   beforeEach(() => {
     mockModel = { name: 'test-model' };
     provider = new VercelProvider(mockModel, {});
+    jest.clearAllMocks();
   });
 
   describe('createAIMetrics', () => {
@@ -117,76 +119,6 @@ describe('VercelProvider', () => {
     });
   });
 
-  describe('convertToZodSchema', () => {
-    it('converts simple object structure to Zod schema', () => {
-      const responseStructure = {
-        name: 'string',
-        age: 0,
-        isActive: true,
-      };
-
-      const schema = VercelProvider.convertToZodSchema(responseStructure);
-
-      expect(schema).toBeDefined();
-      expect(typeof schema.parse).toBe('function');
-    });
-
-    it('converts nested object structure to Zod schema', () => {
-      const responseStructure = {
-        user: {
-          name: 'string',
-          age: 0,
-        },
-        settings: {
-          theme: 'string',
-          notifications: true,
-        },
-      };
-
-      const schema = VercelProvider.convertToZodSchema(responseStructure);
-
-      expect(schema).toBeDefined();
-      expect(typeof schema.parse).toBe('function');
-    });
-
-    it('converts array structure to Zod schema', () => {
-      const responseStructure = {
-        items: ['string'],
-        numbers: [0],
-        booleans: [true],
-      };
-
-      const schema = VercelProvider.convertToZodSchema(responseStructure);
-
-      expect(schema).toBeDefined();
-      expect(typeof schema.parse).toBe('function');
-    });
-
-    it('handles empty array structure', () => {
-      const responseStructure = {
-        items: [],
-      };
-
-      const schema = VercelProvider.convertToZodSchema(responseStructure);
-
-      expect(schema).toBeDefined();
-      expect(typeof schema.parse).toBe('function');
-    });
-
-    it('handles null and undefined values', () => {
-      const responseStructure = {
-        nullable: null,
-        undefined,
-        string: 'string',
-      };
-
-      const schema = VercelProvider.convertToZodSchema(responseStructure);
-
-      expect(schema).toBeDefined();
-      expect(typeof schema.parse).toBe('function');
-    });
-  });
-
   describe('invokeModel', () => {
     it('invokes Vercel AI generateText and returns response', async () => {
       const mockResponse = {
@@ -247,6 +179,30 @@ describe('VercelProvider', () => {
         },
       });
     });
+
+    it('handles errors and returns failure metrics', async () => {
+      const mockError = new Error('API call failed');
+      (generateText as jest.Mock).mockRejectedValue(mockError);
+
+      const mockLogger = {
+        warn: jest.fn(),
+      };
+      provider = new VercelProvider(mockModel, {}, mockLogger as any);
+
+      const messages = [{ role: 'user' as const, content: 'Hello!' }];
+      const result = await provider.invokeModel(messages);
+
+      expect(mockLogger.warn).toHaveBeenCalledWith('Vercel AI model invocation failed:', mockError);
+      expect(result).toEqual({
+        message: {
+          role: 'assistant',
+          content: '',
+        },
+        metrics: {
+          success: false,
+        },
+      });
+    });
   });
 
   describe('invokeStructuredModel', () => {
@@ -278,8 +234,9 @@ describe('VercelProvider', () => {
       expect(generateObject).toHaveBeenCalledWith({
         model: mockModel,
         messages: [{ role: 'user', content: 'Generate user data' }],
-        schema: expect.any(Object), // Zod schema
+        schema: responseStructure,
       });
+      expect(jsonSchema).toHaveBeenCalledWith(responseStructure);
 
       expect(result).toEqual({
         data: {
@@ -332,6 +289,35 @@ describe('VercelProvider', () => {
         },
       });
     });
+
+    it('handles errors and returns failure metrics', async () => {
+      const mockError = new Error('API call failed');
+      (generateObject as jest.Mock).mockRejectedValue(mockError);
+
+      const mockLogger = {
+        warn: jest.fn(),
+      };
+      provider = new VercelProvider(mockModel, {}, mockLogger as any);
+
+      const messages = [{ role: 'user' as const, content: 'Generate result' }];
+      const responseStructure = {
+        result: 'string',
+      };
+
+      const result = await provider.invokeStructuredModel(messages, responseStructure);
+
+      expect(mockLogger.warn).toHaveBeenCalledWith(
+        'Vercel AI structured model invocation failed:',
+        mockError,
+      );
+      expect(result).toEqual({
+        data: {},
+        rawResponse: '',
+        metrics: {
+          success: false,
+        },
+      });
+    });
   });
 
   describe('getModel', () => {