added ai generation on backend

Author: Aswanth-c

packages/api/ai/generate.mts

@@ -6,6 +6,7 @@ import {
   type CodeCellType,
   randomid,
   type CellWithPlaceholderType,
+  type MarkdownCellType,
 } from '@srcbook/shared';
 import { type SessionType } from '../types.mjs';
 import { readFileSync } from 'node:fs';
@@ -79,17 +80,19 @@ ${diagnostics}
 };
 
 const makeGenerateCellEditSystemPrompt = (language: CodeLanguageType) => {
+  console.log(`code-updater-${language}.txt`);
   return readFileSync(Path.join(PROMPTS_DIR, `code-updater-${language}.txt`), 'utf-8');
 };
 
 const makeGenerateCellEditUserPrompt = (
   query: string,
   session: SessionType,
-  cell: CodeCellType,
+  cell: CodeCellType | MarkdownCellType,
 ) => {
+  const cellLanguage = cell.type === 'markdown' ? 'markdown' : session.language;
   // Intentionally not passing in tsconfig.json here as that doesn't need to be in the prompt.
   const inlineSrcbook = encode(
-    { cells: session.cells, language: session.language },
+    { cells: session.cells, language: cellLanguage as CodeLanguageType },
     { inline: true },
   );
 
@@ -98,7 +101,7 @@ ${inlineSrcbook}
 ==== END SRCBOOK ====
 
 ==== BEGIN CODE CELL ====
-${cell.source}
+${cell.type === 'markdown' ? (cell as MarkdownCellType).text : (cell as CodeCellType).source}
 ==== END CODE CELL ====
 
 ==== BEGIN USER REQUEST ====
@@ -180,10 +183,14 @@ export async function generateCells(
   }
 }
 
-export async function generateCellEdit(query: string, session: SessionType, cell: CodeCellType) {
+export async function generateCellEdit(
+  query: string,
+  session: SessionType,
+  cell: CodeCellType | MarkdownCellType,
+) {
   const model = await getModel();
-
-  const systemPrompt = makeGenerateCellEditSystemPrompt(session.language);
+  const cellLanguage = cell.type === 'markdown' ? 'markdown' : session.language;
+  const systemPrompt = makeGenerateCellEditSystemPrompt(cellLanguage as CodeLanguageType);
   const userPrompt = makeGenerateCellEditUserPrompt(query, session, cell);
   const result = await generateText({
     model,

packages/api/prompts/code-updater-markdown.txt

@@ -0,0 +1,84 @@
+<!-- srcbook:{"language":"markdown"} -->
+
+## Instructions Context
+
+You are tasked with editing a **Markdown cell** in a Srcbook.
+
+A Srcbook is a **Markdown-compatible notebook**, used for documentation or text-based content.
+
+### Srcbook Spec
+
+The structure of a Srcbook:
+0. The language comment: `<!-- srcbook:{"language":"markdown"} -->`
+1. Title cell (heading 1)
+2. N more cells, which are either:
+   - **Markdown cells** (GitHub flavored Markdown)
+   -  Markdown cells, which have a filename and source content.
+
+#### Important Note:
+Markdown cells cannot use h1 or h6 heading, as these are reserved for Srcbook.
+
+The user is already working on an existing Srcbook and is asking you to edit a specific Markdown cell.  
+The Srcbook contents will be passed to you as context, as well as the user's request about the intended edits for the Markdown cell.
+
+---
+
+## Example Srcbook
+
+<!-- srcbook:{"language":"markdown"} -->
+
+### Getting Started
+
+#### What are Srcbooks?
+
+Srcbooks are an interactive way of organizing and presenting information. They are similar to other notebooks but unique in their flexibility and format.
+
+#### Dependencies
+
+You can include any necessary information, resources, or links to external content.
+
+##### Introduction
+
+This is a Markdown cell showcasing various Markdown features.
+
+#### Features Overview
+
+##### Text Formatting
+
+- **Bold text**
+- *Italic text*
+- ~~Strikethrough text~~
+
+##### Lists
+
+- **Unordered List:**
+  - Item 1
+  - Item 2
+
+- **Ordered List:**
+  1. First item
+  2. Second item
+
+##### Code Blocks
+
+Inline code: `console.log("Hello, Markdown!")`
+
+##### Links
+
+[Click here to visit Google](https://www.google.com)
+
+##### Images
+
+![Alt text](image.png)
+
+---
+
+## Final Instructions
+
+The user's Srcbook will be passed to you, surrounded with `==== BEGIN SRCBOOK ====` and `==== END SRCBOOK ====`.  
+The specific **Markdown cell** they want updated will also be passed to you, surrounded with `==== BEGIN MARKDOWN CELL ====` and `==== END MARKDOWN CELL ====`.  
+The user's intent will be passed to you between `==== BEGIN USER REQUEST ====` and `==== END USER REQUEST ====`.
+
+Your job is to edit the cell based on the contents of the Srcbook and the user's intent.  
+Act as a **Markdown expert**, writing the best possible content you can. Focus on being **elegant, concise, and clear**.  
+**ONLY RETURN THE MARKDOWN CONTENT, NO PREAMBULE, NO SUFFIX, ONLY THE MARKDOWN**.

packages/shared/src/schemas/cells.ts

@@ -22,7 +22,7 @@ export const PackageJsonCellSchema = z.object({
 
 export const CodeCellSchema = z.object({
   id: z.string(),
-  type: z.literal('code'),
+  type: z.enum(['code', 'markdown']),
   source: z.string(),
   language: z.enum(['javascript', 'typescript']),
   filename: z.string(),