coupling? I've never heard of it

Author: benceruleanlu

src/components/graph/vueWidgets/WidgetChart.vue

@@ -13,19 +13,19 @@ import type { ChartData } from 'chart.js'
 import Chart from 'primevue/chart'
 import { computed } from 'vue'
 
+import type { ChartInputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import type { SimplifiedWidget } from '@/types/simplifiedWidget'
 
+type ChartWidgetOptions = NonNullable<ChartInputSpec['options']>
+
 const value = defineModel<ChartData>({ required: true })
 
 const props = defineProps<{
-  widget: SimplifiedWidget<ChartData>
+  widget: SimplifiedWidget<ChartData, ChartWidgetOptions>
   readonly?: boolean
 }>()
 
-const chartType = computed(() => {
-  const type = props.widget.options?.type
-  return type === 'line' ? 'line' : 'bar'
-})
+const chartType = computed(() => props.widget.options?.type ?? 'line')
 
 const chartData = computed(() => value.value || { labels: [], datasets: [] })
 

src/composables/widgets/useChartWidget.ts

@@ -1,19 +1,26 @@
 import type { LGraphNode } from '@comfyorg/litegraph'
 import type { IChartWidget } from '@comfyorg/litegraph/dist/types/widgets'
 
-import type {
-  ChartInputSpec,
-  InputSpec as InputSpecV2
+import {
+  type ChartInputSpec,
+  type InputSpec as InputSpecV2,
+  isChartInputSpec
 } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import type { ComfyWidgetConstructorV2 } from '@/scripts/widgets'
 
 export const useChartWidget = (): ComfyWidgetConstructorV2 => {
   return (node: LGraphNode, inputSpec: InputSpecV2): IChartWidget => {
+    if (!isChartInputSpec(inputSpec)) {
+      throw new Error('Invalid input spec for chart widget')
+    }
+
     const { name, options = {} } = inputSpec as ChartInputSpec
 
+    const chartType = options.type || 'line'
+
     const widget = node.addWidget('chart', name, options.data || {}, () => {}, {
       serialize: true,
-      type: options.type || 'line',
+      type: chartType,
       ...options
     }) as IChartWidget