Schema improvement and output typing fix

None of the changes are ideal, but they fix bugs and prevent us from
being locked into a bad format in the future

Author: AustinMroz

src/core/graph/widgets/dynamicWidgets.ts

@@ -166,6 +166,7 @@ export function applyDynamicInputs(
   inputSpec: InputSpecV2
 ): boolean {
   if (!(inputSpec.type in dynamicInputs)) return false
+  //TODO: perform parsing/validation of inputSpec here?
   dynamicInputs[inputSpec.type](node, inputSpec)
   return true
 }
@@ -233,7 +234,9 @@ export function applyMatchType(node: LGraphNode, inputSpec: InputSpecV2) {
   const { addNodeInput } = useLitegraphService()
   const name = inputSpec.name
   const { allowed_types, template_id } = (
-    inputSpec as { template: { allowed_types: string; template_id: string } }
+    inputSpec as InputSpecV2 & {
+      template: { allowed_types: string; template_id: string }
+    }
   ).template
   const typedSpec = { ...inputSpec, type: allowed_types }
   addNodeInput(node, typedSpec)
@@ -259,6 +262,8 @@ export function applyMatchType(node: LGraphNode, inputSpec: InputSpecV2) {
         if (!matchGroup) return
         if (iscon && linf) {
           const { output, subgraphInput } = linf.resolve(this.graph)
+          //TODO: fix this bug globally. A link type (and therefore color)
+          //should be the combinedType of origin and target type
           const connectingType = (output ?? subgraphInput)?.type
           if (connectingType) linf.type = connectingType
         }
@@ -302,6 +307,22 @@ export function applyMatchType(node: LGraphNode, inputSpec: InputSpecV2) {
   }
   augmentedNode.comfyMatchType[template_id] ??= {}
   augmentedNode.comfyMatchType[template_id][name] = allowed_types
+
+  //TODO: instead apply on output add?
+  //ensure outputs get updated
+  const index = node.inputs.length - 1
+  const input = node.inputs.at(-1)!
+  setTimeout(
+    () =>
+      node.onConnectionsChange!(
+        LiteGraph.INPUT,
+        index,
+        false,
+        undefined,
+        input
+      ),
+    50
+  )
 }
 
 export function applyAutoGrow(node: LGraphNode, inputSpec: InputSpecV2) {

src/schemas/nodeDefSchema.ts

@@ -14,10 +14,6 @@ const zRemoteWidgetConfig = z.object({
   timeout: z.number().gte(0).optional(),
   max_retries: z.number().gte(0).optional()
 })
-const zWidgetTemplate = z.object({
-  template_id: z.string(),
-  allowed_types: z.string().optional()
-})
 const zMultiSelectOption = z.object({
   placeholder: z.string().optional(),
   chip: z.boolean().optional()
@@ -32,7 +28,6 @@ export const zBaseInputOptions = z
     hidden: z.boolean().optional(),
     advanced: z.boolean().optional(),
     widgetType: z.string().optional(),
-    template: zWidgetTemplate.optional(),
     /** Backend-only properties. */
     rawLink: z.boolean().optional(),
     lazy: z.boolean().optional()