llama_bot_rails 0.1.0

data/bin/rails

@@ -0,0 +1,26 @@
+#!/usr/bin/env ruby
+# This command will automatically be run when you run "rails" with Rails gems
+# installed from the root of your application.
+
+ENGINE_ROOT = File.expand_path("..", __dir__)
+ENGINE_PATH = File.expand_path("../lib/llama_bot_rails/engine", __dir__)
+APP_PATH = File.expand_path("../spec/dummy/config/application", __dir__)
+
+# Set up gems listed in the Gemfile.
+ENV["BUNDLE_GEMFILE"] ||= File.expand_path("../Gemfile", __dir__)
+require "bundler/setup" if File.exist?(ENV["BUNDLE_GEMFILE"])
+
+require "rails"
+# Pick the frameworks you want:
+require "active_model/railtie"
+require "active_job/railtie"
+require "active_record/railtie"
+require "active_storage/engine"
+require "action_controller/railtie"
+# require "action_mailer/railtie"
+# require "action_mailbox/engine"
+# require "action_text/engine"
+require "action_view/railtie"
+require "action_cable/engine"
+# require "rails/test_unit/railtie"
+require "rails/engine/commands"

data/bin/rubocop

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+require "rubygems"
+require "bundler/setup"
+
+# explicit rubocop config increases performance slightly while avoiding config confusion.
+ARGV.unshift("--config", File.expand_path("../.rubocop.yml", __dir__))
+
+load Gem.bin_path("rubocop", "rubocop")

data/config/initializers/llama_bot_rails.rb

@@ -0,0 +1,2 @@
+# Ensure LlamaBotRails files are loaded
+Rails.application.config.paths.add 'lib', eager_load: true 

data/config/routes.rb

@@ -0,0 +1,6 @@
+LlamaBotRails::Engine.routes.draw do
+  post "agent/command", to: "agent#command"
+  get "agent/chat", to: "agent#chat"
+  get "agent/threads", to: "agent#threads"
+  get "agent/chat-history/:thread_id", to: "agent#chat_history"
+end

data/lib/llama_bot_rails/agent_state_builder.rb

@@ -0,0 +1,17 @@
+module LlamaBotRails
+    class AgentStateBuilder
+      def initialize(params:, context:)
+        @params = params
+        @context = context
+      end
+  
+      def build
+        {
+          user_message: @params[:message], # Rails param from JS/chat UI
+          thread_id: @context[:thread_id],
+          api_token: @context[:api_token],
+          agent_name: "llamabot" #Very important. This routes to the appropriate LangGraph agent as defined in langgraph.json
+        }
+      end
+    end
+  end

data/lib/llama_bot_rails/engine.rb

@@ -0,0 +1,23 @@
+module LlamaBotRails
+  require 'llama_bot_rails/agent_state_builder'
+
+  class Engine < ::Rails::Engine
+    isolate_namespace LlamaBotRails
+
+    config.generators do |g|
+      g.test_framework :rspec
+      g.fixture_replacement :factory_bot
+      g.factory_bot dir: 'spec/factories'
+    end
+
+    config.llama_bot_rails = ActiveSupport::OrderedOptions.new
+
+    initializer "llama_bot_rails.assets.precompile" do |app|
+      app.config.assets.precompile += %w( llama_bot_rails/application.js )
+    end
+
+    initializer "llama_bot_rails.defaults" do |app|
+      app.config.llama_bot_rails.state_builder_class = "LlamaBotRails::AgentStateBuilder"
+    end
+  end
+end

data/lib/llama_bot_rails/llama_bot.rb

@@ -0,0 +1,25 @@
+require 'net/http'
+require 'json'
+require 'uri'
+
+module LlamaBotRails
+  class LlamaBot
+    def self.get_threads
+      uri = URI('http://localhost:8000/threads')
+      response = Net::HTTP.get_response(uri)
+      JSON.parse(response.body)
+    rescue => e
+      Rails.logger.error "Error fetching threads: #{e.message}"
+      []
+    end
+
+    def self.get_chat_history(thread_id)
+      uri = URI("http://localhost:8000/chat-history/#{thread_id}")
+      response = Net::HTTP.get_response(uri)
+      JSON.parse(response.body)
+    rescue => e
+      Rails.logger.error "Error fetching chat history: #{e.message}"
+      []
+    end
+  end
+end

data/lib/llama_bot_rails/tools/rails_console_tool.rb

@@ -0,0 +1,20 @@
+module LlamaBotRails
+    module Tools
+        class RailsConsoleTool
+            tool_name "rails_console"
+            description "Run a Rails console through MCP Tool calling"
+
+            arguments do 
+                required(:command).filled(:string).description("The ruby code to run")
+            end
+
+            def call(command:)
+                # VERY basic, sandbox for real use!
+                result = eval(command)
+                { result: result.inspect }
+            rescue => e
+                { error: e.class.name, message: e.message }
+            end
+        end
+    end
+end

data/lib/llama_bot_rails/version.rb

@@ -0,0 +1,3 @@
+module LlamaBotRails
+  VERSION = "0.1.0"
+end

data/lib/llama_bot_rails.rb

@@ -0,0 +1,10 @@
+require "llama_bot_rails/version"
+require "llama_bot_rails/engine"
+
+module LlamaBotRails
+  class << self
+    def config
+      Rails.application.config.llama_bot_rails
+    end
+  end
+end

data/lib/tasks/llama_bot_rails_tasks.rake

@@ -0,0 +1,4 @@
+# desc "Explaining what the task does"
+# task :llama_bot_rails do
+#   # Task goes here
+# end

metadata

@@ -0,0 +1,128 @@
+--- !ruby/object:Gem::Specification
+name: llama_bot_rails
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Kody Kendall
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+- !ruby/object:Gem::Dependency
+  name: actioncable
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+- !ruby/object:Gem::Dependency
+  name: async-websocket
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: async-http
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: LlamaBotRails is a gem that turns your existing Rails App into an AI
+  Agent by connecting it to an open source LangGraph agent, LlamaBot.
+email:
+- kody@llamapress.ai
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- MIT-LICENSE
+- README.md
+- Rakefile
+- app/assets/config/llama_bot_rails_manifest.js
+- app/assets/javascripts/llama_bot_rails/application.js
+- app/assets/javascripts/llama_bot_rails/chat.js
+- app/assets/stylesheets/llama_bot_rails/application.css
+- app/channels/llama_bot_rails/application_cable/channel.rb
+- app/channels/llama_bot_rails/application_cable/connection.rb
+- app/channels/llama_bot_rails/chat_channel.rb
+- app/controllers/llama_bot_rails/agent_controller.rb
+- app/controllers/llama_bot_rails/application_controller.rb
+- app/helpers/llama_bot_rails/application_helper.rb
+- app/javascript/channels/consumer.js
+- app/jobs/llama_bot_rails/application_job.rb
+- app/models/llama_bot_rails/application_record.rb
+- app/views/layouts/llama_bot_rails/application.html.erb
+- app/views/llama_bot_rails/agent/chat.html.erb
+- bin/rails
+- bin/rubocop
+- config/initializers/llama_bot_rails.rb
+- config/routes.rb
+- lib/llama_bot_rails.rb
+- lib/llama_bot_rails/agent_state_builder.rb
+- lib/llama_bot_rails/engine.rb
+- lib/llama_bot_rails/llama_bot.rb
+- lib/llama_bot_rails/tools/rails_console_tool.rb
+- lib/llama_bot_rails/version.rb
+- lib/tasks/llama_bot_rails_tasks.rake
+homepage: https://llamapress.ai
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://llamapress.ai
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.7
+specification_version: 4
+summary: LlamaBotRails is a gem that turns your existing Rails App into an AI Agent
+  by connecting it to an open source LangGraph agent, LlamaBot.
+test_files: []