Shun Miyazawa 10 месяцев назад
Родитель
Сommit
8d0effb569
1 измененных файлов с 3 добавлено и 0 удалено
  1. 3 0
      apps/app/src/features/openai/client/services/editor-assistant.tsx

+ 3 - 0
apps/app/src/features/openai/client/services/editor-assistant.tsx

@@ -144,6 +144,7 @@ export const useEditorAssistant: UseEditorAssistant = () => {
   // Refs
   // const positionRef = useRef<number>(0);
   const lineRef = useRef<number>(0);
+  const isApplyingDiffRef = useRef<boolean>(false);
 
   // States
   const [detectedDiff, setDetectedDiff] = useState<DetectedDiff>();
@@ -210,8 +211,10 @@ export const useEditorAssistant: UseEditorAssistant = () => {
   const processMessage: ProcessMessage = useCallback((data, handler) => {
     handleIfSuccessfullyParsed(data, SseMessageSchema, (data: SseMessage) => {
       handler.onMessage(data);
+      isApplyingDiffRef.current = false;
     });
     handleIfSuccessfullyParsed(data, SseDetectedDiffSchema, (data: SseDetectedDiff) => {
+      isApplyingDiffRef.current = true;
       mutateIsEnableUnifiedMergeView(true);
       setDetectedDiff((prev) => {
         const newData = { data, applied: false, id: crypto.randomUUID() };