chore: simplified the default prompt

jeasonstudio · Jun 25, 2024 · 0d3f948 · 0d3f948
1 parent 635d7af
commit 0d3f948
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 6 deletions.
diff --git a/app/components/prompt-card.tsx b/app/components/prompt-card.tsx
@@ -19,13 +19,19 @@ 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
           className="resize-none w-full"
           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
@@ -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
@@ -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
@@ -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 (