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): add nodesAllowlist to config

Browse Source
This commit is contained in:
psychedelicious
2023-09-07 11:11:34 +10:00
committed by Kent Keirsey
parent cc92ce3da5
commit 74cc409c72
4 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -61,13 +61,21 @@ const isNotInDenylist = (schema: InvocationSchemaObject) =>
export const parseSchema = (
openAPI: OpenAPIV3.Document,
nodesDenylistExtra: string[] = []
nodesAllowlistExtra: string[] | undefined = undefined,
nodesDenylistExtra: string[] | undefined = undefined
): Record<string, InvocationTemplate> => {
const filteredSchemas = Object.values(openAPI.components?.schemas ?? {})
.filter(isInvocationSchemaObject)
.filter(isNotInDenylist)
.filter(
(schema) => !nodesDenylistExtra.includes(schema.properties.type.default)
.filter((schema) =>
nodesAllowlistExtra
? nodesAllowlistExtra.includes(schema.properties.type.default)
: true
)
.filter((schema) =>
nodesDenylistExtra
? !nodesDenylistExtra.includes(schema.properties.type.default)
: true
);
const invocations = filteredSchemas.reduce<