standardrb fixes

Author: andreibondarev

lib/qdrant/client.rb

@@ -24,7 +24,7 @@ def initialize(
     def connection
       @connection ||= Faraday.new(url: url) do |faraday|
         if api_key
-          faraday.headers['api-key'] = api_key
+          faraday.headers["api-key"] = api_key
         end
         faraday.request :json
         faraday.response :json, content_type: /\bjson$/
@@ -56,4 +56,4 @@ def points
       @points ||= Qdrant::Points.new(client: self)
     end
   end
-end
+end

lib/qdrant/clusters.rb

@@ -22,7 +22,7 @@ def remove_peer(
       force: nil
     )
       response = client.connection.post("#{PATH}/recover") do |req|
-        req.params['force'] = force if force
+        req.params["force"] = force if force
       end
       response.body
     end

lib/qdrant/collections.rb

@@ -32,16 +32,16 @@ def create(
     )
       response = client.connection.put("#{PATH}/#{collection_name}") do |req|
         req.body = {}
-        req.body['vectors'] = vectors
-        req.body['shard_number'] = shard_number unless shard_number.nil?
-        req.body['replication_factor'] = replication_factor unless replication_factor.nil?
-        req.body['write_consistency_factor'] = write_consistency_factor unless write_consistency_factor.nil?
-        req.body['on_disk_payload'] = on_disk_payload unless on_disk_payload.nil?
-        req.body['hnsw_config'] = hnsw_config unless hnsw_config.nil?
-        req.body['wal_config'] = wal_config unless wal_config.nil?
-        req.body['optimizers_config'] = optimizers_config unless optimizers_config.nil?
-        req.body['init_from'] = init_from unless init_from.nil?
-        req.body['quantization_config'] = quantization_config unless quantization_config.nil?
+        req.body["vectors"] = vectors
+        req.body["shard_number"] = shard_number unless shard_number.nil?
+        req.body["replication_factor"] = replication_factor unless replication_factor.nil?
+        req.body["write_consistency_factor"] = write_consistency_factor unless write_consistency_factor.nil?
+        req.body["on_disk_payload"] = on_disk_payload unless on_disk_payload.nil?
+        req.body["hnsw_config"] = hnsw_config unless hnsw_config.nil?
+        req.body["wal_config"] = wal_config unless wal_config.nil?
+        req.body["optimizers_config"] = optimizers_config unless optimizers_config.nil?
+        req.body["init_from"] = init_from unless init_from.nil?
+        req.body["quantization_config"] = quantization_config unless quantization_config.nil?
       end
 
       response.body
@@ -95,7 +95,7 @@ def create_index(
         req.body = {
           field_name: field_name
         }
-        req.body['field_schema'] = field_schema unless field_schema.nil?
+        req.body["field_schema"] = field_schema unless field_schema.nil?
       end
 
       response.body
@@ -136,7 +136,7 @@ def download_snapshot(
       filepath:
     )
       response = client.connection.get("#{PATH}/#{collection_name}/snapshots/#{snapshot_name}")
-      File.open(File.expand_path(filepath), 'wb+') { |fp| fp.write(response.body) }
+      File.open(File.expand_path(filepath), "wb+") { |fp| fp.write(response.body) }
     end
 
     # Delete snapshot for a collection
@@ -172,12 +172,12 @@ def restore_snapshot(
       wait: nil
     )
       response = client.connection.post("#{PATH}/#{collection_name}/snapshots/recover") do |req|
-        req.params['wait'] = wait unless wait.nil?
+        req.params["wait"] = wait unless wait.nil?
 
         req.body = {
           location: filepath
         }
-        req.body['priority'] = priority unless priority.nil?
+        req.body["priority"] = priority unless priority.nil?
       end
       response.body
     end

lib/qdrant/points.rb

@@ -46,9 +46,8 @@ def upsert(
     # Delete points
     def delete(
       collection_name:,
-      wait: nil,
-      ordering: nil,
-      points:
+      points:, wait: nil,
+      ordering: nil
     )
       response = client.connection.post("collections/#{collection_name}/#{PATH}/delete") do |req|
         req.params["wait"] = wait unless wait.nil?
@@ -96,9 +95,8 @@ def set_payload(
     # Replace full payload of points with new one
     def overwrite_payload(
       collection_name:,
-      wait: nil,
+      payload:, wait: nil,
       ordering: nil,
-      payload:,
       points: nil,
       filter: nil
     )
@@ -111,14 +109,14 @@ def overwrite_payload(
         req.body["points"] = points unless points.nil?
         req.body["filter"] = filter unless filter.nil?
       end
+      response.body
     end
 
     # Delete specified key payload for points
     def clear_payload_keys(
       collection_name:,
-      wait: nil,
+      keys:, wait: nil,
       ordering: nil,
-      keys:,
       points: nil,
       filter: nil
     )
@@ -129,7 +127,7 @@ def clear_payload_keys(
         req.body = {}
         req.body["keys"] = keys
         req.body["points"] = points unless points.nil?
-        req.body["filter"] = filter unless filter.nil?        
+        req.body["filter"] = filter unless filter.nil?
       end
       response.body
     end

lib/qdrant/service.rb

@@ -22,7 +22,6 @@ def metrics(
       response.body
     end
 
-
     # Set lock options. If write is locked, all write operations and collection creation are forbidden. Returns previous lock options
     def set_lock(
       write:,

lib/qdrant/snapshots.rb

@@ -30,7 +30,7 @@ def download(
       filepath:
     )
       response = client.connection.get("#{PATH}/#{snapshot_name}")
-      File.open(File.expand_path(filepath), 'wb+') { |fp| fp.write(response.body) }
+      File.open(File.expand_path(filepath), "wb+") { |fp| fp.write(response.body) }
     end
   end
 end

spec/qdrant/clusters_spec.rb

@@ -17,15 +17,14 @@
   }
 
   describe "#info" do
-
     before do
       allow_any_instance_of(Faraday::Connection).to receive(:get)
         .with("cluster")
         .and_return(response)
     end
 
     it "return the cluster info" do
-      expect(client.clusters.info.dig('status')).to eq('ok')
+      expect(client.clusters.info.dig("status")).to eq("ok")
     end
   end
 
@@ -37,7 +36,7 @@
     end
 
     it "return the data" do
-      expect(client.clusters.recover.dig('status')).to eq('ok')
+      expect(client.clusters.recover.dig("status")).to eq("ok")
     end
   end
 

spec/qdrant/collections_spec.rb

@@ -25,7 +25,7 @@
     end
 
     it "returns collections" do
-      expect(collections.list.dig('result', 'collections').count).to eq(1)
+      expect(collections.list.dig("result", "collections").count).to eq(1)
     end
   end
 
@@ -40,7 +40,7 @@
 
     it "returns the collection" do
       response = collections.get(collection_name: "test_collection")
-      expect(response.dig('result', 'status')).to eq('green')
+      expect(response.dig("result", "status")).to eq("green")
     end
   end
 
@@ -77,8 +77,8 @@
 
     it "returns the schema" do
       response = collections.delete(collection_name: "test_collection")
-      expect(response.dig('status')).to eq('ok')
-      expect(response.dig('result')).to eq(true)
+      expect(response.dig("status")).to eq("ok")
+      expect(response.dig("result")).to eq(true)
     end
   end
 
@@ -98,8 +98,8 @@
           replication_factor: 1
         }
       )
-      expect(response.dig('status')).to eq('ok')
-      expect(response.dig('result')).to eq(true)
+      expect(response.dig("status")).to eq("ok")
+      expect(response.dig("result")).to eq(true)
     end
   end
 
@@ -116,13 +116,13 @@
       response = collections.update_aliases(
         actions: [{
           create_alias: {
-            collection_name: 'test_collection',
-            alias_name: 'alias_test_collection'
+            collection_name: "test_collection",
+            alias_name: "alias_test_collection"
           }
         }]
       )
-      expect(response.dig('status')).to eq('ok')
-      expect(response.dig('result')).to eq(true)
+      expect(response.dig("status")).to eq("ok")
+      expect(response.dig("result")).to eq(true)
     end
   end
 
@@ -137,7 +137,7 @@
 
     it "returns the schema" do
       response = collections.aliases(collection_name: "test_collection")
-      expect(response.dig('result', 'aliases').count).to eq(1)
+      expect(response.dig("result", "aliases").count).to eq(1)
     end
   end
 
@@ -156,8 +156,8 @@
         field_name: "description",
         field_schema: "text"
       )
-      expect(response.dig('status')).to eq('ok')
-      expect(response.dig('result')).to eq(true)
+      expect(response.dig("status")).to eq("ok")
+      expect(response.dig("result")).to eq(true)
     end
   end
 
@@ -175,8 +175,8 @@
         collection_name: "test_collection",
         field_name: "description"
       )
-      expect(response.dig('status')).to eq('ok')
-      expect(response.dig('result')).to eq(true)
+      expect(response.dig("status")).to eq("ok")
+      expect(response.dig("result")).to eq(true)
     end
   end
 
@@ -193,7 +193,7 @@
       response = collections.cluster_info(
         collection_name: "test_collection"
       )
-      expect(response.dig('result', 'peer_id')).to eq(111)
+      expect(response.dig("result", "peer_id")).to eq(111)
     end
   end
 
@@ -210,13 +210,13 @@
       response = collections.update_cluster(
         collection_name: "test_collection",
         move_shard: {
-            shard_id: 0,
-            to_peer_id: 222,
-            from_peer_id: 111
+          shard_id: 0,
+          to_peer_id: 222,
+          from_peer_id: 111
         }
       )
-      expect(response.dig('status')).to eq('ok')
-      expect(response.dig('result')).to eq(true)
+      expect(response.dig("status")).to eq("ok")
+      expect(response.dig("result")).to eq(true)
     end
   end
 
@@ -233,7 +233,7 @@
       response = collections.list_snapshots(
         collection_name: "test_collection"
       )
-      expect(response.dig('result').count).to eq(2)
+      expect(response.dig("result").count).to eq(2)
     end
   end
 
@@ -252,8 +252,8 @@
       response = collections.create_snapshot(
         collection_name: "test_collection"
       )
-      expect(response.dig('status')).to eq('ok')
-      expect(response.dig('result', 'name')).to eq("test_collection-6106351684939824381-2023-04-06-20-43-03.snapshot")
+      expect(response.dig("status")).to eq("ok")
+      expect(response.dig("result", "name")).to eq("test_collection-6106351684939824381-2023-04-06-20-43-03.snapshot")
     end
   end
 
@@ -271,8 +271,8 @@
         collection_name: "test_collection",
         snapshot_name: "test_collection-6106351684939824381-2023-04-06-20-43-03.snapshot"
       )
-      expect(response.dig('status')).to eq('ok')
-      expect(response.dig('result')).to eq(true)
+      expect(response.dig("status")).to eq("ok")
+      expect(response.dig("result")).to eq(true)
     end
   end
 
@@ -283,8 +283,8 @@
       allow_any_instance_of(Faraday::Connection).to receive(:get)
         .with("collections/test_collection/snapshots/test_collection-6106351684939824381-2023-04-06-20-43-03.snapshot")
         .and_return(response)
-      
-      allow(File).to receive(:open).with("/dir/snapshot.txt", 'wb+').and_return(999)
+
+      allow(File).to receive(:open).with("/dir/snapshot.txt", "wb+").and_return(999)
     end
 
     it "returns the schema" do
@@ -311,8 +311,8 @@
         collection_name: "test_collection",
         filepath: "test_collection-6106351684939824381-2023-04-06-20-43-03.snapshot"
       )
-      expect(response.dig('status')).to eq('ok')
-      expect(response.dig('result')).to eq(true)
+      expect(response.dig("status")).to eq("ok")
+      expect(response.dig("result")).to eq(true)
     end
   end
 end