Add Qdrant generator (#21)

Author: andreibondarev

README.md

@@ -115,6 +115,11 @@ rails generate langchainrb_rails:pgvector --model=Product --llm=openai
 rails generate langchainrb_rails:pinecone --model=Product --llm=openai
 ```
 
+### Qdrant Generator - adds vectorsearch to your ActiveRecord model
+```bash
+rails generate langchainrb_rails:qdrant --model=Product --llm=openai
+```
+
 Available `--llm` options: `cohere`, `google_palm`, `hugging_face`, `llama_cpp`, `ollama`, `openai`, and `replicate`. The selected LLM will be used to generate embeddings and completions.
 
 The `--model` option is used to specify which ActiveRecord model vectorsearch capabilities will be added to.

lib/langchainrb_rails.rb

@@ -19,6 +19,7 @@ module Generators
     autoload :BaseGenerator, "langchainrb_rails/generators/langchainrb_rails/base_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"
   end
 
   class << self

lib/langchainrb_rails/active_record/hooks.rb

@@ -112,7 +112,7 @@ def similarity_search(query, k: 1)
         # @return [String] The answer to the question
         def ask(question, k: 4, &block)
           class_variable_get(:@@provider).ask(
-            question,
+            question: question,
             k: k,
             &block
           ).completion

lib/langchainrb_rails/generators/langchainrb_rails/qdrant_generator.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module LangchainrbRails
+  module Generators
+    #
+    # PineconeGenerator does the following:
+    # 1. Creates the `langchainrb_rails.rb` initializer file
+    # 2. Adds necessary code to the ActiveRecord model to enable vectorsearch
+    # 3. Adds `qdrant-ruby` gem to the Gemfile
+    #
+    # Usage:
+    #     rails generate langchainrb_rails:qdrant --model=Product --llm=openai
+    #
+    class QdrantGenerator < LangchainrbRails::Generators::BaseGenerator
+      desc "This generator adds Qdrant vectorsearch integration to your ActiveRecord model"
+      source_root File.join(__dir__, "templates")
+
+      # Creates the `langchainrb_rails.rb` initializer file
+      def create_initializer_file
+        template "qdrant_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 `qdrant-ruby` gem to the Gemfile
+      # TODO: Can we automatically run `bundle install`?
+      def add_to_gemfile
+        gem "qdrant-ruby"
+      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

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

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+LangchainrbRails.configure do |config|
+  config.vectorsearch = Langchain::Vectorsearch::Qdrant.new(
+    llm: <%= llm_class %>.new(api_key: ENV["<%= llm.upcase %>_API_KEY"]),
+    url: ENV["QDRANT_URL"],
+    api_key: ENV["QDRANT_API_KEY"],
+    index_name: ""
+  )
+end

lib/langchainrb_rails/railtie.rb

@@ -12,6 +12,7 @@ class Railtie < Rails::Railtie
       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"
     end
   end
 end