llama_bot_rails 0.1.8 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8a6895de16432a4a2200c1623a25c68c09897b2a6fb3bf55977a2824f3578310
-  data.tar.gz: c39ee31cf9fa2166de1d48434ab3228a6db960328a6565826242e57e5a22a678
+  metadata.gz: a76c323b0d08220d2952ce3c1c3962dd203e645556123306fd9beb58c6ae17e6
+  data.tar.gz: 9e7c143edb324d49097ddabad7006635027ea6150cb07407702110546f0503c0
 SHA512:
-  metadata.gz: 9711a2a8cfe8e485210b84314860eb098113de321ccfd1a93be15608240312d45ff54d852b965eb60f0d84b530620e856d837d28dc7a8b8a3f958e470859b6a2
-  data.tar.gz: e175b5d7df84b42fff07dec9a0daed0384bf77891fc42611ab83ba087131b879a97aec66b77d154ccd9c5f89ed3db2df3e53052d0a937f5e80c8b84cc778c34e
+  metadata.gz: 232cd0a6410caffa47b25d0fd0e62a69ba9fdfdc779f7adc9ded2cab80879544dffc1de5a4a65e1dabfeadad6b4c0a5ff410d26d71e0ed44101ec9ab00ef0af2
+  data.tar.gz: 730c59884b06bbb671955a2d128c31c6e6b85c24fe6b70120bb7d67af5249d88696eb7069d99689598ed35dc2aaa422626111fe8d3f6b154d2c68dde3d72cda6

data/app/channels/llama_bot_rails/chat_channel.rb

@@ -133,10 +133,9 @@ module LlamaBotRails
         # Forward the processed data to the LlamaBot Backend Socket
         message = data["message"]
 
-        # 1. Instantiate the builder
         builder = state_builder_class.new(
-          params: { message: data["message"] },
-          context: { thread_id: data["thread_id"], api_token: @api_token }
+          params: data,
+          context: { api_token: @api_token }
         )
 
         # 2. Construct the LangGraph-ready state
@@ -171,7 +170,27 @@ module LlamaBotRails
     private
 
     def state_builder_class
-      LlamaBotRails.config.state_builder_class.constantize
+      builder_class_name = LlamaBotRails.config.state_builder_class || 'LlamaBotRails::AgentStateBuilder'
+
+      begin
+        builder_class_name.constantize
+      rescue NameError => e
+        # If it's not the default class, try to manually load from app/llama_bot
+        if builder_class_name != 'LlamaBotRails::AgentStateBuilder'
+          llama_bot_file = Rails.root.join("app", "llama_bot", "agent_state_builder.rb")
+          if llama_bot_file.exist?
+            Rails.logger.info "[LlamaBot] Autoload failed, attempting to manually load #{llama_bot_file}"
+            begin
+              load llama_bot_file.to_s
+              return builder_class_name.constantize
+            rescue => load_error
+              Rails.logger.error "[LlamaBot] Manual load failed: #{load_error.message}"
+            end
+          end
+        end
+        
+        raise NameError, "Could not load state builder class '#{builder_class_name}'. Make sure it's defined in app/llama_bot/agent_state_builder.rb or is available in your autoload paths. Original error: #{e.message}"
+      end
     end
 
     def setup_external_websocket(connection_id)

data/app/controllers/llama_bot_rails/agent_controller.rb

@@ -85,8 +85,8 @@ module LlamaBotRails
 
             # 1. Instantiate the builder
             builder = state_builder_class.new(
-                params: { message: params[:message] },
-                context: { thread_id: params[:thread_id], api_token: @api_token }
+                params: params,
+                context: { api_token: @api_token }
             )
     
             # 2. Construct the LangGraph-ready state

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

@@ -14,9 +14,15 @@ module LlamaBotRails
           end
         end
 
-        def create_config_file
-            empty_directory "config/llama_bot"
-            copy_file "agent_prompt.txt", "config/llama_bot/agent_prompt.txt"
+        def create_agent_prompt
+            empty_directory "app/llama_bot/prompts"
+            copy_file "agent_prompt.txt", "app/llama_bot/prompts/agent_prompt.txt"
+        end
+
+        def create_agent_state_builder
+            empty_directory "app/llama_bot"
+            template "agent_state_builder.rb.erb", "app/llama_bot/agent_state_builder.rb"
+            say_status("created", "app/llama_bot/agent_state_builder.rb", :green)
         end
   
         def mount_engine
@@ -46,9 +52,17 @@ module LlamaBotRails
         def create_initializer
           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.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?
+
+              # ------------------------------------------------------------------------
+              # Custom State Builder
+              # ------------------------------------------------------------------------
+              # The gem uses `LlamaBotRails::AgentStateBuilder` by default.
+              # Uncomment this line to use the builder in app/llama_bot/
+              #
+              # config.llama_bot_rails.state_builder_class = "#{app_name}::AgentStateBuilder"
             end
           RUBY
         end
@@ -56,6 +70,12 @@ module LlamaBotRails
         def finish
           say "\n✅ LlamaBotRails installed! Visit http://localhost:3000/llama_bot/agent/chat\n", :green
         end
+
+        private
+
+        def app_name
+          Rails.application.class.module_parent_name
+        end
       end
     end
   end

data/lib/generators/llama_bot_rails/install/templates/agent_state_builder.rb.erb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+#
+# Customize the params sent to your LangGraph agent here.
+# Uncomment the line in the initializer to activate this builder.
+module <%= app_name %>
+  class AgentStateBuilder
+    def initialize(params:, context:)
+      @params = params
+      @context = context
+    end
+
+    def build
+      {
+        message: @params[:message], # Rails param from JS/chat UI. This is the user's message to the agent.
+        thread_id: @context[:thread_id], # This is the thread id for the agent. It is used to track the conversation history.
+        api_token: @context[:api_token], # This is an authenticated API token for the agent, so that it can authenticate with us. (It may need access to resources on our Rails app, such as the Rails Console.)
+        agent_prompt: LlamaBotRails.agent_prompt_text, # System prompt instructions for the agent. Can be customized in app/llama_bot/prompts/agent_prompt.txt
+        agent_name: "llamabot" # This routes to the appropriate LangGraph agent as defined in LlamaBot/langgraph.json, and enables us to access different agents on our LlamaBot server.
+      }
+    end
+  end
+end 

data/lib/llama_bot_rails/agent_state_builder.rb

@@ -13,9 +13,9 @@ module LlamaBotRails
       def build 
         {
           message: @params[:message], # Rails param from JS/chat UI. This is the user's message to the agent.
-          thread_id: @context[:thread_id], # This is the thread id for the agent. It is used to track the conversation history.
+          thread_id: @params[:thread_id], # This is the thread id for the agent. It is used to track the conversation history.
           api_token: @context[:api_token], # This is an authenticated API token for the agent, so that it can authenticate with us. (It may need access to resources on our Rails app, such as the Rails Console.)
-          agent_prompt: LlamaBotRails.agent_prompt_text, # System prompt instructions for the agent. Can be customized in config/agent_prompt.txt
+          agent_prompt: LlamaBotRails.agent_prompt_text, # System prompt instructions for the agent. Can be customized in app/llama_bot/prompts/agent_prompt.txt
           agent_name: "llamabot" #This routes to the appropriate LangGraph agent as defined in LlamaBot/langgraph.json, and enables us to access different agents on our LlamaBot server.
         }
       end

data/lib/llama_bot_rails/engine.rb

@@ -20,7 +20,7 @@ module LlamaBotRails
     end
 
     initializer "llama_bot_rails.defaults" do |app|
-      app.config.llama_bot_rails.state_builder_class = "LlamaBotRails::AgentStateBuilder"
+      app.config.llama_bot_rails.state_builder_class ||= "LlamaBotRails::AgentStateBuilder"
     end
   end
 end

data/lib/llama_bot_rails/version.rb

@@ -1,3 +1,3 @@
 module LlamaBotRails
-  VERSION = "0.1.8"
+  VERSION = "0.1.10"
 end

data/lib/llama_bot_rails.rb

@@ -1,6 +1,9 @@
 require "llama_bot_rails/version"
 require "llama_bot_rails/engine"
+
+# require "llama_bot_rails/railtie" # We don't need this, as we're loading the LlamaBot path directly in the engine.
 require "llama_bot_rails/llama_bot"
+require "llama_bot_rails/agent_state_builder"
 
 module LlamaBotRails
   class << self
@@ -9,7 +12,7 @@ module LlamaBotRails
     end
 
     def agent_prompt_path
-      Rails.root.join("config", "llama_bot", "agent_prompt.txt")
+      Rails.root.join("app", "llama_bot", "prompts", "agent_prompt.txt")
     end
 
     def agent_prompt_text

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llama_bot_rails
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.10
 platform: ruby
 authors:
 - Kody Kendall
@@ -124,6 +124,7 @@ files:
 - config/routes.rb
 - lib/generators/llama_bot_rails/install/install_generator.rb
 - lib/generators/llama_bot_rails/install/templates/agent_prompt.txt
+- lib/generators/llama_bot_rails/install/templates/agent_state_builder.rb.erb
 - lib/llama_bot_rails.rb
 - lib/llama_bot_rails/agent_state_builder.rb
 - lib/llama_bot_rails/engine.rb