class AnndbApi::TextDataset
Public Class Methods
new(server_addr, rpc_metadata)
click to toggle source
# File lib/anndb_api/text_dataset.rb, line 9 def initialize(server_addr, rpc_metadata) @rpc_metadata = rpc_metadata @stub = AnndbApiPb::TextDataset::Stub.new(server_addr, GRPC::Core::ChannelCredentials.new()) end
Public Instance Methods
insert(text, metadata = {})
click to toggle source
# File lib/anndb_api/text_dataset.rb, line 14 def insert(text, metadata = {}) format_dataset_op_result_single @stub.insert(AnndbApiPb::TextDatasetRequest.new(items: [ AnndbApiPb::TextItem.new( text: text, metadata: format_req_metadata(metadata) ) ]), { metadata: @rpc_metadata }) end
insert_batch(items)
click to toggle source
# File lib/anndb_api/text_dataset.rb, line 23 def insert_batch(items) format_dataset_op_result @stub.insert(AnndbApiPb::TextDatasetRequest.new(items: items.map { |item| AnndbApiPb::TextItem.new( text: item[:text], metadata: format_req_metadata(item.fetch(:metadata, {})) ) }), { metadata: @rpc_metadata }) end
search(query, n)
click to toggle source
# File lib/anndb_api/text_dataset.rb, line 53 def search(query, n) format_search_result @stub.search(AnndbApiPb::TextSearchRequest.new(query: query, n: n), { metadata: @rpc_metadata }) end
update(id, text, metadata = {})
click to toggle source
# File lib/anndb_api/text_dataset.rb, line 33 def update(id, text, metadata = {}) format_dataset_op_result_single @stub.update(AnndbApiPb::TextDatasetRequest.new(items: [ AnndbApiPb::TextItem.new( id: AnndbApiPb::UUID.new(data: AnndbApi::Util.uuid_string_to_bytes(id)), text: text, metadata: format_req_metadata(metadata) ) ]), { metadata: @rpc_metadata }) end
update_batch(items)
click to toggle source
# File lib/anndb_api/text_dataset.rb, line 43 def update_batch(items) format_dataset_op_result @stub.update(AnndbApiPb::TextDatasetRequest.new(items: items.map { |item| AnndbApiPb::TextItem.new( id: AnndbApiPb::UUID.new(data: AnndbApi::Util.uuid_string_to_bytes(item[:id])), text: item[:text], metadata: format_req_metadata(item.fetch(:metadata, {})) ) }), { metadata: @rpc_metadata }) end