Sha256: bc230403ae3dbd9922f2f95b2f2b5bdaebe1da12a9392a3f1dc667d9d06b559a
Contents?: true
Size: 1.62 KB
Versions: 8
Compression:
Stored size: 1.62 KB
Contents
# frozen_string_literal: true
module LangchainrbRails
module Generators
#
# ChromaGenerator does the following:
# 1. Creates the `langchainrb_rails.rb` initializer file
# 2. Adds necessary code to the ActiveRecord model to enable vectorsearch
# 3. Adds `chroma-db` gem to the Gemfile
#
# Usage:
# rails generate langchainrb_rails:chrome --model=Product --llm=openai
#
class ChromaGenerator < LangchainrbRails::Generators::BaseGenerator
desc "This generator adds Chroma vectorsearch integration to your ActiveRecord model"
source_root File.join(__dir__, "templates")
# Creates the `langchainrb_rails.rb` initializer file
def create_initializer_file
template "chroma_initializer.rb", "config/initializers/langchainrb_rails.rb"
end
# Adds `vectorsearch` class method to the model and `after_save` callback that calls `upsert_to_vectorsearch()`
def add_to_model
inject_into_class "app/models/#{model_name.downcase}.rb", model_name do
" vectorsearch\n\n after_save :upsert_to_vectorsearch\n\n"
end
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"
end
private
# @return [String] Name of the model
def model_name
options["model"]
end
# @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
Version data entries
8 entries across 8 versions & 1 rubygems