1+ import { generateText } from 'ai' ;
2+
13import { VercelProvider } from '../src/VercelProvider' ;
24
35// Mock Vercel AI SDK
@@ -70,7 +72,6 @@ describe('VercelProvider', () => {
7072
7173 describe ( 'invokeModel' , ( ) => {
7274 it ( 'invokes Vercel AI generateText and returns response' , async ( ) => {
73- const { generateText } = require ( 'ai' ) ;
7475 const mockResponse = {
7576 text : 'Hello! How can I help you today?' ,
7677 usage : {
@@ -82,9 +83,7 @@ describe('VercelProvider', () => {
8283
8384 ( generateText as jest . Mock ) . mockResolvedValue ( mockResponse ) ;
8485
85- const messages = [
86- { role : 'user' as const , content : 'Hello!' } ,
87- ] ;
86+ const messages = [ { role : 'user' as const , content : 'Hello!' } ] ;
8887
8988 const result = await provider . invokeModel ( messages ) ;
9089
@@ -110,16 +109,13 @@ describe('VercelProvider', () => {
110109 } ) ;
111110
112111 it ( 'handles response without usage data' , async ( ) => {
113- const { generateText } = require ( 'ai' ) ;
114112 const mockResponse = {
115113 text : 'Hello! How can I help you today?' ,
116114 } ;
117115
118116 ( generateText as jest . Mock ) . mockResolvedValue ( mockResponse ) ;
119117
120- const messages = [
121- { role : 'user' as const , content : 'Hello!' } ,
122- ] ;
118+ const messages = [ { role : 'user' as const , content : 'Hello!' } ] ;
123119
124120 const result = await provider . invokeModel ( messages ) ;
125121
@@ -172,7 +168,7 @@ describe('VercelProvider', () => {
172168 } ;
173169
174170 await expect ( VercelProvider . createVercelModel ( mockAiConfig ) ) . rejects . toThrow (
175- 'Unsupported Vercel AI provider: unsupported'
171+ 'Unsupported Vercel AI provider: unsupported' ,
176172 ) ;
177173 } ) ;
178174 } ) ;
0 commit comments