RubyGems - llama_bot_rails - Versions diffs - 0.1.0 - Mend

llama_bot_rails 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (31) hide show

checksums.yaml +7 -0
data/MIT-LICENSE +20 -0
data/README.md +249 -0
data/Rakefile +8 -0
data/app/assets/config/llama_bot_rails_manifest.js +1 -0
data/app/assets/javascripts/llama_bot_rails/application.js +7 -0
data/app/assets/javascripts/llama_bot_rails/chat.js +13 -0
data/app/assets/stylesheets/llama_bot_rails/application.css +15 -0
data/app/channels/llama_bot_rails/application_cable/channel.rb +8 -0
data/app/channels/llama_bot_rails/application_cable/connection.rb +13 -0
data/app/channels/llama_bot_rails/chat_channel.rb +306 -0
data/app/controllers/llama_bot_rails/agent_controller.rb +72 -0
data/app/controllers/llama_bot_rails/application_controller.rb +4 -0
data/app/helpers/llama_bot_rails/application_helper.rb +4 -0
data/app/javascript/channels/consumer.js +4 -0
data/app/jobs/llama_bot_rails/application_job.rb +4 -0
data/app/models/llama_bot_rails/application_record.rb +5 -0
data/app/views/layouts/llama_bot_rails/application.html.erb +17 -0
data/app/views/llama_bot_rails/agent/chat.html.erb +962 -0
data/bin/rails +26 -0
data/bin/rubocop +8 -0
data/config/initializers/llama_bot_rails.rb +2 -0
data/config/routes.rb +6 -0
data/lib/llama_bot_rails/agent_state_builder.rb +17 -0
data/lib/llama_bot_rails/engine.rb +23 -0
data/lib/llama_bot_rails/llama_bot.rb +25 -0
data/lib/llama_bot_rails/tools/rails_console_tool.rb +20 -0
data/lib/llama_bot_rails/version.rb +3 -0
data/lib/llama_bot_rails.rb +10 -0
data/lib/tasks/llama_bot_rails_tasks.rake +4 -0
metadata +128 -0

data/app/controllers/llama_bot_rails/agent_controller.rb ADDED Viewed

@@ -0,0 +1,72 @@
+require 'llama_bot_rails/llama_bot'
+module LlamaBotRails
+    class AgentController < ActionController::Base
+        skip_before_action :verify_authenticity_token, only: [:command]
+        before_action :authenticate_agent!, only: [:command]
+        # POST /agent/command
+        def command
+            input = params[:command]
+            result = safety_eval(input)
+            render json: { result: result.inspect }
+        rescue => e
+            render json: { error: e.class.name, message: e.message }, status: :unprocessable_entity
+        end
+        def index
+            @llama_bot = LlamaBot.new
+        end
+        # GET /agent/chat
+        def chat
+            # Render chat.html.erb
+        end
+        def threads
+            begin
+                threads = LlamaBotRails::LlamaBot.get_threads
+                render json: threads
+            rescue => e
+                Rails.logger.error "Error in threads action: #{e.message}"
+                render json: { error: "Failed to fetch threads" }, status: :internal_server_error
+            end
+        end
+        def chat_history
+            begin
+                thread_id = params[:thread_id]
+                Rails.logger.info "Fetching chat history for thread: #{thread_id}"
+                if thread_id == 'undefined' || thread_id.blank?
+                    render json: []
+                    return
+                end
+                history = LlamaBotRails::LlamaBot.get_chat_history(thread_id)
+                render json: history
+            rescue => e
+                Rails.logger.error "Error in chat_history action: #{e.message}"
+                render json: { error: "Failed to fetch chat history" }, status: :internal_server_error
+            end
+        end
+        private
+        def safety_eval(input)
+            # VERY basic, sandbox for real use!
+            result = eval(input)
+            { result: result.inspect }
+        rescue => e
+            { error: e.class.name, message: e.message }
+        end
+        def authenticate_agent!
+            auth_header = request.headers["Authorization"]
+            token = auth_header&.split("Bearer ")&.last  # Extract token after "Bearer "
+            @session_payload = Rails.application.message_verifier(:llamabot_ws).verify(token)
+        rescue ActiveSupport::MessageVerifier::InvalidSignature
+            head :unauthorized
+        end
+    end
+end

data/app/controllers/llama_bot_rails/application_controller.rb ADDED Viewed

@@ -0,0 +1,4 @@
+module LlamaBotRails
+  class ApplicationController < ActionController::Base
+  end
+end

data/app/helpers/llama_bot_rails/application_helper.rb ADDED Viewed

@@ -0,0 +1,4 @@
+module LlamaBotRails
+  module ApplicationHelper
+  end
+end

data/app/javascript/channels/consumer.js ADDED Viewed

@@ -0,0 +1,4 @@
+// app/javascript/channels/consumer.js
+import { createConsumer } from "@rails/actioncable"
+export default createConsumer()

data/app/jobs/llama_bot_rails/application_job.rb ADDED Viewed

@@ -0,0 +1,4 @@
+module LlamaBotRails
+  class ApplicationJob < ActiveJob::Base
+  end
+end

data/app/models/llama_bot_rails/application_record.rb ADDED Viewed

@@ -0,0 +1,5 @@
+module LlamaBotRails
+  class ApplicationRecord < ActiveRecord::Base
+    self.abstract_class = true
+  end
+end

data/app/views/layouts/llama_bot_rails/application.html.erb ADDED Viewed

@@ -0,0 +1,17 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Llama bot rails</title>
+  <%= csrf_meta_tags %>
+  <%= csp_meta_tag %>
+  <%= yield :head %>
+  <%= stylesheet_link_tag    "llama_bot_rails/application", media: "all" %>
+</head>
+<body>
+<%= yield %>
+</body>
+</html>