feat: added listing params for attachment support (#44) (#34)

.gitignore

@@ -2,4 +2,5 @@
 .idea/
 .vscode/
 build/
-bin/
+bin/
+*.log

src/main/java/com/epam/aidial/core/config/Deployment.java

@@ -2,6 +2,7 @@
 
 import lombok.Data;
 
+import java.util.List;
 import java.util.Set;
 
 @Data
@@ -23,4 +24,6 @@ public abstract class Deployment {
      */
     private boolean forwardApiKey = true;
     private String rateEndpoint;
+    private List<String> inputAttachmentTypes;
+    private Integer maxInputAttachments;
 }

src/main/java/com/epam/aidial/core/controller/ApplicationController.java

@@ -57,6 +57,8 @@ private static ApplicationData createApplication(Application application) {
         data.setDisplayName(application.getDisplayName());
         data.setIconUrl(application.getIconUrl());
         data.setDescription(application.getDescription());
+        data.setInputAttachmentTypes(application.getInputAttachmentTypes());
+        data.setMaxInputAttachments(application.getMaxInputAttachments());
         return data;
     }
 }

src/main/java/com/epam/aidial/core/controller/AssistantController.java

@@ -58,6 +58,8 @@ private static AssistantData createAssistant(Assistant assistant) {
         data.setIconUrl(assistant.getIconUrl());
         data.setDescription(assistant.getDescription());
         data.setAddons(assistant.getAddons());
+        data.setInputAttachmentTypes(assistant.getInputAttachmentTypes());
+        data.setMaxInputAttachments(assistant.getMaxInputAttachments());
         return data;
     }
 }

src/main/java/com/epam/aidial/core/controller/ModelController.java

@@ -70,6 +70,8 @@ private static ModelData createModel(Model model) {
         }
 
         data.setLimits(createLimits(model.getLimits()));
+        data.setInputAttachmentTypes(model.getInputAttachmentTypes());
+        data.setMaxInputAttachments(model.getMaxInputAttachments());
         return data;
     }
 

src/main/java/com/epam/aidial/core/data/DeploymentData.java

@@ -5,6 +5,8 @@
 import com.fasterxml.jackson.databind.annotation.JsonNaming;
 import lombok.Data;
 
+import java.util.List;
+
 @Data
 @JsonInclude(JsonInclude.Include.NON_NULL)
 @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
@@ -23,4 +25,6 @@ public class DeploymentData {
     private long createdAt = 1672534800;
     private long updatedAt = 1672534800;
     private ScaleSettingsData scaleSettings = new ScaleSettingsData();
+    private List<String> inputAttachmentTypes;
+    private Integer maxInputAttachments;
 }

src/main/resources/aidial.config.json

@@ -35,7 +35,9 @@
       "endpoint": "http://localhost:7001/openai/deployments/10k/chat/completions",
       "displayName": "10k",
       "iconUrl": "http://localhost:7001/logo10k.png",
-      "description": "Some description of the application for testing"
+      "description": "Some description of the application for testing",
+      "inputAttachmentTypes": ["*/*"],
+      "maxInputAttachments": 1
     }
   },
   "models": {