langchainrb_rails 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 54ab2d33d89929b9222202d42cc5616aac73ca65fa81c3e4f2bc3d516a3a05c6
-  data.tar.gz: 4ce553994a4219904875168369ab598de11162db2cd6810d202af005acb10638
+  metadata.gz: 6bb3982f419d252f6e94018548685566d9775993ad6349957211a4d686e9a10c
+  data.tar.gz: 783c3d1683d654903f745aa6b1fd973787b03581792a1292a0d5724c6983eda8
 SHA512:
-  metadata.gz: bd93a8976a54120bcec4f8c10ff5bfc5c36d8bccd9f172b071a7732d675ade00ddc3021f93ecb080f508728fdb06b45e5c086933b0bd5522dd094b08d9334d93
-  data.tar.gz: 0fd5d062e69a441c50c4963749e31c6773f38ba3e063d21574983b5fd4579c495c3e1b95d340540754c8f89b110c4e1f6ea3f0399265bcf315a0d26f5b6b8e36
+  metadata.gz: f1fe4ad36c5c12c7260cff5bf29046330d2e649a0f8d96707b13831f09c2c185cf88c5a202f3621eb03cb87fae1df6f7bb75a6bdac5cd9471a4e4637daaafa5f
+  data.tar.gz: f0effcbf9679194a32ece6ff84cc26cf0c5cea5956c9a51ee1a33ba3e32bf4ea56807bba9a463b9edc227d1cb572508bfad70e3a1e8063c5f374e12e1e97d834

data/CHANGELOG.md

@@ -1,5 +1,8 @@
 ## [Unreleased]
 
+## [0.1.5] - 2023-11-30
+- Qdrant vectorsearch generator
+
 ## [0.1.4] - 2023-11-20
 - Bugfix AR integration when using vectorsearch other than Pgvector
 

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    langchainrb_rails (0.1.4)
+    langchainrb_rails (0.1.5)
       langchainrb (~> 0.7.0)
 
 GEM

data/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.

data/lib/langchainrb_rails/active_record/hooks.rb

@@ -112,7 +112,7 @@ module LangchainrbRails
         # @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

data/lib/langchainrb_rails/generators/langchainrb_rails/pgvector_generator.rb

@@ -4,7 +4,7 @@ module LangchainrbRails
   module Generators
     #
     # Usage:
-    #     rails g langchain:pgvector -model=Product -llm=openai
+    #     rails generate langchainrb_rails:pgvector --model=Product --llm=openai
     #
     class PgvectorGenerator < LangchainrbRails::Generators::BaseGenerator
       desc "This generator adds Pgvector vectorsearch integration to your ActiveRecord model"

data/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

data/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

data/lib/langchainrb_rails/railtie.rb

@@ -12,6 +12,7 @@ module LangchainrbRails
       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

data/lib/langchainrb_rails/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LangchainrbRails
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

data/lib/langchainrb_rails.rb

@@ -19,6 +19,7 @@ module LangchainrbRails
     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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb_rails
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-11-20 00:00:00.000000000 Z
+date: 2023-11-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: langchainrb
@@ -90,11 +90,13 @@ files:
 - lib/langchainrb_rails/generators/langchainrb_rails/chroma_generator.rb
 - lib/langchainrb_rails/generators/langchainrb_rails/pgvector_generator.rb
 - lib/langchainrb_rails/generators/langchainrb_rails/pinecone_generator.rb
+- lib/langchainrb_rails/generators/langchainrb_rails/qdrant_generator.rb
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/add_vector_column_template.rb.tt
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/chroma_initializer.rb.tt
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/enable_vector_extension_template.rb.tt
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/pgvector_initializer.rb.tt
 - lib/langchainrb_rails/generators/langchainrb_rails/templates/pinecone_initializer.rb.tt
+- lib/langchainrb_rails/generators/langchainrb_rails/templates/qdrant_initializer.rb.tt
 - lib/langchainrb_rails/railtie.rb
 - lib/langchainrb_rails/version.rb
 - sig/langchainrb_rails.rbs