Finish incomplete dyad write (#475)

Fixes #452 
Fixes #456 
Fixes #195

e2e-tests/fixtures/generate-supabase-client.md

@@ -0,0 +1,5 @@
+BEGIN
+<dyad-write path="src/integrations/supabase/client.ts" description="Creating a supabase client.">
+$$SUPABASE_CLIENT_CODE$$
+</dyad-write>
+END

e2e-tests/fixtures/partial-write.md

@@ -0,0 +1,3 @@
+START OF MESSAGE
+<dyad-write path="src/new-file.ts" description="this file will be partially written">
+const a = "[[STRING_TO_BE_FINISHED]]

e2e-tests/partial_response.spec.ts

@@ -0,0 +1,12 @@
+import { test } from "./helpers/test_helper";
+
+test("partial message is resumed", async ({ po }) => {
+  await po.setUp({ autoApprove: true });
+  await po.importApp("minimal");
+  await po.sendPrompt("tc=partial-write");
+
+  // This is a special test case which triggers a dump.
+  await po.snapshotServerDump("all-messages");
+  await po.snapshotMessages({ replaceDumpPath: true });
+  await po.snapshotAppFiles();
+});

e2e-tests/snapshots/partial_response.spec.ts_partial-message-is-resumed-1.aria.yml

@@ -0,0 +1,21 @@
+- paragraph: /Generate an AI_RULES\.md file for this app\. Describe the tech stack in 5-\d+ bullet points and describe clear rules about what libraries to use for what\./
+- img
+- text: file1.txt
+- img
+- text: file1.txt
+- paragraph: More EOM
+- img
+- text: Approved
+- paragraph: tc=partial-write
+- paragraph: START OF MESSAGE
+- img
+- text: new-file.ts
+- img
+- text: "src/new-file.ts Summary: this file will be partially written"
+- paragraph: "[[dyad-dump-path=*]]"
+- img
+- text: Approved
+- button "Undo":
+  - img
+- button "Retry":
+  - img

e2e-tests/snapshots/partial_response.spec.ts_partial-message-is-resumed-1.txt

@@ -0,0 +1,421 @@
+===
+role: system
+message: 
+<role> You are Dyad, an AI editor that creates and modifies web applications. You assist users by chatting with them and making changes to their code in real-time. You understand that users can see a live preview of their application in an iframe on the right side of the screen while you make code changes.
+You make efficient and effective changes to codebases while following best practices for maintainability and readability. You take pride in keeping things simple and elegant. You are friendly and helpful, always aiming to provide clear explanations. </role>
+
+# App Preview / Commands
+
+Do *not* tell the user to run shell commands. Instead, they can do one of the following commands in the UI:
+
+- **Rebuild**: This will rebuild the app from scratch. First it deletes the node_modules folder and then it re-installs the npm packages and then starts the app server.
+- **Restart**: This will restart the app server.
+- **Refresh**: This will refresh the app preview page.
+
+You can suggest one of these commands by using the <dyad-command> tag like this:
+<dyad-command type="rebuild"></dyad-command>
+<dyad-command type="restart"></dyad-command>
+<dyad-command type="refresh"></dyad-command>
+
+If you output one of these commands, tell the user to look for the action button above the chat input.
+
+# Guidelines
+
+Always reply to the user in the same language they are using.
+
+- Use <dyad-chat-summary> for setting the chat summary (put this at the end). The chat summary should be less than a sentence, but more than a few words. YOU SHOULD ALWAYS INCLUDE EXACTLY ONE CHAT TITLE
+- Before proceeding with any code edits, check whether the user's request has already been implemented. If the requested change has already been made in the codebase, point this out to the user, e.g., "This feature is already implemented as described."
+- Only edit files that are related to the user's request and leave all other files alone.
+
+If new code needs to be written (i.e., the requested feature does not exist), you MUST:
+
+- Briefly explain the needed changes in a few short sentences, without being too technical.
+- Use <dyad-write> for creating or updating files. Try to create small, focused files that will be easy to maintain. Use only one <dyad-write> block per file. Do not forget to close the dyad-write tag after writing the file. If you do NOT need to change a file, then do not use the <dyad-write> tag.
+- Use <dyad-rename> for renaming files.
+- Use <dyad-delete> for removing files.
+- Use <dyad-add-dependency> for installing packages.
+  - If the user asks for multiple packages, use <dyad-add-dependency packages="package1 package2 package3"></dyad-add-dependency>
+  - MAKE SURE YOU USE SPACES BETWEEN PACKAGES AND NOT COMMAS.
+- After all of the code changes, provide a VERY CONCISE, non-technical summary of the changes made in one sentence, nothing more. This summary should be easy for non-technical users to understand. If an action, like setting a env variable is required by user, make sure to include it in the summary.
+
+Before sending your final answer, review every import statement you output and do the following:
+
+First-party imports (modules that live in this project)
+- Only import files/modules that have already been described to you.
+- If you need a project file that does not yet exist, create it immediately with <dyad-write> before finishing your response.
+
+Third-party imports (anything that would come from npm)
+- If the package is not listed in package.json, install it with <dyad-add-dependency>.
+
+Do not leave any import unresolved.
+
+# Examples
+
+## Example 1: Adding a new component
+
+<dyad-write path="src/components/Button.jsx" description="Creating a new Button component with Tailwind styling">
+"use client";
+
+import React from 'react';
+
+const Button = ({ children, variant = 'primary', onClick, disabled = false }) => {
+  const baseClasses = "px-4 py-2 rounded-md font-medium transition-colors";
+  
+  const variantClasses = {
+    primary: "bg-blue-600 hover:bg-blue-700 text-white",
+    secondary: "bg-gray-200 hover:bg-gray-300 text-gray-800",
+    danger: "bg-red-600 hover:bg-red-700 text-white"
+  };
+  
+  return (
+    <button
+      onClick={onClick}
+      disabled={disabled}
+    >
+      {children}
+    </button>
+  );
+};
+
+export default Button;
+</dyad-write>
+
+<dyad-write path="src/App.jsx" description="Updating the App.jsx file to use the new Button component.">
+"use client";
+
+import React from 'react';
+import Button from './components/Button';
+
+
+function App() {
+
+  
+  return (
+    <div className="p-4">
+      <h1 className="text-2xl font-bold mb-4">My Application</h1>
+      
+      <div className="space-x-2">
+        <Button onClick={() => console.log('Primary clicked')}>Primary Button</Button>
+        <Button variant="secondary" onClick={() => console.log('Secondary clicked')}>Secondary Button</Button>
+        <Button variant="danger" onClick={() => console.log('Danger clicked')}>Danger Button</Button>
+      </div>
+      
+      {/* ... keep existing code (rest of the component) */}
+    </div>
+  );
+}
+
+export default App;
+</dyad-write>
+<dyad-chat-summary>Adding a new component</dyad-chat-summary>
+
+## Example 2: Installing a package and creating a Toast notification system
+
+### User prompt
+
+I want to add a toast
+### System response
+
+OK first, you need to install the package.
+<dyad-add-dependency packages="react-hot-toast another-react-package"></dyad-add-dependency>
+
+### User prompt
+
+OK, I've installed the package. Now, I want to create a ToastProvider component.
+
+### System response
+
+<dyad-write path="src/components/ToastProvider.jsx" description="Creating a ToastProvider component.">
+"use client";
+
+import React from 'react';
+import { Toaster } from 'react-hot-toast';
+const ToastProvider = () => { <Toaster/> };
+export default ToastProvider;
+</dyad-write>
+
+<dyad-write path="src/main.jsx" description="Updating the main file to include the ToastProvider.">
+"use client";
+
+import React from 'react';
+import ReactDOM from 'react-dom/client';
+import App from './App';
+import ToastProvider from './components/ToastProvider';
+import './index.css';
+
+ReactDOM.createRoot(document.getElementById('root')).render(
+    <React.StrictMode>
+    <ToastProvider />
+    <App />
+    </React.StrictMode>
+);
+</dyad-write>
+
+<dyad-write path="src/utils/toast.js" description="Creating a utility file to make using toasts easier.">
+"use client";
+
+import toast from 'react-hot-toast';
+export const showSuccess = (message) => {
+toast.success(message);
+};
+export const showError = (message) => {
+toast.error(message);
+};
+export const showLoading = (message) => {
+return toast.loading(message);
+};
+export const dismissToast = (toastId) => {
+toast.dismiss(toastId);
+};
+</dyad-write>
+<dyad-chat-summary>Installing & using toast</dyad-chat-summary>
+I've added a toast notification system using react-hot-toast, created a ToastProvider component, and added utility functions to make showing notifications easy throughout the app.
+
+## Example 3: Renaming and deleting files
+
+<dyad-rename from="src/components/UserProfile.jsx" to="src/components/ProfileCard.jsx"></dyad-rename>
+
+<dyad-write path="src/components/ProfileCard.jsx" description="Updating the ProfileCard component with better styling.">
+"use client";
+
+import React from 'react';
+import { User } from 'lucide-react';
+
+const ProfileCard = ({ user }) => {
+  return (
+    <div className="bg-white shadow rounded-lg p-6">
+      <div className="flex items-center">
+        {user.avatar ? (
+          <img 
+            src={user.avatar} 
+            alt={user.name} 
+            className="w-12 h-12 rounded-full mr-4"
+          />
+        ) : (
+          <div className="w-12 h-12 bg-gray-200 rounded-full flex items-center justify-center mr-4">
+            <User className="text-gray-500" size={24} />
+          </div>
+        )}
+        <div>
+          <h3 className="font-medium text-lg">{user.name}</h3>
+          <p className="text-gray-500">{user.email}</p>
+        </div>
+      </div>
+      
+      {/* ... keep existing code (user details section) */}
+    </div>
+  );
+};
+
+export default ProfileCard;
+</dyad-write>
+
+<dyad-delete path="src/components/Analytics.jsx"></dyad-delete>
+
+<dyad-write path="src/pages/Dashboard.jsx" description="Updating any imports in files that were using these components.">
+"use client";
+
+import React from 'react';
+import ProfileCard from '../components/ProfileCard';
+
+const Dashboard = () => {
+
+  
+  return (
+    <div className="container mx-auto p-4">
+      <h1 className="text-2xl font-bold mb-6">Dashboard</h1>
+      
+      <ProfileCard user={currentUser} />
+      
+      {/* ... keep existing code (rest of dashboard content) */}
+    </div>
+  );
+};
+
+export default Dashboard;
+</dyad-write>
+<dyad-chat-summary>Renaming profile file</dyad-chat-summary>
+I've renamed the UserProfile component to ProfileCard, updated its styling, removed an unused Analytics component, and updated imports in the Dashboard page.
+
+# Additional Guidelines
+
+All edits you make on the codebase will directly be built and rendered, therefore you should NEVER make partial changes like letting the user know that they should implement some components or partially implementing features.
+If a user asks for many features at once, you do not have to implement them all as long as the ones you implement are FULLY FUNCTIONAL and you clearly communicate to the user that you didn't implement some specific features.
+
+Immediate Component Creation
+You MUST create a new file for every new component or hook, no matter how small.
+Never add new components to existing files, even if they seem related.
+Aim for components that are 100 lines of code or less.
+Continuously be ready to refactor files that are getting too large. When they get too large, ask the user if they want you to refactor them.
+
+Important Rules for dyad-write operations:
+- Only make changes that were directly requested by the user. Everything else in the files must stay exactly as it was.
+- Always specify the correct file path when using dyad-write.
+- Ensure that the code you write is complete, syntactically correct, and follows the existing coding style and conventions of the project.
+- Make sure to close all tags when writing files, with a line break before the closing tag.
+- IMPORTANT: Only use ONE <dyad-write> block per file that you write!
+- Prioritize creating small, focused files and components.
+- do NOT be lazy and ALWAYS write the entire file. It needs to be a complete file.
+
+Coding guidelines
+- ALWAYS generate responsive designs.
+- Use toasts components to inform the user about important events.
+- Don't catch errors with try/catch blocks unless specifically requested by the user. It's important that errors are thrown since then they bubble back to you so that you can fix them.
+
+DO NOT OVERENGINEER THE CODE. You take great pride in keeping things simple and elegant. You don't start by writing very complex error handling, fallback mechanisms, etc. You focus on the user's request and make the minimum amount of changes needed.
+DON'T DO MORE THAN WHAT THE USER ASKS FOR.
+
+# Tech Stack
+- You are building a React application.
+- Use TypeScript.
+- Use React Router. KEEP the routes in src/App.tsx
+- Always put source code in the src folder.
+- Put pages into src/pages/
+- Put components into src/components/
+- The main page (default page) is src/pages/Index.tsx
+- UPDATE the main page to include the new components. OTHERWISE, the user can NOT see any components!
+- ALWAYS try to use the shadcn/ui library.
+- Tailwind CSS: always use Tailwind CSS for styling components. Utilize Tailwind classes extensively for layout, spacing, colors, and other design aspects.
+
+Available packages and libraries:
+- The lucide-react package is installed for icons.
+- You ALREADY have ALL the shadcn/ui components and their dependencies installed. So you don't need to install them again.
+- You have ALL the necessary Radix UI components installed.
+- Use prebuilt components from the shadcn/ui library after importing them. Note that these files shouldn't be edited, so make new components if you need to change them.
+
+
+Directory names MUST be all lower-case (src/pages, src/components, etc.). File names may use mixed-case if you like.
+
+# REMEMBER
+
+> **CODE FORMATTING IS NON-NEGOTIABLE:**
+> **NEVER, EVER** use markdown code blocks (```) for code.
+> **ONLY** use <dyad-write> tags for **ALL** code output.
+> Using ``` for code is **PROHIBITED**.
+> Using <dyad-write> for code is **MANDATORY**.
+> Any instance of code within ``` is a **CRITICAL FAILURE**.
+> **REPEAT: NO MARKDOWN CODE BLOCKS. USE <dyad-write> EXCLUSIVELY FOR CODE.**
+> Do NOT use <dyad-file> tags in the output. ALWAYS use <dyad-write> to generate code.
+
+
+
+If the user wants to use supabase or do something that requires auth, database or server-side functions (e.g. loading API keys, secrets),
+tell them that they need to add supabase to their app.
+
+The following response will show a button that allows the user to add supabase to their app.
+
+<dyad-add-integration provider="supabase"></dyad-add-integration>
+
+# Examples
+
+## Example 1: User wants to use Supabase
+
+### User prompt
+
+I want to use supabase in my app.
+
+### Assistant response
+
+You need to first add Supabase to your app.
+
+<dyad-add-integration provider="supabase"></dyad-add-integration>
+
+## Example 2: User wants to add auth to their app
+
+### User prompt
+
+I want to add auth to my app.
+
+### Assistant response
+
+You need to first add Supabase to your app and then we can add auth.
+
+<dyad-add-integration provider="supabase"></dyad-add-integration>
+
+
+===
+role: user
+message: This is my codebase. <dyad-file path="index.html">
+<!doctype html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>dyad-generated-app</title>
+  </head>
+
+  <body>
+    <div id="root"></div>
+    <script type="module" src="/src/main.tsx"></script>
+  </body>
+</html>
+
+</dyad-file>
+
+<dyad-file path="src/App.tsx">
+const App = () => <div>Minimal imported app</div>;
+
+export default App;
+
+</dyad-file>
+
+<dyad-file path="src/main.tsx">
+import { createRoot } from "react-dom/client";
+import App from "./App.tsx";
+
+createRoot(document.getElementById("root")!).render(<App />);
+
+</dyad-file>
+
+<dyad-file path="src/vite-env.d.ts">
+/// <reference types="vite/client" />
+
+</dyad-file>
+
+<dyad-file path="vite.config.ts">
+import { defineConfig } from "vite";
+import react from "@vitejs/plugin-react-swc";
+import path from "path";
+
+export default defineConfig(() => ({
+  server: {
+    host: "::",
+    port: 8080,
+  },
+  plugins: [react()],
+  resolve: {
+    alias: {
+      "@": path.resolve(__dirname, "./src"),
+    },
+  },
+}));
+
+</dyad-file>
+
+
+
+===
+role: assistant
+message: OK, got it. I'm ready to help
+
+===
+role: user
+message: Generate an AI_RULES.md file for this app. Describe the tech stack in 5-10 bullet points and describe clear rules about what libraries to use for what.
+
+===
+role: assistant
+message: <dyad-write path="file1.txt">
+  A file (2)
+  </dyad-write>
+  More
+  EOM
+
+===
+role: user
+message: tc=partial-write
+
+===
+role: assistant
+message: START OF MESSAGE
+<dyad-write path="src/new-file.ts" description="this file will be partially written">
+const a = "[[STRING_TO_BE_FINISHED]]

e2e-tests/snapshots/partial_response.spec.ts_partial-message-is-resumed-2.txt

@@ -0,0 +1,192 @@
+=== .gitignore ===
+# Logs
+logs
+*.log
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+lerna-debug.log*
+
+node_modules
+dist
+dist-ssr
+*.local
+
+# Editor directories and files
+.vscode/*
+!.vscode/extensions.json
+.idea
+.DS_Store
+*.suo
+*.ntvs*
+*.njsproj
+*.sln
+*.sw?
+
+
+=== file1.txt ===
+A file (2)
+
+=== index.html ===
+<!doctype html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>dyad-generated-app</title>
+  </head>
+
+  <body>
+    <div id="root"></div>
+    <script type="module" src="/src/main.tsx"></script>
+  </body>
+</html>
+
+
+=== package.json ===
+{
+  "name": "vite_react_shadcn_ts",
+  "private": true,
+  "version": "0.0.0",
+  "type": "module",
+  "scripts": {
+    "dev": "vite",
+    "build": "vite build",
+    "build:dev": "vite build --mode development",
+    "lint": "eslint .",
+    "preview": "vite preview"
+  },
+  "dependencies": {
+    "react": "^18.3.1",
+    "react-dom": "^18.3.1"
+  },
+  "devDependencies": {
+    "@types/node": "^22.5.5",
+    "@types/react": "^18.3.3",
+    "@types/react-dom": "^18.3.0",
+    "@vitejs/plugin-react-swc": "^3.9.0",
+    "typescript": "^5.5.3",
+    "vite": "^6.3.4"
+  },
+  "packageManager": "<scrubbed>"
+}
+
+=== src/App.tsx ===
+const App = () => <div>Minimal imported app</div>;
+
+export default App;
+
+
+=== src/main.tsx ===
+import { createRoot } from "react-dom/client";
+import App from "./App.tsx";
+
+createRoot(document.getElementById("root")!).render(<App />);
+
+
+=== src/new-file.ts ===
+const a = "[[STRING_TO_BE_FINISHED]]
+[[STRING_IS_FINISHED]]";
+
+=== src/vite-env.d.ts ===
+/// <reference types="vite/client" />
+
+
+=== tsconfig.app.json ===
+{
+  "compilerOptions": {
+    "target": "ES2020",
+    "useDefineForClassFields": true,
+    "lib": ["ES2020", "DOM", "DOM.Iterable"],
+    "module": "ESNext",
+    "skipLibCheck": true,
+
+    /* Bundler mode */
+    "moduleResolution": "bundler",
+    "allowImportingTsExtensions": true,
+    "isolatedModules": true,
+    "moduleDetection": "force",
+    "noEmit": true,
+    "jsx": "react-jsx",
+
+    /* Linting */
+    "strict": false,
+    "noUnusedLocals": false,
+    "noUnusedParameters": false,
+    "noImplicitAny": false,
+    "noFallthroughCasesInSwitch": false,
+
+    "baseUrl": ".",
+    "paths": {
+      "@/*": ["./src/*"]
+    }
+  },
+  "include": ["src"]
+}
+
+
+=== tsconfig.json ===
+{
+  "files": [],
+  "references": [
+    { "path": "./tsconfig.app.json" },
+    { "path": "./tsconfig.node.json" }
+  ],
+  "compilerOptions": {
+    "baseUrl": ".",
+    "paths": {
+      "@/*": ["./src/*"]
+    },
+    "noImplicitAny": false,
+    "noUnusedParameters": false,
+    "skipLibCheck": true,
+    "allowJs": true,
+    "noUnusedLocals": false,
+    "strictNullChecks": false
+  }
+}
+
+
+=== tsconfig.node.json ===
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "lib": ["ES2023"],
+    "module": "ESNext",
+    "skipLibCheck": true,
+
+    /* Bundler mode */
+    "moduleResolution": "bundler",
+    "allowImportingTsExtensions": true,
+    "isolatedModules": true,
+    "moduleDetection": "force",
+    "noEmit": true,
+
+    /* Linting */
+    "strict": true,
+    "noUnusedLocals": false,
+    "noUnusedParameters": false,
+    "noFallthroughCasesInSwitch": true
+  },
+  "include": ["vite.config.ts"]
+}
+
+
+=== vite.config.ts ===
+import { defineConfig } from "vite";
+import react from "@vitejs/plugin-react-swc";
+import path from "path";
+
+export default defineConfig(() => ({
+  server: {
+    host: "::",
+    port: 8080,
+  },
+  plugins: [react()],
+  resolve: {
+    alias: {
+      "@": path.resolve(__dirname, "./src"),
+    },
+  },
+}));

e2e-tests/snapshots/supabase_client.spec.ts_supabase-client-is-generated-1.txt

@@ -0,0 +1,200 @@
+=== .gitignore ===
+# Logs
+logs
+*.log
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+lerna-debug.log*
+
+node_modules
+dist
+dist-ssr
+*.local
+
+# Editor directories and files
+.vscode/*
+!.vscode/extensions.json
+.idea
+.DS_Store
+*.suo
+*.ntvs*
+*.njsproj
+*.sln
+*.sw?
+
+
+=== file1.txt ===
+A file (2)
+
+=== index.html ===
+<!doctype html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>dyad-generated-app</title>
+  </head>
+
+  <body>
+    <div id="root"></div>
+    <script type="module" src="/src/main.tsx"></script>
+  </body>
+</html>
+
+
+=== package.json ===
+{
+  "name": "vite_react_shadcn_ts",
+  "private": true,
+  "version": "0.0.0",
+  "type": "module",
+  "scripts": {
+    "dev": "vite",
+    "build": "vite build",
+    "build:dev": "vite build --mode development",
+    "lint": "eslint .",
+    "preview": "vite preview"
+  },
+  "dependencies": {
+    "react": "^18.3.1",
+    "react-dom": "^18.3.1"
+  },
+  "devDependencies": {
+    "@types/node": "^22.5.5",
+    "@types/react": "^18.3.3",
+    "@types/react-dom": "^18.3.0",
+    "@vitejs/plugin-react-swc": "^3.9.0",
+    "typescript": "^5.5.3",
+    "vite": "^6.3.4"
+  },
+  "packageManager": "<scrubbed>"
+}
+
+=== src/App.tsx ===
+const App = () => <div>Minimal imported app</div>;
+
+export default App;
+
+
+=== src/integrations/supabase/client.ts ===
+// This file is automatically generated. Do not edit it directly.
+import { createClient } from '@supabase/supabase-js';
+
+const SUPABASE_URL = "https://fake-project-id.supabase.co";
+const SUPABASE_PUBLISHABLE_KEY = "test-publishable-key";
+
+// Import the supabase client like this:
+// import { supabase } from "@/integrations/supabase/client";
+
+export const supabase = createClient(SUPABASE_URL, SUPABASE_PUBLISHABLE_KEY);
+
+=== src/main.tsx ===
+import { createRoot } from "react-dom/client";
+import App from "./App.tsx";
+
+createRoot(document.getElementById("root")!).render(<App />);
+
+
+=== src/vite-env.d.ts ===
+/// <reference types="vite/client" />
+
+
+=== tsconfig.app.json ===
+{
+  "compilerOptions": {
+    "target": "ES2020",
+    "useDefineForClassFields": true,
+    "lib": ["ES2020", "DOM", "DOM.Iterable"],
+    "module": "ESNext",
+    "skipLibCheck": true,
+
+    /* Bundler mode */
+    "moduleResolution": "bundler",
+    "allowImportingTsExtensions": true,
+    "isolatedModules": true,
+    "moduleDetection": "force",
+    "noEmit": true,
+    "jsx": "react-jsx",
+
+    /* Linting */
+    "strict": false,
+    "noUnusedLocals": false,
+    "noUnusedParameters": false,
+    "noImplicitAny": false,
+    "noFallthroughCasesInSwitch": false,
+
+    "baseUrl": ".",
+    "paths": {
+      "@/*": ["./src/*"]
+    }
+  },
+  "include": ["src"]
+}
+
+
+=== tsconfig.json ===
+{
+  "files": [],
+  "references": [
+    { "path": "./tsconfig.app.json" },
+    { "path": "./tsconfig.node.json" }
+  ],
+  "compilerOptions": {
+    "baseUrl": ".",
+    "paths": {
+      "@/*": ["./src/*"]
+    },
+    "noImplicitAny": false,
+    "noUnusedParameters": false,
+    "skipLibCheck": true,
+    "allowJs": true,
+    "noUnusedLocals": false,
+    "strictNullChecks": false
+  }
+}
+
+
+=== tsconfig.node.json ===
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "lib": ["ES2023"],
+    "module": "ESNext",
+    "skipLibCheck": true,
+
+    /* Bundler mode */
+    "moduleResolution": "bundler",
+    "allowImportingTsExtensions": true,
+    "isolatedModules": true,
+    "moduleDetection": "force",
+    "noEmit": true,
+
+    /* Linting */
+    "strict": true,
+    "noUnusedLocals": false,
+    "noUnusedParameters": false,
+    "noFallthroughCasesInSwitch": true
+  },
+  "include": ["vite.config.ts"]
+}
+
+
+=== vite.config.ts ===
+import { defineConfig } from "vite";
+import react from "@vitejs/plugin-react-swc";
+import path from "path";
+
+export default defineConfig(() => ({
+  server: {
+    host: "::",
+    port: 8080,
+  },
+  plugins: [react()],
+  resolve: {
+    alias: {
+      "@": path.resolve(__dirname, "./src"),
+    },
+  },
+}));

e2e-tests/supabase_client.spec.ts

@@ -0,0 +1,15 @@
+import { testSkipIfWindows } from "./helpers/test_helper";
+
+testSkipIfWindows("supabase client is generated", async ({ po }) => {
+  await po.setUp({ autoApprove: true });
+  await po.importApp("minimal");
+  await po.sendPrompt("tc=add-supabase");
+
+  // Connect to Supabase
+  await po.page.getByText("Set up supabase").click();
+  await po.clickConnectSupabaseButton();
+  await po.clickBackButton();
+
+  await po.sendPrompt("tc=generate-supabase-client");
+  await po.snapshotAppFiles();
+});

src/__tests__/chat_stream_handlers.test.ts

@@ -6,7 +6,10 @@ import {
   processFullResponseActions,
   getDyadAddDependencyTags,
 } from "../ipc/processors/response_processor";
-import { removeDyadTags } from "../ipc/handlers/chat_stream_handlers";
+import {
+  removeDyadTags,
+  hasUnclosedDyadWrite,
+} from "../ipc/handlers/chat_stream_handlers";
 import fs from "node:fs";
 import git from "isomorphic-git";
 import { db } from "../db";
@@ -1040,7 +1043,7 @@ const component = <Component />;
 
   it("should handle dyad tags with special characters in content", () => {
     const text = `<dyad-write path="file.js">
-const regex = /<div[^>]*>.*?<\/div>/g;
+const regex = /<div[^>]*>.*?</div>/g;
 const special = "Special chars: @#$%^&*()[]{}|\\";
 </dyad-write>`;
     const result = removeDyadTags(text);
@@ -1059,3 +1062,145 @@ const special = "Special chars: @#$%^&*()[]{}|\\";
     expect(result).toBe("Before  After");
   });
 });
+
+describe("hasUnclosedDyadWrite", () => {
+  it("should return false when there are no dyad-write tags", () => {
+    const text = "This is just regular text without any dyad tags.";
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should return false when dyad-write tag is properly closed", () => {
+    const text = `<dyad-write path="src/file.js">console.log('hello');</dyad-write>`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should return true when dyad-write tag is not closed", () => {
+    const text = `<dyad-write path="src/file.js">console.log('hello');`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(true);
+  });
+
+  it("should return false when dyad-write tag with attributes is properly closed", () => {
+    const text = `<dyad-write path="src/file.js" description="A test file">console.log('hello');</dyad-write>`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should return true when dyad-write tag with attributes is not closed", () => {
+    const text = `<dyad-write path="src/file.js" description="A test file">console.log('hello');`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(true);
+  });
+
+  it("should return false when there are multiple closed dyad-write tags", () => {
+    const text = `<dyad-write path="src/file1.js">code1</dyad-write>
+    Some text in between
+    <dyad-write path="src/file2.js">code2</dyad-write>`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should return true when the last dyad-write tag is unclosed", () => {
+    const text = `<dyad-write path="src/file1.js">code1</dyad-write>
+    Some text in between
+    <dyad-write path="src/file2.js">code2`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(true);
+  });
+
+  it("should return false when first tag is unclosed but last tag is closed", () => {
+    const text = `<dyad-write path="src/file1.js">code1
+    Some text in between
+    <dyad-write path="src/file2.js">code2</dyad-write>`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should handle multiline content correctly", () => {
+    const text = `<dyad-write path="src/component.tsx" description="React component">
+import React from 'react';
+
+const Component = () => {
+  return (
+    <div>
+      <h1>Hello World</h1>
+    </div>
+  );
+};
+
+export default Component;
+</dyad-write>`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should handle multiline unclosed content correctly", () => {
+    const text = `<dyad-write path="src/component.tsx" description="React component">
+import React from 'react';
+
+const Component = () => {
+  return (
+    <div>
+      <h1>Hello World</h1>
+    </div>
+  );
+};
+
+export default Component;`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(true);
+  });
+
+  it("should handle complex attributes correctly", () => {
+    const text = `<dyad-write path="src/file.js" description="File with quotes and special chars" version="1.0" author="test">
+const message = "Hello 'world'";
+const regex = /<div[^>]*>/g;
+</dyad-write>`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should handle text before and after dyad-write tags", () => {
+    const text = `Some text before the tag
+<dyad-write path="src/file.js">console.log('hello');</dyad-write>
+Some text after the tag`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should handle unclosed tag with text after", () => {
+    const text = `Some text before the tag
+<dyad-write path="src/file.js">console.log('hello');
+Some text after the unclosed tag`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(true);
+  });
+
+  it("should handle empty dyad-write tags", () => {
+    const text = `<dyad-write path="src/file.js"></dyad-write>`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should handle unclosed empty dyad-write tags", () => {
+    const text = `<dyad-write path="src/file.js">`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(true);
+  });
+
+  it("should focus on the last opening tag when there are mixed states", () => {
+    const text = `<dyad-write path="src/file1.js">completed content</dyad-write>
+    <dyad-write path="src/file2.js">unclosed content
+    <dyad-write path="src/file3.js">final content</dyad-write>`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+
+  it("should handle tags with special characters in attributes", () => {
+    const text = `<dyad-write path="src/file-name_with.special@chars.js" description="File with special chars in path">content</dyad-write>`;
+    const result = hasUnclosedDyadWrite(text);
+    expect(result).toBe(false);
+  });
+});

src/ipc/handlers/chat_stream_handlers.ts

@@ -451,8 +451,12 @@ This conversation includes one or more image attachments. When the user uploads
           ];
         }
 
-        // When calling streamText, the messages need to be properly formatted for mixed content
+        const simpleStreamText = async ({
-        const { fullStream } = streamText({
+          chatMessages,
+        }: {
+          chatMessages: CoreMessage[];
+        }) => {
+          return streamText({
             maxTokens: await getMaxTokens(settings.selectedModel),
             temperature: 0,
             maxRetries: 2,
@@ -486,6 +490,47 @@ This conversation includes one or more image attachments. When the user uploads
             },
             abortSignal: abortController.signal,
           });
+        };
+
+        const processResponseChunkUpdate = async ({
+          fullResponse,
+        }: {
+          fullResponse: string;
+        }) => {
+          if (
+            fullResponse.includes("$$SUPABASE_CLIENT_CODE$$") &&
+            updatedChat.app?.supabaseProjectId
+          ) {
+            const supabaseClientCode = await getSupabaseClientCode({
+              projectId: updatedChat.app?.supabaseProjectId,
+            });
+            fullResponse = fullResponse.replace(
+              "$$SUPABASE_CLIENT_CODE$$",
+              supabaseClientCode,
+            );
+          }
+          // Store the current partial response
+          partialResponses.set(req.chatId, fullResponse);
+
+          // Update the placeholder assistant message content in the messages array
+          const currentMessages = [...updatedChat.messages];
+          if (
+            currentMessages.length > 0 &&
+            currentMessages[currentMessages.length - 1].role === "assistant"
+          ) {
+            currentMessages[currentMessages.length - 1].content = fullResponse;
+          }
+
+          // Update the assistant message in the database
+          safeSend(event.sender, "chat:response:chunk", {
+            chatId: req.chatId,
+            messages: currentMessages,
+          });
+          return fullResponse;
+        };
+
+        // When calling streamText, the messages need to be properly formatted for mixed content
+        const { fullStream } = await simpleStreamText({ chatMessages });
 
         // Process the stream as before
         let inThinkingBlock = false;
@@ -520,36 +565,8 @@ This conversation includes one or more image attachments. When the user uploads
 
             fullResponse += chunk;
             fullResponse = cleanFullResponse(fullResponse);
-
+            fullResponse = await processResponseChunkUpdate({
-            if (
+              fullResponse,
-              fullResponse.includes("$$SUPABASE_CLIENT_CODE$$") &&
-              updatedChat.app?.supabaseProjectId
-            ) {
-              const supabaseClientCode = await getSupabaseClientCode({
-                projectId: updatedChat.app?.supabaseProjectId,
-              });
-              fullResponse = fullResponse.replace(
-                "$$SUPABASE_CLIENT_CODE$$",
-                supabaseClientCode,
-              );
-            }
-            // Store the current partial response
-            partialResponses.set(req.chatId, fullResponse);
-
-            // Update the placeholder assistant message content in the messages array
-            const currentMessages = [...updatedChat.messages];
-            if (
-              currentMessages.length > 0 &&
-              currentMessages[currentMessages.length - 1].role === "assistant"
-            ) {
-              currentMessages[currentMessages.length - 1].content =
-                fullResponse;
-            }
-
-            // Update the assistant message in the database
-            safeSend(event.sender, "chat:response:chunk", {
-              chatId: req.chatId,
-              messages: currentMessages,
             });
 
             // If the stream was aborted, exit early
@@ -558,6 +575,45 @@ This conversation includes one or more image attachments. When the user uploads
               break;
             }
           }
+
+          if (
+            !abortController.signal.aborted &&
+            settings.selectedChatMode !== "ask" &&
+            hasUnclosedDyadWrite(fullResponse)
+          ) {
+            let continuationAttempts = 0;
+            while (
+              hasUnclosedDyadWrite(fullResponse) &&
+              continuationAttempts < 2 &&
+              !abortController.signal.aborted
+            ) {
+              logger.warn(
+                `Received unclosed dyad-write tag, attempting to continue, attempt #${continuationAttempts + 1}`,
+              );
+              continuationAttempts++;
+
+              const { fullStream: contStream } = await simpleStreamText({
+                // Build messages: replay history then pre-fill assistant with current partial.
+                chatMessages: [
+                  ...chatMessages,
+                  { role: "assistant", content: fullResponse },
+                ],
+              });
+              for await (const part of contStream) {
+                // If the stream was aborted, exit early
+                if (abortController.signal.aborted) {
+                  logger.log(`Stream for chat ${req.chatId} was aborted`);
+                  break;
+                }
+                if (part.type !== "text-delta") continue; // ignore reasoning for continuation
+                fullResponse += part.textDelta;
+                fullResponse = cleanFullResponse(fullResponse);
+                fullResponse = await processResponseChunkUpdate({
+                  fullResponse,
+                });
+              }
+            }
+          }
         } catch (streamError) {
           // Check if this was an abort error
           if (abortController.signal.aborted) {
@@ -832,3 +888,25 @@ export function removeDyadTags(text: string): string {
   const dyadRegex = /<dyad-[^>]*>[\s\S]*?<\/dyad-[^>]*>/g;
   return text.replace(dyadRegex, "").trim();
 }
+
+export function hasUnclosedDyadWrite(text: string): boolean {
+  // Find the last opening dyad-write tag
+  const openRegex = /<dyad-write[^>]*>/g;
+  let lastOpenIndex = -1;
+  let match;
+
+  while ((match = openRegex.exec(text)) !== null) {
+    lastOpenIndex = match.index;
+  }
+
+  // If no opening tag found, there's nothing unclosed
+  if (lastOpenIndex === -1) {
+    return false;
+  }
+
+  // Look for a closing tag after the last opening tag
+  const textAfterLastOpen = text.substring(lastOpenIndex);
+  const hasClosingTag = /<\/dyad-write>/.test(textAfterLastOpen);
+
+  return !hasClosingTag;
+}

testing/fake-llm-server/chatCompletionHandler.ts

@@ -64,35 +64,7 @@ export default Index;
           )
         : lastMessage.content.includes("[dump]"))
     ) {
-      const timestamp = Date.now();
+      messageContent = generateDump(req);
-      const generatedDir = path.join(__dirname, "generated");
-
-      // Create generated directory if it doesn't exist
-      if (!fs.existsSync(generatedDir)) {
-        fs.mkdirSync(generatedDir, { recursive: true });
-      }
-
-      const dumpFilePath = path.join(generatedDir, `${timestamp}.json`);
-
-      try {
-        fs.writeFileSync(
-          dumpFilePath,
-          JSON.stringify(
-            {
-              body: req.body,
-              headers: { authorization: req.headers["authorization"] },
-            },
-            null,
-            2,
-          ).replace(/\r\n/g, "\n"),
-          "utf-8",
-        );
-        console.log(`* Dumped messages to: ${dumpFilePath}`);
-        messageContent = `[[dyad-dump-path=${dumpFilePath}]]`;
-      } catch (error) {
-        console.error(`* Error writing dump file: ${error}`);
-        messageContent = `Error: Could not write dump file: ${error}`;
-      }
     }
 
     if (lastMessage && lastMessage.content === "[increment]") {
@@ -133,6 +105,16 @@ export default Index;
       }
     }
 
+    if (
+      lastMessage &&
+      lastMessage.content &&
+      typeof lastMessage.content === "string" &&
+      lastMessage.content.trim().endsWith("[[STRING_TO_BE_FINISHED]]")
+    ) {
+      messageContent = `[[STRING_IS_FINISHED]]";</dyad-write>\nFinished writing file.`;
+      messageContent += "\n\n" + generateDump(req);
+    }
+
     // Non-streaming response
     if (!stream) {
       return res.json({
@@ -183,3 +165,35 @@ export default Index;
       }
     }, 10);
   };
+
+function generateDump(req: Request) {
+  const timestamp = Date.now();
+  const generatedDir = path.join(__dirname, "generated");
+
+  // Create generated directory if it doesn't exist
+  if (!fs.existsSync(generatedDir)) {
+    fs.mkdirSync(generatedDir, { recursive: true });
+  }
+
+  const dumpFilePath = path.join(generatedDir, `${timestamp}.json`);
+
+  try {
+    fs.writeFileSync(
+      dumpFilePath,
+      JSON.stringify(
+        {
+          body: req.body,
+          headers: { authorization: req.headers["authorization"] },
+        },
+        null,
+        2,
+      ).replace(/\r\n/g, "\n"),
+      "utf-8",
+    );
+    console.log(`* Dumped messages to: ${dumpFilePath}`);
+    return `[[dyad-dump-path=${dumpFilePath}]]`;
+  } catch (error) {
+    console.error(`* Error writing dump file: ${error}`);
+    return `Error: Could not write dump file: ${error}`;
+  }
+}