[AXON-810] Fix replay stream bug (#710)

* AXON-810: send Sent Prompt signal before handling replay

* AXON-810: get rid of debugging override

* AXON-810: move sent to prompt inside replay func

Author: cabella-dot

src/react/atlascode/rovo-dev/rovoDevView.tsx

@@ -69,9 +69,7 @@ const RovoDevView: React.FC = () => {
     const [currentMessage, setCurrentMessage] = useState<DefaultMessage | null>(null);
     const [curThinkingMessages, setCurThinkingMessages] = useState<ChatMessage[]>([]);
 
-    const [currentState, setCurrentState] = useState(
-        process.env.ROVODEV_BBY ? State.GeneratingResponse : State.WaitingForPrompt,
-    );
+    const [currentState, setCurrentState] = useState(State.WaitingForPrompt);
 
     const [promptText, setPromptText] = useState('');
     const [pendingToolCallMessage, setPendingToolCallMessage] = useState('');

src/rovo-dev/rovoDevWebviewProvider.ts

@@ -499,14 +499,18 @@ export class RovoDevWebviewProvider extends Disposable implements WebviewViewPro
     private async sendUserPromptToView({ text, enable_deep_plan, context }: RovoDevPrompt) {
         const webview = this._webView!;
 
-        await webview.postMessage({
+        return await webview.postMessage({
             type: RovoDevProviderMessageType.UserChatMessage,
             message: {
                 text: text,
                 source: 'User',
                 context: context,
             },
         });
+    }
+
+    private async sendPromptSentToView({ text, enable_deep_plan, context }: RovoDevPrompt) {
+        const webview = this._webView!;
 
         return await webview.postMessage({
             type: RovoDevProviderMessageType.PromptSent,
@@ -675,6 +679,8 @@ ${message}`;
             context,
         };
 
+        await this.sendPromptSentToView({ text, enable_deep_plan, context });
+
         let payloadToSend = this.addUndoContextToPrompt(text);
         payloadToSend = this.addContextToPrompt(payloadToSend, context);
 
@@ -842,6 +848,7 @@ ${message}`;
 
     private async executeReplay(): Promise<void> {
         this.beginNewPrompt('replay');
+        await this.sendPromptSentToView({ text: '', enable_deep_plan: false, context: undefined });
 
         await this.executeApiWithErrorHandling(async (client) => {
             return this.processChatResponse('replay', client.replay());