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
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