llama_bot_rails 0.1.4 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9bd4c2d542ae02f6d430d6c6233792445b79dcff44b265d508c9d985db22b09b
-  data.tar.gz: '09ad72a1840000cf121133633dc104517e5937a843be0cba48dadd0a3bf60a66'
+  metadata.gz: 44750a5b4dc8e228f8b3422c91f875024e8994a6800bccefd7a3c5303f1e0487
+  data.tar.gz: 4cb58030cd0100dbaf1d6a64f75ad2e12bc1840153f9d50661a6258aa429a55d
 SHA512:
-  metadata.gz: 192fe783f3c6e91170bb3109dbb116d21ca7bb6c68ecb064a92435d416efa16a1e765e5ff59d07af5978e348724e9f70569dbbf2faaeab8ac9a432f7df1712e6
-  data.tar.gz: 02362ff59435fd5a0b88038f276e2cc69d8fd70fac868b16ee4c1fa98bfb4ae36c298e541aec388a37856de636b9ee841bff81be971fbae50949925d2c9e3cd2
+  metadata.gz: f29e9c1d3ce80499e237bcbcfabe0bc6cd4ba6ab2271cef4b1d8f5bb4bec5d70753e66589368be68ea3f71b2f35dd613f2849a9c2f8beb2dc9cff530bcbc8fcf
+  data.tar.gz: e1eb3bf985687e22321804d7b572c027370ac57303fc82ef17ac088ec36af68a6d3030147ecad1a32fdc8cfe20e35bd4b72fdfa3195023db3570eeafa55072ff

data/README.md

@@ -12,7 +12,7 @@ Chat with a powerful agent that has access to your models, your application cont
 
 ## 🎥 **See it in action** (30-Second Demo)
 
-👉 [Insert GIF or YouTube link here]
+👉 ![LlamaBot Demo](https://llamapress-ai-image-uploads.s3.us-west-2.amazonaws.com/pp1s3l4iskwabnq0gi5tx8mi9mue)
 “Welcome to the future of Rails + AI.”
 
 ### The agent can:
@@ -38,6 +38,8 @@ rails generate llama_bot_rails:install
 # 3. Clone & run the LangGraph backend
 git clone https://github.com/kodykendall/llamabot
 
+cd llamabot
+
 # 4. Set up your environment
 python3 -m venv venv
 

data/app/controllers/llama_bot_rails/agent_controller.rb

@@ -7,9 +7,25 @@ module LlamaBotRails
         # POST /agent/command
         def command
             input = params[:command]
+            
+            # Capture both stdout and return value
+            output = StringIO.new
+            result = nil
+            
+            $stdout = output
             result = safety_eval(input)
-            render json: { result: result.inspect }
+            $stdout = STDOUT
+            
+            # If result is a string and output has content, prefer output
+            final_result = if output.string.present?
+                             output.string.strip
+                           else
+                             result
+                           end
+            
+            render json: { result: final_result }
         rescue => e
+            $stdout = STDOUT  # Reset stdout on error
             render json: { error: e.class.name, message: e.message }, status: :unprocessable_entity
         end
 
@@ -53,12 +69,11 @@ module LlamaBotRails
         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 }
+            # Change to Rails root directory for file operations
+            Dir.chdir(Rails.root) do
+                # Create a safer evaluation context
+                binding.eval(input)
+            end
         end
 
         def authenticate_agent!

data/app/views/llama_bot_rails/agent/chat.html.erb

@@ -509,9 +509,17 @@
       }
     }
   </style>
-  <%= javascript_importmap_tags %>
+
+  <% if defined?(javascript_importmap_tags) %> <!-- Rails 7+ -->
+    <%= javascript_importmap_tags %>
+  <% else %> <!-- Rails 6 -->
+    <%= javascript_include_tag "application" %>
+  <% end %>
+  
   <%= javascript_include_tag "llama_bot_rails/application" %>
-  <%= action_cable_meta_tag %>
+  <% if defined?(action_cable_meta_tag) %>
+    <%= action_cable_meta_tag %>
+  <% end %>
   <!-- Add Snarkdown CDN -->
   <script src="https://unpkg.com/snarkdown/dist/snarkdown.umd.js"></script>
 </head>

data/lib/generators/llama_bot_rails/install/install_generator.rb

@@ -3,6 +3,11 @@ module LlamaBotRails
     module Generators
       class InstallGenerator < Rails::Generators::Base
         source_root File.expand_path("templates", __dir__)
+
+        def create_config_file
+            empty_directory "config/llama_bot"
+            copy_file "agent_prompt.txt", "config/llama_bot/agent_prompt.txt"
+        end
   
         def mount_engine
             say <<~MSG, :yellow
@@ -32,6 +37,7 @@ module LlamaBotRails
           create_file "config/initializers/llama_bot_rails.rb", <<~RUBY
             Rails.application.configure do
               config.llama_bot_rails.websocket_url = ENV.fetch("LLAMABOT_WEBSOCKET_URL", "ws://localhost:8000/ws")
+              config.llama_bot_rails.llamabot_api_url = ENV.fetch("LLAMABOT_API_URL", "http://localhost:8000")
               config.llama_bot_rails.enable_console_tool = !Rails.env.production?
             end
           RUBY

data/lib/generators/llama_bot_rails/install/templates/agent_prompt.txt

@@ -0,0 +1 @@
+You are LlamaBot, a helpful assistant inside a Ruby on Rails app.

data/lib/llama_bot_rails/agent_state_builder.rb

@@ -10,6 +10,7 @@ module LlamaBotRails
           user_message: @params[:message], # Rails param from JS/chat UI
           thread_id: @context[:thread_id],
           api_token: @context[:api_token],
+          agent_prompt: LlamaBotRails.agent_prompt_text,
           agent_name: "llamabot" #Very important. This routes to the appropriate LangGraph agent as defined in langgraph.json
         }
       end

data/lib/llama_bot_rails/engine.rb

@@ -14,7 +14,7 @@ module LlamaBotRails
     config.llama_bot_rails.websocket_url = 'ws://localhost:8000/ws' # <-- default
     config.llama_bot_rails.llamabot_api_url = "http://localhost:8000"
     config.llama_bot_rails.enable_console_tool = true
-
+    
     initializer "llama_bot_rails.assets.precompile" do |app|
       app.config.assets.precompile += %w( llama_bot_rails/application.js )
     end

data/lib/llama_bot_rails/version.rb

@@ -1,3 +1,3 @@
 module LlamaBotRails
-  VERSION = "0.1.4"
+  VERSION = "0.1.6"
 end

data/lib/llama_bot_rails.rb

@@ -6,5 +6,22 @@ module LlamaBotRails
     def config
       Rails.application.config.llama_bot_rails
     end
+
+    def agent_prompt_path
+      Rails.root.join("config", "llama_bot", "agent_prompt.txt")
+    end
+
+    def agent_prompt_text
+      if File.exist?(agent_prompt_path)
+        File.read(agent_prompt_path)
+      else
+        "You are LlamaBot, a helpful assistant." #Fallback default.
+      end
+    end
+
+    def add_instruction_to_agent_prompt!(new_instruction)
+      FileUtils.mkdir_p(agent_prompt_path.dirname)
+      File.write(agent_prompt_path, "\n#{new_instruction}", mode: 'a')
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llama_bot_rails
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.6
 platform: ruby
 authors:
 - Kody Kendall
@@ -15,7 +15,7 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: '6.0'
     - - "<"
       - !ruby/object:Gem::Version
         version: '9.0'
@@ -25,7 +25,7 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: '6.0'
     - - "<"
       - !ruby/object:Gem::Version
         version: '9.0'
@@ -35,7 +35,7 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: '6.0'
     - - "<"
       - !ruby/object:Gem::Version
         version: '9.0'
@@ -45,7 +45,7 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: '6.0'
     - - "<"
       - !ruby/object:Gem::Version
         version: '9.0'
@@ -122,6 +122,7 @@ files:
 - config/initializers/llama_bot_rails.rb
 - config/routes.rb
 - lib/generators/llama_bot_rails/install/install_generator.rb
+- lib/generators/llama_bot_rails/install/templates/agent_prompt.txt
 - lib/llama_bot_rails.rb
 - lib/llama_bot_rails/agent_state_builder.rb
 - lib/llama_bot_rails/engine.rb