Use the parent generator class

Author: andreibondarev

langchainrb_rails.gemspec

@@ -30,8 +30,7 @@ Gem::Specification.new do |spec|
   spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
-  # TODO - add back ... loading locally
-  #spec.add_dependency "langchainrb", "~> 0.7.0"
+  spec.add_dependency "langchainrb", "~> 0.7.0"
 
   spec.add_development_dependency "pry-byebug", "~> 3.10.0"
   spec.add_development_dependency "yard", "~> 0.9.34"

lib/langchainrb_rails/generators/langchainrb_rails/pgvector_generator.rb

@@ -1,30 +1,15 @@
-require "rails/generators/active_record"
+# frozen_string_literal: true
 
 module LangchainrbRails
   module Generators
     #
     # Usage:
     #     rails g langchain:pgvector -model=Product -llm=openai
     #
-    class PgvectorGenerator < Rails::Generators::Base
+    class PgvectorGenerator < LangchainrbRails::Generators::BaseGenerator
       desc "This generator adds Pgvector vectorsearch integration to your ActiveRecord model"
-
-      include ::ActiveRecord::Generators::Migration
       source_root File.join(__dir__, "templates")
 
-      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"
-
-      LLMS = {
-        "cohere" => "Langchain::LLM::Cohere",
-        "google_palm" => "Langchain::LLM::GooglePalm",
-        "hugging_face" => "Langchain::LLM::HuggingFace",
-        "llama_cpp" => "Langchain::LLM::LlamaCpp",
-        "ollama" => "Langchain::LLM::Ollama",
-        "openai" => "Langchain::LLM::OpenAI",
-        "replicate" => "Langchain::LLM::Replicate"
-      }
-
       def copy_migration
         migration_template "enable_vector_extension_template.rb", "db/migrate/enable_vector_extension.rb", migration_version: migration_version
         migration_template "add_vector_column_template.rb", "db/migrate/add_vector_column_to_#{table_name}.rb", migration_version: migration_version