bx_builder_chain 0.1.10 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d16cadffa1ec3539c314ae22581283f341ab129ebb7015377163d862f0a4c634
-  data.tar.gz: 8b5f037a98477791dad85e7c2a257d927f0c8a41a0c78e7b0c399e645bbf3b6c
+  metadata.gz: eb8dc7d40d37072877b376103879593077531cecc15e4487197892eea0101c0f
+  data.tar.gz: d49614a734f1add1ead4e75d71fdead7b20437a41ed1a01d85780766dd75dd81
 SHA512:
-  metadata.gz: 03ff418896064cb8bacae0d9e21a18a92a6a10ad56216ebb553c79d4d1742ec0d7d35f7b8399529f5aed2adf0a3bff5ea2564536bb343b10bf93e67456d803c7
-  data.tar.gz: '0810e1e449e075f1516d88d4916655729850778668f4ada4e82073e32a3972a3e46675b32d122f505c06fdd35d2ad4106016b48e7bdd144e664f1383f7dfc59e'
+  metadata.gz: a847b3db61c0e705b06208061288690da66d2036449be89519d146b304a657ac31e1d2bd88dcae0c2ee91d8471e768b3ef7a8576222e3884199b15b9082bbb9e
+  data.tar.gz: eae9012d1602ad2cedf44b5173edc08eda4c17c3081a1b88808989bc1eee7540b36a2a2d01839e6a01076cab4ec26829bc569cc5a5044aa0fcb07651b0bbf66f

data/lib/bx_builder_chain/llm/open_ai.rb

@@ -46,7 +46,13 @@ module BxBuilderChain::Llm
       validate_max_tokens(text, parameters[:model])
 
       response = client.embeddings(parameters: parameters.merge(params))
-      response.dig("data").first.dig("embedding")
+      embedding = response.dig("data", 0, "embedding")
+
+      return embedding if embedding
+
+      puts response
+      raise "Error: #{response.dig("data")}"
+
     end
 
     #

data/lib/bx_builder_chain/version.rb

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

data/lib/generators/bx_builder_chain/install_generator.rb

@@ -3,7 +3,7 @@ require 'rails/generators/active_record'
 
   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"
 

data/lib/generators/bx_builder_chain/templates/initializer.rb

@@ -1,22 +1,26 @@
 BxBuilderChain.configure do |config|
-  config.openai_api_key = ENV['OPENAI_API_KEY']
+  begin
+    config.openai_api_key = ENV['OPENAI_API_KEY']
 
-  # for db use this
-  config.pg_url = ENV['DB_URL'] || nil # eg 'postgres://postgres:password@localhost:5432/my_db'
+    # for db use this
+    config.pg_url = ENV['DB_URL'] || nil # eg 'postgres://postgres:password@localhost:5432/my_db'
+  
+    # or this - pg_url with take preference
+    config.database_host = ENV['TEMPLATE_DATABASE_HOSTNAME'] || 'localhost'
+    config.database_name = ENV['TEMPLATEAPP_DATABASE']
+    config.database_user = ENV['TEMPLATEAPP_DATABASE_USER']
+    config.database_password = ENV['TEMPLATEAPP_DATABASE_PASSWORD']
+    config.database_port = '5432'
 
-  # or this - pg_url with take preference
-  config.database_host = ENV['DB_HOSTNAME'] || 'localhost'
-  config.database_name = ENV['DB_NAME']
-  config.database_user = ENV['DB_USER']
-  config.database_password = ENV['DB_PASSWORD']
-  config.database_port = ENV['DB_PORT'] || '5432' # Defaulting to 5432 if not set
-
-  config.public_namespace = "public"
-  config.threshold = 0.25
-  config.default_prompt_template = "Context information is below
-  --------------------
-  %{context}
-  --------------------
-  Given the context information and not prior knowledge
-  answer the question: %{question}"
+    config.public_namespace = "public" # namespace everyone has access to
+    config.threshold = 0.25 # return content with a match of less than this.
+    config.default_prompt_template = "Context information is below
+    --------------------
+    %{context}
+    --------------------
+    Given the context information and not prior knowledge
+    answer the question: %{question}"
+  rescue StandardError => e
+    logger.error("Failed to set configuration: #{e.message}")
+  end
 end

data/lib/sequel/plugins/pgvector.rb

@@ -10,7 +10,7 @@ module Sequel
       end
 
       module DatasetMethods
-        def nearest_neighbors(column, value, distance:)
+        def nearest_neighbors(column, value, distance:, threshold: nil)
           value = ::Pgvector.encode(value) unless value.is_a?(String)
           quoted_column = quote_identifier(column)
           distance = distance.to_s
@@ -36,9 +36,15 @@ module Sequel
               order
             end
 
-          select_append(Sequel.lit("#{neighbor_distance} AS neighbor_distance", value))
-            .exclude(column => nil)
-            .order(Sequel.lit(order, value))
+
+          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
       end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bx_builder_chain
 version: !ruby/object:Gem::Version
-  version: 0.1.10
+  version: 0.1.12
 platform: ruby
 authors:
 - Paul Ketelle