Add description (optional) to SQL query

src/ipc/handlers/proposal_handlers.ts

@@ -3,6 +3,7 @@ import type {
   CodeProposal,
   FileChange,
   ProposalResult,
+  SqlQuery,
 } from "../../lib/schemas";
 import { db } from "../../db";
 import { messages } from "../../db/schema";
@@ -105,7 +106,11 @@ const getProposalHandler = async (
         })),
       ];
       // Check if we have enough information to create a proposal
-      if (filesChanged.length > 0 || packagesAdded.length > 0) {
+      if (
+        filesChanged.length > 0 ||
+        packagesAdded.length > 0 ||
+        proposalExecuteSqlQueries.length > 0
+      ) {
         const proposal: CodeProposal = {
           type: "code-proposal",
           // Use parsed title or a default title if summary tag is missing but write tags exist
@@ -113,7 +118,10 @@ const getProposalHandler = async (
           securityRisks: [], // Keep empty
           filesChanged,
           packagesAdded,
-          sqlQueries: proposalExecuteSqlQueries,
+          sqlQueries: proposalExecuteSqlQueries.map((query) => ({
+            content: query.content,
+            description: query.description,
+          })),
         };
         logger.log(
           "Generated code proposal. title=",

src/ipc/processors/response_processor.ts

@@ -15,6 +15,7 @@ import {
   executeSupabaseSql,
 } from "../../supabase_admin/supabase_management_client";
 import { isServerFunction } from "../../supabase_admin/supabase_utils";
+import { SqlQuery } from "../../lib/schemas";
 
 const readFile = fs.promises.readFile;
 const logger = log.scope("response_processor");
@@ -108,14 +109,18 @@ export function getDyadChatSummaryTag(fullResponse: string): string | null {
   return null;
 }
 
-export function getDyadExecuteSqlTags(fullResponse: string): string[] {
+export function getDyadExecuteSqlTags(fullResponse: string): SqlQuery[] {
   const dyadExecuteSqlRegex =
-    /<dyad-execute-sql>([\s\S]*?)<\/dyad-execute-sql>/g;
+    /<dyad-execute-sql([^>]*)>([\s\S]*?)<\/dyad-execute-sql>/g;
+  const descriptionRegex = /description="([^"]+)"/;
   let match;
-  const queries: string[] = [];
+  const queries: { content: string; description?: string }[] = [];
 
   while ((match = dyadExecuteSqlRegex.exec(fullResponse)) !== null) {
-    let content = match[1].trim();
+    const attributesString = match[1] || "";
+    let content = match[2].trim();
+    const descriptionMatch = descriptionRegex.exec(attributesString);
+    const description = descriptionMatch?.[1];
 
     // Handle markdown code blocks if present
     const contentLines = content.split("\n");
@@ -127,7 +132,7 @@ export function getDyadExecuteSqlTags(fullResponse: string): string[] {
     }
     content = contentLines.join("\n");
 
-    queries.push(content);
+    queries.push({ content, description });
   }
 
   return queries;
@@ -209,11 +214,11 @@ export async function processFullResponseActions(
         try {
           const result = await executeSupabaseSql({
             supabaseProjectId: chatWithApp.app.supabaseProjectId!,
-            query,
+            query: query.content,
           });
         } catch (error) {
           errors.push({
-            message: `Failed to execute SQL query: ${query}`,
+            message: `Failed to execute SQL query: ${query.content}`,
             error: error,
           });
         }