Merge pull request #6 from andreibondarev/points-get_all

Points #get_all() method

Author: andreibondarev

.env.example

@@ -0,0 +1,2 @@
+QDRANT_URL=
+QDRANT_API_KEY=

README.md

@@ -163,6 +163,14 @@ client.points.get(
     consistency: "int"
 )
 
+# Retrieve full information of points by ids
+client.points.get_all(
+    collection_name: "string", # required
+    ids: "[int]",              # required
+    with_payload: "boolean"
+    with_vector: "boolean"
+)
+
 # Lists all data objects in reverse order of creation. The data will be returned as an array of objects.
 client.points.list(
     collection_name: "string", # required

bin/console

@@ -12,8 +12,8 @@ require "pry"
 # Pry.start
 
 client = Qdrant::Client.new(
-  url: "https://02e23d35-486d-4555-a471-0415ac641516.us-east-1-0.aws.cloud.qdrant.io:6333",
-  api_key: "yCjKEx3UPJXgTUfHgzzmZAP8kXk-4nUy7LoCdFwe8HtcDzXA-8KVVw"
+  url: ENV["QDRANT_URL"],
+  api_key: ENV["QDRANT_API_KEY"]
 )
 
 require "irb"

lib/qdrant/points.rb

@@ -71,6 +71,25 @@ def get(
       response.body
     end
 
+    # Retrieve full information of points by ids
+    def get_all(
+      collection_name:,
+      ids:,
+      consistency: nil,
+      with_payload: nil,
+      with_vector: nil
+    )
+      response = client.connection.post("collections/#{collection_name}/#{PATH}") do |req|
+        req.params["consistency"] = consistency unless consistency.nil?
+
+        req.body = {}
+        req.body["ids"] = ids
+        req.body["with_payload"] = with_payload unless with_payload.nil?
+        req.body["with_vector"] = with_vector unless with_vector.nil?
+      end
+      response.body
+    end
+
     # Set payload values for points
     def set_payload(
       collection_name:,

spec/qdrant/points_spec.rb

@@ -37,23 +37,23 @@
     end
   end
 
-  describe "#get" do
+  describe "#get_all" do
     let(:response) {
-      OpenStruct.new(body: point_fixture)
+      OpenStruct.new(body: points_fixture)
     }
 
     before do
-      allow_any_instance_of(Faraday::Connection).to receive(:get)
-        .with("collections/test_collection/points/1")
+      allow_any_instance_of(Faraday::Connection).to receive(:post)
+        .with("collections/test_collection/points")
         .and_return(response)
     end
 
     it "return the data" do
-      response = client.points.get(
+      response = client.points.get_all(
         collection_name: "test_collection",
-        id: 1
+        ids: [4, 1, 2, 5, 6]
       )
-      expect(response.dig("result", "id")).to eq(1)
+      expect(response.dig("result").count).to eq(5)
     end
   end