Feat: improved system prompt

Author: will-lp1

apps/snow-leopard/app/api/chat/route.ts

@@ -35,9 +35,6 @@ import { webSearch } from '@/lib/ai/tools/web-search';
 
 export const maxDuration = 60;
 
-/**
- * Creates an enhanced system prompt that includes active and mentioned document content
- */
 async function createEnhancedSystemPrompt({
   selectedChatModel,
   activeDocumentId,

apps/snow-leopard/lib/ai/prompts.ts

@@ -1,19 +1,32 @@
 import { ArtifactKind } from '@/components/artifact';
 
-// Document awareness instructions
 const documentAwarenessPrompt = `
-You have access to the CURRENT DOCUMENT. Use its content silently to guide your responses - answer with a few sentences - don't over write or include the edited document content in your response.
-
-- Only invoke internal document operations when the user's request involves document creation or content modifications.
-- When a query requires up-to-date or external information not contained in the CURRENT DOCUMENT, call webSearch with an appropriate query to fetch relevant sources.
-- Use webSearch for: current events, recent statistics, product comparisons, news, or any factual claims that need verification.
-- For all other queries, respond normally without using any tools.
-- When no active document exists, call createDocument first (with title and kind), then streamingDocument to generate and stream initial content.
-- When an active document exists but is empty, call streamingDocument (with title and kind) to fill it with initial content.
-- When an active document exists and has content, call updateDocument with a concise description of the desired edits (NEVER CALL TWICE or MORE THAN ONCE).
-- Never reveal tool names, document IDs, or internal details; keep all updates seamless and invisible to the user.`;
-
-// Dynamically generate the artifact-management tools section
+CURRENT DOCUMENT: Read silently, never quote large chunks in your response - ONLY A THREE SENTENCE SUMMARY OF CHANGES MAX - insightful not lengthy.
+
+• Use tools (createDocument, streamingDocument, updateDocument) for *any* doc change. Do **not** echo the change as chat text.
+• One \`webSearch\` if info is outside the doc; prefer 2025-latest sources.
+
+Lifecycle
+  • No doc → createDocument ⇒ streamingDocument
+  • Empty doc → streamingDocument
+  • Has content → updateDocument (call once)
+
+EXAMPLES
+  1. User: "Start a travel blog outline" ⇒ createDocument(title:"Travel Blog", kind:"text") then streamingDocument.
+  2. User: "Add catchy intro" ⇒ updateDocument(desc:"Add a punchy intro paragraph about sustainable travel.")
+  3. User: "Latest iPhone sales?" ⇒ webSearch("iPhone sales 2025 statistics")
+  4. User: 'Write like me' using the writing style summary and writing style snippet to help ⇒ updateDocument 
+
+Never expose tool names/IDs to the user.`;
+
+const writingQualityPrompt = `
+STYLE
+• Clear, active voice; concise.
+• Use Markdown: headings, bullets, tables when useful.
+• No code fences around normal prose.
+• Cite sources with [^1] style when webSearch used.
+• Respect user's existing style when editing.`;
+
 export function buildArtifactsPrompt(
   tools: Array<'createDocument' | 'streamingDocument' | 'updateDocument' | 'webSearch'>
 ): string {
@@ -41,7 +54,7 @@ export function buildArtifactsPrompt(
 }
 
 export const regularPrompt =
-  'You are a friendly assistant. Keep your responses concise and helpful.';
+  'You are a knowledgeable writing assistant (current year: 2025). Provide helpful, succinct, and well-structured responses.';
 
 export const systemPrompt = ({
   selectedChatModel,
@@ -53,6 +66,8 @@ export const systemPrompt = ({
   const artifactsText = buildArtifactsPrompt(availableTools);
   return `${regularPrompt}
 
+${writingQualityPrompt}
+
 ${artifactsText}
 
 ${documentAwarenessPrompt}`;

apps/snow-leopard/lib/ai/tools/update-document.ts

@@ -5,7 +5,6 @@ import { getDocumentById } from '@/lib/db/queries';
 import { myProvider } from '@/lib/ai/providers';
 
 interface UpdateDocumentProps {
-  // Session is currently not used by this tool, but we keep it for future ACL needs.
   session: Session;
   documentId?: string;
 }
@@ -22,7 +21,6 @@ export const updateDocument = ({ session: _session, documentId: defaultDocumentI
       const documentId = defaultDocumentId;
 
       try {
-        // --- Validation ---
         if (!description.trim()) {
           return { error: 'No update description provided.' };
         }
@@ -47,17 +45,14 @@ export const updateDocument = ({ session: _session, documentId: defaultDocumentI
         }
         const originalContent = document.content || '';
 
-        // Generate full replacement content.  Encourage the model to perform the *smallest* possible change set so that our diff visualisation remains concise.
         const prompt = `You are an expert editor. Here is the ORIGINAL document:\n\n${originalContent}\n\n---\n\nTASK: Apply the following edits.\n- Make only the minimal changes required to satisfy the description.\n- Keep paragraphs, sentences, and words that do **not** need to change exactly as they are.\n- Do **not** paraphrase or re-flow content unless strictly necessary.\n- Preserve existing formatting and line breaks.\n\nReturn ONLY the updated document with no additional commentary.\n\nDESCRIPTION: "${description}"`;
 
         const { text: newContent } = await generateText({
           model: myProvider.languageModel('artifact-model'),
           prompt,
-          // Asking for lower temperature for deterministic updates.
           temperature: 0.2,
         });
 
-        // --- Return Result with Both Contents ---
         return {
           id: documentId,
           title: document.title,
@@ -70,7 +65,6 @@ export const updateDocument = ({ session: _session, documentId: defaultDocumentI
       } catch (error: any) {
         console.error('[AI Tool] updateDocument failed:', error);
         const errorMessage = error instanceof Error ? error.message : String(error);
-        // No dataStream to write to, just return error
         return {
           error: 'Failed to generate document update: ' + errorMessage,
         };