weaviate_record 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: af84dff0fba26b232fd8b537749ff5742f5bcc0197656cc647e8b6791a55bd85
-  data.tar.gz: 9534adbc21943b257d0e8f0a517721dd17bb93a1d1e3c14f284e11f30499f1f9
+  metadata.gz: b8e3a3dbf676df52348c83eb5a472516ca992c7ea956127d6a7ba81ee8fdb311
+  data.tar.gz: 35cf2a6a0369e9e6370e0bbb245c0ada8aa5e82d3f5dcd4c504d65295eacb9ac
 SHA512:
-  metadata.gz: e7d0fda549172607928d6e943426f5091d481d4cbcd4c44bb767a32ef667769091238125f79926c1070358dc1095a191e52af70d20ea915b17acbf4eeac51beb
-  data.tar.gz: dde3def3b42f332182c8d7f8cc115a1516f177be5abedbd516713e00474f9c1b7b9935da51ef134d9010b773df8acb95d2f867a4be043765edef53308e97b201
+  metadata.gz: 1aefd604f54db251167b59e76d2f1542178cbb4e857b7a9b476b89b7d0e503103cd3f4e60777c8b20a9631e0411301d93e052e63d483d536f04faaf3be0a0233
+  data.tar.gz: 2c2fae434083f2cef4a96618e929b73f1392912e80e81a4567329b108496b1fbf528869dde504936f994db9da3da1b198f6e9c08af085cf1ad51bb35cbde6f30

data/lib/weaviate_record/connection.rb

@@ -22,32 +22,46 @@ module WeaviateRecord
     # Returns the present schema of the Weaviate database.
     def schema_list
       client.schema.list.deep_symbolize_keys!
+    rescue Faraday::ConnectionFailed => e
+      raise WeaviateRecord::Errors::DatabaseNotConnected, e.message, cause: nil
     end
 
     # :enddoc:
 
     def find_call(id)
       client.objects.get(class_name: @collection_name, id: id)
+    rescue Faraday::ConnectionFailed => e
+      raise WeaviateRecord::Errors::DatabaseNotConnected, e.message, cause: nil
     end
 
     def create_call(properties)
       client.objects.create(class_name: @collection_name, properties: properties)
+    rescue Faraday::ConnectionFailed => e
+      raise WeaviateRecord::Errors::DatabaseNotConnected, e.message, cause: nil
     end
 
     def update_call(id, properties)
       client.objects.update(class_name: @collection_name, id: id, properties: properties)
+    rescue Faraday::ConnectionFailed => e
+      raise WeaviateRecord::Errors::DatabaseNotConnected, e.message, cause: nil
     end
 
     def delete_call(id)
       client.objects.delete(class_name: @collection_name, id: id)
+    rescue Faraday::ConnectionFailed => e
+      raise WeaviateRecord::Errors::DatabaseNotConnected, e.message, cause: nil
     end
 
     def check_existence(id)
       client.objects.exists?(class_name: @collection_name, id: id)
+    rescue Faraday::ConnectionFailed => e
+      raise WeaviateRecord::Errors::DatabaseNotConnected, e.message, cause: nil
     end
 
     def delete_where(condition)
       client.objects.batch_delete(class_name: @collection_name, where: condition)
+    rescue Faraday::ConnectionFailed => e
+      raise WeaviateRecord::Errors::DatabaseNotConnected, e.message, cause: nil
     end
   end
 end

data/lib/weaviate_record/errors.rb

@@ -65,5 +65,9 @@ module WeaviateRecord # :nodoc: all
     # Raised when required argument is empty
     class EmptyPrompt < StandardError
     end
+
+    # Raised when the weaviate database is not connected
+    class DatabaseNotConnected < StandardError
+    end
   end
 end

data/lib/weaviate_record/queries/order.rb

@@ -6,7 +6,7 @@ module WeaviateRecord
     module Order
       # Sort the records based on the given attributes.
       # You can pass multiple attributes to sort the records.
-      # This sorting specification will be ignored if you are performing keyword (bm25) search.
+      # This sorting specification will be ignored if you are performing +bm25+ search.
       #
       # ==== Example:
       #    Article.order(:title)

data/lib/weaviate_record/queries/where.rb

@@ -88,12 +88,21 @@ module WeaviateRecord
       def create_query_condition(equation)
         return null_condition(equation[0]) if equation[2].nil?
 
+        validate_attribute!(equation[0])
         handle_timestamps_condition(equation)
         "{ path: [\"#{equation[0]}\"], " \
           "operator: #{map_operator(equation[1])}, " \
           "#{map_value_type(equation[2])}: #{equation[2].inspect} }"
       end
 
+      def validate_attribute!(attribute)
+        unless ['id', 'created_at', 'updated_at',
+                *WeaviateRecord::Schema.find_collection(@klass).attributes_list].include?(attribute)
+          raise WeaviateRecord::Errors::InvalidAttributeError,
+                "Invalid attribute #{attribute} used for collection #{@klass}"
+        end
+      end
+
       def handle_timestamps_condition(equation_array)
         return nil unless equation_array[0] == 'created_at' || equation_array[0] == 'updated_at'
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: weaviate_record
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Sriram V
@@ -53,8 +53,8 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '2.4'
 description: An ORM for Weaviate vector database that follows the same conventions
-  as the ActiveRecord. This gem uses weaviate-ruby internally to connect with weaviate
-  database.
+  as the ActiveRecord. Bring the power of Vector database and Retrieval augmented
+  generation (RAG) to your Ruby application.
 email: srira.venkat@gmail.com
 executables: []
 extensions: []