add filter + test

patterns-ai-core · Jan 12, 2024 · 3c9b0e5 · 3c9b0e5
1 parent 070407a
commit 3c9b0e5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/lib/qdrant/points.rb b/lib/qdrant/points.rb
@@ -47,14 +47,16 @@ def upsert(
     def delete(
       collection_name:,
       points:, wait: nil,
-      ordering: nil
+      ordering: nil,
+      filter: nil
     )
       response = client.connection.post("collections/#{collection_name}/#{PATH}/delete") do |req|
         req.params["wait"] = wait unless wait.nil?
         req.params["ordering"] = ordering unless ordering.nil?
 
         req.body = {}
         req.body["points"] = points
+        req.body["filter"] = filter unless filter.nil?
       end
       response.body
     end

diff --git a/spec/qdrant/points_spec.rb b/spec/qdrant/points_spec.rb
@@ -71,7 +71,8 @@
     it "return the data" do
       response = client.points.delete(
         collection_name: "test_collection",
-        points: [3]
+        points: [3],
+        filter: []
       )
       expect(response.dig("status")).to eq("ok")
     end