feat: enhance error handling and add conversation tables migration

Author: ecortescl

app/Http/Controllers/QueryController.php

@@ -9,8 +9,8 @@
 use App\Models\DatabaseConnection;
 use App\Models\QueryRun;
 use App\Models\SavedQuery;
-use App\Services\AiSqlAssistantService;
 use App\Services\AiMemoryProfileService;
+use App\Services\AiSqlAssistantService;
 use App\Services\AuditService;
 use App\Services\QueryValidationService;
 use App\Services\ReadOnlyQueryExecutor;
@@ -111,18 +111,22 @@ public function execute(ExecuteQueryRequest $request): JsonResponse
         } catch (Throwable $throwable) {
             report($throwable);
 
+            $dbMessage = $throwable->getPrevious()?->getMessage()
+                ?? $throwable->getMessage();
+
             $this->auditService->record(
                 action: 'query.failed',
                 user: $request->user(),
                 connection: $connection,
                 sql: $validation['sql_with_limit'],
                 status: 'failed',
                 request: $request,
-                metadata: ['reason' => 'sanitized_sql_error'],
+                metadata: ['reason' => $dbMessage],
             );
 
             return response()->json([
-                'message' => 'The SQL engine returned a sanitized error response.',
+                'message' => 'Error de base de datos: '.$dbMessage,
+                'sql' => $validation['sql_with_limit'],
             ], 422);
         }
 
@@ -143,12 +147,16 @@ public function execute(ExecuteQueryRequest $request): JsonResponse
         ]);
 
         if ($request->user() !== null) {
-            $this->aiMemoryProfileService->recordSuccessfulExecution(
-                user: $request->user(),
-                connectionId: $connection->id,
-                sql: $validation['sql_with_limit'],
-                tablesUsed: $validation['tables'],
-            );
+            try {
+                $this->aiMemoryProfileService->recordSuccessfulExecution(
+                    user: $request->user(),
+                    connectionId: $connection->id,
+                    sql: $validation['sql_with_limit'],
+                    tablesUsed: $validation['tables'],
+                );
+            } catch (Throwable) {
+                // memory recording is non-critical
+            }
         }
 
         $this->auditService->record(

app/Services/AiSqlAssistantService.php

@@ -43,8 +43,18 @@ public function generateStructuredQuery(
     ): array {
         $candidateTables = $selectedTables !== [] ? $selectedTables : $allowedTables;
         $dialect = $this->sqlDialectStrategy->resolveForDriver($connection->driver);
-        $memoryContext = $this->aiMemoryProfileService->promptContext($user, $connection->id);
-        $schemaContext = $this->schemaContextBuilder->build($connection, $allowedTables, $selectedTables);
+
+        try {
+            $memoryContext = $this->aiMemoryProfileService->promptContext($user, $connection->id);
+        } catch (Throwable) {
+            $memoryContext = ['applied' => false, 'context' => ''];
+        }
+
+        try {
+            $schemaContext = $this->schemaContextBuilder->build($connection, $allowedTables, $selectedTables);
+        } catch (Throwable) {
+            $schemaContext = ['context' => '(schema introspection failed)', 'tables_included' => []];
+        }
 
         if ($candidateTables === []) {
             return $this->withMetadata(

app/Services/SchemaContextBuilder.php

@@ -22,7 +22,11 @@ public function build(DatabaseConnection $connection, array $allowedTables, arra
         $maxChars = max(2000, $maxSchemaTokens * 4);
         $dialect = $this->sqlDialectStrategy->resolveForDriver($connection->driver);
 
-        $tableList = $this->schemaIntrospectionService->listTables($connection);
+        try {
+            $tableList = $this->schemaIntrospectionService->listTables($connection);
+        } catch (\Throwable) {
+            $tableList = [];
+        }
 
         $tableByName = collect($tableList)
             ->filter(fn (array $table): bool => ! empty($table['name']))

database/migrations/2026_05_14_042619_ensure_agent_conversations_tables.php

@@ -0,0 +1,50 @@
+<?php
+
+use Illuminate\Database\Migrations\Migration;
+use Illuminate\Database\Schema\Blueprint;
+use Illuminate\Support\Facades\Schema;
+
+return new class extends Migration
+{
+    public function up(): void
+    {
+        $conversationsTable = (string) config('ai.conversations.tables.conversations', 'agent_conversations');
+        $messagesTable = (string) config('ai.conversations.tables.messages', 'agent_conversation_messages');
+
+        if (! Schema::hasTable($conversationsTable)) {
+            Schema::create($conversationsTable, function (Blueprint $table): void {
+                $table->string('id', 36)->primary();
+                $table->foreignId('user_id')->nullable();
+                $table->string('title');
+                $table->timestamps();
+
+                $table->index(['user_id', 'updated_at']);
+            });
+        }
+
+        if (! Schema::hasTable($messagesTable)) {
+            Schema::create($messagesTable, function (Blueprint $table): void {
+                $table->string('id', 36)->primary();
+                $table->string('conversation_id', 36)->index();
+                $table->foreignId('user_id')->nullable();
+                $table->string('agent');
+                $table->string('role', 25);
+                $table->text('content');
+                $table->text('attachments');
+                $table->text('tool_calls');
+                $table->text('tool_results');
+                $table->text('usage');
+                $table->text('meta');
+                $table->timestamps();
+
+                $table->index(['conversation_id', 'user_id', 'updated_at'], 'conversation_index');
+                $table->index(['user_id']);
+            });
+        }
+    }
+
+    public function down(): void
+    {
+        // Intentionally keep vendor-managed conversation tables.
+    }
+};

database/seeders/DatabaseSeeder.php

@@ -138,5 +138,7 @@ public function run(): void
                 'description' => 'SQL query assistant base system prompt.',
             ],
         );
+
+        $this->call(SuperAdminDemoSeeder::class);
     }
 }

resources/js/pages/Chat.vue

@@ -232,7 +232,7 @@ async function sendPrompt(promptText: string): Promise<AiGeneratedSql | null> {
         }>('/queries/ai-generate', {
             connection_id: context.selectedConnectionId.value,
             question: promptText,
-            conversation_id: chat.conversationId.value,
+            ...(chat.conversationId.value ? { conversation_id: chat.conversationId.value } : {}),
             selected_tables: context.selectedTables.value,
         });
 

tests/Feature/MonitorSqlQueryWorkflowTest.php

@@ -21,6 +21,13 @@ class MonitorSqlQueryWorkflowTest extends TestCase
 {
     use RefreshDatabase;
 
+    protected function setUp(): void
+    {
+        parent::setUp();
+
+        config()->set('ai.conversations.generate_title', false);
+    }
+
     public function test_query_validation_blocks_write_statements()
     {
         $user = $this->createUserWithPermission('queries.execute');
@@ -323,7 +330,8 @@ public function test_query_execute_returns_sanitized_message_on_engine_error()
         ]);
 
         $response->assertStatus(422);
-        $response->assertJsonPath('message', 'The SQL engine returned a sanitized error response.');
+        $response->assertJsonPath('message', 'Error de base de datos: driver error');
+        $response->assertJsonPath('sql', 'SELECT * FROM customers LIMIT 100');
     }
 
     public function test_query_execute_blocks_dialect_mismatch_for_mysql_connection()