From 0d3f948ebd830e2de441e287c7f6bb7d91e8ca2b Mon Sep 17 00:00:00 2001
From: Jeason <jitong.zjt@alibaba-inc.com>
Date: Tue, 25 Jun 2024 15:40:02 +0800
Subject: [PATCH] chore: simplified the default prompt

---
 app/components/prompt-card.tsx | 8 +++++++-
 app/page.tsx                   | 6 +++++-
 src/language-model.test.ts     | 6 +++---
 src/language-model.ts          | 2 +-
 4 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/app/components/prompt-card.tsx b/app/components/prompt-card.tsx
index 4b5bb4b..062b378 100644
--- a/app/components/prompt-card.tsx
+++ b/app/components/prompt-card.tsx
@@ -19,6 +19,7 @@ export interface PromptCardProps {
 export const PromptCard = React.forwardRef<HTMLDivElement, PromptCardProps>(
   ({ onPrompt }, ref) => {
     const [prompt, setPrompt] = React.useState('');
+
     return (
       <div className="grid w-full gap-2" ref={ref}>
         <Textarea
@@ -26,6 +27,11 @@ export const PromptCard = React.forwardRef<HTMLDivElement, PromptCardProps>(
           placeholder="Type your message here."
           value={prompt}
           onChange={(event) => setPrompt(event.target.value)}
+          onKeyDown={(event) => {
+            if ((event.metaKey || event.ctrlKey) && event.key === 'Enter') {
+              onPrompt?.(prompt);
+            }
+          }}
         />
         <div className="grid w-full gap-2 grid-cols-12">
           <Select>
@@ -43,7 +49,7 @@ export const PromptCard = React.forwardRef<HTMLDivElement, PromptCardProps>(
               onPrompt?.(prompt);
             }}
           >
-            Send message
+            Send message (⌘ + ↩)
           </Button>
         </div>
       </div>
diff --git a/app/page.tsx b/app/page.tsx
index 868660b..17dc996 100644
--- a/app/page.tsx
+++ b/app/page.tsx
@@ -27,7 +27,11 @@ const HomePage: React.FC<unknown> = () => {
       setError(undefined);
       setLoading(true);
       const startTimestamp = Date.now();
-      const { textStream } = await streamText({ model, prompt });
+      const { textStream } = await streamText({
+        model,
+        prompt,
+        // temperature: 0.8,
+      });
       for await (const textPart of textStream) {
         setResult(textPart);
       }
diff --git a/src/language-model.test.ts b/src/language-model.test.ts
index 0095915..caf3949 100644
--- a/src/language-model.test.ts
+++ b/src/language-model.test.ts
@@ -64,7 +64,7 @@ describe('chrome-ai', () => {
     const result = await generateText({ model: chromeai(), prompt: 'test' });
     expect(result).toMatchObject({
       finishReason: 'stop',
-      text: 'user:\ntest\n\nmodel:\n',
+      text: 'user:\ntest\n',
     });
   });
 
@@ -86,7 +86,7 @@ describe('chrome-ai', () => {
 
     const result = await streamText({ model: chromeai(), prompt: 'test' });
     for await (const textPart of result.textStream) {
-      expect(textPart).toBe('user:\ntest\n\nmodel:\n');
+      expect(textPart).toBe('user:\ntest\n');
     }
   });
 
@@ -146,6 +146,6 @@ describe('chrome-ai', () => {
         },
       ],
     });
-    expect(result.text).toBe('user:\n\n\nmodel:\n');
+    expect(result.text).toBe('user:\n\n');
   });
 });
diff --git a/src/language-model.ts b/src/language-model.ts
index fcbb6e5..1a7ad31 100644
--- a/src/language-model.ts
+++ b/src/language-model.ts
@@ -158,7 +158,7 @@ export class ChromeAIChatLanguageModel implements LanguageModelV1 {
       .join('\n\n');
     debug('format prompt:', prompt);
     debug('formated message:', messages);
-    return messages + `\n\n${this.roleMap.assistant}:\n`;
+    return messages + `\n`;
   };
 
   public doGenerate = async (