@tpitre/story-ui 4.3.0 → 4.4.1

package/README.md

@@ -10,7 +10,7 @@ Story UI revolutionizes component documentation by automatically generating Stor
 ## Why Story UI?
 
 - **Framework Agnostic**: Works with React, Vue, Angular, Svelte, and Web Components
-- **Multi-Provider AI**: Choose between Claude (Anthropic), GPT-4o (OpenAI), or Gemini (Google)
+- **Multi-Provider AI**: Choose between Claude, OpenAI, or Google Gemini - always using the latest models
 - **Design System Aware**: Learns your component library and generates appropriate code
 - **Production Ready**: Deploy as a standalone web app with full MCP integration
 - **Zero Lock-in**: Use any component library - Mantine, Vuetify, Angular Material, Shoelace, or your own
@@ -68,8 +68,8 @@ Story UI will guide you through:
 | Provider | Models | Best For |
 |----------|--------|----------|
 | **Claude** (Anthropic) | claude-opus-4-5, claude-sonnet-4-5, claude-haiku-4-5 | Complex reasoning, code quality |
-| **GPT** (OpenAI) | gpt-4o, gpt-4o-mini, o1 | Versatility, speed |
-| **Gemini** (Google) | gemini-2.0-flash, gemini-1.5-pro | Fast generation, cost efficiency |
+| **GPT** (OpenAI) | gpt-5.2, gpt-5.1, gpt-4o, gpt-4o-mini | Versatility, latest capabilities |
+| **Gemini** (Google) | gemini-2.5-pro, gemini-2.5-flash, gemini-2.0-flash | Fast generation, cost efficiency |
 
 ### Production Deployment
 - **Railway**: Node.js backend with file-based story persistence
@@ -121,7 +121,7 @@ The interactive installer will ask:
    ```
    ? Which AI provider do you prefer?
      > Claude (Anthropic) - Recommended
-       OpenAI (GPT-4o)
+       OpenAI
        Google Gemini
 
    ? Enter your API key:
@@ -262,11 +262,18 @@ The easiest way to connect is via Claude Desktop's built-in connector UI:
 2. Go to **Settings** → **Connectors**
 3. Click **"Add custom connector"**
 4. Enter:
-   - **Name**: `Story UI` (or any name you prefer)
-   - **URL**: `https://story-ui-demo.up.railway.app/mcp-remote/mcp` (production)
+   - **Name**: `Story UI React` (or any descriptive name)
+   - **URL**: Your deployed Railway URL + `/mcp-remote/mcp`
+     - Example: `https://your-app-name.up.railway.app/mcp-remote/mcp`
 5. Click **Add**
 6. **Restart Claude Desktop**
 
+> **Note**: The URL will be your own Railway deployment URL. See [Production Deployment](#production-deployment) to set up your instance.
+
+**Multiple Projects**: If you have multiple Storybook projects, add a separate connector for each:
+- `Story UI React` → `https://my-react-app.up.railway.app/mcp-remote/mcp`
+- `Story UI Vue` → `https://my-vue-app.up.railway.app/mcp-remote/mcp`
+
 Once connected, you'll have access to all Story UI tools directly in your Claude conversations:
 - `generate-story` - Generate Storybook stories from natural language
 - `list-components` - Discover available components
@@ -280,38 +287,52 @@ Once connected, you'll have access to all Story UI tools directly in your Claude
 Connect via Claude Code's built-in MCP support:
 
 ```bash
-# Add remote HTTP MCP server (production)
-claude mcp add --transport http story-ui https://story-ui-demo.up.railway.app/mcp-remote/mcp
+# Add your production Railway deployment
+claude mcp add --transport http story-ui-react https://your-react-app.up.railway.app/mcp-remote/mcp
+
+# Add another project (if needed)
+claude mcp add --transport http story-ui-vue https://your-vue-app.up.railway.app/mcp-remote/mcp
 
-# Or for local development
-claude mcp add --transport http story-ui-local http://localhost:4005/mcp-remote/mcp
+# For local development (default port is 4001)
+claude mcp add --transport http story-ui-local http://localhost:4001/mcp-remote/mcp
 ```
 
 ### Manual Configuration (Advanced)
 
-For advanced users who prefer manual configuration, add to your Claude Desktop config (`~/Library/Application Support/Claude/claude_desktop_config.json` on macOS):
+For running multiple local Story UI instances with different ports, configure your Claude Desktop config (`~/Library/Application Support/Claude/claude_desktop_config.json` on macOS):
 
 ```json
 {
   "mcpServers": {
-    "story-ui": {
+    "story-ui-react": {
+      "command": "npx",
+      "args": ["@tpitre/story-ui", "start", "--port", "4001"]
+    },
+    "story-ui-vue": {
+      "command": "npx",
+      "args": ["@tpitre/story-ui", "start", "--port", "4002"]
+    },
+    "story-ui-angular": {
       "command": "npx",
-      "args": ["@tpitre/story-ui", "mcp"],
-      "env": {
-        "ANTHROPIC_API_KEY": "your-api-key"
-      }
+      "args": ["@tpitre/story-ui", "start", "--port", "4003"]
     }
   }
 }
 ```
 
+> **Note**: When using Claude Desktop, API keys are managed through your Anthropic account - no need to configure them in the MCP server.
+
 ### Starting the Local MCP Server
 
 ```bash
+# Start with default port (4001)
 npx story-ui start
+
+# Or specify a custom port
+npx story-ui start --port 4002
 ```
 
-This starts the Story UI HTTP server with MCP endpoint at `http://localhost:4005/mcp-remote/mcp`.
+This starts the Story UI HTTP server with MCP endpoint at `http://localhost:<port>/mcp-remote/mcp`.
 
 ### Available MCP Commands
 
@@ -325,13 +346,13 @@ Once connected, you can use these commands in Claude Desktop:
 
 ## Production Deployment
 
-Story UI can be deployed as a standalone web application accessible from anywhere.
+Story UI can be deployed as a standalone web application accessible from anywhere. We recommend Railway for its ease of use, but any Node.js hosting platform will work.
 
 ### Architecture
 
 ```
 ┌─────────────────────────────────────────────────────────────┐
-│                    Railway Deployment                        │
+│                    Your Deployment (e.g., Railway)           │
 │  ┌─────────────────────────────────────────────────────────┐│
 │  │              Express MCP Server (Node.js)                ││
 │  │  - Serves Storybook with Story UI addon                  ││
@@ -342,9 +363,9 @@ Story UI can be deployed as a standalone web application accessible from anywher
 └──────────────────────────────────────────────────────────────┘
 ```
 
-### Deploy to Railway
+### Deploy to Railway (Recommended)
 
-Railway provides a straightforward deployment experience with file-based story persistence.
+Railway provides a straightforward deployment experience with automatic HTTPS and file-based story persistence.
 
 **Quick Start:**
 
@@ -353,23 +374,27 @@ Railway provides a straightforward deployment experience with file-based story p
 npm install -g @railway/cli
 railway login
 
-# Initialize and deploy
+# Initialize and deploy from your Storybook project
 railway init
 railway up
 ```
 
 **Environment Variables (set in Railway Dashboard):**
 - `ANTHROPIC_API_KEY` - Required for Claude models
-- `OPENAI_API_KEY` - Optional for OpenAI models
-- `GEMINI_API_KEY` - Optional for Gemini models
+- `OPENAI_API_KEY` - Optional, for OpenAI models
+- `GEMINI_API_KEY` - Optional, for Gemini models
+
+**After Deployment:**
 
-**Connect MCP Clients:**
+Your Railway app will have a URL like `https://your-app-name.up.railway.app`. Use this URL to connect MCP clients:
 
 ```bash
-# Add your Railway deployment as an MCP server
-claude mcp add --transport http story-ui https://your-app.up.railway.app/mcp-remote/mcp
+# In Claude Code
+claude mcp add --transport http story-ui https://your-app-name.up.railway.app/mcp-remote/mcp
 ```
 
+Or add it to Claude Desktop via **Settings** → **Connectors** → **Add custom connector**.
+
 See [DEPLOYMENT.md](DEPLOYMENT.md) for detailed deployment instructions and troubleshooting.
 
 ---
@@ -423,17 +448,14 @@ For simpler setups, use `story-ui-considerations.md`:
 ## CLI Reference
 
 ```bash
-# Initialize Story UI
+# Initialize Story UI in your project
 npx story-ui init
 
-# Start the development server
+# Start the MCP server (default port: 4001)
 npx story-ui start
-npx story-ui start --port 4005
-
-# Deploy to production
-npx story-ui deploy
+npx story-ui start --port 4002  # Custom port
 
-# Run MCP server
+# Run MCP STDIO server (for Claude Desktop local integration)
 npx story-ui mcp
 ```
 

package/dist/mcp-server/index.js

@@ -22,6 +22,26 @@ import { UrlRedirectService } from '../story-generator/urlRedirectService.js';
 import { getProviders, getModels, configureProviderRoute, validateApiKey, setDefaultProvider, setModel, getUISettings, applyUISettings, getSettingsConfig } from './routes/providers.js';
 import { listFrameworks, detectCurrentFramework, getFrameworkDetails, validateStoryForFramework, postProcessStoryForFramework, } from './routes/frameworks.js';
 import mcpRemoteRouter from './routes/mcpRemote.js';
+import { getAdapterRegistry } from '../story-generator/framework-adapters/index.js';
+// Supported story file extensions for all frameworks
+const STORY_EXTENSIONS = ['.stories.tsx', '.stories.ts', '.stories.svelte', '.stories.js'];
+/**
+ * Check if a file is a story file (supports all framework extensions)
+ */
+function isStoryFile(filename) {
+    return STORY_EXTENSIONS.some(ext => filename.endsWith(ext));
+}
+/**
+ * Remove story extension from filename to get base name
+ */
+function removeStoryExtension(filename) {
+    for (const ext of STORY_EXTENSIONS) {
+        if (filename.endsWith(ext)) {
+            return filename.replace(ext, '');
+        }
+    }
+    return filename;
+}
 const app = express();
 // CORS configuration
 // - Allow all origins for /story-ui/* routes (public API for production Storybooks)
@@ -88,19 +108,19 @@ app.get('/mcp/stories', async (req, res) => {
         }
         const files = fs.readdirSync(storiesPath);
         const stories = files
-            .filter(file => file.endsWith('.stories.tsx') || file.endsWith('.stories.ts') || file.endsWith('.stories.svelte'))
+            .filter(file => isStoryFile(file))
             .map(file => {
             const filePath = path.join(storiesPath, file);
             const stats = fs.statSync(filePath);
             const content = fs.readFileSync(filePath, 'utf-8');
             // Extract title from story file
             const titleMatch = content.match(/title:\s*['"]([^'"]+)['"]/);
-            let title = titleMatch ? titleMatch[1].replace('Generated/', '') : file.replace(/\.stories\.(tsx|ts|svelte)$/, '');
+            let title = titleMatch ? titleMatch[1].replace('Generated/', '') : removeStoryExtension(file);
             // Remove hash suffix from display title
             title = title.replace(/\s*\([a-f0-9]{8}\)$/i, '');
             return {
-                id: file.replace(/\.stories\.(tsx|ts|svelte)$/, ''),
-                storyId: file.replace(/\.stories\.(tsx|ts|svelte)$/, ''),
+                id: removeStoryExtension(file),
+                storyId: removeStoryExtension(file),
                 fileName: file,
                 title,
                 lastUpdated: stats.mtime.getTime(),
@@ -326,18 +346,18 @@ app.get('/story-ui/stories', async (req, res) => {
         }
         const files = fs.readdirSync(storiesPath);
         const stories = files
-            .filter(file => file.endsWith('.stories.tsx'))
+            .filter(file => isStoryFile(file))
             .map(file => {
             const filePath = path.join(storiesPath, file);
             const stats = fs.statSync(filePath);
             const content = fs.readFileSync(filePath, 'utf-8');
             // Extract title from story file
             const titleMatch = content.match(/title:\s*['"]([^'"]+)['"]/);
-            let title = titleMatch ? titleMatch[1].replace('Generated/', '') : file.replace('.stories.tsx', '');
+            let title = titleMatch ? titleMatch[1].replace('Generated/', '') : removeStoryExtension(file);
             // Remove hash suffix like " (a1b2c3d4)" from display title - hash is for Storybook uniqueness only
             title = title.replace(/\s*\([a-f0-9]{8}\)$/i, '');
             return {
-                id: file.replace('.stories.tsx', ''),
+                id: removeStoryExtension(file),
                 fileName: file,
                 title,
                 lastUpdated: stats.mtime.getTime(),
@@ -365,7 +385,12 @@ app.post('/story-ui/stories', async (req, res) => {
         if (!fs.existsSync(storiesPath)) {
             fs.mkdirSync(storiesPath, { recursive: true });
         }
-        const fileName = `${id}.stories.tsx`;
+        // Get the correct file extension from the framework adapter
+        const registry = getAdapterRegistry();
+        const frameworkType = (config.componentFramework || 'react');
+        const adapter = registry.getAdapter(frameworkType);
+        const extension = adapter?.defaultExtension || '.stories.tsx';
+        const fileName = `${id}${extension}`;
         const filePath = path.join(storiesPath, fileName);
         fs.writeFileSync(filePath, code, 'utf-8');
         console.log(`✅ Saved story: ${filePath}`);
@@ -388,10 +413,15 @@ app.delete('/story-ui/stories/:id', async (req, res) => {
         const { id } = req.params;
         const storiesPath = config.generatedStoriesPath;
         // Try exact match first (fileName format)
-        // Handle both .tsx and .svelte extensions
+        // Handle all story file extensions
         let fileName = id;
-        if (!id.endsWith('.stories.tsx') && !id.endsWith('.stories.ts') && !id.endsWith('.stories.svelte')) {
-            fileName = `${id}.stories.tsx`;
+        if (!isStoryFile(id)) {
+            // Get the correct file extension from the framework adapter
+            const registry = getAdapterRegistry();
+            const frameworkType = (config.componentFramework || 'react');
+            const adapter = registry.getAdapter(frameworkType);
+            const extension = adapter?.defaultExtension || '.stories.tsx';
+            fileName = `${id}${extension}`;
         }
         const filePath = path.join(storiesPath, fileName);
         if (fs.existsSync(filePath)) {
@@ -433,7 +463,7 @@ app.post('/story-ui/delete', async (req, res) => {
         console.log(`🔍 Searching for story in: ${storiesPath}`);
         if (fs.existsSync(storiesPath)) {
             const files = fs.readdirSync(storiesPath);
-            const matchingFile = files.find(file => file.includes(id) || file.replace('.stories.tsx', '') === id);
+            const matchingFile = files.find(file => file.includes(id) || removeStoryExtension(file) === id);
             if (matchingFile) {
                 const filePath = path.join(storiesPath, matchingFile);
                 fs.unlinkSync(filePath);
@@ -467,9 +497,15 @@ app.post('/story-ui/stories/delete-bulk', async (req, res) => {
         const deleted = [];
         const notFound = [];
         const errors = [];
+        // Get the correct file extension from the framework adapter
+        const registry = getAdapterRegistry();
+        const frameworkType = (config.componentFramework || 'react');
+        const adapter = registry.getAdapter(frameworkType);
+        const extension = adapter?.defaultExtension || '.stories.tsx';
         for (const id of ids) {
             try {
-                const fileName = id.endsWith('.stories.tsx') ? id : `${id}.stories.tsx`;
+                // Check if id already has a story extension
+                const fileName = isStoryFile(id) ? id : `${id}${extension}`;
                 const filePath = path.join(storiesPath, fileName);
                 if (fs.existsSync(filePath)) {
                     fs.unlinkSync(filePath);
@@ -477,7 +513,17 @@ app.post('/story-ui/stories/delete-bulk', async (req, res) => {
                     console.log(`✅ Deleted: ${fileName}`);
                 }
                 else {
-                    notFound.push(id);
+                    // Try to find a matching file with any story extension
+                    const files = fs.readdirSync(storiesPath);
+                    const matchingFile = files.find(f => isStoryFile(f) && (f === id || removeStoryExtension(f) === id));
+                    if (matchingFile) {
+                        fs.unlinkSync(path.join(storiesPath, matchingFile));
+                        deleted.push(id);
+                        console.log(`✅ Deleted: ${matchingFile}`);
+                    }
+                    else {
+                        notFound.push(id);
+                    }
                 }
             }
             catch (err) {
@@ -543,10 +589,8 @@ app.delete('/story-ui/stories', async (req, res) => {
             return res.json({ success: true, deleted: 0, message: 'No stories directory found' });
         }
         const files = fs.readdirSync(storiesPath);
-        // Support all story file extensions: .tsx, .ts, .svelte
-        const storyFiles = files.filter(file => file.endsWith('.stories.tsx') ||
-            file.endsWith('.stories.ts') ||
-            file.endsWith('.stories.svelte'));
+        // Support all story file extensions
+        const storyFiles = files.filter(file => isStoryFile(file));
         let deleted = 0;
         for (const file of storyFiles) {
             try {
@@ -582,24 +626,16 @@ app.post('/story-ui/orphan-stories', async (req, res) => {
             return res.json({ orphans: [], count: 0 });
         }
         const files = fs.readdirSync(storiesPath);
-        const storyFiles = files.filter(file => file.endsWith('.stories.tsx') ||
-            file.endsWith('.stories.ts') ||
-            file.endsWith('.stories.svelte'));
+        const storyFiles = files.filter(file => isStoryFile(file));
         // Find orphans: stories that don't match any chat fileName
         const orphans = storyFiles.filter(storyFile => {
             // Extract the base name without extension for comparison
-            const storyBase = storyFile
-                .replace('.stories.tsx', '')
-                .replace('.stories.ts', '')
-                .replace('.stories.svelte', '');
+            const storyBase = removeStoryExtension(storyFile);
             // Check if any chat fileName matches this story
             return !chatFileNames.some(chatFileName => {
                 if (!chatFileName)
                     return false;
-                const chatBase = chatFileName
-                    .replace('.stories.tsx', '')
-                    .replace('.stories.ts', '')
-                    .replace('.stories.svelte', '');
+                const chatBase = removeStoryExtension(chatFileName);
                 return storyBase === chatBase || storyFile === chatFileName;
             });
         });
@@ -642,22 +678,14 @@ app.delete('/story-ui/orphan-stories', async (req, res) => {
             return res.json({ deleted: [], count: 0 });
         }
         const files = fs.readdirSync(storiesPath);
-        const storyFiles = files.filter(file => file.endsWith('.stories.tsx') ||
-            file.endsWith('.stories.ts') ||
-            file.endsWith('.stories.svelte'));
+        const storyFiles = files.filter(file => isStoryFile(file));
         // Find orphans: stories that don't match any chat fileName
         const orphans = storyFiles.filter(storyFile => {
-            const storyBase = storyFile
-                .replace('.stories.tsx', '')
-                .replace('.stories.ts', '')
-                .replace('.stories.svelte', '');
+            const storyBase = removeStoryExtension(storyFile);
             return !chatFileNames.some(chatFileName => {
                 if (!chatFileName)
                     return false;
-                const chatBase = chatFileName
-                    .replace('.stories.tsx', '')
-                    .replace('.stories.ts', '')
-                    .replace('.stories.svelte', '');
+                const chatBase = removeStoryExtension(chatFileName);
                 return storyBase === chatBase || storyFile === chatFileName;
             });
         });

package/dist/mcp-server/routes/mcpRemote.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcpRemote.d.ts","sourceRoot":"","sources":["../../../mcp-server/routes/mcpRemote.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAiCH,eAAO,MAAM,MAAM,4CAAW,CAAC;AA6mB/B,eAAe,MAAM,CAAC"}
+{"version":3,"file":"mcpRemote.d.ts","sourceRoot":"","sources":["../../../mcp-server/routes/mcpRemote.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAiGH,eAAO,MAAM,MAAM,4CAAW,CAAC;AAwlB/B,eAAe,MAAM,CAAC"}

package/dist/mcp-server/routes/mcpRemote.js

@@ -38,6 +38,63 @@ const HTTP_PORT = process.env.VITE_STORY_UI_PORT || process.env.STORY_UI_HTTP_PO
 const HTTP_BASE_URL = `http://localhost:${HTTP_PORT}`;
 // Load configuration
 const config = loadUserConfig();
+// Get all supported story file extensions based on framework adapters
+// This ensures MCP remote works with all frameworks (React, Vue, Angular, Svelte, Web Components)
+const STORY_EXTENSIONS = ['.stories.tsx', '.stories.ts', '.stories.svelte', '.stories.js'];
+/**
+ * Check if a file is a story file (supports all framework extensions)
+ */
+function isStoryFile(filename) {
+    return STORY_EXTENSIONS.some(ext => filename.endsWith(ext));
+}
+/**
+ * Get the story ID from a filename (works with all extensions)
+ */
+function getStoryIdFromFile(filename) {
+    // Extract hash from filename like "Button-a1b2c3d4.stories.tsx" or "Card-12345678.stories.svelte"
+    const hashMatch = filename.match(/-([a-f0-9]{8})\.stories\.(tsx?|svelte|js)$/);
+    if (hashMatch) {
+        return `story-${hashMatch[1]}`;
+    }
+    // Fall back to filename without extension
+    for (const ext of STORY_EXTENSIONS) {
+        if (filename.endsWith(ext)) {
+            return filename.replace(ext, '');
+        }
+    }
+    return filename;
+}
+/**
+ * Get the base name from a filename (without extension)
+ */
+function getBaseName(filename) {
+    for (const ext of STORY_EXTENSIONS) {
+        if (filename.endsWith(ext)) {
+            return filename.replace(ext, '');
+        }
+    }
+    return filename;
+}
+/**
+ * Check if a file matches a story ID (supports all extensions)
+ */
+function fileMatchesStoryId(filename, storyId) {
+    // Extract hash from storyId like "story-a1b2c3d4"
+    const hashMatch = storyId.match(/^story-([a-f0-9]{8})$/);
+    const hash = hashMatch ? hashMatch[1] : null;
+    if (hash) {
+        // Check if filename contains the hash
+        return filename.includes(`-${hash}.stories.`);
+    }
+    // Check if filename matches storyId with any extension
+    if (filename === storyId)
+        return true;
+    for (const ext of STORY_EXTENSIONS) {
+        if (filename === `${storyId}${ext}`)
+            return true;
+    }
+    return false;
+}
 export const router = Router();
 // Store SSE transports for legacy session management
 const sseTransports = {};
@@ -206,11 +263,10 @@ async function handleToolCall(name, args) {
             if (config.generatedStoriesPath && fs.existsSync(config.generatedStoriesPath)) {
                 const files = fs.readdirSync(config.generatedStoriesPath);
                 fileStories = files
-                    .filter(file => file.endsWith('.stories.tsx'))
+                    .filter(file => isStoryFile(file))
                     .map(file => {
-                    const hash = file.match(/-([a-f0-9]{8})\.stories\.tsx$/)?.[1] || '';
-                    const storyId = hash ? `story-${hash}` : file.replace('.stories.tsx', '');
-                    let title = file.replace('.stories.tsx', '').replace(/-/g, ' ');
+                    const storyId = getStoryIdFromFile(file);
+                    let title = getBaseName(file).replace(/-/g, ' ');
                     try {
                         const filePath = path.join(config.generatedStoriesPath, file);
                         const content = fs.readFileSync(filePath, 'utf-8');
@@ -243,17 +299,7 @@ async function handleToolCall(name, args) {
             const { storyId } = args;
             if (config.generatedStoriesPath && fs.existsSync(config.generatedStoriesPath)) {
                 const files = fs.readdirSync(config.generatedStoriesPath);
-                const hashMatch = storyId.match(/^story-([a-f0-9]{8})$/);
-                const hash = hashMatch ? hashMatch[1] : null;
-                const matchingFile = files.find(file => {
-                    if (hash && file.includes(`-${hash}.stories.tsx`))
-                        return true;
-                    if (file === `${storyId}.stories.tsx`)
-                        return true;
-                    if (file === storyId)
-                        return true;
-                    return false;
-                });
+                const matchingFile = files.find(file => fileMatchesStoryId(file, storyId));
                 if (matchingFile) {
                     const filePath = path.join(config.generatedStoriesPath, matchingFile);
                     const content = fs.readFileSync(filePath, 'utf-8');
@@ -273,17 +319,7 @@ async function handleToolCall(name, args) {
             const { storyId } = args;
             if (config.generatedStoriesPath && fs.existsSync(config.generatedStoriesPath)) {
                 const files = fs.readdirSync(config.generatedStoriesPath);
-                const hashMatch = storyId.match(/^story-([a-f0-9]{8})$/);
-                const hash = hashMatch ? hashMatch[1] : null;
-                const matchingFile = files.find(file => {
-                    if (hash && file.includes(`-${hash}.stories.tsx`))
-                        return true;
-                    if (file === `${storyId}.stories.tsx`)
-                        return true;
-                    if (file === storyId)
-                        return true;
-                    return false;
-                });
+                const matchingFile = files.find(file => fileMatchesStoryId(file, storyId));
                 if (matchingFile) {
                     const filePath = path.join(config.generatedStoriesPath, matchingFile);
                     fs.unlinkSync(filePath);
@@ -325,15 +361,14 @@ async function handleToolCall(name, args) {
             // Find the story to update if no ID provided
             if (!storyId && config.generatedStoriesPath && fs.existsSync(config.generatedStoriesPath)) {
                 const files = fs.readdirSync(config.generatedStoriesPath)
-                    .filter(f => f.endsWith('.stories.tsx'))
+                    .filter(f => isStoryFile(f))
                     .sort((a, b) => {
                     const statA = fs.statSync(path.join(config.generatedStoriesPath, a));
                     const statB = fs.statSync(path.join(config.generatedStoriesPath, b));
                     return statB.mtime.getTime() - statA.mtime.getTime();
                 });
                 if (files.length > 0) {
-                    const hash = files[0].match(/-([a-f0-9]{8})\.stories\.tsx$/)?.[1];
-                    storyId = hash ? `story-${hash}` : files[0].replace('.stories.tsx', '');
+                    storyId = getStoryIdFromFile(files[0]);
                 }
             }
             if (!storyId) {
@@ -350,13 +385,7 @@ async function handleToolCall(name, args) {
             let storyMetadata = {};
             if (config.generatedStoriesPath && fs.existsSync(config.generatedStoriesPath)) {
                 const files = fs.readdirSync(config.generatedStoriesPath);
-                const hashMatch = storyId.match(/^story-([a-f0-9]{8})$/);
-                const hash = hashMatch ? hashMatch[1] : null;
-                const matchingFile = files.find(file => {
-                    if (hash && file.includes(`-${hash}.stories.tsx`))
-                        return true;
-                    return false;
-                });
+                const matchingFile = files.find(file => fileMatchesStoryId(file, storyId));
                 if (matchingFile) {
                     const filePath = path.join(config.generatedStoriesPath, matchingFile);
                     existingCode = fs.readFileSync(filePath, 'utf-8');