Fix CLI new chat handling (#1591)

<!--
Ensure the title clearly reflects what was changed.
Provide a clear and concise description of the changes made. The PR
should only contain the changes related to the issue, and no other
unrelated changes.
-->

Fixes OPS-2979

Author: alexandrudanpop

packages/react-ui/src/app/features/ai/lib/assistant-ui-chat-hook.ts

@@ -6,7 +6,7 @@ import { useChat } from '@ai-sdk/react';
 import { AssistantRuntime } from '@assistant-ui/react';
 import { useAISDKRuntime } from '@assistant-ui/react-ai-sdk';
 import { toast } from '@openops/components/ui';
-import { flowHelper } from '@openops/shared';
+import { flowHelper, FlowVersion } from '@openops/shared';
 import { getFrontendToolDefinitions } from '@openops/ui-kit';
 import { useQuery, useQueryClient } from '@tanstack/react-query';
 import { DefaultChatTransport, ToolSet, UIMessage } from 'ai';
@@ -135,14 +135,10 @@ export const useAssistantChat = ({
           stepDetails &&
           flowId
         ) {
-          const stepId =
-            flowHelper.getStep(context.flowVersion, context.selectedStep)?.id ??
-            context.selectedStep;
-
           return await aiChatApi.open(
             flowId,
             getBlockName(stepDetails),
-            stepId,
+            getStepId(context.flowVersion, context.selectedStep),
             getActionName(stepDetails),
           );
         }
@@ -337,7 +333,16 @@ export const useAssistantChat = ({
           chatMode === ChatMode.StepSettings
         ) {
           await aiChatApi.delete(oldChatId);
+
+          const conversation = await aiChatApi.open(
+            context.flowVersion.flowId,
+            getBlockName(stepDetails),
+            getStepId(context.flowVersion, context.selectedStep),
+            getActionName(stepDetails),
+          );
+
           chat.setMessages([]);
+          onChatIdChange(conversation.chatId);
         } else {
           onChatIdChange(null);
         }
@@ -351,7 +356,7 @@ export const useAssistantChat = ({
         `There was an error canceling the current run and invalidating queries while creating a new chat: ${error}`,
       );
     }
-  }, [chatId, chat, getBuilderState, chatMode, onChatIdChange]);
+  }, [chatId, chat, getBuilderState, chatMode, stepDetails, onChatIdChange]);
 
   return {
     runtime,
@@ -363,3 +368,7 @@ export const useAssistantChat = ({
     chatStatus: chat.status,
   };
 };
+
+const getStepId = (flowVersion: FlowVersion, selectedStep: string) => {
+  return flowHelper.getStep(flowVersion, selectedStep)?.id ?? selectedStep;
+};