test: cases and coverage

app/page.tsx

@@ -5,15 +5,14 @@ import React from 'react';
 
 import { streamText } from 'ai';
 import { chromeai } from 'chrome-ai';
-import { z } from 'zod';
 import { PromptCard } from './components/prompt-card';
 import { ChatCard } from './components/chat-card';
 import { Alert, AlertDescription, AlertTitle } from './components/ui/alert';
 import { AlertCircle } from 'lucide-react';
 import { Footer } from './components/footer';
 import { checkEnv } from './utils';
 
-const model = chromeai();
+const model = chromeai('generic');
 
 const HomePage: React.FC<unknown> = () => {
   const [result, setResult] = React.useState<string | undefined>(undefined);
@@ -29,8 +28,8 @@ const HomePage: React.FC<unknown> = () => {
       const startTimestamp = Date.now();
       const { textStream } = await streamText({
         model,
-        prompt,
-        // temperature: 0.8,
+        system: 'You are a helpful assistant.',
+        messages: [{ role: 'user', content: prompt }],
       });
       for await (const textPart of textStream) {
         setResult(textPart);

src/language-model.test.ts

@@ -1,6 +1,6 @@
 import { describe, it, expect, vi, afterEach } from 'vitest';
 import { chromeai, ChromeAIChatLanguageModel } from './index';
-import { generateText, streamText, generateObject } from 'ai';
+import { generateText, streamText, generateObject, streamObject } from 'ai';
 import {
   LoadSettingError,
   UnsupportedFunctionalityError,
@@ -66,6 +66,17 @@ describe('chrome-ai', () => {
       finishReason: 'stop',
       text: 'test',
     });
+
+    const resultForMessages = await generateText({
+      model: chromeai(),
+      messages: [
+        { role: 'user', content: 'test' },
+        { role: 'assistant', content: 'assistant' },
+      ],
+    });
+    expect(resultForMessages.text).toBe(
+      'user\ntest\nmodel\nassistant\nmodel\n'
+    );
   });
 
   it('should do stream text', async () => {
@@ -146,6 +157,24 @@ describe('chrome-ai', () => {
         },
       ],
     });
-    expect(result.text).toBe('user:\n\n');
+    expect(result.text).toBe('user\n\nmodel\n');
+
+    await expect(() =>
+      generateObject({
+        model: chromeai(),
+        mode: 'grammar',
+        schema: z.object({}),
+        prompt: 'test',
+      })
+    ).rejects.toThrowError(UnsupportedFunctionalityError);
+
+    await expect(() =>
+      streamObject({
+        model: chromeai(),
+        mode: 'grammar',
+        schema: z.object({}),
+        prompt: 'test',
+      })
+    ).rejects.toThrowError(UnsupportedFunctionalityError);
   });
 });