anndb_api 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 40231828a91cb8e7abaf19f7f59106653fd1a2f1879ef296af3b16ff254d0aa9
-  data.tar.gz: f5a4247b51dcbb1c0a0cd847057d3128e80b8230c18262085b1e158a6eb2a8d6
+  metadata.gz: 6a71f0644c3f5ac1a0a0832b3ddf2261656862829d77331697ab72468659cbff
+  data.tar.gz: ccd552eceec9d5c156057d496b543283cc2e83f760d44f67b85daa2dd23f608e
 SHA512:
-  metadata.gz: ae3175f1cb7b5f85d6aba9839f1df6850017ea77aba31ff692ee72cd4244f705bc07ac0e335d6b4c6f3a56ae91eae43059ac56d7b3a831714f1e67ff1530cb10
-  data.tar.gz: 2ce06b8904d9a2e87bfff59e009883c740a0e9ffa4f7a66cadd7a586a71e8147ea2d740f4d55595a7580cb9d289d60a030c4a6f6e83fe7a8710bc7a473353ab7
+  metadata.gz: 2b942bfe8d73f5d81efe16dbe8e92d2e96670569a5c74df0212d30688e97be7ee4dac1441e91a086e3e60642f58704b11276981f5744677f104af2407b8d258e
+  data.tar.gz: b4ad6e9c0c3671686864f8fbd9a7e9cdd4909eb4042a7bf7a61445ba3d69b84d39e9b057976ec7def36819dcbb8ec026cf96c6ae4be6349a91002bac9898712e

data/lib/anndb_api/client.rb

@@ -1,4 +1,6 @@
 require 'anndb_api/image_dataset'
+require 'anndb_api/text_dataset'
+require 'anndb_api/vector_dataset'
 
 module AnndbApi
     SERVER_ADDR = 'grpc-api.anndb.com:1433'

data/lib/anndb_api/image_dataset.rb

@@ -67,10 +67,10 @@ module AnndbApi
         private
 
         def image_proto(img)
-            if img.is_a? Array
-                return AnndbApiPb::Image.new(data: img)
-            elsif img.is_a? String
+            if !img.match(/^http/).nil?
                 return AnndbApiPb::Image.new(url: img)
+            else
+                return AnndbApiPb::Image.new(data: img)
             end
             
             raise TypeError.new('Image has to be either an array or a string')

data/lib/anndb_api/text_dataset.rb

@@ -42,7 +42,7 @@ module AnndbApi
 
         def update_batch(items)
             format_dataset_op_result @stub.update(AnndbApiPb::TextDatasetRequest.new(items: items.map { |item|
-                AnndbApiPb::ImageItem.new(
+                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, {}))

data/lib/anndb_api/vector_dataset.rb

@@ -43,7 +43,7 @@ module AnndbApi
 
         def update_batch(items)
             format_dataset_op_result @stub.update(AnndbApiPb::VectorDatasetRequest.new(items: items.map { |item|
-                AnndbApiPb::ImageItem.new(
+                AnndbApiPb::VectorItem.new(
                     id: AnndbApiPb::UUID.new(data: AnndbApi::Util.uuid_string_to_bytes(item[:id])),
                     vector: AnndbApiPb::Vector.new(values: item[:vector]),
                     metadata: format_req_metadata(item.fetch(:metadata, {}))

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: anndb_api
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Marek Galovic