Update workspace nodes/edges; misc fixes (#394) (#396)

Signed-off-by: Tyler Ohlsen <[email protected]>
(cherry picked from commit c8bdacc)

Co-authored-by: Tyler Ohlsen <[email protected]>

common/constants.ts

@@ -86,6 +86,15 @@ export const OPENAI_DIMENSIONS = {
   [`text-embedding-ada-002`]: 1536,
 };
 
+// Amazon BedRock
+export const BEDROCK_DIMENSIONS = {
+  [`amazon.titan-embed-text-v1`]: 1536,
+  [`amazon.titan-embed-text-v2`]: 1024,
+  [`amazon.titan-embed-image-v1`]: 1024,
+  [`cohere.embed-english-v3`]: 1024, // same as Cohere directly
+  [`cohere.embed-multilingual-v3`]: 1024, // same as Cohere directly
+};
+
 /**
  * Various constants pertaining to Workflow configs
  */
@@ -135,24 +144,13 @@ export enum NODE_CATEGORY {
  * A base set of component classes / types.
  */
 export enum COMPONENT_CLASS {
-  // Indexer-related classes
-  INDEXER = 'indexer',
-  KNN_INDEXER = 'knn_indexer',
-  // Retriever-related classes
-  RETRIEVER = 'retriever',
-  // Transformer-related classes
+  INDEX = 'index',
+  KNN_INDEX = 'knn_index',
   TRANSFORMER = 'transformer',
-  JSON_TO_JSON_TRANSFORMER = 'json_to_json_transformer',
   ML_TRANSFORMER = 'ml_transformer',
-  RESULTS_TRANSFORMER = 'results_transformer',
-  // Query-related classes
-  QUERY = 'query',
-  MATCH_QUERY = 'match_query',
-  NEURAL_QUERY = 'neural_query',
-  // Document-related classes
+  SEARCH_REQUEST = 'search_request',
   DOCUMENT = 'document',
-  // Results-related classes
-  RESULTS = 'results',
+  SEARCH_RESPONSE = 'search_response',
 }
 
 /**

public/component_types/index.ts

@@ -4,5 +4,5 @@
  */
 
 export * from './transformer';
-export * from './indexer';
+export * from './indices';
 export * from './other';

public/component_types/indices/base_index.ts

@@ -0,0 +1,42 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { COMPONENT_CATEGORY, COMPONENT_CLASS } from '../../../common';
+import { BaseComponent } from '../base_component';
+
+/**
+ * A basic index placeholder UI component. Input/output depends on ingest or search context.
+ * Does not have any functionality.
+ */
+export class BaseIndex extends BaseComponent {
+  constructor(category: COMPONENT_CATEGORY) {
+    super();
+    this.type = COMPONENT_CLASS.INDEX;
+    this.label = 'Index';
+    this.description = 'An OpenSearch index';
+    this.inputs = [
+      {
+        id:
+          category === COMPONENT_CATEGORY.INGEST
+            ? 'document'
+            : 'search_request',
+        label:
+          category === COMPONENT_CATEGORY.INGEST
+            ? 'Document'
+            : 'Search Request',
+        acceptMultiple: false,
+      },
+    ];
+    this.outputs =
+      category === COMPONENT_CATEGORY.INGEST
+        ? []
+        : [
+            {
+              id: 'search_response',
+              label: 'Search Response',
+            },
+          ];
+  }
+}

public/component_types/indexer/index.ts → public/component_types/indices/index.ts

@@ -3,5 +3,5 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-export * from './base_indexer';
-export * from './knn_indexer';
+export * from './base_index';
+export * from './knn_index';

public/component_types/indices/knn_index.ts

@@ -0,0 +1,20 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { COMPONENT_CATEGORY, COMPONENT_CLASS } from '../../../common';
+import { BaseIndex } from './base_index';
+
+/**
+ * A basic knn index placeholder UI component. Input/output depends on ingest or search context.
+ * Does not have any functionality.
+ */
+export class KnnIndex extends BaseIndex {
+  constructor(category: COMPONENT_CATEGORY) {
+    super(category);
+    this.type = COMPONENT_CLASS.KNN_INDEX;
+    this.label = 'k-NN Index';
+    this.description = 'A specialized k-NN index';
+  }
+}

public/component_types/other/document.tsx

@@ -7,7 +7,7 @@ import { COMPONENT_CLASS } from '../../../common';
 import { BaseComponent } from '../base_component';
 
 /**
- * A basic Document placeholder UI component.
+ * A basic document placeholder UI component.
  * Does not have any functionality.
  */
 export class Document extends BaseComponent {
@@ -18,8 +18,8 @@ export class Document extends BaseComponent {
     this.description = 'A document to be ingested';
     this.outputs = [
       {
-        id: 'output',
-        label: 'Output',
+        id: 'document',
+        label: 'Document',
       },
     ];
   }

public/component_types/other/index.ts

@@ -4,5 +4,5 @@
  */
 
 export * from './document';
-export * from './results';
-export * from './query';
+export * from './search_response';
+export * from './search_request';

public/component_types/other/search_request.tsx

@@ -0,0 +1,26 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { COMPONENT_CLASS } from '../../../common';
+import { BaseComponent } from '../base_component';
+
+/**
+ * A basic search request placeholder UI component.
+ * Does not have any functionality.
+ */
+export class SearchRequest extends BaseComponent {
+  constructor() {
+    super();
+    this.type = COMPONENT_CLASS.SEARCH_REQUEST;
+    this.label = 'Search Request';
+    this.description = 'An OpenSearch search request';
+    this.outputs = [
+      {
+        id: 'search_request',
+        label: this.label,
+      },
+    ];
+  }
+}

public/component_types/other/search_response.tsx

@@ -0,0 +1,23 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { COMPONENT_CLASS } from '../../../common';
+import { BaseComponent } from '../base_component';
+
+/**
+ * A basic search response placeholder UI component.
+ * Does not have any functionality.
+ */
+export class SearchResponse extends BaseComponent {
+  constructor() {
+    super();
+    this.type = COMPONENT_CLASS.SEARCH_RESPONSE;
+    this.label = 'Search Response';
+    this.description = 'OpenSearch search response';
+    this.inputs = [
+      { id: 'search_response', label: this.label, acceptMultiple: false },
+    ];
+  }
+}