From 37fc1bf8d27d7823793e91a67bc7269dfc95aac0 Mon Sep 17 00:00:00 2001
From: Teddy <teddy.crepineau@gmail.com>
Date: Mon, 21 Jul 2025 08:04:51 +0200
Subject: [PATCH] feat: added AI Platform Config json (#22449)

---
 .../OpenMetadataApplicationConfig.java        |  4 ++
 .../aiPlatformConfiguration.json              | 66 +++++++++++++++++++
 2 files changed, 70 insertions(+)
 create mode 100644 openmetadata-spec/src/main/resources/json/schema/configuration/aiPlatformConfiguration.json

diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplicationConfig.java b/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplicationConfig.java
index afaa2578e0a..cfefb17f0c4 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplicationConfig.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplicationConfig.java
@@ -31,6 +31,7 @@ import org.openmetadata.schema.api.security.AuthenticationConfiguration;
 import org.openmetadata.schema.api.security.AuthorizerConfiguration;
 import org.openmetadata.schema.api.security.OpsConfig;
 import org.openmetadata.schema.api.security.jwt.JWTTokenConfiguration;
+import org.openmetadata.schema.configuration.AiPlatformConfiguration;
 import org.openmetadata.schema.configuration.LimitsConfiguration;
 import org.openmetadata.schema.security.scim.ScimConfiguration;
 import org.openmetadata.schema.security.secrets.SecretsManagerConfiguration;
@@ -151,6 +152,9 @@ public class OpenMetadataApplicationConfig extends Configuration {
   @JsonProperty("scimConfiguration")
   private ScimConfiguration scimConfiguration;
 
+  @JsonProperty("aiPlatformConfiguration")
+  private AiPlatformConfiguration aiPlatformConfiguration;
+
   @Override
   public String toString() {
     return "catalogConfig{"
diff --git a/openmetadata-spec/src/main/resources/json/schema/configuration/aiPlatformConfiguration.json b/openmetadata-spec/src/main/resources/json/schema/configuration/aiPlatformConfiguration.json
new file mode 100644
index 00000000000..baf80176c78
--- /dev/null
+++ b/openmetadata-spec/src/main/resources/json/schema/configuration/aiPlatformConfiguration.json
@@ -0,0 +1,66 @@
+{
+  "$id": "https://open-metadata.org/schema/configuration/aiPlatformConfiguration.json",
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "title": "AiPlatformConfiguration",
+  "description": "This schema defines the JWT Configuration.",
+  "type": "object",
+  "javaType": "org.openmetadata.schema.configuration.AiPlatformConfiguration",
+  "definitions": {
+    "grpcConfiguration": {
+        "javaType": "org.openmetadata.schema.configuration.GrpcConfiguration",
+        "type": "object",
+        "properties": {
+          "port": {
+            "description": "Host for the gRPC server",
+            "type": "integer"
+          },
+          "maxInboundMessageSize": {
+            "description": "Port for the gRPC server",
+            "type": "integer"
+          },
+          "keepAliveTime": {
+            "description": "Keep alive time for the gRPC server",
+            "type": "integer"
+          },
+          "keepAliveTimeout": {
+              "description": "Keep alive timeout for the gRPC server",
+              "type": "integer"
+          }
+        },
+        "required": ["port"]
+    }
+  },
+  "properties": {
+    "enabled": {
+      "description": "Indicates whether the AI Platform is enabled",
+      "type": "boolean",
+      "default": false
+    },
+    "host": {
+      "description": "Host for the AI Platform server",
+      "type": "string"
+    },
+    "port": {
+      "description": "Port for the AI Platform server",
+      "type": "integer"
+    },
+    "tlsCertPath": {
+      "description": "Path to the TLS certificate for the AI Platform server",
+      "type": "string"
+    },
+    "tlsKeyPath": {
+      "description": "Path to the TLS key for the AI Platform server",
+      "type": "string"
+    },
+    "trustedCertsPath": {
+      "description": "Path to the trusted CA certificate for the AI Platform server",
+      "type": "string"
+    },
+    "grpc": {
+      "description": "gRPC configuration for the AI Platform server",
+      "$ref": "#/definitions/grpcConfiguration"
+    }
+  },
+  "required": ["host", "port", "grpc"],
+  "additionalProperties": false
+}
\ No newline at end of file