Merge pull request #3012 from the-deep/feature/add-secondary-tagging-…

…widgets-in-llm Support singleselect, multiselect, scale, organigram LLM integration
the-deep · Dec 27, 2024 · 0b5fd43 · 0b5fd43
2 parents 6c2b8ab + a215d0d
commit 0b5fd43
Showing 1 changed file with 119 additions and 3 deletions.
diff --git a/app/components/LeftPaneEntries/AssistItem/utils.ts b/app/components/LeftPaneEntries/AssistItem/utils.ts
@@ -6,8 +6,16 @@ import {
 import {
     Matrix1dWidget,
     Matrix1dValue,
-    Matrix2dValue,
     Matrix2dWidget,
+    Matrix2dValue,
+    SingleSelectWidget,
+    SingleSelectValue,
+    MultiSelectWidget,
+    MultiSelectValue,
+    ScaleWidget,
+    ScaleValue,
+    OrganigramWidget,
+    OrganigramValue,
 } from '#types/newAnalyticalFramework';
 
 import {
@@ -17,14 +25,17 @@ import {
 import {
     PartialAttributeType,
 } from '#components/entry/schema';
+import {
+    getOrganigramFlatOptions,
+} from '#components/framework/CompactAttributeInput/OrganigramWidgetInput/utils';
 
 type Matrix1dWidgetAttribute = getType<PartialAttributeType, { widgetType: 'MATRIX1D' }>;
 type Matrix2dWidgetAttribute = getType<PartialAttributeType, { widgetType: 'MATRIX2D' }>;
-/*
-type ScaleWidgetAttribute = getType<PartialAttributeType, { widgetType: 'SCALE' }>;
 type SingleSelectWidgetAttribute = getType<PartialAttributeType, { widgetType: 'SELECT' }>;
 type MultiSelectWidgetAttribute = getType<PartialAttributeType, { widgetType: 'MULTISELECT' }>;
 type OrganigramWidgetAttribute = getType<PartialAttributeType, { widgetType: 'ORGANIGRAM' }>;
+type ScaleWidgetAttribute = getType<PartialAttributeType, { widgetType: 'SCALE' }>;
+/*
 type GeoLocationWidgetAttribute = getType<PartialAttributeType, { widgetType: 'GEO' }>;
 */
 
@@ -163,3 +174,108 @@ export function createMatrix2dAttrFromTags(
         data: { value },
     };
 }
+
+// TODO: Write tests
+export function createSingleSelectAttrFromTags(
+    unsafeMapping: SingleSelectValue | undefined,
+    widget: SingleSelectWidget,
+): SingleSelectWidgetAttribute | undefined {
+    const mapping = unsafeMapping as unknown;
+    if (!mapping || typeof mapping !== 'string') {
+        return undefined;
+    }
+
+    const valueOption = widget.properties?.options.find((item) => item.key === mapping);
+
+    if (!valueOption) {
+        return undefined;
+    }
+
+    return {
+        clientId: randomString(),
+        widget: widget.id,
+        widgetVersion: widget.version,
+        widgetType: 'SELECT',
+        data: { value: mapping },
+    };
+}
+
+// TODO: Write tests
+export function createMultiSelectAttrFromTags(
+    unsafeMapping: MultiSelectValue | undefined,
+    widget: MultiSelectWidget,
+): MultiSelectWidgetAttribute | undefined {
+    const mapping = unsafeMapping as unknown;
+    if (!mapping || !isValidStringArray(mapping)) {
+        return undefined;
+    }
+
+    const validKeys = widget.properties?.options?.map((item) => item.key);
+
+    const validOptions = mapping.filter((item) => validKeys?.includes(item));
+
+    if (validOptions.length < 1) {
+        return undefined;
+    }
+
+    return {
+        clientId: randomString(),
+        widget: widget.id,
+        widgetVersion: widget.version,
+        widgetType: 'MULTISELECT',
+        data: { value: mapping },
+    };
+}
+
+// TODO: Write tests
+export function createScaleAttrFromTags(
+    unsafeMapping: ScaleValue | undefined,
+    widget: ScaleWidget,
+): ScaleWidgetAttribute | undefined {
+    const mapping = unsafeMapping as unknown;
+    if (!mapping || typeof mapping !== 'string') {
+        return undefined;
+    }
+
+    const valueOption = widget.properties?.options.find((item) => item.key === mapping);
+
+    if (!valueOption) {
+        return undefined;
+    }
+
+    return {
+        clientId: randomString(),
+        widget: widget.id,
+        widgetVersion: widget.version,
+        widgetType: 'SCALE',
+        data: { value: mapping },
+    };
+}
+
+// TODO: Write tests
+export function createOrganigramAttrFromTags(
+    unsafeMapping: OrganigramValue | undefined,
+    widget: OrganigramWidget,
+): OrganigramWidgetAttribute | undefined {
+    const mapping = unsafeMapping as unknown;
+    if (!mapping || !isValidStringArray(mapping)) {
+        return undefined;
+    }
+
+    const validKeys = getOrganigramFlatOptions(widget.properties?.options)
+        .map((item) => item.key);
+
+    const validOptions = mapping.filter((item) => validKeys?.includes(item));
+
+    if (validOptions.length < 1) {
+        return undefined;
+    }
+
+    return {
+        clientId: randomString(),
+        widget: widget.id,
+        widgetVersion: widget.version,
+        widgetType: 'ORGANIGRAM',
+        data: { value: mapping },
+    };
+}