bx_builder_chain 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b7694b1cbe868aa9aa9ca318e27393cbf42cd640b9ebb19cb64f79866cdfd916
-  data.tar.gz: b72ff57d3e402c6da8d20bce56510a9aaf5183b62f558fd30862e384273be861
+  metadata.gz: 4aa72e8ded8be5ea40f3cb865f989eabe6f6c8acee8fd51b2602e4e61b866c2a
+  data.tar.gz: 0d9ddd1c44ce88ea1fbf3f0970f1e9735116ecded0546a3ad5467ec062312463
 SHA512:
-  metadata.gz: 07f4da8f3ed4110434ad99eb60bfa49db4ddc19f5ebc5fd8780a986cf6943d11404db29ac9500e39e97c19ef179ebcb234e915629c3d164f652c5fa05abb51c0
-  data.tar.gz: 39c074e3f2cabc584d9bb02f77eae892132e33675e8ae0043a4501653b90ac6eab31b469210406b39e1568db6f5c75f1e755f26b0436a99314a270cb0f686fa9
+  metadata.gz: d774555dbd4e0d775c2e0b3e23384a1f20231ee8c118c62c16c7867f2e9d9cebab24ac48512896c9f61a9d2e09cc21941b5b4fcb963eb1f3f891f4fe6a0d9017
+  data.tar.gz: 1570c0677f6b47686b76437daf4ccf8652e0659ffce821ebc68f3857e251db055b02acc3b0a13d9b63a7d79f9d11535934c0d483209e50680e4711894a7dddc1

data/bx_builder_chain.gemspec

@@ -8,8 +8,8 @@ Gem::Specification.new do |spec|
   spec.authors = ["Paul Ketelle"]
   spec.email = ["paul.ketelle@builder.ai"]
 
-  spec.summary = "Write a short summary, because RubyGems requires one."
-  spec.description = "Write a longer description or delete this line."
+  spec.summary = "LangChain but for the builder tech stack."
+  spec.description = "LangChain but for the builder tech stack."
   spec.license = "MIT"
   spec.required_ruby_version = ">= 2.6.0"
 

data/lib/bx_builder_chain/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module BxBuilderChain
-  VERSION = "0.1.8"
+  VERSION = "0.1.9"
 end

data/lib/bx_builder_chain.rb

@@ -9,6 +9,7 @@ Dotenv.load
 loader = Zeitwerk::Loader.for_gem(warn_on_extra_files: false)
 loader.ignore("#{__dir__}/bx_builder_chain.rb")
 loader.ignore("#{__dir__}/bx_builder_chain/configuration.rb")
+loader.ignore("#{__dir__}/generators")
 loader.setup
 
 module BxBuilderChain

data/lib/generators/bx_builder_chain/install_generator.rb

@@ -1,10 +1,9 @@
 require 'rails/generators'
 require 'rails/generators/active_record'
 
-module Generators
   module BxBuilderChain
     class InstallGenerator < Rails::Generators::Base
-      source_root File.expand_path('templates', __dir__)
+      source_root File.expand_path('../../../templates', __dir__)
 
       desc "Creates BxBuilderChain initializer, migration, and copies app templates for your application"
 
@@ -39,4 +38,3 @@ module Generators
       end
     end
   end
-end

data/lib/pgvector/pg.rb

@@ -6,5 +6,23 @@ module Pgvector
       registry.register_type(0, "vector", nil, TextDecoder::Vector)
       registry.register_type(1, "vector", nil, BinaryDecoder::Vector)
     end
+
+    module BinaryDecoder
+      class Vector < ::PG::SimpleDecoder
+        def decode(string, tuple = nil, field = nil)
+          dim, unused = string[0, 4].unpack("nn")
+          raise "expected unused to be 0" if unused != 0
+          string[4..-1].unpack("g#{dim}")
+        end
+      end
+    end
+
+    module TextDecoder
+      class Vector < ::PG::SimpleDecoder
+        def decode(string, tuple = nil, field = nil)
+          Pgvector.decode(string)
+        end
+      end
+    end
   end
-end
+end

data/lib/pgvector.rb

@@ -1,5 +1,4 @@
 module Pgvector
-  autoload :PG, "pgvector/pg"
 
   def self.encode(data)
     "[#{data.to_a.map(&:to_f).join(",")}]"

data/lib/sequel/plugins/pgvector.rb

@@ -1,3 +1,4 @@
+require 'sequel'
 module Sequel
   module Plugins
     module Pgvector
@@ -7,6 +8,75 @@ module Sequel
           model.vector_columns[column.to_sym] = {}
         end
       end
+
+      module DatasetMethods
+        def nearest_neighbors(column, value, distance:)
+          value = ::Pgvector.encode(value) unless value.is_a?(String)
+          quoted_column = quote_identifier(column)
+          distance = distance.to_s
+
+          operator =
+            case distance
+            when "inner_product"
+              "<#>"
+            when "cosine"
+              "<=>"
+            when "euclidean"
+              "<->"
+            end
+
+          raise ArgumentError, "Invalid distance: #{distance}" unless operator
+
+          order = "#{quoted_column} #{operator} ?"
+
+          neighbor_distance =
+            if distance == "inner_product"
+              "(#{order}) * -1"
+            else
+              order
+            end
+
+          select_append(Sequel.lit("#{neighbor_distance} AS neighbor_distance", value))
+            .exclude(column => nil)
+            .order(Sequel.lit(order, value))
+        end
+      end
+
+      module ClassMethods
+        attr_accessor :vector_columns
+
+        Sequel::Plugins.def_dataset_methods(self, :nearest_neighbors)
+
+        Plugins.inherited_instance_variables(self, :@vector_columns => :dup)
+      end
+
+      module InstanceMethods
+        def nearest_neighbors(column, **options)
+          column = column.to_sym
+          # important! check if neighbor attribute before calling send
+          raise ArgumentError, "Invalid column" unless self.class.vector_columns[column]
+
+          self.class
+            .nearest_neighbors(column, self[column], **options)
+            .exclude(primary_key => self[primary_key])
+        end
+
+        def []=(k, v)
+          if self.class.vector_columns.key?(k.to_sym) && !v.is_a?(String)
+            super(k, ::Pgvector.encode(v))
+          else
+            super
+          end
+        end
+
+        def [](k)
+          if self.class.vector_columns.key?(k.to_sym)
+            ::Pgvector.decode(super)
+          else
+            super
+          end
+        end
+      end
     end
   end
-end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bx_builder_chain
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - Paul Ketelle
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-09-27 00:00:00.000000000 Z
+date: 2023-09-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: zeitwerk
@@ -150,7 +150,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 2.8.3
-description: Write a longer description or delete this line.
+description: LangChain but for the builder tech stack.
 email:
 - paul.ketelle@builder.ai
 executables: []
@@ -206,11 +206,7 @@ files:
 - lib/generators/bx_builder_chain/templates/migration.rb
 - lib/pgvector.rb
 - lib/pgvector/pg.rb
-- lib/pgvector/pg/binary_decoder/vector.rb
-- lib/pgvector/pg/text_decoder/vector.rb
 - lib/sequel/plugins/pgvector.rb
-- lib/sequel/plugins/pgvector/class_methods.rb
-- lib/sequel/plugins/pgvector/instance_methods.rb
 - sig/bx_langchain_chat.rbs
 homepage:
 licenses:
@@ -234,5 +230,5 @@ requirements: []
 rubygems_version: 3.1.0
 signing_key:
 specification_version: 4
-summary: Write a short summary, because RubyGems requires one.
+summary: LangChain but for the builder tech stack.
 test_files: []

data/lib/pgvector/pg/binary_decoder/vector.rb

@@ -1,14 +0,0 @@
-module Pgvector
-    module PG
-      module BinaryDecoder
-        class Vector < ::PG::SimpleDecoder
-          def decode(string, tuple = nil, field = nil)
-            dim, unused = string[0, 4].unpack("nn")
-            raise "expected unused to be 0" if unused != 0
-            string[4..-1].unpack("g#{dim}")
-          end
-        end
-      end
-    end
-  end
-  

data/lib/pgvector/pg/text_decoder/vector.rb

@@ -1,12 +0,0 @@
-module Pgvector
-    module PG
-      module TextDecoder
-        class Vector < ::PG::SimpleDecoder
-          def decode(string, tuple = nil, field = nil)
-            Pgvector.decode(string)
-          end
-        end
-      end
-    end
-  end
-  

data/lib/sequel/plugins/pgvector/class_methods.rb

@@ -1,47 +0,0 @@
-module Sequel
-    module Plugins
-        module Pgvector
-            module ClassMethods
-                attr_accessor :vector_columns
-
-                def nearest_neighbors(column, value, distance:, threshold: nil)
-                    value = ::Pgvector.encode(value) unless value.is_a?(String)
-                    quoted_column = dataset.quote_identifier(column)
-                    distance = distance.to_s
-                
-                    operator =
-                        case distance
-                        when "inner_product"
-                        "<#>"
-                        when "cosine"
-                        "<=>"
-                        when "euclidean"
-                        "<->"
-                        else
-                        raise ArgumentError, "Invalid distance: #{distance}"
-                        end
-                
-                    order = "#{quoted_column} #{operator} ?"
-                
-                    neighbor_distance =
-                        if distance == "inner_product"
-                        "(#{order}) * -1"
-                        else
-                        order
-                        end
-                
-                    query = select_append(Sequel.lit("#{neighbor_distance} AS neighbor_distance", value))
-                        .exclude(column => nil)
-                        .order(Sequel.lit(order, value))
-                
-                    # Apply the WHERE condition only if threshold is provided
-                    query = query.where(Sequel.lit("(#{neighbor_distance}) < ?", value, threshold)) if threshold
-                
-                    query
-                end                
-
-                Plugins.inherited_instance_variables(self, :@vector_columns => :dup)
-            end
-        end
-    end
-end

data/lib/sequel/plugins/pgvector/instance_methods.rb

@@ -1,34 +0,0 @@
-module Sequel
-    module Plugins
-      module Pgvector
-        module InstanceMethods
-          def nearest_neighbors(column, **options)
-            column = column.to_sym
-            # important! check if neighbor attribute before calling send
-            raise ArgumentError, "Invalid column" unless self.class.vector_columns[column]
-  
-            self.class
-              .nearest_neighbors(column, self[column], **options)
-              .exclude(primary_key => self[primary_key])
-          end
-  
-          def []=(k, v)
-            if self.class.vector_columns.key?(k.to_sym) && !v.is_a?(String)
-              super(k, ::Pgvector.encode(v))
-            else
-              super
-            end
-          end
-  
-          def [](k)
-            if self.class.vector_columns.key?(k.to_sym)
-              ::Pgvector.decode(super)
-            else
-              super
-            end
-          end
-        end
-      end
-    end
-  end
-