bx_builder_chain 0.1.9 → 0.1.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4aa72e8ded8be5ea40f3cb865f989eabe6f6c8acee8fd51b2602e4e61b866c2a
-  data.tar.gz: 0d9ddd1c44ce88ea1fbf3f0970f1e9735116ecded0546a3ad5467ec062312463
+  metadata.gz: 11f1eadf29ceef8ebd78087f6c49d22c1586e5850d13489c2c6bfca14346dd80
+  data.tar.gz: 8ae63c3ae73284c5248d7f5a427aef2b99fdb6b298de627cd7cd810392d39a37
 SHA512:
-  metadata.gz: d774555dbd4e0d775c2e0b3e23384a1f20231ee8c118c62c16c7867f2e9d9cebab24ac48512896c9f61a9d2e09cc21941b5b4fcb963eb1f3f891f4fe6a0d9017
-  data.tar.gz: 1570c0677f6b47686b76437daf4ccf8652e0659ffce821ebc68f3857e251db055b02acc3b0a13d9b63a7d79f9d11535934c0d483209e50680e4711894a7dddc1
+  metadata.gz: dca990841242d29fa54f01550a562ea8a2df2395761797a9c5f2af4dd85cfa2399ab32e98f58309d2a0ce39cc24f284078878e0b1301f6947a3ba17485ae4d03
+  data.tar.gz: c0ff76ccd8cd4d89be26fa24a22b7db998b5aa93291d0c46a0959aa34b2a8f878b54c271029324a2dbf2ed95cea841a548a2f5a451a4193ac7b2b0c05b03427f

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.9"
+  VERSION = "0.1.11"
 end

data/lib/generators/bx_builder_chain/templates/app/controllers/bx_builder_chain/documents_controller.rb

@@ -44,7 +44,7 @@ module BxBuilderChain
     def document_service
       @service ||= DocumentUploadService.new(
         files: params[:files],
-        current_user_groups: current_user_document_groups,
+        user_groups: current_user_document_groups,
         client_class_name: CLIENT_CLASS_NAME,
         llm_class_name: LLM_CLASS_NAME
       )

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.9
+  version: 0.1.11
 platform: ruby
 authors:
 - Paul Ketelle