feat: Create feature flag for the support of content parts #513 (#563)

epam · Nov 6, 2024 · 7dcf05e · 7dcf05e
1 parent 82e5256
commit 7dcf05e
Show file tree

Hide file tree

Showing 7 changed files with 45 additions and 18 deletions.
diff --git a/config/src/main/java/com/epam/aidial/core/config/Features.java b/config/src/main/java/com/epam/aidial/core/config/Features.java
@@ -33,4 +33,5 @@ public class Features {
     @JsonAlias({"allowResume", "allow_resume"})
     private Boolean allowResume;
     private Boolean accessibleByPerRequestKey;
+    private Boolean contentPartsSupported;
 }
diff --git a/server/src/main/java/com/epam/aidial/core/server/config/FileConfigStore.java b/server/src/main/java/com/epam/aidial/core/server/config/FileConfigStore.java
@@ -194,5 +194,8 @@ private static void setMissingFeatures(Deployment model, Features features) {
         if (modelFeatures.getAccessibleByPerRequestKey() == null) {
             modelFeatures.setAccessibleByPerRequestKey(features.getAccessibleByPerRequestKey());
         }
+        if (modelFeatures.getContentPartsSupported() == null) {
+            modelFeatures.setContentPartsSupported(features.getContentPartsSupported());
+        }
     }
 }
diff --git a/server/src/main/java/com/epam/aidial/core/server/controller/DeploymentController.java b/server/src/main/java/com/epam/aidial/core/server/controller/DeploymentController.java
@@ -155,6 +155,10 @@ static FeaturesData createFeatures(Features features) {
             data.setAccessibleByPerRequestKey(features.getAccessibleByPerRequestKey());
         }
 
+        if (features.getContentPartsSupported() != null) {
+            data.setContentPartsSupported(features.getContentPartsSupported());
+        }
+
         return data;
     }
 }
diff --git a/server/src/main/java/com/epam/aidial/core/server/data/FeaturesData.java b/server/src/main/java/com/epam/aidial/core/server/data/FeaturesData.java
@@ -21,4 +21,5 @@ public class FeaturesData {
     private boolean folderAttachments = false;
     private boolean allowResume = true;
     private boolean accessibleByPerRequestKey = true;
+    private boolean contentPartsSupported = false;
 }
diff --git a/server/src/test/java/com/epam/aidial/core/server/ApplicationDeploymentApiTest.java b/server/src/test/java/com/epam/aidial/core/server/ApplicationDeploymentApiTest.java
@@ -767,7 +767,8 @@ void testOpenAiApi() {
                       "url_attachments" : false,
                       "folder_attachments" : false,
                       "allow_resume" : true,
-                      "accessible_by_per_request_key" : true
+                      "accessible_by_per_request_key" : true,
+                      "content_parts_supported": false
                     },
                     "defaults" : { },
                     "description_keywords" : [ ],
@@ -814,7 +815,8 @@ void testOpenAiApi() {
                       "url_attachments" : false,
                       "folder_attachments" : false,
                       "allow_resume" : true,
-                      "accessible_by_per_request_key" : true
+                      "accessible_by_per_request_key" : true,
+                      "content_parts_supported": false
                     },
                     "defaults" : { },
                     "description_keywords" : [ ]
@@ -842,7 +844,8 @@ void testOpenAiApi() {
                       "url_attachments" : false,
                       "folder_attachments" : false,
                       "allow_resume" : true,
-                      "accessible_by_per_request_key" : true
+                      "accessible_by_per_request_key" : true,
+                      "content_parts_supported": false
                     },
                     "defaults" : { },
                     "description_keywords" : [ ],

diff --git a/server/src/test/java/com/epam/aidial/core/server/CustomApplicationApiTest.java b/server/src/test/java/com/epam/aidial/core/server/CustomApplicationApiTest.java
@@ -334,7 +334,8 @@ void testApplicationSharing() {
                      "url_attachments" : false,
                      "folder_attachments" : false,
                      "allow_resume":true,
-                     "accessible_by_per_request_key": true
+                     "accessible_by_per_request_key": true,
+                     "content_parts_supported": false
                    },
                    "defaults" : { },
                    "description_keywords":[]
@@ -369,7 +370,8 @@ void testApplicationSharing() {
                                 "url_attachments":false,
                                 "folder_attachments":false,
                                 "allow_resume":true,
-                                "accessible_by_per_request_key": true
+                                "accessible_by_per_request_key": true,
+                                "content_parts_supported": false
                                 },
                             "defaults":{},
                             "description_keywords":[]
@@ -398,7 +400,8 @@ void testApplicationSharing() {
                               "url_attachments" : false,
                               "folder_attachments" : false,
                               "allow_resume":true,
-                              "accessible_by_per_request_key": true
+                              "accessible_by_per_request_key": true,
+                              "content_parts_supported": false
                             },
                             "defaults" : { },
                             "description_keywords":[]
@@ -436,7 +439,8 @@ void testApplicationSharing() {
                                 "url_attachments":false,
                                 "folder_attachments":false,
                                 "allow_resume":true,
-                                "accessible_by_per_request_key": true
+                                "accessible_by_per_request_key": true,
+                                "content_parts_supported": false
                                 },
                             "defaults":{},
                             "description_keywords":[]
@@ -465,7 +469,8 @@ void testApplicationSharing() {
                               "url_attachments" : false,
                               "folder_attachments" : false,
                               "allow_resume":true,
-                              "accessible_by_per_request_key": true
+                              "accessible_by_per_request_key": true,
+                              "content_parts_supported": false
                             },
                             "defaults" : { },
                             "description_keywords":[]
@@ -576,7 +581,8 @@ void testApplicationPublication() {
                                 "url_attachments":false,
                                 "folder_attachments":false,
                                 "allow_resume":true,
-                                "accessible_by_per_request_key": true
+                                "accessible_by_per_request_key": true,
+                                "content_parts_supported": false
                                 },
                             "defaults":{},
                             "description_keywords":[]
@@ -605,7 +611,8 @@ void testApplicationPublication() {
                               "url_attachments" : false,
                               "folder_attachments" : false,
                               "allow_resume":true,
-                              "accessible_by_per_request_key": true
+                              "accessible_by_per_request_key": true,
+                              "content_parts_supported": false
                             },
                             "defaults" : { },
                             "description_keywords":[]
@@ -646,7 +653,8 @@ void testOpenAiApplicationListing() {
                                 "url_attachments":false,
                                 "folder_attachments":false,
                                 "allow_resume": true,
-                                "accessible_by_per_request_key": true
+                                "accessible_by_per_request_key": true,
+                                "content_parts_supported": false
                                 },
                             "defaults":{},
                             "description_keywords":[]
@@ -699,7 +707,8 @@ void testOpenAiApplicationListing() {
                         "url_attachments":false,
                         "folder_attachments":false,
                         "allow_resume": true,
-                        "accessible_by_per_request_key": true
+                        "accessible_by_per_request_key": true,
+                        "content_parts_supported": false
                     },
                     "defaults":{},
                     "description_keywords":[]
@@ -734,7 +743,8 @@ void testOpenAiApplicationListing() {
                                 "url_attachments":false,
                                 "folder_attachments":false,
                                 "allow_resume": true,
-                                "accessible_by_per_request_key": true
+                                "accessible_by_per_request_key": true,
+                                "content_parts_supported": false
                                 },
                             "defaults":{},
                             "description_keywords":[]
@@ -763,7 +773,8 @@ void testOpenAiApplicationListing() {
                               "url_attachments" : false,
                               "folder_attachments" : false,
                               "allow_resume": true,
-                              "accessible_by_per_request_key": true
+                              "accessible_by_per_request_key": true,
+                              "content_parts_supported": false
                             },
                             "defaults" : { },
                             "description_keywords":[]

diff --git a/server/src/test/java/com/epam/aidial/core/server/ListingTest.java b/server/src/test/java/com/epam/aidial/core/server/ListingTest.java
@@ -51,7 +51,8 @@ void testFeaturesEmbedding(Vertx vertx, VertxTestContext context) {
                     { "rate": false, "tokenize": false, "truncate_prompt": false
                     , "system_prompt": true, "tools": false, "seed": false
                     , "url_attachments": false, "folder_attachments": false
-                    , "configuration": false, "allow_resume": true, "accessible_by_per_request_key": true
+                    , "configuration": false, "allow_resume": true, "accessible_by_per_request_key": true,
+                    "content_parts_supported": false
                     }
                 """));
     }
@@ -62,7 +63,8 @@ void testFeaturesModel(Vertx vertx, VertxTestContext context) {
                     { "rate": true, "tokenize": true, "truncate_prompt": true
                     , "system_prompt": true, "tools": true, "seed": true
                     , "url_attachments": true, "folder_attachments": false
-                    , "configuration": false, "allow_resume": true, "accessible_by_per_request_key": true
+                    , "configuration": false, "allow_resume": true, "accessible_by_per_request_key": true,
+                    "content_parts_supported": false
                     }
                 """));
     }
@@ -73,7 +75,8 @@ void testFeaturesApplication(Vertx vertx, VertxTestContext context) {
                     { "rate": true, "tokenize": false, "truncate_prompt": false
                     , "system_prompt": false, "tools": false, "seed": false
                     , "url_attachments": false, "folder_attachments": false
-                    , "configuration": true, "allow_resume": true, "accessible_by_per_request_key": true
+                    , "configuration": true, "allow_resume": true, "accessible_by_per_request_key": true,
+                    "content_parts_supported": false
                     }
                 """));
     }
@@ -84,7 +87,8 @@ void testFeaturesAssistant(Vertx vertx, VertxTestContext context) {
                     { "rate": true, "tokenize": false, "truncate_prompt": false
                     , "system_prompt": true, "tools": false, "seed": false
                     , "url_attachments": false,  "folder_attachments": false
-                    , "configuration": false, "allow_resume": true, "accessible_by_per_request_key": true
+                    , "configuration": false, "allow_resume": true, "accessible_by_per_request_key": true,
+                    "content_parts_supported": false
                     }
                 """));
     }