fix that table name and file name are correctly named (#107)

* fix that table name and file name are correctly underscored and tableized

* standardrb fixes

Author: sergiobayona

langchainrb_rails.gemspec

@@ -35,4 +35,5 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "pry-byebug", "~> 3.10.0"
   spec.add_development_dependency "yard", "~> 0.9.34"
   spec.add_development_dependency "rails", "> 6.0.0"
+  spec.add_development_dependency "generator_spec"
 end

lib/langchainrb_rails/generators/langchainrb_rails/pgvector_generator.rb

@@ -24,7 +24,7 @@ def migration_version
       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 @@ def model_name
 
       # @return [String] Table name of the model
       def table_name
-        model_name.downcase.pluralize
+        model_name.tableize
       end
 
       # @return [String] LLM provider to use

spec/langchainrb_rails/generators/langchainrb_rails/pgvector_generator_spec.rb

@@ -0,0 +1,47 @@
+require "spec_helper"
+
+TMP_ROOT = Pathname.new(Dir.mktmpdir("tmp"))
+
+RSpec.describe LangchainrbRails::Generators::PgvectorGenerator, type: :generator do
+  destination TMP_ROOT
+  arguments %w[--model=SupportArticle --llm=openai]
+
+  before(:all) do
+    prepare_destination
+    model_file = TMP_ROOT.join("app/models/support_article.rb")
+    gemfile = TMP_ROOT.join("Gemfile")
+    write_file(model_file, "class SupportArticle < ApplicationRecord\nend")
+    write_file(gemfile, "")
+    run_generator
+  end
+
+  after(:all) { delete_directory(TMP_ROOT) }
+
+  it "creates the initializer file" do
+    assert_file "config/initializers/langchainrb_rails.rb"
+  end
+
+  it "creates the vector extension migration" do
+    assert_migration "db/migrate/enable_vector_extension.rb"
+  end
+
+  it "adds the embedding column to the model" do
+    assert_migration "db/migrate/add_vector_column_to_support_articles.rb" do |migration|
+      assert_match(/add_column :support_articles, :embedding, :vector/, migration)
+    end
+  end
+
+  it "adds the vectorsearch module to the model" do
+    assert_file "app/models/support_article.rb" do |model|
+      assert_match(/vectorsearch/, model)
+      assert_match(/after_save :upsert_to_vectorsearch/, model)
+    end
+  end
+
+  it "adds the necessary gems to the Gemfile" do
+    assert_file "Gemfile" do |gemfile|
+      assert_match(/gem "neighbor"/, gemfile)
+      assert_match(/gem "ruby-openai"/, gemfile)
+    end
+  end
+end

spec/langchainrb_rails/helpers.rb

@@ -0,0 +1,12 @@
+module Helpers
+  module FileSystem
+    def write_file(file_name, contents)
+      FileUtils.mkdir_p(File.dirname(file_name)) unless File.directory?(File.dirname(file_name))
+      ::File.write(file_name.to_s, contents)
+    end
+
+    def delete_directory(name)
+      FileUtils.rm_rf(name)
+    end
+  end
+end

spec/spec_helper.rb

@@ -2,6 +2,10 @@
 
 require "rails"
 require "langchainrb_rails"
+require "rails/generators"
+require "rails/generators/test_case"
+require "generator_spec"
+require "langchainrb_rails/helpers"
 
 RSpec.configure do |config|
   # Enable flags like --only-failures and --next-failure
@@ -13,4 +17,6 @@
   config.expect_with :rspec do |c|
     c.syntax = :expect
   end
+
+  config.include Helpers::FileSystem
 end