Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): move field output template builder to own file

Browse Source
This commit is contained in:
psychedelicious 2023-11-26 21:06:38 +11:00
parent a703e1b3d3
commit ad9c954a58
2 changed files with 33 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
invokeai/frontend/web/src/features/nodes/util/buildFieldOutputTemplate.ts Normal file
View File

@ -0,0 +1,24 @@
import { startCase } from 'lodash-es';
import { FieldOutputTemplate, FieldType } from '../types/field';
import { InvocationFieldSchema } from '../types/openapi';
export const buildFieldOutputTemplate = (
fieldSchema: InvocationFieldSchema,
fieldName: string,
fieldType: FieldType
): FieldOutputTemplate => {
const { title, description, ui_hidden, ui_type, ui_order } = fieldSchema;
const fieldOutputTemplate: FieldOutputTemplate = {
fieldKind: 'output',
name: fieldName,
title: title ?? (fieldName ? startCase(fieldName) : ''),
description: description ?? '',
type: fieldType,
ui_hidden,
ui_type,
ui_order,
};
return fieldOutputTemplate;
};

23
invokeai/frontend/web/src/features/nodes/util/parseSchema.ts
View File

@ -1,7 +1,9 @@
import { logger } from 'app/logging/logger';
import { parseify } from 'common/util/serialize';
import { reduce, startCase } from 'lodash-es';
import { t } from 'i18next';
import { reduce } from 'lodash-es';
import { OpenAPIV3_1 } from 'openapi-types';
import { FieldTypeParseError, UnsupportedFieldTypeError } from '../types/error';
import { FieldInputTemplate, FieldOutputTemplate } from '../types/field';
import { InvocationTemplate } from '../types/invocation';
import {
@ -11,9 +13,8 @@ import {
isInvocationSchemaObject,
} from '../types/openapi';
import { buildFieldInputTemplate } from './buildFieldInputTemplate';
import { buildFieldOutputTemplate } from './buildFieldOutputTemplate';
import { parseFieldType } from './parseFieldType';
import { FieldTypeParseError, UnsupportedFieldTypeError } from '../types/error';
import { t } from 'i18next';
const RESERVED_INPUT_FIELD_NAMES = ['id', 'type', 'use_cache'];
const RESERVED_OUTPUT_FIELD_NAMES = ['type'];
@ -209,17 +210,11 @@ export const parseSchema = (
return outputsAccumulator;
}
const fieldOutputTemplate: FieldOutputTemplate = {
fieldKind: 'output',
name: propertyName,
title:
property.title ?? (propertyName ? startCase(propertyName) : ''),
description: property.description ?? '',
type: fieldType,
ui_hidden: property.ui_hidden ?? false,
ui_type: property.ui_type,
ui_order: property.ui_order,
};
const fieldOutputTemplate = buildFieldOutputTemplate(
property,
propertyName,
fieldType
);
outputsAccumulator[propertyName] = fieldOutputTemplate;
} catch (e) {