rails_console_ai 0.28.0 → 0.30.0

data/lib/rails_console_ai.rb

@@ -55,6 +55,39 @@ module RailsConsoleAi
       @current_user = name
     end
 
+    # Enqueue an agent run. Returns the Integer session id immediately;
+    # the actual work is picked up by `rake rails_console_ai:agents`.
+    def run_agent(query, name: nil, user_name: nil)
+      require 'rails_console_ai/session_logger'
+      id = SessionLogger.log(
+        query: query,
+        conversation: [],
+        mode: 'agent_api',
+        name: name,
+        user_name: user_name,
+        status: 'queued',
+        executed: false
+      )
+      raise 'Failed to enqueue agent run (session logging disabled or table missing)' unless id
+      id
+    end
+
+    # Returns the current status string for an enqueued agent run, or nil
+    # if the session id is not found. Status is one of:
+    # 'queued' | 'running' | 'ready' | 'failed'.
+    def check_agent(session_id)
+      Session.where(id: session_id).pluck(:status).first
+    end
+
+    # Returns a hash describing an agent run:
+    #   { status:, result:, error: }
+    # All three keys are nil when the session id is not found.
+    def get_agent_response(session_id)
+      row = Session.where(id: session_id).select(:status, :result, :error_message).first
+      return { status: nil, result: nil, error: nil } unless row
+      { status: row.status, result: row.result, error: row.error_message }
+    end
+
     def status
       c = configuration
       key = c.resolved_api_key
@@ -123,11 +156,183 @@ module RailsConsoleAi
         $stdout.puts "\e[32mRailsConsoleAi: created #{table} table.\e[0m"
       end
 
+      setup_skills_tables!(conn)
+      setup_memories_tables!(conn)
+      setup_agents_tables!(conn)
+
       migrate!
     rescue => e
       $stderr.puts "\e[31mRailsConsoleAi setup failed: #{e.class}: #{e.message}\e[0m"
     end
 
+    def setup_skills_tables!(conn)
+      skills_table   = 'rails_console_ai_skills'
+      versions_table = 'rails_console_ai_skill_versions'
+
+      unless conn.table_exists?(skills_table)
+        conn.create_table(skills_table) do |t|
+          t.string   :name,        limit: 255, null: false
+          t.text     :description
+          t.text     :body
+          t.text     :tags
+          t.text     :bypass_guards_for_methods
+          t.string   :status,      limit: 20,  default: 'proposed', null: false
+          t.string   :approved_by, limit: 255
+          t.datetime :approved_at
+          t.integer  :use_count,   default: 0, null: false
+          t.datetime :last_used_at
+          t.datetime :created_at,  null: false
+          t.datetime :updated_at,  null: false
+        end
+        conn.add_index(skills_table, :name, unique: true)
+        conn.add_index(skills_table, :status)
+        $stdout.puts "\e[32mRailsConsoleAi: created #{skills_table} table.\e[0m"
+      else
+        # Idempotent column-add probes for existing installs.
+        unless conn.column_exists?(skills_table, :status)
+          conn.add_column(skills_table, :status, :string, limit: 20, default: 'proposed', null: false)
+          conn.add_index(skills_table, :status) unless conn.index_exists?(skills_table, :status)
+        end
+        unless conn.column_exists?(skills_table, :approved_by)
+          conn.add_column(skills_table, :approved_by, :string, limit: 255)
+        end
+        unless conn.column_exists?(skills_table, :approved_at)
+          conn.add_column(skills_table, :approved_at, :datetime)
+        end
+        unless conn.column_exists?(skills_table, :use_count)
+          conn.add_column(skills_table, :use_count, :integer, default: 0, null: false)
+        end
+        unless conn.column_exists?(skills_table, :last_used_at)
+          conn.add_column(skills_table, :last_used_at, :datetime)
+        end
+      end
+
+      unless conn.table_exists?(versions_table)
+        conn.create_table(versions_table) do |t|
+          t.integer  :skill_id
+          t.string   :name,        limit: 255
+          t.text     :description
+          t.text     :body
+          t.text     :tags
+          t.text     :bypass_guards_for_methods
+          t.string   :status,      limit: 20
+          t.string   :edited_by,   limit: 255
+          t.text     :change_note
+          t.datetime :created_at,  null: false
+        end
+        conn.add_index(versions_table, :skill_id)
+        conn.add_index(versions_table, :created_at)
+        $stdout.puts "\e[32mRailsConsoleAi: created #{versions_table} table.\e[0m"
+      else
+        unless conn.column_exists?(versions_table, :status)
+          conn.add_column(versions_table, :status, :string, limit: 20)
+        end
+      end
+    end
+
+    def setup_memories_tables!(conn)
+      memories_table = 'rails_console_ai_memories'
+      versions_table = 'rails_console_ai_memory_versions'
+
+      unless conn.table_exists?(memories_table)
+        conn.create_table(memories_table) do |t|
+          t.string   :name,        limit: 255, null: false
+          t.text     :description
+          t.text     :tags
+          t.integer  :use_count,   default: 0, null: false
+          t.datetime :last_used_at
+          t.datetime :created_at,  null: false
+          t.datetime :updated_at,  null: false
+        end
+        conn.add_index(memories_table, :name, unique: true)
+        $stdout.puts "\e[32mRailsConsoleAi: created #{memories_table} table.\e[0m"
+      else
+        unless conn.column_exists?(memories_table, :use_count)
+          conn.add_column(memories_table, :use_count, :integer, default: 0, null: false)
+        end
+        unless conn.column_exists?(memories_table, :last_used_at)
+          conn.add_column(memories_table, :last_used_at, :datetime)
+        end
+      end
+
+      unless conn.table_exists?(versions_table)
+        conn.create_table(versions_table) do |t|
+          t.integer  :memory_id
+          t.string   :name,        limit: 255
+          t.text     :description
+          t.text     :tags
+          t.string   :edited_by,   limit: 255
+          t.text     :change_note
+          t.datetime :created_at,  null: false
+        end
+        conn.add_index(versions_table, :memory_id)
+        conn.add_index(versions_table, :created_at)
+        $stdout.puts "\e[32mRailsConsoleAi: created #{versions_table} table.\e[0m"
+      end
+    end
+
+    def setup_agents_tables!(conn)
+      agents_table   = 'rails_console_ai_agents'
+      versions_table = 'rails_console_ai_agent_versions'
+
+      unless conn.table_exists?(agents_table)
+        conn.create_table(agents_table) do |t|
+          t.string   :name,        limit: 255, null: false
+          t.text     :description
+          t.text     :body
+          t.integer  :max_rounds
+          t.string   :model,       limit: 100
+          t.text     :tools
+          t.string   :status,      limit: 20,  default: 'proposed', null: false
+          t.string   :approved_by, limit: 255
+          t.datetime :approved_at
+          t.integer  :use_count,   default: 0, null: false
+          t.datetime :last_used_at
+          t.datetime :created_at,  null: false
+          t.datetime :updated_at,  null: false
+        end
+        conn.add_index(agents_table, :name, unique: true)
+        conn.add_index(agents_table, :status)
+        $stdout.puts "\e[32mRailsConsoleAi: created #{agents_table} table.\e[0m"
+      else
+        unless conn.column_exists?(agents_table, :status)
+          conn.add_column(agents_table, :status, :string, limit: 20, default: 'proposed', null: false)
+          conn.add_index(agents_table, :status) unless conn.index_exists?(agents_table, :status)
+        end
+        unless conn.column_exists?(agents_table, :approved_by)
+          conn.add_column(agents_table, :approved_by, :string, limit: 255)
+        end
+        unless conn.column_exists?(agents_table, :approved_at)
+          conn.add_column(agents_table, :approved_at, :datetime)
+        end
+        unless conn.column_exists?(agents_table, :use_count)
+          conn.add_column(agents_table, :use_count, :integer, default: 0, null: false)
+        end
+        unless conn.column_exists?(agents_table, :last_used_at)
+          conn.add_column(agents_table, :last_used_at, :datetime)
+        end
+      end
+
+      unless conn.table_exists?(versions_table)
+        conn.create_table(versions_table) do |t|
+          t.integer  :agent_id
+          t.string   :name,        limit: 255
+          t.text     :description
+          t.text     :body
+          t.integer  :max_rounds
+          t.string   :model,       limit: 100
+          t.text     :tools
+          t.string   :status,      limit: 20
+          t.string   :edited_by,   limit: 255
+          t.text     :change_note
+          t.datetime :created_at,  null: false
+        end
+        conn.add_index(versions_table, :agent_id)
+        conn.add_index(versions_table, :created_at)
+        $stdout.puts "\e[32mRailsConsoleAi: created #{versions_table} table.\e[0m"
+      end
+    end
+
     def migrate!
       conn = session_connection
       table = 'rails_console_ai_sessions'
@@ -156,6 +361,50 @@ module RailsConsoleAi
         migrations << 'slack_channel_name'
       end
 
+      unless conn.column_exists?(table, :status)
+        conn.add_column(table, :status, :string, limit: 20)
+        migrations << 'status'
+      end
+
+      unless conn.column_exists?(table, :result)
+        conn.add_column(table, :result, :text)
+        migrations << 'result'
+      end
+
+      unless conn.column_exists?(table, :error_message)
+        conn.add_column(table, :error_message, :text)
+        migrations << 'error_message'
+      end
+
+      unless conn.index_exists?(table, [:mode, :status], name: 'idx_rca_sessions_mode_status')
+        conn.add_index(table, [:mode, :status], name: 'idx_rca_sessions_mode_status')
+        migrations << 'idx_rca_sessions_mode_status'
+      end
+
+      # Bring skills/memories/agents tables fully up to date. Each setup_* method is
+      # internally idempotent (guards both `create_table` and every `add_column` /
+      # `add_index`), so running it on an existing install adds any missing columns
+      # (e.g. `status`, `approved_by`, `approved_at`) and indexes without disturbing
+      # data. Note: we always call these — the previous version skipped them when
+      # the base table already existed, which meant column probes never ran on
+      # upgrade and methods like Skill#status hit NameError. See:
+      # https://github.com/cortfr/rails_console_ai/issues (whichever issue you file)
+      pre_columns = {
+        skills:   table_columns(conn, 'rails_console_ai_skills'),
+        memories: table_columns(conn, 'rails_console_ai_memories'),
+        agents:   table_columns(conn, 'rails_console_ai_agents')
+      }
+
+      setup_skills_tables!(conn)
+      setup_memories_tables!(conn)
+      setup_agents_tables!(conn)
+
+      [[:skills, 'rails_console_ai_skills'], [:memories, 'rails_console_ai_memories'], [:agents, 'rails_console_ai_agents']].each do |key, name|
+        post = table_columns(conn, name)
+        added = post - pre_columns[key]
+        migrations.concat(added.map { |c| "#{name}.#{c}" }) unless added.empty?
+      end
+
       if migrations.empty?
         $stdout.puts "\e[32mRailsConsoleAi: #{table} is up to date.\e[0m"
       else
@@ -214,6 +463,13 @@ module RailsConsoleAi
         ActiveRecord::Base.connection
       end
     end
+
+    def table_columns(conn, table_name)
+      return [] unless conn.table_exists?(table_name)
+      conn.columns(table_name).map { |c| c.name }
+    rescue
+      []
+    end
   end
 end
 

data/lib/tasks/rails_console_ai.rake

@@ -4,4 +4,11 @@ namespace :rails_console_ai do
     require 'rails_console_ai/slack_bot'
     RailsConsoleAi::SlackBot.new.start
   end
+
+  desc "Run the RailsConsoleAi agent runner (polls DB for queued agent runs)"
+  task agents: :environment do
+    require 'rails_console_ai/agent_runner'
+    concurrency = Integer(ENV['AGENT_CONCURRENCY'] || RailsConsoleAi::AgentRunner::DEFAULT_CONCURRENCY)
+    RailsConsoleAi::AgentRunner.new(concurrency: concurrency).start
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails_console_ai
 version: !ruby/object:Gem::Version
-  version: 0.28.0
+  version: 0.30.0
 platform: ruby
 authors:
 - Cortfr
@@ -79,6 +79,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '12.0'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.4'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.4'
 description: An LLM-powered agent for your Rails console. Ask questions in natural
   language, get executable Ruby code.
 email: cortfr@gmail.com
@@ -89,20 +103,59 @@ files:
 - CHANGELOG.md
 - LICENSE
 - README.md
+- app/controllers/rails_console_ai/agent_versions_controller.rb
+- app/controllers/rails_console_ai/agents_controller.rb
 - app/controllers/rails_console_ai/application_controller.rb
+- app/controllers/rails_console_ai/memories_controller.rb
+- app/controllers/rails_console_ai/memory_versions_controller.rb
 - app/controllers/rails_console_ai/sessions_controller.rb
+- app/controllers/rails_console_ai/skill_versions_controller.rb
+- app/controllers/rails_console_ai/skills_controller.rb
+- app/helpers/rails_console_ai/diff_helper.rb
 - app/helpers/rails_console_ai/sessions_helper.rb
+- app/models/rails_console_ai/agent.rb
+- app/models/rails_console_ai/agent_version.rb
+- app/models/rails_console_ai/memory.rb
+- app/models/rails_console_ai/memory_version.rb
 - app/models/rails_console_ai/session.rb
+- app/models/rails_console_ai/skill.rb
+- app/models/rails_console_ai/skill_version.rb
 - app/views/layouts/rails_console_ai/application.html.erb
+- app/views/rails_console_ai/agent_versions/index.html.erb
+- app/views/rails_console_ai/agent_versions/show.html.erb
+- app/views/rails_console_ai/agents/_form.html.erb
+- app/views/rails_console_ai/agents/diff.html.erb
+- app/views/rails_console_ai/agents/edit.html.erb
+- app/views/rails_console_ai/agents/index.html.erb
+- app/views/rails_console_ai/agents/new.html.erb
+- app/views/rails_console_ai/agents/show.html.erb
+- app/views/rails_console_ai/memories/_form.html.erb
+- app/views/rails_console_ai/memories/diff.html.erb
+- app/views/rails_console_ai/memories/edit.html.erb
+- app/views/rails_console_ai/memories/index.html.erb
+- app/views/rails_console_ai/memories/new.html.erb
+- app/views/rails_console_ai/memories/show.html.erb
+- app/views/rails_console_ai/memory_versions/index.html.erb
+- app/views/rails_console_ai/memory_versions/show.html.erb
 - app/views/rails_console_ai/sessions/index.html.erb
 - app/views/rails_console_ai/sessions/show.html.erb
+- app/views/rails_console_ai/skill_versions/index.html.erb
+- app/views/rails_console_ai/skill_versions/show.html.erb
+- app/views/rails_console_ai/skills/_form.html.erb
+- app/views/rails_console_ai/skills/diff.html.erb
+- app/views/rails_console_ai/skills/edit.html.erb
+- app/views/rails_console_ai/skills/index.html.erb
+- app/views/rails_console_ai/skills/new.html.erb
+- app/views/rails_console_ai/skills/show.html.erb
 - config/routes.rb
 - lib/generators/rails_console_ai/install_generator.rb
 - lib/generators/rails_console_ai/templates/initializer.rb
 - lib/rails_console_ai.rb
 - lib/rails_console_ai/agent_loader.rb
+- lib/rails_console_ai/agent_runner.rb
 - lib/rails_console_ai/agents/explore-data.md
 - lib/rails_console_ai/agents/investigate-code.md
+- lib/rails_console_ai/channel/api.rb
 - lib/rails_console_ai/channel/base.rb
 - lib/rails_console_ai/channel/console.rb
 - lib/rails_console_ai/channel/slack.rb
@@ -126,6 +179,7 @@ files:
 - lib/rails_console_ai/skill_loader.rb
 - lib/rails_console_ai/slack_bot.rb
 - lib/rails_console_ai/storage/base.rb
+- lib/rails_console_ai/storage/database_storage.rb
 - lib/rails_console_ai/storage/file_storage.rb
 - lib/rails_console_ai/sub_agent.rb
 - lib/rails_console_ai/tools/code_tools.rb