ruby_llm-agents 3.0.0 → 3.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a72ac8e976da462e8a77ff8b50641acfab95fa061daacf1f38a70d5ca25dc80a
-  data.tar.gz: 61b3595c0ef91fa70dfa776ef6deddce257c96098146a94deb902d0705a483bd
+  metadata.gz: 552fe8427ac0b47f41b14e4652ac65eac1c7b0389a603b71576f8171aba504f7
+  data.tar.gz: d356d7fc391f93194bc0d4dc1cd179178361af5db8c6b00270c8cbd9bcea19b8
 SHA512:
-  metadata.gz: 0c3827e5607ba8376976a84048ae2a9486fe5797720c8842e371ad3793190aaa2a8ec6bea0015b94cbb1c34c8bd516502a6a052dca823e68b5291665d5ae63b8
-  data.tar.gz: 6ac84e773905ee65d8ca0f482633247fb7254c64a13b3b25d83698503a9c70f70070a3ceb1127fc8d96271611e4270616030b1f64837c431b4553991b3038209
+  metadata.gz: 8d7d722cc72e9baa30a3da29f234ed039965f12743f541ae354f7d477aa7a9a5798c0639ddf37872fa1b67c58c7244b7490d0d9f548a03f3734e101a717153a9
+  data.tar.gz: 33f0e37172dbd4b607f86ca27d27b2be997b1650544dcbf41877dd00e094f66bde86b81afaa7aacbf31d7049eda2f9567f3a27b3970fad64c01113a6e436785b

data/app/models/ruby_llm/agents/execution.rb

@@ -74,7 +74,7 @@ module RubyLLM
               foreign_key: :execution_id, dependent: :destroy
 
       # Delegations so existing code keeps working transparently
-      delegate :system_prompt, :user_prompt, :response, :error_message,
+      delegate :system_prompt, :user_prompt, :assistant_prompt, :response, :error_message,
                :messages_summary, :tool_calls, :attempts, :fallback_chain,
                :parameters, :routed_to, :classification_result,
                :cached_at, :cache_creation_tokens,

data/app/views/layouts/ruby_llm/agents/application.html.erb

@@ -19,20 +19,151 @@
       })();
     </script>
 
+    <!-- Gruvbox-aware color palette: CSS custom properties swapped in .dark -->
+    <style>
+      :root {
+        /* Gray (Tailwind defaults) */
+        --c-gray-50: 249 250 251; --c-gray-100: 243 244 246; --c-gray-200: 229 231 235;
+        --c-gray-300: 209 213 219; --c-gray-400: 156 163 175; --c-gray-500: 107 114 128;
+        --c-gray-600: 75 85 99; --c-gray-700: 55 65 81; --c-gray-800: 31 41 55;
+        --c-gray-900: 17 24 39; --c-gray-950: 3 7 18;
+        /* Blue */
+        --c-blue-50: 239 246 255; --c-blue-100: 219 234 254; --c-blue-200: 191 219 254;
+        --c-blue-300: 147 197 253; --c-blue-400: 96 165 250; --c-blue-500: 59 130 246;
+        --c-blue-600: 37 99 235; --c-blue-700: 29 78 216; --c-blue-800: 30 64 175;
+        --c-blue-900: 30 58 138; --c-blue-950: 23 37 84;
+        /* Red */
+        --c-red-50: 254 242 242; --c-red-100: 254 226 226; --c-red-200: 254 202 202;
+        --c-red-300: 252 165 165; --c-red-400: 248 113 113; --c-red-500: 239 68 68;
+        --c-red-600: 220 38 38; --c-red-700: 185 28 28; --c-red-800: 153 27 27;
+        --c-red-900: 127 29 29; --c-red-950: 69 10 10;
+        /* Green */
+        --c-green-50: 240 253 244; --c-green-100: 220 252 231; --c-green-200: 187 247 208;
+        --c-green-300: 134 239 172; --c-green-400: 74 222 128; --c-green-500: 34 197 94;
+        --c-green-600: 22 163 74; --c-green-700: 21 128 61; --c-green-800: 22 101 52;
+        --c-green-900: 20 83 45; --c-green-950: 5 46 22;
+        /* Yellow */
+        --c-yellow-50: 254 252 232; --c-yellow-100: 254 249 195; --c-yellow-200: 254 240 138;
+        --c-yellow-300: 253 224 71; --c-yellow-400: 250 204 21; --c-yellow-500: 234 179 8;
+        --c-yellow-600: 202 138 4; --c-yellow-700: 161 98 7; --c-yellow-800: 133 77 14;
+        --c-yellow-900: 113 63 18; --c-yellow-950: 66 32 6;
+        /* Orange */
+        --c-orange-50: 255 247 237; --c-orange-100: 255 237 213; --c-orange-200: 254 215 170;
+        --c-orange-300: 253 186 116; --c-orange-400: 251 146 60; --c-orange-500: 249 115 22;
+        --c-orange-600: 234 88 12; --c-orange-700: 194 65 12; --c-orange-800: 154 52 18;
+        --c-orange-900: 124 45 18; --c-orange-950: 67 20 7;
+        /* Purple */
+        --c-purple-50: 250 245 255; --c-purple-100: 243 232 255; --c-purple-200: 233 213 255;
+        --c-purple-300: 216 180 254; --c-purple-400: 192 132 252; --c-purple-500: 168 85 247;
+        --c-purple-600: 147 51 234; --c-purple-700: 126 34 206; --c-purple-800: 107 33 168;
+        --c-purple-900: 88 28 135; --c-purple-950: 59 7 100;
+        /* Cyan */
+        --c-cyan-50: 236 254 255; --c-cyan-100: 207 250 254; --c-cyan-200: 165 243 252;
+        --c-cyan-300: 103 232 249; --c-cyan-400: 34 211 238; --c-cyan-500: 6 182 212;
+        --c-cyan-600: 8 145 178; --c-cyan-700: 14 116 144; --c-cyan-800: 21 94 117;
+        --c-cyan-900: 22 78 99; --c-cyan-950: 8 51 68;
+        /* Pink */
+        --c-pink-50: 253 242 248; --c-pink-100: 252 231 243; --c-pink-200: 251 207 232;
+        --c-pink-300: 249 168 212; --c-pink-400: 244 114 182; --c-pink-500: 236 72 153;
+        --c-pink-600: 219 39 119; --c-pink-700: 190 24 93; --c-pink-800: 157 23 77;
+        --c-pink-900: 131 24 67; --c-pink-950: 80 7 36;
+        /* Amber */
+        --c-amber-50: 255 251 235; --c-amber-100: 254 243 199; --c-amber-200: 253 230 138;
+        --c-amber-300: 252 211 77; --c-amber-400: 251 191 36; --c-amber-500: 245 158 11;
+        --c-amber-600: 217 119 6; --c-amber-700: 180 83 9; --c-amber-800: 146 64 14;
+        --c-amber-900: 120 53 15; --c-amber-950: 69 26 3;
+      }
+      .dark {
+        /* Gray → Gruvbox bg/fg */
+        --c-gray-50: 251 241 199; --c-gray-100: 235 219 178; --c-gray-200: 213 196 161;
+        --c-gray-300: 189 174 147; --c-gray-400: 189 174 147; --c-gray-500: 168 153 132;
+        --c-gray-600: 146 131 116; --c-gray-700: 102 92 84; --c-gray-800: 80 73 69;
+        --c-gray-900: 60 56 54; --c-gray-950: 40 40 40;
+        /* Blue → Gruvbox aqua (#83a598 / #458588) */
+        --c-blue-50: 195 215 205; --c-blue-100: 174 199 188; --c-blue-200: 153 182 170;
+        --c-blue-300: 131 165 152; --c-blue-400: 100 149 144; --c-blue-500: 69 133 136;
+        --c-blue-600: 58 112 115; --c-blue-700: 48 93 95; --c-blue-800: 38 73 75;
+        --c-blue-900: 28 55 57; --c-blue-950: 20 40 42;
+        /* Red → Gruvbox red (#fb4934 / #cc241d) */
+        --c-red-50: 253 165 155; --c-red-100: 253 135 122; --c-red-200: 252 104 87;
+        --c-red-300: 251 73 52; --c-red-400: 228 55 41; --c-red-500: 251 89 66;
+        --c-red-600: 204 36 29; --c-red-700: 136 24 19; --c-red-800: 102 18 14;
+        --c-red-900: 72 12 10; --c-red-950: 50 8 7;
+        /* Green → Gruvbox green (#b8bb26 / #98971a) */
+        --c-green-50: 225 227 145; --c-green-100: 213 215 112; --c-green-200: 199 201 75;
+        --c-green-300: 184 187 38; --c-green-400: 168 169 32; --c-green-500: 152 151 26;
+        --c-green-600: 126 125 22; --c-green-700: 101 100 17; --c-green-800: 77 77 13;
+        --c-green-900: 55 55 9; --c-green-950: 38 38 6;
+        /* Yellow → Gruvbox yellow (#fabd2f / #d79921) */
+        --c-yellow-50: 253 225 155; --c-yellow-100: 252 215 121; --c-yellow-200: 251 202 84;
+        --c-yellow-300: 250 189 47; --c-yellow-400: 233 171 40; --c-yellow-500: 215 153 33;
+        --c-yellow-600: 179 127 27; --c-yellow-700: 143 102 22; --c-yellow-800: 109 78 17;
+        --c-yellow-900: 77 55 12; --c-yellow-950: 53 38 8;
+        /* Orange → Gruvbox orange (#fe8019 / #d65d0e) */
+        --c-orange-50: 255 197 140; --c-orange-100: 255 172 102; --c-orange-200: 254 150 64;
+        --c-orange-300: 254 128 25; --c-orange-400: 234 111 20; --c-orange-500: 214 93 14;
+        --c-orange-600: 178 77 12; --c-orange-700: 142 62 9; --c-orange-800: 108 47 7;
+        --c-orange-900: 76 33 5; --c-orange-950: 52 23 3;
+        /* Purple → Gruvbox purple (#d3869b / #b16286) */
+        --c-purple-50: 235 186 200; --c-purple-100: 227 168 184; --c-purple-200: 219 151 170;
+        --c-purple-300: 211 134 155; --c-purple-400: 194 116 145; --c-purple-500: 177 98 134;
+        --c-purple-600: 147 81 111; --c-purple-700: 118 65 89; --c-purple-800: 90 50 68;
+        --c-purple-900: 63 35 48; --c-purple-950: 43 24 33;
+        /* Cyan → Gruvbox green-cyan (#8ec07c / #689d6a) */
+        --c-cyan-50: 194 228 184; --c-cyan-100: 177 216 164; --c-cyan-200: 160 204 144;
+        --c-cyan-300: 142 192 124; --c-cyan-400: 123 175 115; --c-cyan-500: 104 157 106;
+        --c-cyan-600: 86 131 88; --c-cyan-700: 69 105 70; --c-cyan-800: 53 80 54;
+        --c-cyan-900: 37 57 38; --c-cyan-950: 26 40 27;
+        /* Pink → Gruvbox purple (no pink in Gruvbox) */
+        --c-pink-50: 235 186 200; --c-pink-100: 227 168 184; --c-pink-200: 219 151 170;
+        --c-pink-300: 211 134 155; --c-pink-400: 194 116 145; --c-pink-500: 177 98 134;
+        --c-pink-600: 147 81 111; --c-pink-700: 118 65 89; --c-pink-800: 90 50 68;
+        --c-pink-900: 63 35 48; --c-pink-950: 43 24 33;
+        /* Amber → Gruvbox yellow-orange */
+        --c-amber-50: 254 224 155; --c-amber-100: 253 213 121; --c-amber-200: 252 198 84;
+        --c-amber-300: 251 184 47; --c-amber-400: 235 162 37; --c-amber-500: 215 140 26;
+        --c-amber-600: 179 116 22; --c-amber-700: 143 93 17; --c-amber-800: 109 71 13;
+        --c-amber-900: 77 50 9; --c-amber-950: 53 35 6;
+      }
+    </style>
+
     <!-- Tailwind CSS via CDN -->
     <script src="https://cdn.tailwindcss.com"></script>
 
     <script>
       tailwind.config = {
-        darkMode: 'class'
+        darkMode: 'class',
+        theme: {
+          extend: {
+            colors: (function() {
+              var families = ['gray','blue','red','green','yellow','orange','purple','cyan','pink','amber'];
+              var shades = [50,100,200,300,400,500,600,700,800,900,950];
+              var c = {};
+              families.forEach(function(f) {
+                c[f] = {};
+                shades.forEach(function(s) {
+                  c[f][s] = 'rgb(var(--c-' + f + '-' + s + ') / <alpha-value>)';
+                });
+              });
+              return c;
+            })()
+          }
+        }
       }
     </script>
 
     <!-- Highcharts for charts -->
     <script src="https://code.highcharts.com/highcharts.js"></script>
 
-    <!-- Configure Highcharts defaults -->
+    <!-- Chart color helpers + Highcharts defaults -->
     <script>
+      function chartColor(lightHex, darkHex) {
+        return document.documentElement.classList.contains('dark') ? darkHex : lightHex;
+      }
+      function chartColorAlpha(lightRgba, darkR, darkG, darkB, alpha) {
+        return document.documentElement.classList.contains('dark')
+          ? 'rgba(' + darkR + ',' + darkG + ',' + darkB + ',' + alpha + ')' : lightRgba;
+      }
       Highcharts.setOptions({
         credits: { enabled: false },
         chart: {
@@ -41,22 +172,22 @@
         },
         title: { text: null },
         xAxis: {
-          labels: { style: { color: '#6B7280' } },
-          lineColor: 'rgba(107, 114, 128, 0.1)',
-          tickColor: 'rgba(107, 114, 128, 0.1)'
+          labels: { style: { color: chartColor('#6B7280', '#928374') } },
+          lineColor: chartColorAlpha('rgba(107, 114, 128, 0.1)', 146, 131, 116, 0.1),
+          tickColor: chartColorAlpha('rgba(107, 114, 128, 0.1)', 146, 131, 116, 0.1)
         },
         yAxis: {
-          labels: { style: { color: '#6B7280' } },
-          gridLineColor: 'rgba(107, 114, 128, 0.1)'
+          labels: { style: { color: chartColor('#6B7280', '#928374') } },
+          gridLineColor: chartColorAlpha('rgba(107, 114, 128, 0.1)', 146, 131, 116, 0.1)
         },
         legend: {
-          itemStyle: { color: '#6B7280' },
-          itemHoverStyle: { color: '#9CA3AF' }
+          itemStyle: { color: chartColor('#6B7280', '#928374') },
+          itemHoverStyle: { color: chartColor('#9CA3AF', '#bdae93') }
         },
         tooltip: {
-          backgroundColor: 'rgba(0, 0, 0, 0.85)',
+          backgroundColor: chartColorAlpha('rgba(0, 0, 0, 0.85)', 40, 40, 40, 0.95),
           borderColor: 'transparent',
-          style: { color: '#E5E7EB', fontFamily: 'ui-monospace, monospace' }
+          style: { color: chartColor('#E5E7EB', '#ebdbb2'), fontFamily: 'ui-monospace, monospace' }
         }
       });
     </script>

data/app/views/ruby_llm/agents/agents/show.html.erb

@@ -109,7 +109,7 @@
     title: { text: null },
     xAxis: {
       type: 'datetime',
-      labels: { style: { color: '#6B7280', fontSize: '9px', fontFamily: 'ui-monospace, monospace' }, format: '{value:%b %d}' },
+      labels: { style: { color: chartColor('#6B7280', '#928374'), fontSize: '9px', fontFamily: 'ui-monospace, monospace' }, format: '{value:%b %d}' },
       lineColor: 'transparent',
       tickLength: 0,
       gridLineWidth: 0
@@ -118,19 +118,19 @@
       title: { text: null },
       min: 0,
       allowDecimals: false,
-      labels: { style: { color: '#6B7280', fontSize: '9px', fontFamily: 'ui-monospace, monospace' } },
-      gridLineColor: 'rgba(107, 114, 128, 0.08)'
+      labels: { style: { color: chartColor('#6B7280', '#928374'), fontSize: '9px', fontFamily: 'ui-monospace, monospace' } },
+      gridLineColor: chartColorAlpha('rgba(107, 114, 128, 0.08)', 146, 131, 116, 0.08)
     },
     legend: { enabled: false },
     credits: { enabled: false },
     tooltip: {
-      backgroundColor: 'rgba(0, 0, 0, 0.85)',
+      backgroundColor: chartColorAlpha('rgba(0, 0, 0, 0.85)', 40, 40, 40, 0.95),
       borderColor: 'transparent',
       borderRadius: 3,
-      style: { color: '#E5E7EB', fontSize: '10px', fontFamily: 'ui-monospace, monospace' },
+      style: { color: chartColor('#E5E7EB', '#ebdbb2'), fontSize: '10px', fontFamily: 'ui-monospace, monospace' },
       shared: true,
       formatter: function() {
-        let html = '<span style="color:#9CA3AF">' + Highcharts.dateFormat('%b %d', this.x) + '</span>';
+        let html = '<span style="color:' + chartColor('#9CA3AF', '#bdae93') + '">' + Highcharts.dateFormat('%b %d', this.x) + '</span>';
         let cost = trendData.find(d => new Date(d.date).getTime() === this.x);
         this.points.forEach(p => html += '<br/>' + p.series.name + ': <b>' + p.y + '</b>');
         if (cost) html += '<br/>cost: <b>$' + cost.cost.toFixed(4) + '</b>';
@@ -148,14 +148,14 @@
       {
         name: 'errors',
         data: errorData,
-        color: '#EF4444',
-        fillColor: { linearGradient: { x1: 0, y1: 0, x2: 0, y2: 1 }, stops: [[0, 'rgba(239, 68, 68, 0.08)'], [1, 'rgba(239, 68, 68, 0)']] }
+        color: chartColor('#EF4444', '#fb4934'),
+        fillColor: { linearGradient: { x1: 0, y1: 0, x2: 0, y2: 1 }, stops: [[0, chartColorAlpha('rgba(239, 68, 68, 0.08)', 251, 73, 52, 0.08)], [1, chartColorAlpha('rgba(239, 68, 68, 0)', 251, 73, 52, 0)]] }
       },
       {
         name: 'success',
         data: successData,
-        color: '#10B981',
-        fillColor: { linearGradient: { x1: 0, y1: 0, x2: 0, y2: 1 }, stops: [[0, 'rgba(16, 185, 129, 0.08)'], [1, 'rgba(16, 185, 129, 0)']] }
+        color: chartColor('#10B981', '#b8bb26'),
+        fillColor: { linearGradient: { x1: 0, y1: 0, x2: 0, y2: 1 }, stops: [[0, chartColorAlpha('rgba(16, 185, 129, 0.08)', 184, 187, 38, 0.08)], [1, chartColorAlpha('rgba(16, 185, 129, 0)', 184, 187, 38, 0)]] }
       }
     ]
   });

data/app/views/ruby_llm/agents/dashboard/index.html.erb

@@ -60,7 +60,7 @@
               type: 'datetime',
               min: now - getTimeRangeMs(range),
               max: now,
-              labels: { style: { color: '#6B7280', fontSize: '9px', fontFamily: 'ui-monospace, monospace' }, format: '{value:' + fmt + '}' },
+              labels: { style: { color: chartColor('#6B7280', '#928374'), fontSize: '9px', fontFamily: 'ui-monospace, monospace' }, format: '{value:' + fmt + '}' },
               lineColor: 'transparent',
               tickLength: 0,
               gridLineWidth: 0
@@ -69,19 +69,19 @@
               title: { text: null },
               min: 0,
               allowDecimals: false,
-              labels: { style: { color: '#6B7280', fontSize: '9px', fontFamily: 'ui-monospace, monospace' } },
-              gridLineColor: 'rgba(107, 114, 128, 0.08)'
+              labels: { style: { color: chartColor('#6B7280', '#928374'), fontSize: '9px', fontFamily: 'ui-monospace, monospace' } },
+              gridLineColor: chartColorAlpha('rgba(107, 114, 128, 0.08)', 146, 131, 116, 0.08)
             },
             legend: { enabled: false },
             credits: { enabled: false },
             tooltip: {
-              backgroundColor: 'rgba(0, 0, 0, 0.85)',
+              backgroundColor: chartColorAlpha('rgba(0, 0, 0, 0.85)', 40, 40, 40, 0.95),
               borderColor: 'transparent',
               borderRadius: 3,
-              style: { color: '#E5E7EB', fontSize: '10px', fontFamily: 'ui-monospace, monospace' },
+              style: { color: chartColor('#E5E7EB', '#ebdbb2'), fontSize: '10px', fontFamily: 'ui-monospace, monospace' },
               shared: true,
               formatter: function() {
-                let html = '<span style="color:#9CA3AF">' + Highcharts.dateFormat(fmt, this.x) + '</span>';
+                let html = '<span style="color:' + chartColor('#9CA3AF', '#bdae93') + '">' + Highcharts.dateFormat(fmt, this.x) + '</span>';
                 this.points.forEach(p => html += '<br/>' + p.series.name + ': <b>' + p.y + '</b>');
                 return html;
               }
@@ -97,14 +97,14 @@
               {
                 name: 'errors',
                 data: toDatetimePoints(data.series[1].data, range),
-                color: '#EF4444',
-                fillColor: { linearGradient: { x1: 0, y1: 0, x2: 0, y2: 1 }, stops: [[0, 'rgba(239, 68, 68, 0.08)'], [1, 'rgba(239, 68, 68, 0)']] }
+                color: chartColor('#EF4444', '#fb4934'),
+                fillColor: { linearGradient: { x1: 0, y1: 0, x2: 0, y2: 1 }, stops: [[0, chartColorAlpha('rgba(239, 68, 68, 0.08)', 251, 73, 52, 0.08)], [1, chartColorAlpha('rgba(239, 68, 68, 0)', 251, 73, 52, 0)]] }
               },
               {
                 name: 'success',
                 data: toDatetimePoints(data.series[0].data, range),
-                color: '#10B981',
-                fillColor: { linearGradient: { x1: 0, y1: 0, x2: 0, y2: 1 }, stops: [[0, 'rgba(16, 185, 129, 0.08)'], [1, 'rgba(16, 185, 129, 0)']] }
+                color: chartColor('#10B981', '#b8bb26'),
+                fillColor: { linearGradient: { x1: 0, y1: 0, x2: 0, y2: 1 }, stops: [[0, chartColorAlpha('rgba(16, 185, 129, 0.08)', 184, 187, 38, 0.08)], [1, chartColorAlpha('rgba(16, 185, 129, 0)', 184, 187, 38, 0)]] }
               }
             ]
           });

data/app/views/ruby_llm/agents/executions/show.html.erb

@@ -439,6 +439,19 @@
   <pre id="user-prompt-content" class="hidden bg-gray-50 dark:bg-gray-900 text-gray-900 dark:text-gray-100 rounded p-4 text-xs overflow-x-auto max-h-96 font-mono whitespace-pre-wrap"><%= @execution.user_prompt %></pre>
 <% end %>
 
+<!-- ── assistant prompt ──────────────────── -->
+<% if @execution.respond_to?(:assistant_prompt) && @execution.assistant_prompt.present? %>
+  <div class="flex items-center gap-3 mt-6 mb-3">
+    <span class="text-[10px] font-medium text-gray-400 dark:text-gray-600 uppercase tracking-widest font-mono">assistant prompt</span>
+    <div class="flex-1 border-t border-gray-200 dark:border-gray-800"></div>
+    <button type="button" onclick="togglePrompt('assistant')" class="font-mono text-xs text-gray-400 dark:text-gray-500 hover:text-gray-700 dark:hover:text-gray-300">
+      <span id="assistant-prompt-toggle">expand</span>
+    </button>
+  </div>
+  <p id="assistant-prompt-preview" class="text-xs text-gray-600 dark:text-gray-300 font-mono bg-gray-50 dark:bg-gray-900 rounded p-3 truncate"><%= @execution.assistant_prompt.truncate(150) %></p>
+  <pre id="assistant-prompt-content" class="hidden bg-gray-50 dark:bg-gray-900 text-gray-900 dark:text-gray-100 rounded p-4 text-xs overflow-x-auto max-h-96 font-mono whitespace-pre-wrap"><%= @execution.assistant_prompt %></pre>
+<% end %>
+
 <!-- ── conversation ──────────────────── -->
 <% if @execution.respond_to?(:messages_count) && @execution.messages_count.to_i > 0 %>
   <%

data/lib/generators/ruby_llm_agents/templates/add_assistant_prompt_migration.rb.tt

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+class AddAssistantPromptToExecutionDetails < ActiveRecord::Migration<%= migration_version %>
+  def change
+    unless column_exists?(:ruby_llm_agents_execution_details, :assistant_prompt)
+      add_column :ruby_llm_agents_execution_details, :assistant_prompt, :text
+    end
+  end
+end

data/lib/generators/ruby_llm_agents/templates/split_execution_details_migration.rb.tt

@@ -9,7 +9,7 @@
 class SplitExecutionDetailsFromExecutions < ActiveRecord::Migration<%= migration_version %>
   # Columns that belong on execution_details, not executions
   DETAIL_COLUMNS = %i[
-    error_message system_prompt user_prompt response messages_summary
+    error_message system_prompt user_prompt assistant_prompt response messages_summary
     tool_calls attempts fallback_chain parameters routed_to
     classification_result cached_at cache_creation_tokens
   ].freeze
@@ -50,6 +50,7 @@ class SplitExecutionDetailsFromExecutions < ActiveRecord::Migration<%= migration
       t.text     :error_message
       t.text     :system_prompt
       t.text     :user_prompt
+      t.text     :assistant_prompt
       t.json     :response,             default: {}
       t.json     :messages_summary,     default: {}, null: false
       t.json     :tool_calls,           default: [], null: false

data/lib/generators/ruby_llm_agents/upgrade_generator.rb

@@ -60,6 +60,25 @@ module RubyLlmAgents
       )
     end
 
+    # Add assistant_prompt column to execution_details (v3.0 -> v3.1 upgrade)
+    def create_add_assistant_prompt_migration
+      if column_exists?(:ruby_llm_agents_execution_details, :assistant_prompt)
+        say_status :skip, "assistant_prompt column already exists on execution_details", :yellow
+        return
+      end
+
+      unless table_exists?(:ruby_llm_agents_execution_details)
+        say_status :skip, "execution_details table does not exist yet", :yellow
+        return
+      end
+
+      say_status :upgrade, "Adding assistant_prompt to execution_details", :blue
+      migration_template(
+        "add_assistant_prompt_migration.rb.tt",
+        File.join(db_migrate_path, "add_assistant_prompt_to_execution_details.rb")
+      )
+    end
+
     def suggest_config_consolidation
       ruby_llm_initializer = File.join(destination_root, "config/initializers/ruby_llm.rb")
       agents_initializer = File.join(destination_root, "config/initializers/ruby_llm_agents.rb")
@@ -116,7 +135,7 @@ module RubyLlmAgents
 
     # Detail columns that should only exist on execution_details, not executions
     DETAIL_COLUMNS = %i[
-      error_message system_prompt user_prompt response messages_summary
+      error_message system_prompt user_prompt assistant_prompt response messages_summary
       tool_calls attempts fallback_chain parameters routed_to
       classification_result cached_at cache_creation_tokens
     ].freeze

data/lib/ruby_llm/agents/base_agent.rb

@@ -380,7 +380,8 @@ module RubyLLM
         base_data.merge(
           model: model,
           system_prompt: system_prompt,
-          user_prompt: user_prompt
+          user_prompt: user_prompt,
+          assistant_prompt: assistant_prompt
         )
       end
 

data/lib/ruby_llm/agents/core/instrumentation.rb

@@ -275,6 +275,9 @@ module RubyLLM
           system_prompt: config.persist_prompts ? stored_system_prompt : nil,
           user_prompt: config.persist_prompts ? stored_user_prompt : nil
         }
+        if config.persist_prompts && assistant_prompt_column_exists?
+          detail_data[:assistant_prompt] = stored_assistant_prompt
+        end
         detail_data.merge!(@_pending_detail_data) if @_pending_detail_data
         @_pending_detail_data = nil
 
@@ -528,7 +531,9 @@ module RubyLLM
           user_prompt: safe_user_prompt,
           messages_summary: config.persist_messages_summary ? messages_summary : {},
           error_message: error&.message
-        }.merge(detail_fields || {})
+        }
+        detail_data[:assistant_prompt] = safe_assistant_prompt if assistant_prompt_column_exists?
+        detail_data.merge!(detail_fields || {})
 
         execution_data[:_detail_data] = detail_data
 
@@ -634,6 +639,23 @@ module RubyLLM
         nil
       end
 
+      # Safely captures assistant prompt, handling errors gracefully
+      #
+      # @return [String, nil] The assistant prompt or nil if unavailable
+      def safe_assistant_prompt
+        respond_to?(:assistant_prompt) ? assistant_prompt&.to_s : nil
+      rescue StandardError => e
+        Rails.logger.warn("[RubyLLM::Agents] Could not capture assistant_prompt: #{e.message}")
+        nil
+      end
+
+      # Returns the assistant prompt for storage
+      #
+      # @return [String, nil] The assistant prompt
+      def stored_assistant_prompt
+        safe_assistant_prompt
+      end
+
       # Safely extracts a value from response object
       #
       # @param response [Object] The response object
@@ -923,6 +945,22 @@ module RubyLLM
         end
       end
 
+      # Checks if the assistant_prompt column exists on execution_details
+      #
+      # Memoized to avoid repeated schema queries. Returns false for older installs
+      # that haven't run the migration yet.
+      #
+      # @return [Boolean] true if the column exists
+      def assistant_prompt_column_exists?
+        return @_assistant_prompt_column_exists if defined?(@_assistant_prompt_column_exists)
+
+        @_assistant_prompt_column_exists = begin
+          RubyLLM::Agents::ExecutionDetail.column_names.include?("assistant_prompt")
+        rescue StandardError
+          false
+        end
+      end
+
       # Emergency fallback to mark execution as failed
       #
       # Uses update_all to bypass ActiveRecord callbacks and validations,

data/lib/ruby_llm/agents/core/version.rb

@@ -4,6 +4,6 @@ module RubyLLM
   module Agents
     # Current version of the RubyLLM::Agents gem
     # @return [String] Semantic version string
-    VERSION = "3.0.0"
+    VERSION = "3.1.0"
   end
 end

data/lib/ruby_llm/agents/pipeline/middleware/instrumentation.rb

@@ -253,6 +253,13 @@ module RubyLLM
 
             detail_data = {}
 
+            if global_config.persist_prompts
+              exec_opts = context.options[:options] || {}
+              detail_data[:system_prompt] = exec_opts[:system_prompt]
+              detail_data[:user_prompt] = context.input.to_s.presence
+              detail_data[:assistant_prompt] = exec_opts[:assistant_prefill] if assistant_prompt_column_exists?
+            end
+
             if context.error
               detail_data[:error_message] = truncate_error_message(context.error.message)
             end
@@ -349,6 +356,12 @@ module RubyLLM
 
             # Store detail data for separate creation
             detail_data = { parameters: sanitize_parameters(context) }
+            if global_config.persist_prompts
+              exec_opts = context.options[:options] || {}
+              detail_data[:system_prompt] = exec_opts[:system_prompt]
+              detail_data[:user_prompt] = context.input.to_s.presence
+              detail_data[:assistant_prompt] = exec_opts[:assistant_prefill] if assistant_prompt_column_exists?
+            end
             detail_data[:error_message] = truncate_error_message(context.error.message) if context.error
             detail_data[:tool_calls] = context[:tool_calls] if context[:tool_calls].present?
             detail_data[:attempts] = context[:reliability_attempts] if context[:reliability_attempts].present?
@@ -496,6 +509,22 @@ module RubyLLM
             false
           end
 
+          # Checks if the assistant_prompt column exists on execution_details
+          #
+          # Memoized to avoid repeated schema queries.
+          #
+          # @return [Boolean]
+          def assistant_prompt_column_exists?
+            return @_assistant_prompt_column_exists if defined?(@_assistant_prompt_column_exists)
+
+            @_assistant_prompt_column_exists = begin
+              defined?(RubyLLM::Agents::ExecutionDetail) &&
+                RubyLLM::Agents::ExecutionDetail.column_names.include?("assistant_prompt")
+            rescue StandardError
+              false
+            end
+          end
+
           # Returns whether the Execution model is available
           #
           # @return [Boolean]

data/lib/ruby_llm/agents/pipeline/middleware/tenant.rb

@@ -42,6 +42,7 @@ module RubyLLM
           # @return [Context] The context with tenant fields populated
           def call(context)
             resolve_tenant!(context)
+            ensure_tenant_record!(context)
             apply_api_configuration!(context)
             @app.call(context)
           end
@@ -84,6 +85,83 @@ module RubyLLM
             end
           end
 
+          # Ensures a Tenant record exists in the database for the resolved tenant.
+          #
+          # When a host model (e.g., Organization) with LLMTenant is passed as
+          # tenant: to an agent, the after_create callback only fires for new records.
+          # Pre-existing records won't have a Tenant row yet. This method auto-creates
+          # it on first use so budget tracking and the dashboard work correctly.
+          #
+          # @param context [Context] The execution context
+          def ensure_tenant_record!(context)
+            return unless context.tenant_id.present?
+            return unless tenant_table_exists?
+
+            tenant_object = context.tenant_object
+
+            # Only auto-create when the tenant object uses the LLMTenant concern
+            if tenant_object.respond_to?(:llm_tenant_id) && tenant_object.is_a?(::ActiveRecord::Base)
+              ensure_tenant_for_model!(tenant_object)
+            else
+              # For hash-based or string tenants, ensure a minimal record exists
+              RubyLLM::Agents::Tenant.find_or_create_by!(tenant_id: context.tenant_id)
+            end
+          rescue StandardError => e
+            # Don't fail the execution if tenant record creation fails
+            log_tenant_warning("ensure tenant record", e)
+          end
+
+          # Creates a Tenant record linked to the host model if one doesn't exist
+          #
+          # @param tenant_object [ActiveRecord::Base] The host model with LLMTenant
+          def ensure_tenant_for_model!(tenant_object)
+            # Check polymorphic link first, then tenant_id
+            existing = RubyLLM::Agents::Tenant.find_by(tenant_record: tenant_object) ||
+                       RubyLLM::Agents::Tenant.find_by(tenant_id: tenant_object.llm_tenant_id)
+            return if existing
+
+            options = tenant_object.class.try(:llm_tenant_options) || {}
+            limits = options[:limits] || {}
+            name_method = options[:name] || :to_s
+
+            RubyLLM::Agents::Tenant.create!(
+              tenant_id: tenant_object.llm_tenant_id,
+              name: tenant_object.send(name_method).to_s,
+              tenant_record: tenant_object,
+              daily_limit: limits[:daily_cost],
+              monthly_limit: limits[:monthly_cost],
+              daily_token_limit: limits[:daily_tokens],
+              monthly_token_limit: limits[:monthly_tokens],
+              daily_execution_limit: limits[:daily_executions],
+              monthly_execution_limit: limits[:monthly_executions],
+              enforcement: options[:enforcement]&.to_s || "soft",
+              inherit_global_defaults: options.fetch(:inherit_global, true)
+            )
+          end
+
+          # Checks if the tenants table exists (memoized)
+          #
+          # @return [Boolean]
+          def tenant_table_exists?
+            return @tenant_table_exists if defined?(@tenant_table_exists)
+
+            @tenant_table_exists = ::ActiveRecord::Base.connection.table_exists?(:ruby_llm_agents_tenants)
+          rescue StandardError
+            @tenant_table_exists = false
+          end
+
+          # Logs a warning without failing the execution
+          #
+          # @param action [String] What was being attempted
+          # @param error [StandardError] The error
+          def log_tenant_warning(action, error)
+            return unless defined?(Rails) && Rails.respond_to?(:logger)
+
+            Rails.logger.warn(
+              "[RubyLLM::Agents] Failed to #{action}: #{error.message}"
+            )
+          end
+
           # Applies API configuration to RubyLLM based on resolved tenant
           #
           # @param context [Context] The execution context

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-agents
 version: !ruby/object:Gem::Version
-  version: 3.0.0
+  version: 3.1.0
 platform: ruby
 authors:
 - adham90
@@ -146,6 +146,7 @@ files:
 - lib/generators/ruby_llm_agents/multi_tenancy_generator.rb
 - lib/generators/ruby_llm_agents/restructure_generator.rb
 - lib/generators/ruby_llm_agents/speaker_generator.rb
+- lib/generators/ruby_llm_agents/templates/add_assistant_prompt_migration.rb.tt
 - lib/generators/ruby_llm_agents/templates/add_attempts_migration.rb.tt
 - lib/generators/ruby_llm_agents/templates/add_caching_migration.rb.tt
 - lib/generators/ruby_llm_agents/templates/add_execution_type_migration.rb.tt