Fix the post install instructions and the env var name (#81)

Author: andreibondarev

lib/langchainrb_rails/generators/langchainrb_rails/base_generator.rb

@@ -19,6 +19,15 @@ class BaseGenerator < Rails::Generators::Base
         "openai" => "Langchain::LLM::OpenAI",
         "replicate" => "Langchain::LLM::Replicate"
       }
+
+      def post_install_message
+        say "Please do the following to start Q&A with your #{model_name} records:", :green
+        say "1. Run `bundle install` to install the new gems."
+        say "2. Set an environment variable ENV['#{llm.upcase}_API_KEY'] for your #{llm_class}."
+        say "3. Run `rails db:migrate` to apply the database migrations to enable pgvector and add the embedding column."
+        say "4. In Rails console, run `#{model_name}.embed!` to set the embeddings for all records."
+        say "5. Ask a question in the Rails console, ie: `#{model_name}.ask('[YOUR QUESTION]')`"
+      end
     end
   end
 end

lib/langchainrb_rails/generators/langchainrb_rails/pgvector_generator.rb

@@ -35,15 +35,6 @@ def add_to_gemfile
         gem "ruby-openai"
       end
 
-      def post_install_message
-        say "Please do the following to start Q&A with your #{model_name} records:", :green
-        say "1. Run `bundle install` to install the new gems."
-        say "2. Set `OPENAI_API_KEY` environment variable to your OpenAI API key."
-        say "3. Run `rails db:migrate` to apply the database migrations to enable pgvector and add the embedding column."
-        say "4. In Rails console, run `#{model_name}.embed!` to set the embeddings for all records."
-        say "5. Ask a question in the Rails console, ie: `#{model_name}.ask('[YOUR QUESTION]')`"
-      end
-
       private
 
       # @return [String] Name of the model

lib/langchainrb_rails/generators/langchainrb_rails/templates/pgvector_initializer.rb.tt

@@ -2,6 +2,6 @@
 
 LangchainrbRails.configure do |config|
   config.vectorsearch = Langchain::Vectorsearch::Pgvector.new(
-    llm: <%= llm_class %>.new(api_key: ENV["OPENAI_API_KEY"])
+    llm: <%= llm_class %>.new(api_key: ENV["<%= llm.upcase %>_API_KEY"])
   )
 end