Fix/bedrock support 1.9.1 (#22908)

* enable bedrock embedding service

* Update generated TypeScript types

* add configs for embedding provider

* Update generated TypeScript types

* make embedding dimension dynamic

* Update generated TypeScript types

---------

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

conf/openmetadata.yaml

@@ -314,10 +314,13 @@ elasticsearch:
   searchIndexFactoryClassName : org.openmetadata.service.search.SearchIndexFactory
   naturalLanguageSearch:
     enabled: false
+    embeddingProvider: ${EMBEDDING_PROVIDER:-bedrock}
     providerClass: ${NATURAL_LANGUAGE_SEARCH_PROVIDER_CLASS:-org.openmetadata.service.search.nlq.NoOpNLQService}
     bedrock:
       region: ${AWS_BEDROCK_REGION:-""}
       modelId: ${AWS_BEDROCK_MODEL_ID:-""}
+      embeddingModelId: ${AWS_BEDROCK_EMBED_MODEL_ID:-""}
+      embeddingDimension: ${AWS_BEDROCK_EMBEDDING_DIMENSION:-""}
       accessKey: ${AWS_BEDROCK_ACCESS_KEY:-""}
       secretKey: ${AWS_BEDROCK_SECRET_KEY:-""}
       useIamRole: ${AWS_BEDROCK_USE_IAM:-"false"}

openmetadata-spec/src/main/resources/json/schema/configuration/elasticSearchConfiguration.json

@@ -102,6 +102,11 @@
           "type": "boolean",
           "default": false
         },
+        "embeddingProvider": {
+          "description": "The provider to use for generating vector embeddings (e.g., bedrock, openai).",
+          "type": "string",
+          "default": "bedrock"
+        },
         "providerClass": {
           "description": "Fully qualified class name of the NLQService implementation to use",
           "type": "string",
@@ -121,6 +126,16 @@
               "type": "string",
               "default": "anthropic.claude-v2"
             },
+            "embeddingModelId": {
+              "description": "Bedrock embedding model identifier to use for vector search",
+              "type": "string",
+              "default": "amazon.titan-embed-text-v2:0"
+            },
+            "embeddingDimension": {
+              "description": "Dimension of the embedding vector",
+              "type": "integer",
+              "default": 524
+            },
             "accessKey": {
               "description": "AWS access key for Bedrock service authentication",
               "type": "string"

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;
+    /**
+     * The provider to use for generating vector embeddings (e.g., bedrock, openai).
+     */
+    embeddingProvider?: string;
     /**
      * Enable or disable natural language search
      */
@@ -107,6 +111,14 @@ export interface Bedrock {
      * AWS access key for Bedrock service authentication
      */
     accessKey?: string;
+    /**
+     * Dimension of the embedding vector
+     */
+    embeddingDimension?: number;
+    /**
+     * Bedrock embedding model identifier to use for vector search
+     */
+    embeddingModelId?: string;
     /**
      * Bedrock model identifier to use for query transformation
      */

openmetadata-ui/src/main/resources/ui/src/generated/settings/settings.ts

@@ -1328,6 +1328,10 @@ export interface NaturalLanguageSearch {
      * AWS Bedrock configuration for natural language processing
      */
     bedrock?: Bedrock;
+    /**
+     * The provider to use for generating vector embeddings (e.g., bedrock, openai).
+     */
+    embeddingProvider?: string;
     /**
      * Enable or disable natural language search
      */
@@ -1346,6 +1350,14 @@ export interface Bedrock {
      * AWS access key for Bedrock service authentication
      */
     accessKey?: string;
+    /**
+     * Dimension of the embedding vector
+     */
+    embeddingDimension?: number;
+    /**
+     * Bedrock embedding model identifier to use for vector search
+     */
+    embeddingModelId?: string;
     /**
      * Bedrock model identifier to use for query transformation
      */