diff --git a/Gemfile.lock b/Gemfile.lock index 7085f26..3b75475 100644 --- a/Gemfile.lock +++ b/Gemfile.lock @@ -1,7 +1,7 @@ PATH remote: . specs: - chromable (0.3.1) + chromable (0.3.2) chroma-db (>= 0.6.0) GEM diff --git a/README.md b/README.md index 7e6390d..c1a7cc4 100644 --- a/README.md +++ b/README.md @@ -73,7 +73,7 @@ Post.query( is_query: true # `is_query` here will be passed to `Post.embed` as an option. ) -Post.first.neighbors(results: 20) # => [#, ...] +Post.first.neighbors(results: 20) # => [#, ...] ``` Also, `chromable` provides the following methods for each model instance: diff --git a/lib/chromable.rb b/lib/chromable.rb index 3d4741e..c51f97e 100644 --- a/lib/chromable.rb +++ b/lib/chromable.rb @@ -56,8 +56,7 @@ def chroma_query(text:, results: 10, where: {}, where_document: {}, **embedder_o query_embeddings: [send(chromable_settings.embedder, text, **embedder_options)], results: results, where: where, - where_document: where_document, - include: include + where_document: where_document ).map(&:id)) end end @@ -96,8 +95,7 @@ def chroma_neighbors(results: 10, where: {}, where_document: {}) query_embeddings: [embedding], results: results, where: where, - where_document: where_document, - include: include + where_document: where_document ).map(&:id)) end diff --git a/lib/chromable/version.rb b/lib/chromable/version.rb index 917c4b3..810a7bd 100644 --- a/lib/chromable/version.rb +++ b/lib/chromable/version.rb @@ -1,5 +1,5 @@ # frozen_string_literal: true module Chromable - VERSION = '0.3.1' + VERSION = '0.3.2' end