langchainrb_rails 0.1.11 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ac7837afbfe4347fbafa41c1cec9f9c6c14a3efe938e75ae611a2ec986a5fe64
-  data.tar.gz: 55b1a04a984a306d1fdea91c0975b0b4f14a91f0035848032c573f2defdc1b6f
+  metadata.gz: 8f7c8d8bece5767488618f3707e15d6df1582c767d72ecfea63511bbf94f5b28
+  data.tar.gz: 95ab1ee21ea0dd9dfb35051d6b4e95cd19bfebc0cc789dc6132decb63b4c8b39
 SHA512:
-  metadata.gz: 795dd248bf74cfa20e45d2457d43e1833a538a5c399854f41fede624c82fe44d1c7f8e9d7a877776492770910e2431293ab8ab824e447ff2c37aebddc0bef1ac
-  data.tar.gz: ed8a5c720b6420e5ae8bb9a51b51c11909afd32e6e1726a084554609096a9d051d7da59df4bb06f0fb343f18540843c375a6840db685f3774ccb82e926c2314b
+  metadata.gz: 4791c533d1c67ee0f0b4fb49bc6b7cde0c523af86ec9d24d1a121ef7a8050b28cc782be97d8939f69c0b90784920d4df8935bf91715fd261b9d553d222d4c373
+  data.tar.gz: 502c1f1b25a3dc257e537fc4b6a37df74a366efe0f357ec245c509f42faa2e92f54b40b0f8992e66f5c40a2ee4f8c0f5c4fd2321651123dbf3baaf89fc341a6d

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 ## [Unreleased]
 
+## [0.1.12] - 2024-09-20
+- Adding `rails g langchainrb_rails:assistant --llm=...` generator
+- Adding `rails g langchainrb_rails:prompt` generator
+
 ## [0.1.11] - 2024-06-16
 - Add destroy_from_vectorsearch hook
 

data/Gemfile

@@ -9,10 +9,11 @@ gem "rake", "~> 13.2"
 
 gem "rspec", "~> 3.13"
 
-gem "bundler-audit", require: false
 gem "brakeman", require: false
+gem "bundler-audit", require: false
 gem "rubocop", require: false
 
 gem "standardrb"
 
+gem "generator_spec"
 gem "langchainrb"

data/Gemfile.lock

@@ -1,41 +1,41 @@
 PATH
   remote: .
   specs:
-    langchainrb_rails (0.1.11)
-      langchainrb (>= 0.7, < 0.14)
+    langchainrb_rails (0.1.12)
+      langchainrb (>= 0.7, < 0.17)
 
 GEM
   remote: https://rubygems.org/
   specs:
-    actioncable (7.1.3.4)
-      actionpack (= 7.1.3.4)
-      activesupport (= 7.1.3.4)
+    actioncable (7.1.4)
+      actionpack (= 7.1.4)
+      activesupport (= 7.1.4)
       nio4r (~> 2.0)
       websocket-driver (>= 0.6.1)
       zeitwerk (~> 2.6)
-    actionmailbox (7.1.3.4)
-      actionpack (= 7.1.3.4)
-      activejob (= 7.1.3.4)
-      activerecord (= 7.1.3.4)
-      activestorage (= 7.1.3.4)
-      activesupport (= 7.1.3.4)
+    actionmailbox (7.1.4)
+      actionpack (= 7.1.4)
+      activejob (= 7.1.4)
+      activerecord (= 7.1.4)
+      activestorage (= 7.1.4)
+      activesupport (= 7.1.4)
       mail (>= 2.7.1)
       net-imap
       net-pop
       net-smtp
-    actionmailer (7.1.3.4)
-      actionpack (= 7.1.3.4)
-      actionview (= 7.1.3.4)
-      activejob (= 7.1.3.4)
-      activesupport (= 7.1.3.4)
+    actionmailer (7.1.4)
+      actionpack (= 7.1.4)
+      actionview (= 7.1.4)
+      activejob (= 7.1.4)
+      activesupport (= 7.1.4)
       mail (~> 2.5, >= 2.5.4)
       net-imap
       net-pop
       net-smtp
       rails-dom-testing (~> 2.2)
-    actionpack (7.1.3.4)
-      actionview (= 7.1.3.4)
-      activesupport (= 7.1.3.4)
+    actionpack (7.1.4)
+      actionview (= 7.1.4)
+      activesupport (= 7.1.4)
       nokogiri (>= 1.8.5)
       racc
       rack (>= 2.2.4)
@@ -43,35 +43,35 @@ GEM
       rack-test (>= 0.6.3)
       rails-dom-testing (~> 2.2)
       rails-html-sanitizer (~> 1.6)
-    actiontext (7.1.3.4)
-      actionpack (= 7.1.3.4)
-      activerecord (= 7.1.3.4)
-      activestorage (= 7.1.3.4)
-      activesupport (= 7.1.3.4)
+    actiontext (7.1.4)
+      actionpack (= 7.1.4)
+      activerecord (= 7.1.4)
+      activestorage (= 7.1.4)
+      activesupport (= 7.1.4)
       globalid (>= 0.6.0)
       nokogiri (>= 1.8.5)
-    actionview (7.1.3.4)
-      activesupport (= 7.1.3.4)
+    actionview (7.1.4)
+      activesupport (= 7.1.4)
       builder (~> 3.1)
       erubi (~> 1.11)
       rails-dom-testing (~> 2.2)
       rails-html-sanitizer (~> 1.6)
-    activejob (7.1.3.4)
-      activesupport (= 7.1.3.4)
+    activejob (7.1.4)
+      activesupport (= 7.1.4)
       globalid (>= 0.3.6)
-    activemodel (7.1.3.4)
-      activesupport (= 7.1.3.4)
-    activerecord (7.1.3.4)
-      activemodel (= 7.1.3.4)
-      activesupport (= 7.1.3.4)
+    activemodel (7.1.4)
+      activesupport (= 7.1.4)
+    activerecord (7.1.4)
+      activemodel (= 7.1.4)
+      activesupport (= 7.1.4)
       timeout (>= 0.4.0)
-    activestorage (7.1.3.4)
-      actionpack (= 7.1.3.4)
-      activejob (= 7.1.3.4)
-      activerecord (= 7.1.3.4)
-      activesupport (= 7.1.3.4)
+    activestorage (7.1.4)
+      actionpack (= 7.1.4)
+      activejob (= 7.1.4)
+      activerecord (= 7.1.4)
+      activesupport (= 7.1.4)
       marcel (~> 1.0)
-    activesupport (7.1.3.4)
+    activesupport (7.1.4)
       base64
       bigdecimal
       concurrent-ruby (~> 1.0, >= 1.0.2)
@@ -81,48 +81,47 @@ GEM
       minitest (>= 5.1)
       mutex_m
       tzinfo (~> 2.0)
-    addressable (2.8.6)
-      public_suffix (>= 2.0.2, < 6.0)
+    addressable (2.8.7)
+      public_suffix (>= 2.0.2, < 7.0)
     ast (2.4.2)
-    baran (0.1.11)
+    baran (0.1.12)
     base64 (0.2.0)
     bigdecimal (3.1.8)
-    brakeman (6.1.2)
+    brakeman (6.2.1)
       racc
     builder (3.3.0)
-    bundler-audit (0.9.1)
+    bundler-audit (0.9.2)
       bundler (>= 1.2.0, < 3)
       thor (~> 1.0)
     byebug (11.1.3)
     coderay (1.1.3)
-    colorize (1.1.0)
-    concurrent-ruby (1.3.3)
+    concurrent-ruby (1.3.4)
     connection_pool (2.4.1)
     crass (1.0.6)
     date (3.3.4)
     diff-lcs (1.5.1)
     drb (2.2.1)
-    erubi (1.12.0)
+    erubi (1.13.0)
+    generator_spec (0.10.0)
+      activesupport (>= 3.0.0)
+      railties (>= 3.0.0)
     globalid (1.2.1)
       activesupport (>= 6.1)
-    i18n (1.14.5)
+    i18n (1.14.6)
       concurrent-ruby (~> 1.0)
     io-console (0.7.2)
-    irb (1.13.1)
+    irb (1.14.0)
       rdoc (>= 4.0.0)
       reline (>= 0.4.2)
     json (2.7.2)
-    json-schema (4.3.0)
+    json-schema (4.3.1)
       addressable (>= 2.8)
-    langchainrb (0.11.4)
-      activesupport (>= 7.0.8)
+    langchainrb (0.16.0)
       baran (~> 0.1.9)
-      colorize (~> 1.1.0)
       json-schema (~> 4)
       matrix
       pragmatic_segmenter (~> 0.3.0)
-      tiktoken_ruby (~> 0.0.8)
-      to_bool (~> 2.0.0)
+      rainbow (~> 3.1.0)
       zeitwerk (~> 2.5)
     language_server-protocol (3.17.0.3)
     lint_roller (1.1.0)
@@ -139,9 +138,9 @@ GEM
     method_source (1.0.0)
     mini_mime (1.1.5)
     mini_portile2 (2.8.7)
-    minitest (5.23.1)
+    minitest (5.25.1)
     mutex_m (0.2.0)
-    net-imap (0.4.12)
+    net-imap (0.4.16)
       date
       net-protocol
     net-pop (0.1.2)
@@ -151,19 +150,18 @@ GEM
     net-smtp (0.5.0)
       net-protocol
     nio4r (2.7.3)
-    nokogiri (1.16.5)
+    nokogiri (1.16.7)
       mini_portile2 (~> 2.8.2)
       racc (~> 1.4)
-    nokogiri (1.16.5-x86_64-darwin)
+    nokogiri (1.16.7-x86_64-darwin)
       racc (~> 1.4)
-    nokogiri (1.16.5-x86_64-linux)
+    nokogiri (1.16.7-x86_64-linux)
       racc (~> 1.4)
-    parallel (1.24.0)
-    parser (3.3.1.0)
+    parallel (1.26.3)
+    parser (3.3.5.0)
       ast (~> 2.4.1)
       racc
-    pragmatic_segmenter (0.3.23)
-      unicode
+    pragmatic_segmenter (0.3.24)
     pry (0.14.2)
       coderay (~> 1.1)
       method_source (~> 1.0)
@@ -172,9 +170,9 @@ GEM
       pry (>= 0.13, < 0.15)
     psych (5.1.2)
       stringio
-    public_suffix (5.0.5)
-    racc (1.8.0)
-    rack (3.0.11)
+    public_suffix (6.0.1)
+    racc (1.8.1)
+    rack (3.1.7)
     rack-session (2.0.0)
       rack (>= 3.0.0)
     rack-test (2.1.0)
@@ -182,20 +180,20 @@ GEM
     rackup (2.1.0)
       rack (>= 3)
       webrick (~> 1.8)
-    rails (7.1.3.4)
-      actioncable (= 7.1.3.4)
-      actionmailbox (= 7.1.3.4)
-      actionmailer (= 7.1.3.4)
-      actionpack (= 7.1.3.4)
-      actiontext (= 7.1.3.4)
-      actionview (= 7.1.3.4)
-      activejob (= 7.1.3.4)
-      activemodel (= 7.1.3.4)
-      activerecord (= 7.1.3.4)
-      activestorage (= 7.1.3.4)
-      activesupport (= 7.1.3.4)
+    rails (7.1.4)
+      actioncable (= 7.1.4)
+      actionmailbox (= 7.1.4)
+      actionmailer (= 7.1.4)
+      actionpack (= 7.1.4)
+      actiontext (= 7.1.4)
+      actionview (= 7.1.4)
+      activejob (= 7.1.4)
+      activemodel (= 7.1.4)
+      activerecord (= 7.1.4)
+      activestorage (= 7.1.4)
+      activesupport (= 7.1.4)
       bundler (>= 1.15.0)
-      railties (= 7.1.3.4)
+      railties (= 7.1.4)
     rails-dom-testing (2.2.0)
       activesupport (>= 5.0.0)
       minitest
@@ -203,9 +201,9 @@ GEM
     rails-html-sanitizer (1.6.0)
       loofah (~> 2.21)
       nokogiri (~> 1.14)
-    railties (7.1.3.4)
-      actionpack (= 7.1.3.4)
-      activesupport (= 7.1.3.4)
+    railties (7.1.4)
+      actionpack (= 7.1.4)
+      activesupport (= 7.1.4)
       irb
       rackup (>= 1.0.0)
       rake (>= 12.2)
@@ -213,14 +211,11 @@ GEM
       zeitwerk (~> 2.6)
     rainbow (3.1.1)
     rake (13.2.1)
-    rb_sys (0.9.96)
     rdoc (6.7.0)
       psych (>= 4.0.0)
     regexp_parser (2.9.2)
-    reline (0.5.8)
+    reline (0.5.10)
       io-console (~> 0.5)
-    rexml (3.2.8)
-      strscan (>= 3.0.9)
     rspec (3.13.0)
       rspec-core (~> 3.13.0)
       rspec-expectations (~> 3.13.0)
@@ -234,18 +229,17 @@ GEM
       diff-lcs (>= 1.2.0, < 2.0)
       rspec-support (~> 3.13.0)
     rspec-support (3.13.0)
-    rubocop (1.64.0)
+    rubocop (1.66.1)
       json (~> 2.3)
       language_server-protocol (>= 3.17.0)
       parallel (~> 1.10)
       parser (>= 3.3.0.2)
       rainbow (>= 2.2.2, < 4.0)
-      regexp_parser (>= 1.8, < 3.0)
-      rexml (>= 3.2.5, < 4.0)
-      rubocop-ast (>= 1.31.1, < 2.0)
+      regexp_parser (>= 2.4, < 3.0)
+      rubocop-ast (>= 1.32.2, < 2.0)
       ruby-progressbar (~> 1.7)
       unicode-display_width (>= 2.4.0, < 3.0)
-    rubocop-ast (1.31.3)
+    rubocop-ast (1.32.3)
       parser (>= 3.3.1.0)
     rubocop-performance (1.20.1)
       rubocop (>= 1.48.1, < 2.0)
@@ -265,25 +259,18 @@ GEM
       rubocop-performance (~> 1.20.1)
     standardrb (1.0.1)
       standard
-    stringio (3.1.0)
-    strscan (3.1.0)
-    thor (1.3.1)
-    tiktoken_ruby (0.0.8)
-      rb_sys (>= 0.9.86)
-    tiktoken_ruby (0.0.8-x86_64-darwin)
-    tiktoken_ruby (0.0.8-x86_64-linux)
+    stringio (3.1.1)
+    thor (1.3.2)
     timeout (0.4.1)
-    to_bool (2.0.0)
     tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
-    unicode (0.4.4.4)
     unicode-display_width (2.5.0)
     webrick (1.8.1)
     websocket-driver (0.7.6)
       websocket-extensions (>= 0.1.0)
     websocket-extensions (0.1.5)
-    yard (0.9.36)
-    zeitwerk (2.6.15)
+    yard (0.9.37)
+    zeitwerk (2.6.18)
 
 PLATFORMS
   ruby
@@ -294,6 +281,7 @@ PLATFORMS
 DEPENDENCIES
   brakeman
   bundler-audit
+  generator_spec
   langchainrb
   langchainrb_rails!
   pry-byebug (~> 3.10.0)

data/README.md

@@ -129,3 +129,25 @@ Pinecone Generator does the following:
 2. Adds necessary code to the ActiveRecord model to enable vectorsearch
 3. Adds `pinecone` gem to the Gemfile
 
+### Prompt Generator - adds prompt templating capabilities to your ActiveRecord model
+```bash
+rails generate langchainrb_rails:prompt
+```
+
+This generator adds the following files to your Rails project:
+1. An ActiveRecord `Prompt` model at `app/models/prompt.rb`
+2. A rails migration to create the `prompts` table
+
+You can then use the `Prompt` model to create and manage prompts for your model.
+
+Example usage:
+```ruby
+prompt = Prompt.create!(template: "Tell me a {adjective} joke about {subject}.")
+prompt.render(adjective: "funny", subject: "elephants")
+# => "Tell me a funny joke about elephants."
+```
+
+### Assistant Generator - adds assistant capabilities to your ActiveRecord model
+```bash
+rails generate langchainrb_rails:assistant
+```

data/lib/langchainrb_overrides/assistant.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require "active_record"
+
+module Langchain
+  class Assistant
+    attr_accessor :id
+
+    alias_method :original_initialize, :initialize
+
+    def initialize(id: nil, **kwargs) # rubocop:disable Style/ArgumentsForwarding
+      @id = id
+      original_initialize(**kwargs) # rubocop:disable Style/ArgumentsForwarding
+    end
+
+    def save
+      ::ActiveRecord::Base.transaction do
+        ar_assistant = if id
+          self.class.find_assistant(id)
+        else
+          ::Assistant.new
+        end
+
+        ar_assistant.update!(
+          instructions: instructions,
+          tool_choice: tool_choice,
+          tools: tools.map(&:class).map(&:name)
+        )
+
+        messages.each do |message|
+          ar_message = ar_assistant.messages.find_or_initialize_by(id: message.id)
+          ar_message.update!(
+            role: message.role,
+            content: message.content,
+            tool_calls: message.tool_calls,
+            tool_call_id: message.tool_call_id
+          )
+          message.id = ar_message.id
+        end
+
+        @id = ar_assistant.id
+        true
+      end
+    end
+
+    class << self
+      def find_assistant(id)
+        ::Assistant.find(id)
+      end
+
+      def load(id)
+        ar_assistant = find_assistant(id)
+
+        tools = ar_assistant.tools.map { |tool_name| Object.const_get(tool_name).new }
+
+        assistant = Langchain::Assistant.new(
+          id: ar_assistant.id,
+          llm: ar_assistant.llm,
+          tools: tools,
+          instructions: ar_assistant.instructions,
+          tool_choice: ar_assistant.tool_choice
+        )
+
+        ar_assistant.messages.each do |ar_message|
+          messages = assistant.add_message(
+            role: ar_message.role,
+            content: ar_message.content,
+            tool_calls: ar_message.tool_calls,
+            tool_call_id: ar_message.tool_call_id
+          )
+          messages.last.id = ar_message.id
+        end
+
+        assistant
+      end
+    end
+  end
+end

data/lib/langchainrb_overrides/message.rb

@@ -0,0 +1,7 @@
+module Langchain
+  module Messages
+    class Base
+      attr_accessor :id
+    end
+  end
+end

data/lib/langchainrb_rails/generators/langchainrb_rails/assistant_generator.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+require "rails/generators/active_record"
+
+module LangchainrbRails
+  module Generators
+    #
+    # Usage:
+    #     rails generate langchainrb_rails:assistant --llm=openai
+    #
+    class AssistantGenerator < Rails::Generators::Base
+      include ::ActiveRecord::Generators::Migration
+
+      # TODO: Move constant this to a shared place
+      LLMS = {
+        "anthropic" => "Langchain::LLM::Anthropic",
+        "cohere" => "Langchain::LLM::Cohere",
+        "google_palm" => "Langchain::LLM::GooglePalm",
+        "google_gemini" => "Langchain::LLM::GoogleGemini",
+        "google_vertex_ai" => "Langchain::LLM::GoogleVertexAI",
+        "hugging_face" => "Langchain::LLM::HuggingFace",
+        "llama_cpp" => "Langchain::LLM::LlamaCpp",
+        "mistral_ai" => "Langchain::LLM::MistralAI",
+        "ollama" => "Langchain::LLM::Ollama",
+        "openai" => "Langchain::LLM::OpenAI",
+        "replicate" => "Langchain::LLM::Replicate"
+      }.freeze
+
+      class_option :llm,
+        type: :string,
+        required: true,
+        default: "openai",
+        desc: "LLM provider that will be used to generate embeddings and completions",
+        enum: LLMS.keys
+
+      desc "This generator adds Assistant and Message models and tables to your Rails app"
+      source_root File.join(__dir__, "templates")
+
+      def copy_migration
+        migration_template "assistant/migrations/create_assistants.rb", "db/migrate/create_assistants.rb", migration_version: migration_version
+        migration_template "assistant/migrations/create_messages.rb", "db/migrate/create_messages.rb", migration_version: migration_version
+      end
+
+      def create_model_file
+        template "assistant/models/assistant.rb", "app/models/assistant.rb"
+        template "assistant/models/message.rb", "app/models/message.rb"
+      end
+
+      def migration_version
+        "[#{::ActiveRecord::VERSION::MAJOR}.#{::ActiveRecord::VERSION::MINOR}]"
+      end
+
+      # TODO: Depending on the LLM provider, we may need to add additional gems
+      # def add_to_gemfile
+      # end
+
+      private
+
+      # @return [String] LLM provider to use
+      def llm
+        options["llm"]
+      end
+
+      # @return [Langchain::LLM::*] LLM class
+      def llm_class
+        Langchain::LLM.const_get(LLMS[llm])
+      end
+    end
+  end
+end

data/lib/langchainrb_rails/generators/langchainrb_rails/base_generator.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "rails/generators"
 require "rails/generators/active_record"
 
@@ -6,19 +8,38 @@ module LangchainrbRails
     class BaseGenerator < Rails::Generators::Base
       include ::ActiveRecord::Generators::Migration
 
-      class_option :model, type: :string, required: true, desc: "ActiveRecord Model to add vectorsearch to", aliases: "-m"
-      class_option :llm, type: :string, required: true, desc: "LLM provider that will be used to generate embeddings and completions"
-
       # Available LLM providers to be passed in as --llm option
       LLMS = {
+        "anthropic" => "Langchain::LLM::Anthropic",
         "cohere" => "Langchain::LLM::Cohere",
         "google_palm" => "Langchain::LLM::GooglePalm",
+        "google_gemini" => "Langchain::LLM::GoogleGemini",
+        "google_vertex_ai" => "Langchain::LLM::GoogleVertexAI",
         "hugging_face" => "Langchain::LLM::HuggingFace",
         "llama_cpp" => "Langchain::LLM::LlamaCpp",
+        "mistral_ai" => "Langchain::LLM::MistralAI",
         "ollama" => "Langchain::LLM::Ollama",
         "openai" => "Langchain::LLM::OpenAI",
         "replicate" => "Langchain::LLM::Replicate"
-      }
+      }.freeze
+
+      class_option :model,
+        type: :string,
+        required: true,
+        aliases: "-m",
+        desc: "ActiveRecord Model to add vectorsearch to"
+
+      class_option :llm,
+        type: :string,
+        required: true,
+        default: "openai",
+        desc: "LLM provider that will be used to generate embeddings and completions",
+        enum: LLMS.keys
+
+      # Run bundle install after running the generator
+      def after_generate
+        run "bundle install"
+      end
 
       def post_install_message
         say "Please do the following to start Q&A with your #{model_name} records:", :green

data/lib/langchainrb_rails/generators/langchainrb_rails/chroma_generator.rb

@@ -28,9 +28,8 @@ module LangchainrbRails
       end
 
       # Adds `chroma-db` gem to the Gemfile
-      # TODO: Can we automatically run `bundle install`?
       def add_to_gemfile
-        gem "chroma-db", version: "~> 0.6.0"
+        gem "chroma-db"
       end
 
       private

data/lib/langchainrb_rails/generators/langchainrb_rails/pgvector_generator.rb

@@ -24,7 +24,7 @@ module LangchainrbRails
       end
 
       def add_to_model
-        inject_into_class "app/models/#{model_name.downcase}.rb", model_name do
+        inject_into_class "app/models/#{model_name.underscore}.rb", model_name do
           "  vectorsearch\n\n  after_save :upsert_to_vectorsearch\n\n"
         end
       end
@@ -44,7 +44,7 @@ module LangchainrbRails
 
       # @return [String] Table name of the model
       def table_name
-        model_name.downcase.pluralize
+        model_name.tableize
       end
 
       # @return [String] LLM provider to use

data/lib/langchainrb_rails/generators/langchainrb_rails/pinecone_generator.rb

@@ -28,9 +28,8 @@ module LangchainrbRails
       end
 
       # Adds `pinecone` gem to the Gemfile
-      # TODO: Can we automatically run `bundle install`?
       def add_to_gemfile
-        gem "pinecone", version: "~> 0.1.6"
+        gem "pinecone"
       end
 
       private

data/lib/langchainrb_rails/generators/langchainrb_rails/prompt_generator.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+require "rails/generators/active_record"
+
+module LangchainrbRails
+  module Generators
+    # Adds prompt templating capabilities to your ActiveRecord model
+    #
+    # Usage:
+    #     rails generate langchainrb_rails:prompt
+    #
+    class PromptGenerator < Rails::Generators::Base
+      include ::ActiveRecord::Generators::Migration
+
+      source_root File.join(__dir__, "templates")
+
+      def create_model_file
+        template "prompt_model.rb", "app/models/prompt.rb"
+      end
+
+      def copy_migration
+        migration_template "create_prompts.rb", "db/migrate/create_prompts.rb", migration_version: migration_version
+      end
+
+      def migration_version
+        "[#{::ActiveRecord::VERSION::MAJOR}.#{::ActiveRecord::VERSION::MINOR}]"
+      end
+    end
+  end
+end

data/lib/langchainrb_rails/generators/langchainrb_rails/qdrant_generator.rb

@@ -28,7 +28,6 @@ module LangchainrbRails
       end
 
       # Adds `qdrant-ruby` gem to the Gemfile
-      # TODO: Can we automatically run `bundle install`?
       def add_to_gemfile
         gem "qdrant-ruby"
       end

data/lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/migrations/create_assistants.rb.tt

@@ -0,0 +1,10 @@
+class <%= migration_class_name %> < ActiveRecord::Migration<%= migration_version %>
+  def change
+    create_table :assistants do |t|
+      t.string :instructions
+      t.string :tool_choice
+      t.json :tools
+      t.timestamps
+    end
+  end
+end

data/lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/migrations/create_messages.rb.tt

@@ -0,0 +1,12 @@
+class <%= migration_class_name %> < ActiveRecord::Migration<%= migration_version %>
+  def change
+    create_table :messages do |t|
+      t.references :assistant, foreign_key: true
+      t.string :role
+      t.text :content
+      t.json :tool_calls
+      t.string :tool_call_id
+      t.timestamps
+    end
+  end
+end

data/lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/models/assistant.rb.tt

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+class Assistant < ActiveRecord::Base
+  has_many :messages
+
+  def llm
+    <%= llm_class %>.new(api_key: ENV["<%= llm.upcase %>_API_KEY"])
+  end
+end

data/lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/models/message.rb.tt

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+class Message < ActiveRecord::Base
+  belongs_to :assistant
+end

data/lib/langchainrb_rails/generators/langchainrb_rails/templates/create_prompts.rb.tt

@@ -0,0 +1,9 @@
+class CreatePrompts < ActiveRecord::Migration<%= migration_version %>
+  def change
+    create_table :prompts do |t|
+      t.text :template, null: false
+
+      t.timestamps
+    end
+  end
+end

data/lib/langchainrb_rails/generators/langchainrb_rails/templates/prompt_model.rb.tt

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+class Prompt < ApplicationRecord
+  include LangchainrbRails::Prompting
+end

data/lib/langchainrb_rails/prompting.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module LangchainrbRails
+  module Prompting
+    extend ActiveSupport::Concern
+
+    included do
+      validates :template, presence: true
+
+      def render(variables = {})
+        template.gsub(/\{(\w+)\}/) do |match|
+          variable = ::Regexp.last_match(1).to_sym
+          variables.fetch(variable, match)
+        end
+      end
+
+      def template_variables
+        template.scan(/\{(\w+)\}/).flatten.uniq
+      end
+    end
+  end
+end

data/lib/langchainrb_rails/railtie.rb

@@ -9,10 +9,12 @@ module LangchainrbRails
     end
 
     generators do
+      require_relative "generators/langchainrb_rails/assistant_generator"
       require_relative "generators/langchainrb_rails/chroma_generator"
       require_relative "generators/langchainrb_rails/pinecone_generator"
       require_relative "generators/langchainrb_rails/pgvector_generator"
       require_relative "generators/langchainrb_rails/qdrant_generator"
+      require_relative "generators/langchainrb_rails/prompt_generator"
     end
   end
 end

data/lib/langchainrb_rails/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LangchainrbRails
-  VERSION = "0.1.11"
+  VERSION = "0.1.12"
 end

data/lib/langchainrb_rails.rb

@@ -3,10 +3,15 @@
 require "forwardable"
 require "langchain"
 require "rails"
-require_relative "langchainrb_rails/version"
-require "langchainrb_rails/railtie"
+
 require "langchainrb_rails/config"
+require "langchainrb_rails/prompting"
+require "langchainrb_rails/railtie"
+require "langchainrb_rails/version"
+
 require_relative "langchainrb_overrides/vectorsearch/pgvector"
+require_relative "langchainrb_overrides/assistant"
+require_relative "langchainrb_overrides/message"
 
 module LangchainrbRails
   class Error < StandardError; end
@@ -17,9 +22,11 @@ module LangchainrbRails
 
   module Generators
     autoload :BaseGenerator, "langchainrb_rails/generators/langchainrb_rails/base_generator"
+    autoload :AssistantGenerator, "langchainrb_rails/generators/langchainrb_rails/assistant_generator"
     autoload :ChromaGenerator, "langchainrb_rails/generators/langchainrb_rails/chroma_generator"
     autoload :PgvectorGenerator, "langchainrb_rails/generators/langchainrb_rails/pgvector_generator"
     autoload :QdrantGenerator, "langchainrb_rails/generators/langchainrb_rails/qdrant_generator"
+    autoload :PromptGenerator, "langchainrb_rails/generators/langchainrb_rails/prompt_generator"
   end
 
   class << self

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb_rails
 version: !ruby/object:Gem::Version
-  version: 0.1.11
+  version: 0.1.12
 platform: ruby
 authors:
 - Andrei Bondarev
@@ -19,7 +19,7 @@ dependencies:
         version: '0.7'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '0.14'
+        version: '0.17'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -29,7 +29,7 @@ dependencies:
         version: '0.7'
     - - "<"
       - !ruby/object:Gem::Version
-        version: '0.14'
+        version: '0.17'
 - !ruby/object:Gem::Dependency
   name: pry-byebug
   requirement: !ruby/object:Gem::Requirement
@@ -72,6 +72,20 @@ dependencies:
     - - ">"
       - !ruby/object:Gem::Version
         version: 6.0.0
+- !ruby/object:Gem::Dependency
+  name: generator_spec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Rails wrapper for langchainrb gem
 email:
 - andrei.bondarev13@gmail.com
@@ -88,21 +102,32 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- lib/langchainrb_overrides/assistant.rb
+- lib/langchainrb_overrides/message.rb
 - lib/langchainrb_overrides/vectorsearch/pgvector.rb
 - lib/langchainrb_rails.rb
 - lib/langchainrb_rails/active_record/hooks.rb
 - lib/langchainrb_rails/config.rb
+- lib/langchainrb_rails/generators/langchainrb_rails/assistant_generator.rb
 - lib/langchainrb_rails/generators/langchainrb_rails/base_generator.rb
 - lib/langchainrb_rails/generators/langchainrb_rails/chroma_generator.rb
 - lib/langchainrb_rails/generators/langchainrb_rails/pgvector_generator.rb
 - lib/langchainrb_rails/generators/langchainrb_rails/pinecone_generator.rb
+- lib/langchainrb_rails/generators/langchainrb_rails/prompt_generator.rb
 - lib/langchainrb_rails/generators/langchainrb_rails/qdrant_generator.rb
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/add_vector_column_template.rb.tt
+- lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/migrations/create_assistants.rb.tt
+- lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/migrations/create_messages.rb.tt
+- lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/models/assistant.rb.tt
+- lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/models/message.rb.tt
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/chroma_initializer.rb.tt
+- lib/langchainrb_rails/generators/langchainrb_rails/templates/create_prompts.rb.tt
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/enable_vector_extension_template.rb.tt
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/pgvector_initializer.rb.tt
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/pinecone_initializer.rb.tt
+- lib/langchainrb_rails/generators/langchainrb_rails/templates/prompt_model.rb.tt
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/qdrant_initializer.rb.tt
+- lib/langchainrb_rails/prompting.rb
 - lib/langchainrb_rails/railtie.rb
 - lib/langchainrb_rails/version.rb
 - sig/langchainrb_rails.rbs