feat: error handling

Author: nicosammito

src/packages/ce/src/ai/components/AIChatComponent.tsx

@@ -43,7 +43,7 @@ export interface AIChatComponentProps {
     projectId: NamespaceProject['id']
     flowId?: Flow['id']
     prompt?: string
-    onData?: (data: any) => void
+    onData?: (data: any) => string | void
 
 }
 
@@ -72,8 +72,13 @@ export const AIChatComponent: React.FC<AIChatComponentProps> = (props) => {
         onData: (data) => {
             setIsAIActive(true)
             if (data.data.data?.aiGenerateFlow?.flow) {
-                onData?.(data.data.data?.aiGenerateFlow)
-                setPromptState("")
+                const result = onData?.(data.data.data?.aiGenerateFlow)
+                if (typeof result === "string") {
+                    setAiErrorMessage(result)
+                } else {
+                    setPromptState("")
+                }
+                setExecutionIdentifier(null)
             } else if (data.data.data?.aiGenerateFlow?.flow === null) {
                 setExecutionIdentifier(null)
                 setAiErrorMessage("Generation failed. Try another model.")
@@ -83,6 +88,7 @@ export const AIChatComponent: React.FC<AIChatComponentProps> = (props) => {
         onError: () => {
             setIsAIActive(false)
             setAiErrorMessage("Generation failed. Try another model.")
+            setExecutionIdentifier(null)
         },
     })
 

src/packages/ce/src/flow/components/panels/FlowPanelControlComponent.tsx

@@ -74,18 +74,18 @@ export const FlowPanelControlComponent: React.FC<FlowPanelControlComponentProps>
 
     const onAIData = React.useCallback((payload: AiGenerateFlowSubscriptionPayload) => {
         const aiFlow = payload?.flow
-        if (!aiFlow) return
+        if (!aiFlow) return "No flow returned. Try again."
 
         const currentFlow = flowService.getById(flowId, {namespaceId, projectId})
-        if (!currentFlow) return
+        if (!currentFlow) return "Flow not found. Try again."
 
         const currentFlowName = currentFlow.name ?? undefined
         const existingNames = (flowService.values({namespaceId, projectId}) ?? [])
             .map(f => f.name)
             .filter((n): n is string => !!n && n !== currentFlowName)
 
         const flowInput = mapAiGenerationFlowToFlowInput(aiFlow, {existingNames})
-        if (!flowInput) return
+        if (!flowInput) return "Invalid flow type. Try another model."
 
         const oldFlowSnapshot: FlowView = JSON.parse(JSON.stringify(currentFlow))
 

src/packages/ce/src/flow/pages/FlowOverviewPage.tsx

@@ -61,14 +61,14 @@ export const FlowOverviewPage: React.FC = () => {
 
     const onAIData = React.useCallback((payload: AiGenerateFlowSubscriptionPayload) => {
         const aiFlow = payload?.flow
-        if (!aiFlow) return
+        if (!aiFlow) return "No flow returned. Try again."
 
         const existingNames = (flowService.values({namespaceId, projectId}) ?? [])
             .map(f => f.name)
             .filter((n): n is string => !!n)
 
         const flowInput = mapAiGenerationFlowToFlowInput(aiFlow, {existingNames})
-        if (!flowInput) return
+        if (!flowInput) return "Invalid flow type. Try another model."
 
         flowService.flowCreate({
             flow: flowInput,