From 62b37477efbb4a0404746679d9204909e0668040 Mon Sep 17 00:00:00 2001
From: Teddy <teddy.crepineau@gmail.com>
Date: Mon, 6 Oct 2025 10:47:50 +0200
Subject: [PATCH] MINOR - djl local embedding (#23563)

* feat: config for djl local embedding

* Update generated TypeScript types

---------

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Pere Miquel Brull <peremiquelbrull@gmail.com>
---
 .../configuration/elasticSearchConfiguration.json  | 11 +++++++++++
 .../configuration/elasticSearchConfiguration.ts    | 14 ++++++++++++++
 .../ui/src/generated/settings/settings.ts          | 14 ++++++++++++++
 3 files changed, 39 insertions(+)

diff --git a/openmetadata-spec/src/main/resources/json/schema/configuration/elasticSearchConfiguration.json b/openmetadata-spec/src/main/resources/json/schema/configuration/elasticSearchConfiguration.json
index 94d79668fc3..74b05034f40 100644
--- a/openmetadata-spec/src/main/resources/json/schema/configuration/elasticSearchConfiguration.json
+++ b/openmetadata-spec/src/main/resources/json/schema/configuration/elasticSearchConfiguration.json
@@ -152,6 +152,17 @@
             }
           },
           "additionalProperties": false
+        },
+        "djl": {
+          "description": "Embedding generation using Deep Java Library (DJL)",
+          "type": "object",
+          "properties": {
+            "embeddingModel": {
+              "description": "DJL model name for embedding generation",
+              "type": "string"
+            }
+          },
+          "additionalProperties": false
         }
       },
       "additionalProperties": false
diff --git a/openmetadata-ui/src/main/resources/ui/src/generated/configuration/elasticSearchConfiguration.ts b/openmetadata-ui/src/main/resources/ui/src/generated/configuration/elasticSearchConfiguration.ts
index af9d2cdea7a..694f4032741 100644
--- a/openmetadata-ui/src/main/resources/ui/src/generated/configuration/elasticSearchConfiguration.ts
+++ b/openmetadata-ui/src/main/resources/ui/src/generated/configuration/elasticSearchConfiguration.ts
@@ -89,6 +89,10 @@ export interface NaturalLanguageSearch {
      * AWS Bedrock configuration for natural language processing
      */
     bedrock?: Bedrock;
+    /**
+     * Embedding generation using Deep Java Library (DJL)
+     */
+    djl?: Djl;
     /**
      * The provider to use for generating vector embeddings (e.g., bedrock, openai).
      */
@@ -137,6 +141,16 @@ export interface Bedrock {
     useIamRole?: boolean;
 }
 
+/**
+ * Embedding generation using Deep Java Library (DJL)
+ */
+export interface Djl {
+    /**
+     * DJL model name for embedding generation
+     */
+    embeddingModel?: string;
+}
+
 /**
  * This schema defines the language options available for search index mappings.
  */
diff --git a/openmetadata-ui/src/main/resources/ui/src/generated/settings/settings.ts b/openmetadata-ui/src/main/resources/ui/src/generated/settings/settings.ts
index c5205d9587e..703b048f8e3 100644
--- a/openmetadata-ui/src/main/resources/ui/src/generated/settings/settings.ts
+++ b/openmetadata-ui/src/main/resources/ui/src/generated/settings/settings.ts
@@ -1847,6 +1847,10 @@ export interface NaturalLanguageSearch {
      * AWS Bedrock configuration for natural language processing
      */
     bedrock?: Bedrock;
+    /**
+     * Embedding generation using Deep Java Library (DJL)
+     */
+    djl?: Djl;
     /**
      * The provider to use for generating vector embeddings (e.g., bedrock, openai).
      */
@@ -1895,6 +1899,16 @@ export interface Bedrock {
     useIamRole?: boolean;
 }
 
+/**
+ * Embedding generation using Deep Java Library (DJL)
+ */
+export interface Djl {
+    /**
+     * DJL model name for embedding generation
+     */
+    embeddingModel?: string;
+}
+
 /**
  * Configuration for Natural Language Query capabilities
  */