enable with_vector flag in api #50

qdrant · Jan 30, 2022 · 1a909e6 · 1a909e6
1 parent 376c2e0
commit 1a909e6
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 2 deletions.
diff --git a/docs/redoc/master/openapi.json b/docs/redoc/master/openapi.json
@@ -1359,6 +1359,10 @@
                 "nullable": true
               }
             ]
+          },
+          "with_vector": {
+            "nullable": true,
+            "type": "boolean"
           }
         },
         "required": [

diff --git a/docs/redoc/openapi.json b/docs/redoc/openapi.json
@@ -1359,6 +1359,10 @@
                 "nullable": true
               }
             ]
+          },
+          "with_vector": {
+            "nullable": true,
+            "type": "boolean"
           }
         },
         "required": [

diff --git a/openapi/models.json b/openapi/models.json
@@ -1340,6 +1340,10 @@
                                 "nullable": true
                             }
                         ]
+                    },
+                    "with_vector": {
+                        "type": "boolean",
+                        "nullable": true
                     }
                 }
             },

diff --git a/openapi/openapi-merged.yaml b/openapi/openapi-merged.yaml
@@ -2104,6 +2104,9 @@ components:
           anyOf:
             - $ref: '#/components/schemas/WithPayloadInterface'
             - nullable: true
+        with_vector:
+          nullable: true
+          type: boolean
       required:
         - ids
       type: object

diff --git a/src/actix/api/retrieve_api.rs b/src/actix/api/retrieve_api.rs
@@ -16,6 +16,7 @@ use crate::actix::helpers::process_response;
 pub struct PointRequest {
     pub ids: Vec<PointIdType>,
     pub with_payload: Option<WithPayloadInterface>,
+    pub with_vector: Option<bool>,
 }
 
 async fn do_get_point(
@@ -37,8 +38,13 @@ async fn do_get_points(
         .with_payload
         .unwrap_or(WithPayloadInterface::Bool(true));
     let with_payload = WithPayload::from(with_payload_interface);
-    toc.retrieve(collection_name, &request.ids, &with_payload, true)
-        .await
+    toc.retrieve(
+        collection_name,
+        &request.ids,
+        &with_payload,
+        request.with_vector.unwrap_or(false),
+    )
+    .await
 }
 
 async fn scroll_get_points(