diff --git a/Gemfile.lock b/Gemfile.lock
index d16ac38..78036e8 100644
--- a/Gemfile.lock
+++ b/Gemfile.lock
@@ -2,7 +2,7 @@ PATH
   remote: .
   specs:
     langchainrb_rails (0.1.12)
-      langchainrb (>= 0.7, < 0.17)
+      langchainrb (>= 0.7, < 0.19)
 
 GEM
   remote: https://rubygems.org/
@@ -116,12 +116,11 @@ GEM
     json (2.7.2)
     json-schema (4.3.1)
       addressable (>= 2.8)
-    langchainrb (0.16.0)
+    langchainrb (0.18.0)
       baran (~> 0.1.9)
       json-schema (~> 4)
       matrix
       pragmatic_segmenter (~> 0.3.0)
-      rainbow (~> 3.1.0)
       zeitwerk (~> 2.5)
     language_server-protocol (3.17.0.3)
     lint_roller (1.1.0)
diff --git a/langchainrb_rails.gemspec b/langchainrb_rails.gemspec
index 6101e62..edcb3b2 100644
--- a/langchainrb_rails.gemspec
+++ b/langchainrb_rails.gemspec
@@ -30,7 +30,7 @@ Gem::Specification.new do |spec|
   spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
-  spec.add_dependency "langchainrb", ">= 0.7", "< 0.17"
+  spec.add_dependency "langchainrb", ">= 0.7", "< 0.19"
 
   spec.add_development_dependency "pry-byebug", "~> 3.10.0"
   spec.add_development_dependency "yard", "~> 0.9.34"
diff --git a/lib/langchainrb_overrides/message.rb b/lib/langchainrb_overrides/message.rb
index 6c58cb7..c60bbbf 100644
--- a/lib/langchainrb_overrides/message.rb
+++ b/lib/langchainrb_overrides/message.rb
@@ -1,7 +1,11 @@
+# frozen_string_literal: true
+
 module Langchain
-  module Messages
-    class Base
-      attr_accessor :id
+  class Assistant
+    module Messages
+      class Base
+        attr_accessor :id
+      end
     end
   end
 end
diff --git a/spec/langchainrb_overrides/message_spec.rb b/spec/langchainrb_overrides/message_spec.rb
index ff23cb2..6c0c9ac 100644
--- a/spec/langchainrb_overrides/message_spec.rb
+++ b/spec/langchainrb_overrides/message_spec.rb
@@ -1,6 +1,6 @@
 require "spec_helper"
 
-RSpec.describe Langchain::Messages::Base do
+RSpec.describe Langchain::Assistant::Messages::Base do
   describe "#id" do
     it "allows setting and getting an id" do
       message = described_class.new