Updated Get API to add _source_include_vectors

Mikep86 · Sep 23, 2024 · 07bca98 · 07bca98
1 parent bef691c
commit 07bca98
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/rest-api-spec/src/main/resources/rest-api-spec/api/get.json b/rest-api-spec/src/main/resources/rest-api-spec/api/get.json
@@ -68,6 +68,10 @@
         "type":"list",
         "description":"A list of fields to extract and return from the _source field"
       },
+      "_source_include_vectors": {
+        "type":"boolean",
+        "description":"True or false to include vector fields in _source"
+      },
       "version":{
         "type":"number",
         "description":"Explicit version number for concurrency control"

diff --git a/server/src/main/java/org/elasticsearch/search/fetch/subphase/FetchSourceContext.java b/server/src/main/java/org/elasticsearch/search/fetch/subphase/FetchSourceContext.java
@@ -199,6 +199,7 @@ public static FetchSourceContext parseFromRestRequest(RestRequest request) {
         Boolean fetchSource = null;
         String[] sourceExcludes = null;
         String[] sourceIncludes = null;
+        Boolean includeVectors = null;
 
         String source = request.param("_source");
         if (source != null) {
@@ -221,8 +222,13 @@ public static FetchSourceContext parseFromRestRequest(RestRequest request) {
             sourceExcludes = Strings.splitStringByCommaToArray(sExcludes);
         }
 
-        if (fetchSource != null || sourceIncludes != null || sourceExcludes != null) {
-            return FetchSourceContext.of(fetchSource == null || fetchSource, sourceIncludes, sourceExcludes);
+        String sIncludeVectors = request.param("_source_include_vectors");
+        if (sIncludeVectors != null) {
+            includeVectors = Booleans.parseBoolean(sIncludeVectors);
+        }
+
+        if (fetchSource != null || sourceIncludes != null || sourceExcludes != null || includeVectors != null) {
+            return FetchSourceContext.of(fetchSource == null || fetchSource, sourceIncludes, sourceExcludes, includeVectors);
         }
         return null;
     }