Adding specs

.gitignore

@@ -9,3 +9,5 @@
 
 # rspec failure tracking
 .rspec_status
+
+/volumes

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    milvus (0.1.0)
+    milvus (0.9.0)
       faraday (~> 2.7.0)
 
 GEM
@@ -39,6 +39,7 @@ GEM
 
 PLATFORMS
   x86_64-darwin-19
+  x86_64-darwin-22
 
 DEPENDENCIES
   milvus!

README.md

@@ -35,6 +35,8 @@ client = Milvus::Client.new(
 ### Using the Collections endpoints
 
 ```ruby
+# Data types: "boolean", "int8", "int16", "int32", "int64", "float", "double", "string", "varchar", "binary_vector", "float_vector"
+
 # Creating a new collection schema
 client.collections.create(
   collection_name: "book",
@@ -71,19 +73,19 @@ client.collections.create(
 ```
 ```ruby
 # Get the collection info
-client.collections.get(collection_name: "recipes")
+client.collections.get(collection_name: "book")
 ```
 ```ruby
 # Delete the collection
-client.collections.delete(collection_name: "recipes")
+client.collections.delete(collection_name: "book")
 ```
 ```ruby
 # Load the collection to memory before a search or a query
-client.collections.load(collection_name: "recipes")
+client.collections.load(collection_name: "book")
 ```
 ```ruby
 # Release a collection from memory after a search or a query to reduce memory usage
-client.collections.release(collection_name: "recipes")
+client.collections.release(collection_name: "book")
 ```
 
 ### Inserting Data
@@ -159,7 +161,22 @@ client.indices.delete(
 ### Search & Querying
 ```ruby
 client.search(
-
+  collection_name: "book",
+  output_fields: ["book_id"], # optional
+  anns_field: "book_intro",
+  top_k: "2",
+  params: "{\"nprobe\": 10}",
+  metric_type: "L2",
+  round_decimal: "-1",
+  vectors: [ [0.1,0.2] ],
+  dsl_type: 1
+)
+```
+```ruby
+client.query(
+  collection_name: "book",
+  output_fields: ["book_id", "book_intro"],
+  expr: "book_id in [2,4,6,8]"
 )
 ```
 
@@ -215,4 +232,5 @@ Bug reports and pull requests are welcome on GitHub at https://github.com/andrei
 
 ## License
 
-The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+`milvus` is licensed under the Apache License, Version 2.0. View a copy of the License file.
+

bin/console

@@ -8,7 +8,7 @@ require "milvus"
 # with your gem easier. You can also use a different console, if you like.
 
 # (If you use this, don't forget to add pry to your Gemfile!)
-require "pry"
+# require "pry"
 # Pry.start
 
 client = Milvus::Client.new(

lib/milvus.rb

@@ -12,4 +12,6 @@ module Milvus
   autoload :Health, "milvus/health"
   autoload :Indices, "milvus/indices"
   autoload :Partitions, "milvus/partitions"
+  autoload :Search, "milvus/search"
+  autoload :Query, "milvus/query"
 end

lib/milvus/client.rb

@@ -36,12 +36,12 @@ def indices
       @indices ||= Milvus::Indices.new(client: self)
     end
 
-    def search
-      @search ||= Milvus::Search.new(client: self).post
+    def search(...)
+      @search ||= Milvus::Search.new(client: self).post(...)
     end
 
-    def query
-      @query ||= Milvus::Query.new(client: self).post
+    def query(...)
+      @query ||= Milvus::Query.new(client: self).post(...)
     end
 
     def connection

lib/milvus/collections.rb

@@ -42,7 +42,7 @@ def delete(collection_name:)
           collection_name: collection_name
         }.to_json
       end
-      response.body || true
+      response.body.empty? ? true : response.body
     end
 
     # Load the collection to memory before a search or a query
@@ -52,17 +52,17 @@ def load(collection_name:)
           collection_name: collection_name
         }.to_json
       end
-      response.body
+      response.body.empty? ? true : response.body
     end
 
     # Release a collection from memory after a search or a query to reduce memory usage
     def release(collection_name:)
-      response = client.connection.delete("#{PATH}/release") do |req|
+      response = client.connection.delete("#{PATH}/load") do |req|
         req.body = {
           collection_name: collection_name
         }.to_json
       end
-      response.body
+      response.body.empty? ? true : response.body
     end
   end
 end

lib/milvus/entities.rb

@@ -20,7 +20,7 @@ def insert(
 
         req.body['partition_name'] = partition_name if partition_name
       end
-      response.body
+      response.body.empty? ? true : response.body
     end
 
     # Delete the entities with the boolean expression you created
@@ -34,7 +34,7 @@ def delete(
           expr: expression
         }.to_json
       end
-      response.body
+      response.body.empty? ? true : response.body
     end
 
     # Compact data manually
@@ -46,7 +46,7 @@ def compact!(
           collectionID: collection_id
         }.to_json
       end
-      response.body
+      response.body.empty? ? true : response.body
     end
 
     # Check compaction status
@@ -58,7 +58,7 @@ def compact_status(
           compactionID: compaction_id
         }.to_json
       end
-      response.body
+      response.body.empty? ? true : response.body
     end
   end
 end

lib/milvus/partitions.rb

@@ -6,10 +6,12 @@ class Partitions < Base
 
     # Create a partition
     def create(collection_name:, partition_name:)
-      response = client.connection.post(PATH, {
-        collection_name: collection_name,
-        partition_name: partition_name
-      })
+      response = client.connection.post(PATH) do |req|
+        req.body = {
+          collection_name: collection_name,
+          partition_name: partition_name
+        }
+      end
       response.body.empty? ? true : response.body
     end
 
@@ -53,13 +55,13 @@ def load(
 
     def release(
       collection_name:,
-      partition_name:,
+      partition_names:,
       replica_number: nil
     )
       response = client.connection.delete("#{PATH}s/load") do |req|
         req.body = {
           collection_name: collection_name,
-          partition_name: partition_name
+          partition_names: partition_names
         }
         req.body[:replica_number] = replica_number if replica_number
       end

lib/milvus/query.rb

@@ -3,5 +3,22 @@
 module Milvus
   class Query < Base
     PATH = "query".freeze
+
+    def post(
+      collection_name:,
+      output_fields:,
+      expr:
+    )
+      response = client.connection.post(PATH) do |req|
+        req.body = {
+          collection_name: collection_name,
+          expr: expr
+        }
+        if output_fields
+          req.body[:output_fields] = output_fields
+        end
+      end
+      response.body.empty? ? true : response.body
+    end
   end
 end

lib/milvus/search.rb

@@ -6,15 +6,35 @@ class Search < Base
 
     def post(
       collection_name:,
-      output_fileds:,
-      search_params:
+      output_fields: nil,
+      anns_field:,
+      top_k:,
+      params:,
+      metric_type:,
+      round_decimal: nil,
+      vectors:,
+      dsl_type:
     )
       response = client.connection.post(PATH) do |req|
         req.body = {
           collection_name: collection_name,
-          output_fileds: output_fileds,
-          search_params: search_params
-        }.to_json
+          search_params: [
+            { key: "anns_field", value: anns_field },
+            { key: "topk", value: top_k },
+            { key: "params", value: params },
+            { key: "metric_type", value: metric_type },
+          ],
+          dsl_type: dsl_type,
+          vectors: vectors
+        }
+        if round_decimal
+          req.body[:search_params].push(
+            { key: "round_decimal", value: round_decimal }
+          )
+        end
+        if output_fields
+          req.body[:output_fields] = output_fields
+        end
       end
       response.body.empty? ? true : response.body
     end

lib/milvus/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Milvus
-  VERSION = "0.1.0".freeze
+  VERSION = "0.9.0".freeze
 end

spec/fixtures/collection.json

@@ -0,0 +1,50 @@
+{
+  "status": {},
+  "schema": {
+    "name": "book",
+    "description": "Test book search",
+    "fields": [
+      {
+        "fieldID": 100,
+        "name": "book_id",
+        "is_primary_key": true,
+        "description": "book id",
+        "data_type": 5
+      },
+      {
+        "fieldID": 101,
+        "name": "word_count",
+        "description": "count of words",
+        "data_type": 5
+      },
+      {
+        "fieldID": 102,
+        "name": "book_intro",
+        "description": "embedded vector of book introduction",
+        "data_type": 101,
+        "type_params": [
+          {
+            "key": "dim",
+            "value": "2"
+          }
+        ]
+      }
+    ]
+  },
+  "collectionID": 440928616022607200,
+  "virtual_channel_names": [
+    "by-dev-rootcoord-dml_14_440928616022607200v0",
+    "by-dev-rootcoord-dml_15_440928616022607200v1"
+  ],
+  "physical_channel_names": [
+    "by-dev-rootcoord-dml_14",
+    "by-dev-rootcoord-dml_15"
+  ],
+  "created_timestamp": 440932111134752770,
+  "created_utc_timestamp": 1682022518672,
+  "shards_num": 2,
+  "collection_name": "book"
+}
+
+
+

spec/fixtures/compaction.json

@@ -0,0 +1,4 @@
+{
+  "status": {},
+  "compactionID": 434262132129005569
+}

spec/fixtures/entities_created.json

@@ -0,0 +1,25 @@
+{
+  "status": {},
+  "IDs": {
+    "IdField": {
+      "IntId": {
+        "data": [
+          1,
+          2,
+          3,
+          4,
+          5
+        ]
+      }
+    }
+  },
+  "succ_index": [
+    0,
+    1,
+    2,
+    3,
+    4
+  ],
+  "insert_cnt": 5,
+  "timestamp": 440950902901702658
+}

spec/fixtures/entities_deleted.json

@@ -0,0 +1,15 @@
+{
+  "status": {},
+  "IDs": {
+    "IdField": {
+      "IntId": {
+        "data": [
+          0,
+          1
+        ]
+      }
+    }
+  },
+  "delete_cnt": 2,
+  "timestamp": 440950991152218116
+}