ruby_llm-agents 0.3.4 → 0.3.5

data/app/views/ruby_llm/agents/shared/_executions_table.html.erb

@@ -0,0 +1,251 @@
+<%# Show tenant column when multi-tenancy is enabled and no specific tenant is selected %>
+<% show_tenant_column = tenant_filter_enabled? && current_tenant_id.blank? %>
+
+<% if executions.empty? %>
+  <p class="text-gray-500 dark:text-gray-400 text-center py-8">No executions found.</p>
+<% else %>
+  <div class="overflow-x-auto">
+    <table class="min-w-full text-sm">
+      <thead class="bg-gray-50 dark:bg-gray-900/50">
+        <tr>
+          <th scope="col" class="px-4 py-3 text-left text-xs font-medium text-gray-500 dark:text-gray-400 uppercase tracking-wider">Status</th>
+          <th scope="col" class="px-4 py-3 text-left text-xs font-medium text-gray-500 dark:text-gray-400 uppercase tracking-wider">Agent</th>
+          <% if show_tenant_column %>
+            <th scope="col" class="px-4 py-3 text-left text-xs font-medium text-gray-500 dark:text-gray-400 uppercase tracking-wider">Tenant</th>
+          <% end %>
+          <th scope="col" class="px-4 py-3 text-left text-xs font-medium text-gray-500 dark:text-gray-400 uppercase tracking-wider">Model</th>
+          <th scope="col" class="px-4 py-3 text-right text-xs font-medium text-gray-500 dark:text-gray-400 uppercase tracking-wider">Duration</th>
+          <th scope="col" class="px-4 py-3 text-right text-xs font-medium text-gray-500 dark:text-gray-400 uppercase tracking-wider">Tokens</th>
+          <th scope="col" class="px-4 py-3 text-right text-xs font-medium text-gray-500 dark:text-gray-400 uppercase tracking-wider">Cost</th>
+          <th scope="col" class="px-4 py-3 text-right text-xs font-medium text-gray-500 dark:text-gray-400 uppercase tracking-wider">Time</th>
+        </tr>
+      </thead>
+      <tbody class="bg-white dark:bg-gray-800 divide-y divide-gray-100 dark:divide-gray-700">
+        <% executions.each do |execution| %>
+          <%
+            is_workflow = execution.workflow_type.present?
+            children = execution.child_executions.sort_by(&:created_at)
+            has_children = children.any?
+          %>
+
+          <%# Parent/Main Row %>
+          <tr class="hover:bg-gray-50 dark:hover:bg-gray-700/50 transition-colors">
+            <%# Status %>
+            <td class="px-4 py-3 whitespace-nowrap">
+              <%= render "ruby_llm/agents/shared/status_badge", status: execution.status, size: :sm %>
+            </td>
+
+            <%# Agent Name with Workflow Badge %>
+            <td class="px-4 py-3 whitespace-nowrap">
+              <div class="flex items-center gap-2">
+                <%= link_to ruby_llm_agents.execution_path(execution), class: "font-medium text-gray-900 dark:text-gray-100 hover:text-blue-600 dark:hover:text-blue-400" do %>
+                  <%= execution.agent_type.gsub(/Agent$/, "") %>
+                <% end %>
+                <% if is_workflow %>
+                  <% badge_style = case execution.workflow_type
+                       when "pipeline" then "text-indigo-600 dark:text-indigo-400"
+                       when "parallel" then "text-cyan-600 dark:text-cyan-400"
+                       when "router" then "text-amber-600 dark:text-amber-400"
+                     end %>
+                  <% badge_icon = case execution.workflow_type
+                       when "pipeline" then "→"
+                       when "parallel" then "⫴"
+                       when "router" then "⑂"
+                     end %>
+                  <span class="text-xs <%= badge_style %>"><%= badge_icon %></span>
+                <% end %>
+              </div>
+            </td>
+
+            <%# Tenant (only when viewing all tenants) %>
+            <% if show_tenant_column %>
+              <td class="px-4 py-3 whitespace-nowrap">
+                <% if execution.tenant_id.present? %>
+                  <a href="<%= url_for(request.query_parameters.merge(tenant_id: execution.tenant_id)) %>"
+                     class="inline-flex items-center px-2 py-0.5 rounded text-xs font-medium bg-blue-50 dark:bg-blue-900/30 text-blue-700 dark:text-blue-300 hover:bg-blue-100 dark:hover:bg-blue-900/50">
+                    <%= truncate(execution.tenant_id, length: 15) %>
+                  </a>
+                <% else %>
+                  <span class="text-gray-400 dark:text-gray-500 text-xs">—</span>
+                <% end %>
+              </td>
+            <% end %>
+
+            <%# Model %>
+            <td class="px-4 py-3 whitespace-nowrap text-gray-500 dark:text-gray-400 font-mono text-xs">
+              <%= execution.model_id %>
+            </td>
+
+            <%# Duration %>
+            <td class="px-4 py-3 whitespace-nowrap text-right text-gray-900 dark:text-gray-100 tabular-nums">
+              <%= execution.duration_ms ? "#{number_with_delimiter(execution.duration_ms)}ms" : "-" %>
+            </td>
+
+            <%# Tokens %>
+            <td class="px-4 py-3 whitespace-nowrap text-right text-gray-900 dark:text-gray-100 tabular-nums">
+              <%= number_with_delimiter(execution.total_tokens || 0) %>
+            </td>
+
+            <%# Cost %>
+            <td class="px-4 py-3 whitespace-nowrap text-right text-gray-900 dark:text-gray-100 tabular-nums">
+              $<%= number_with_precision(execution.total_cost || 0, precision: 4) %>
+            </td>
+
+            <%# Time %>
+            <td class="px-4 py-3 whitespace-nowrap text-right text-gray-500 dark:text-gray-400">
+              <%= time_ago_in_words(execution.created_at) %> ago
+            </td>
+          </tr>
+
+          <%# Error Row (if applicable) %>
+          <% if execution.status_error? && execution.error_message.present? %>
+            <tr class="bg-red-50/50 dark:bg-red-900/20">
+              <td></td>
+              <td colspan="<%= show_tenant_column ? 7 : 6 %>" class="px-4 py-2">
+                <p class="text-xs text-red-600 dark:text-red-400">
+                  <span class="font-medium"><%= execution.error_class %>:</span>
+                  <%= truncate(execution.error_message, length: 120) %>
+                </p>
+              </td>
+            </tr>
+          <% end %>
+
+          <%# Child Rows (for workflows) %>
+          <% if has_children %>
+            <% children.each_with_index do |child, index| %>
+              <% is_last = index == children.size - 1 %>
+              <tr class="bg-gray-50/50 dark:bg-gray-900/30 hover:bg-gray-100/50 dark:hover:bg-gray-800/50 transition-colors">
+                <%# Status with tree line %>
+                <td class="px-4 py-2 whitespace-nowrap">
+                  <div class="flex items-center">
+                    <span class="text-gray-300 dark:text-gray-600 mr-2 font-mono text-xs"><%= is_last ? "└─" : "├─" %></span>
+                    <% case child.status
+                       when "success" %>
+                      <span class="text-green-600 dark:text-green-400">✓</span>
+                    <% when "error" %>
+                      <span class="text-red-600 dark:text-red-400">✗</span>
+                    <% when "timeout" %>
+                      <span class="text-orange-600 dark:text-orange-400">⏱</span>
+                    <% when "running" %>
+                      <span class="text-blue-600 dark:text-blue-400 animate-pulse">●</span>
+                    <% else %>
+                      <span class="text-gray-400">○</span>
+                    <% end %>
+                  </div>
+                </td>
+
+                <%# Step/Branch Name %>
+                <td class="px-4 py-2 whitespace-nowrap">
+                  <%= link_to ruby_llm_agents.execution_path(child), class: "text-gray-600 dark:text-gray-300 hover:text-blue-600 dark:hover:text-blue-400" do %>
+                    <% if execution.pipeline_workflow? %>
+                      <span class="text-gray-400 dark:text-gray-500 text-xs"><%= index + 1 %>.</span>
+                    <% end %>
+                    <%= child.workflow_step || child.agent_type.gsub(/Agent$/, "") %>
+                  <% end %>
+                </td>
+
+                <%# Tenant - empty for child rows %>
+                <% if show_tenant_column %>
+                  <td class="px-4 py-2"></td>
+                <% end %>
+
+                <%# Model %>
+                <td class="px-4 py-2 whitespace-nowrap text-gray-400 dark:text-gray-500 font-mono text-xs">
+                  <%= child.model_id %>
+                </td>
+
+                <%# Duration %>
+                <td class="px-4 py-2 whitespace-nowrap text-right text-gray-600 dark:text-gray-300 tabular-nums text-xs">
+                  <%= child.duration_ms ? "#{number_with_delimiter(child.duration_ms)}ms" : "-" %>
+                </td>
+
+                <%# Tokens %>
+                <td class="px-4 py-2 whitespace-nowrap text-right text-gray-600 dark:text-gray-300 tabular-nums text-xs">
+                  <%= number_with_delimiter(child.total_tokens || 0) %>
+                </td>
+
+                <%# Cost %>
+                <td class="px-4 py-2 whitespace-nowrap text-right text-gray-600 dark:text-gray-300 tabular-nums text-xs">
+                  $<%= number_with_precision(child.total_cost || 0, precision: 4) %>
+                </td>
+
+                <%# Time - empty for children %>
+                <td class="px-4 py-2"></td>
+              </tr>
+
+              <%# Child Error Row %>
+              <% if child.status_error? && child.error_message.present? %>
+                <tr class="bg-red-50/30 dark:bg-red-900/10">
+                  <td></td>
+                  <td colspan="<%= show_tenant_column ? 7 : 6 %>" class="px-4 py-1.5 pl-8">
+                    <p class="text-xs text-red-500 dark:text-red-400">
+                      <%= truncate(child.error_message, length: 100) %>
+                    </p>
+                  </td>
+                </tr>
+              <% end %>
+            <% end %>
+          <% end %>
+        <% end %>
+      </tbody>
+    </table>
+  </div>
+
+  <%# Pagination %>
+  <% if pagination[:total_pages] > 1 %>
+    <%
+      current_page = pagination[:current_page]
+      total_pages = pagination[:total_pages]
+      total_count = pagination[:total_count]
+      per_page = pagination[:per_page]
+
+      from_record = ((current_page - 1) * per_page) + 1
+      to_record = [current_page * per_page, total_count].min
+    %>
+    <div class="mt-4 flex items-center justify-between border-t border-gray-100 dark:border-gray-700 pt-4">
+      <p class="text-sm text-gray-500 dark:text-gray-400">
+        Showing <%= from_record %>-<%= to_record %> of <%= number_with_delimiter(total_count) %> executions
+      </p>
+      <nav class="flex items-center space-x-1">
+        <% if current_page > 1 %>
+          <%= link_to "Previous", url_for(request.query_parameters.merge(page: current_page - 1)), class: "px-3 py-1.5 text-sm font-medium text-gray-700 dark:text-gray-200 bg-white dark:bg-gray-800 border border-gray-300 dark:border-gray-600 rounded-md hover:bg-gray-50 dark:hover:bg-gray-700" %>
+        <% else %>
+          <span class="px-3 py-1.5 text-sm font-medium text-gray-400 dark:text-gray-500 bg-gray-100 dark:bg-gray-800 border border-gray-200 dark:border-gray-700 rounded-md cursor-not-allowed">Previous</span>
+        <% end %>
+
+        <%
+          window = 2
+          left_edge = 1
+          right_edge = 1
+
+          pages_to_show = []
+          (1..total_pages).each do |page|
+            if page <= left_edge ||
+               page > total_pages - right_edge ||
+               (page >= current_page - window && page <= current_page + window)
+              pages_to_show << page
+            elsif pages_to_show.last != :gap
+              pages_to_show << :gap
+            end
+          end
+        %>
+
+        <% pages_to_show.each do |page| %>
+          <% if page == :gap %>
+            <span class="px-2 py-1.5 text-sm text-gray-500 dark:text-gray-400">...</span>
+          <% elsif page == current_page %>
+            <span class="px-3 py-1.5 text-sm font-medium text-white bg-blue-600 border border-blue-600 rounded-md"><%= page %></span>
+          <% else %>
+            <%= link_to page, url_for(request.query_parameters.merge(page: page)), class: "px-3 py-1.5 text-sm font-medium text-gray-700 dark:text-gray-200 bg-white dark:bg-gray-800 border border-gray-300 dark:border-gray-600 rounded-md hover:bg-gray-50 dark:hover:bg-gray-700" %>
+          <% end %>
+        <% end %>
+
+        <% if current_page < total_pages %>
+          <%= link_to "Next", url_for(request.query_parameters.merge(page: current_page + 1)), class: "px-3 py-1.5 text-sm font-medium text-gray-700 dark:text-gray-200 bg-white dark:bg-gray-800 border border-gray-300 dark:border-gray-600 rounded-md hover:bg-gray-50 dark:hover:bg-gray-700" %>
+        <% else %>
+          <span class="px-3 py-1.5 text-sm font-medium text-gray-400 dark:text-gray-500 bg-gray-100 dark:bg-gray-800 border border-gray-200 dark:border-gray-700 rounded-md cursor-not-allowed">Next</span>
+        <% end %>
+      </nav>
+    </div>
+  <% end %>
+<% end %>

data/app/views/{rubyllm → ruby_llm}/agents/shared/_filter_dropdown.html.erb

@@ -1,7 +1,7 @@
 <%
   # Multi-select filter dropdown with Alpine.js
   # Usage:
-  #   render "rubyllm/agents/shared/filter_dropdown",
+  #   render "ruby_llm/agents/shared/filter_dropdown",
   #     name: "statuses[]",
   #     filter_id: "statuses",
   #     label: "Status",

data/app/views/{rubyllm → ruby_llm}/agents/shared/_select_dropdown.html.erb

@@ -1,7 +1,7 @@
 <%
   # Single-select filter dropdown with Alpine.js
   # Usage:
-  #   render "rubyllm/agents/shared/select_dropdown",
+  #   render "ruby_llm/agents/shared/select_dropdown",
   #     name: "days",
   #     filter_id: "days",
   #     options: [

data/app/views/{rubyllm → ruby_llm}/agents/shared/_status_badge.html.erb

@@ -1,6 +1,6 @@
 <%
   # Status badge with icon and color coding
-  # Usage: render "rubyllm/agents/shared/status_badge", status: execution.status
+  # Usage: render "ruby_llm/agents/shared/status_badge", status: execution.status
   # Options: size: :sm (default), :md, :lg
 
   size = local_assigns[:size] || :sm

data/app/views/{rubyllm → ruby_llm}/agents/shared/_status_dot.html.erb

@@ -1,6 +1,6 @@
 <%
   # Simple status dot indicator
-  # Usage: render "rubyllm/agents/shared/status_dot", status: execution.status
+  # Usage: render "ruby_llm/agents/shared/status_dot", status: execution.status
 
   config = case status.to_s
     when "running"

data/app/views/ruby_llm/agents/shared/_tenant_filter.html.erb

@@ -0,0 +1,26 @@
+<% if tenant_filter_enabled? && available_tenants.any? %>
+  <div class="tenant-filter">
+    <%= form_with url: request.path, method: :get, local: true, class: "tenant-filter-form" do |f| %>
+      <label for="tenant_id">Tenant:</label>
+      <select name="tenant_id" id="tenant_id" onchange="this.form.submit()">
+        <option value="">All Tenants</option>
+        <% available_tenants.each do |tenant| %>
+          <option value="<%= tenant %>" <%= 'selected' if tenant == current_tenant_id %>>
+            <%= tenant %>
+          </option>
+        <% end %>
+      </select>
+
+      <%# Preserve other filter params %>
+      <% params.except(:tenant_id, :controller, :action).each do |key, value| %>
+        <% if value.is_a?(Array) %>
+          <% value.each do |v| %>
+            <input type="hidden" name="<%= key %>[]" value="<%= v %>">
+          <% end %>
+        <% else %>
+          <input type="hidden" name="<%= key %>" value="<%= value %>">
+        <% end %>
+      <% end %>
+    <% end %>
+  </div>
+<% end %>

data/app/views/ruby_llm/agents/shared/_workflow_type_badge.html.erb

@@ -0,0 +1,61 @@
+<%
+  # Workflow type badge with icon and color coding
+  # Usage: render "ruby_llm/agents/shared/workflow_type_badge", workflow_type: "pipeline"
+  # Options:
+  #   size: :xs, :sm (default), :md
+  #   show_label: true (default) or false for icon-only mode
+
+  workflow_type = local_assigns[:workflow_type]
+  size = local_assigns[:size] || :sm
+  show_label = local_assigns.fetch(:show_label, true)
+
+  config = case workflow_type.to_s
+    when "pipeline"
+      {
+        icon: "arrow-right",
+        label: "Pipeline",
+        bg: "bg-indigo-100 dark:bg-indigo-900/50",
+        text: "text-indigo-700 dark:text-indigo-300",
+        icon_char: "→"
+      }
+    when "parallel"
+      {
+        icon: "parallel",
+        label: "Parallel",
+        bg: "bg-cyan-100 dark:bg-cyan-900/50",
+        text: "text-cyan-700 dark:text-cyan-300",
+        icon_char: "⫿"
+      }
+    when "router"
+      {
+        icon: "router",
+        label: "Router",
+        bg: "bg-amber-100 dark:bg-amber-900/50",
+        text: "text-amber-700 dark:text-amber-300",
+        icon_char: "⌂"
+      }
+    else
+      {
+        icon: "workflow",
+        label: "Workflow",
+        bg: "bg-gray-100 dark:bg-gray-700",
+        text: "text-gray-700 dark:text-gray-300",
+        icon_char: "⚙"
+      }
+    end
+
+  size_classes = case size
+    when :xs
+      { badge: "px-1.5 py-0.5", icon: "text-[10px]", text: "text-[10px]" }
+    when :md
+      { badge: "px-2.5 py-1", icon: "text-sm", text: "text-sm" }
+    else # :sm
+      { badge: "px-2 py-0.5", icon: "text-xs", text: "text-xs" }
+    end
+%>
+<span class="inline-flex items-center gap-1 rounded-md font-medium <%= config[:bg] %> <%= config[:text] %> <%= size_classes[:badge] %>">
+  <span class="<%= size_classes[:icon] %>" aria-hidden="true"><%= config[:icon_char] %></span>
+  <% if show_label %>
+    <span class="<%= size_classes[:text] %>"><%= config[:label] %></span>
+  <% end %>
+</span>

data/lib/generators/ruby_llm_agents/multi_tenancy_generator.rb

@@ -0,0 +1,97 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+require "rails/generators/active_record"
+
+module RubyLlmAgents
+  # Multi-tenancy generator for ruby_llm-agents
+  #
+  # Usage:
+  #   rails generate ruby_llm_agents:multi_tenancy
+  #
+  # This will create migrations for:
+  #   - ruby_llm_agents_tenant_budgets table for per-tenant budget configuration
+  #   - Adding tenant_id column to ruby_llm_agents_executions
+  #
+  class MultiTenancyGenerator < ::Rails::Generators::Base
+    include ::ActiveRecord::Generators::Migration
+
+    source_root File.expand_path("templates", __dir__)
+
+    desc "Adds multi-tenancy support to RubyLLM::Agents"
+
+    def create_tenant_budgets_migration
+      if table_exists?(:ruby_llm_agents_tenant_budgets)
+        say_status :skip, "ruby_llm_agents_tenant_budgets table already exists", :yellow
+        return
+      end
+
+      migration_template(
+        "create_tenant_budgets_migration.rb.tt",
+        File.join(db_migrate_path, "create_ruby_llm_agents_tenant_budgets.rb")
+      )
+    end
+
+    def create_add_tenant_to_executions_migration
+      if column_exists?(:ruby_llm_agents_executions, :tenant_id)
+        say_status :skip, "tenant_id column already exists", :yellow
+        return
+      end
+
+      migration_template(
+        "add_tenant_to_executions_migration.rb.tt",
+        File.join(db_migrate_path, "add_tenant_id_to_ruby_llm_agents_executions.rb")
+      )
+    end
+
+    def show_post_install_message
+      say ""
+      say "Multi-tenancy migrations created!", :green
+      say ""
+      say "Next steps:"
+      say "  1. Run: rails db:migrate"
+      say "  2. Configure multi-tenancy in your initializer:"
+      say ""
+      say "     RubyLLM::Agents.configure do |config|"
+      say "       config.multi_tenancy_enabled = true"
+      say "       config.tenant_resolver = -> { Current.tenant&.id }"
+      say "     end"
+      say ""
+      say "  3. Set Current.tenant in your ApplicationController"
+      say ""
+      say "  4. Create tenant budgets:"
+      say ""
+      say "     RubyLLM::Agents::TenantBudget.create!("
+      say "       tenant_id: 'acme_corp',"
+      say "       daily_limit: 50.0,"
+      say "       monthly_limit: 500.0,"
+      say "       enforcement: 'hard'"
+      say "     )"
+      say ""
+    end
+
+    private
+
+    def migration_version
+      "[#{::ActiveRecord::VERSION::STRING.to_f}]"
+    end
+
+    def db_migrate_path
+      "db/migrate"
+    end
+
+    def table_exists?(table)
+      ActiveRecord::Base.connection.table_exists?(table)
+    rescue StandardError
+      false
+    end
+
+    def column_exists?(table, column)
+      return false unless ActiveRecord::Base.connection.table_exists?(table)
+
+      ActiveRecord::Base.connection.column_exists?(table, column)
+    rescue StandardError
+      false
+    end
+  end
+end

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

@@ -8,8 +8,8 @@
 # Run with: rails db:migrate
 class AddAttemptsToRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migration_version %>
   def change
-    # Add attempts JSONB array for storing per-attempt details
-    add_column :ruby_llm_agents_executions, :attempts, :jsonb, null: false, default: []
+    # Add attempts JSON array for storing per-attempt details
+    add_column :ruby_llm_agents_executions, :attempts, :json, null: false, default: []
 
     # Add counter for quick access to attempt count
     add_column :ruby_llm_agents_executions, :attempts_count, :integer, null: false, default: 0
@@ -18,7 +18,7 @@ class AddAttemptsToRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migratio
     add_column :ruby_llm_agents_executions, :chosen_model_id, :string
 
     # Add fallback chain (list of models that were configured to try)
-    add_column :ruby_llm_agents_executions, :fallback_chain, :jsonb, null: false, default: []
+    add_column :ruby_llm_agents_executions, :fallback_chain, :json, null: false, default: []
 
     # Add indexes for common queries
     add_index :ruby_llm_agents_executions, :attempts_count

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

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+# Migration to add tenant_id column to executions for multi-tenancy support
+#
+# This migration adds a tenant_id column to track which tenant each execution
+# belongs to, enabling:
+# - Filtering executions by tenant
+# - Tenant-scoped analytics and reporting
+# - Per-tenant budget tracking and circuit breakers
+#
+# Run with: rails db:migrate
+class AddTenantIdToRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migration_version %>
+  def change
+    # Add tenant_id column (nullable for backward compatibility)
+    add_column :ruby_llm_agents_executions, :tenant_id, :string
+
+    # Add indexes for efficient tenant-scoped queries
+    add_index :ruby_llm_agents_executions, :tenant_id
+    add_index :ruby_llm_agents_executions, [:tenant_id, :created_at]
+    add_index :ruby_llm_agents_executions, [:tenant_id, :agent_type]
+    add_index :ruby_llm_agents_executions, [:tenant_id, :status]
+  end
+end

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

@@ -15,9 +15,9 @@
 # Run with: rails db:migrate
 class AddToolCallsToRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migration_version %>
   def change
-    # Add tool_calls JSONB array for storing tool call details
+    # Add tool_calls JSON array for storing tool call details
     # Each tool call contains: id, name, arguments
-    add_column :ruby_llm_agents_executions, :tool_calls, :jsonb, null: false, default: []
+    add_column :ruby_llm_agents_executions, :tool_calls, :json, null: false, default: []
 
     # Add counter for quick access to tool call count
     add_column :ruby_llm_agents_executions, :tool_calls_count, :integer, null: false, default: 0

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

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+# Migration to add workflow orchestration columns to executions
+#
+# This migration adds columns for tracking workflow executions (Pipeline,
+# Parallel, Router patterns) and linking child executions to their
+# parent workflow.
+#
+# Workflow patterns supported:
+# - Pipeline: Sequential execution with data flowing between steps
+# - Parallel: Concurrent execution with result aggregation
+# - Router: Conditional dispatch based on classification
+#
+# Run with: rails db:migrate
+class AddWorkflowToRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migration_version %>
+  def change
+    # Unique identifier for the workflow execution
+    # All steps/branches share the same workflow_id
+    add_column :ruby_llm_agents_executions, :workflow_id, :string
+
+    # Type of workflow: "pipeline", "parallel", "router", or nil for regular agents
+    add_column :ruby_llm_agents_executions, :workflow_type, :string
+
+    # Name of the step/branch within the workflow
+    add_column :ruby_llm_agents_executions, :workflow_step, :string
+
+    # For routers: the route that was selected
+    add_column :ruby_llm_agents_executions, :routed_to, :string
+
+    # For routers: classification details (route, method, time)
+    add_column :ruby_llm_agents_executions, :classification_result, :json
+
+    # Add indexes for efficient querying
+    add_index :ruby_llm_agents_executions, :workflow_id
+    add_index :ruby_llm_agents_executions, :workflow_type
+    add_index :ruby_llm_agents_executions, [:workflow_id, :workflow_step]
+  end
+end

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

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+# Migration to create the tenant_budgets table for multi-tenancy support
+#
+# This table stores per-tenant budget configuration, allowing different
+# tenants to have their own budget limits and enforcement modes.
+#
+# Features:
+# - Per-tenant daily and monthly budget limits
+# - Per-agent budget limits within a tenant
+# - Configurable enforcement mode (none, soft, hard)
+# - Option to inherit global defaults for unset limits
+#
+# Run with: rails db:migrate
+class CreateRubyLLMAgentsTenantBudgets < ActiveRecord::Migration<%= migration_version %>
+  def change
+    create_table :ruby_llm_agents_tenant_budgets do |t|
+      # Unique identifier for the tenant (e.g., organization ID, workspace ID)
+      t.string :tenant_id, null: false
+
+      # Global budget limits for this tenant
+      t.decimal :daily_limit, precision: 12, scale: 6
+      t.decimal :monthly_limit, precision: 12, scale: 6
+
+      # Per-agent budget limits (JSON hash)
+      # Format: { "AgentName" => limit_value }
+      t.json :per_agent_daily, null: false, default: {}
+      t.json :per_agent_monthly, null: false, default: {}
+
+      # Enforcement mode for this tenant: "none", "soft", or "hard"
+      # - none: no enforcement, only tracking
+      # - soft: log warnings when limits exceeded
+      # - hard: block execution when limits exceeded
+      t.string :enforcement, default: "soft"
+
+      # Whether to inherit from global config for unset limits
+      t.boolean :inherit_global_defaults, default: true
+
+      t.timestamps
+    end
+
+    # Ensure unique tenant IDs
+    add_index :ruby_llm_agents_tenant_budgets, :tenant_id, unique: true
+  end
+end

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

@@ -54,10 +54,10 @@ class CreateRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migration_versi
       t.decimal :output_cost, precision: 12, scale: 6
       t.decimal :total_cost, precision: 12, scale: 6
 
-      # Data (JSONB for PostgreSQL, JSON for others)
-      t.jsonb :parameters, null: false, default: {}
-      t.jsonb :response, default: {}
-      t.jsonb :metadata, null: false, default: {}
+      # Data (JSON - works with PostgreSQL, MySQL, SQLite3)
+      t.json :parameters, null: false, default: {}
+      t.json :response, default: {}
+      t.json :metadata, null: false, default: {}
 
       # Error tracking
       t.string :error_class
@@ -68,9 +68,16 @@ class CreateRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migration_versi
       t.text :user_prompt
 
       # Tool calls tracking
-      t.jsonb :tool_calls, null: false, default: []
+      t.json :tool_calls, null: false, default: []
       t.integer :tool_calls_count, null: false, default: 0
 
+      # Workflow orchestration
+      t.string :workflow_id
+      t.string :workflow_type
+      t.string :workflow_step
+      t.string :routed_to
+      t.json :classification_result
+
       t.timestamps
     end
 
@@ -96,6 +103,11 @@ class CreateRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migration_versi
     # Tool calls index
     add_index :ruby_llm_agents_executions, :tool_calls_count
 
+    # Workflow indexes
+    add_index :ruby_llm_agents_executions, :workflow_id
+    add_index :ruby_llm_agents_executions, :workflow_type
+    add_index :ruby_llm_agents_executions, [:workflow_id, :workflow_step]
+
     # Foreign keys for execution hierarchy
     add_foreign_key :ruby_llm_agents_executions, :ruby_llm_agents_executions,
                     column: :parent_execution_id, on_delete: :nullify