weaviate_record 0.0.4 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d802d205f85ad643cd6549f4f0283e7a96577a3ff598f11016e999bf9c106d85
-  data.tar.gz: fc776f9c01eed6cdca251cd17d976c6a1259c5a3af2bc53959a70d3c416b3204
+  metadata.gz: 9d763c92aedebf48af6010dd279b83e50505efd1c2d754588826d90316bf0e71
+  data.tar.gz: ee07e2f4a8fba0243802c28383084cd0aa76a4259e780e3623c983a6a6475106
 SHA512:
-  metadata.gz: 34321036e3cb5a2813f11984dc860983fbab09bd381f34be506963644b336854b40fa414fdb80fcc934a5fdafb3ec0140e8ab8273d987987f725e5f01c3bf948
-  data.tar.gz: 51015663e59b6e0677dcfef1afaff37c679c3d62b4f48ff932658c228b0f77a6031ddce2de161604459e163e5b197fbd06d15f1433040ef54730ec36f66ad074
+  metadata.gz: 65de891828ef016b2fff0a2971d765b140e2f0e573503feeb4e1ec2cc9a930410db22836bc9935681e4b1490851b76e66f99c1673840a56a274212f83666e064
+  data.tar.gz: cc284a067506e2fee9ddf45cbae30722d2e77caef7303ef68aa4fcfd283dfddc134728c8d6c2e0259aa501cf0e20ced0556a4647b8cf9782cae11b143ef56225

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,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: weaviate_record
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.1.1
 platform: ruby
 authors:
 - Sriram V
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-06-08 00:00:00.000000000 Z
+date: 2024-07-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activemodel
@@ -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 and brings the power of Vector database and Retrieval augmented
+  generation (RAG) to your Ruby application.
 email: srira.venkat@gmail.com
 executables: []
 extensions: []