@gendive/chatllm 0.17.32 → 0.17.33

package/dist/react/index.js

@@ -3190,42 +3190,14 @@ ${result.content}
       if (activeChecklistRef.current) {
         const checklist = activeChecklistRef.current;
         const stepIndex = checklist.currentStep;
-        checklist.stepResults.push(accumulatedContent);
-        setSessions(
-          (prev) => prev.map((s) => {
-            if (s.id !== checklist.sessionId) return s;
-            return {
-              ...s,
-              messages: s.messages.map((m) => {
-                if (m.id !== checklist.messageId || !m.checklistBlock) return m;
-                const updatedItems = m.checklistBlock.items.map((it, idx) => ({
-                  ...it,
-                  status: idx <= stepIndex ? "done" : it.status,
-                  result: idx === stepIndex ? accumulatedContent : it.result
-                }));
-                return {
-                  ...m,
-                  checklistBlock: { ...m.checklistBlock, items: updatedItems, currentStep: stepIndex + 1 }
-                };
-              })
-            };
-          })
-        );
-        setSessions(
-          (prev) => prev.map((s) => {
-            if (s.id !== capturedSessionId) return s;
-            return {
-              ...s,
-              messages: s.messages.map((m) => {
-                if (m.id !== assistantMessageId) return m;
-                return { ...m, hidden: true };
-              })
-            };
-          })
-        );
-        const nextStep = stepIndex + 1;
-        if (nextStep < checklist.items.length) {
-          checklist.currentStep = nextStep;
+        const hasPollTag = /<poll[\s>]/i.test(accumulatedContent);
+        if (hasPollTag && !checklist.waitingForPoll) {
+          checklist.waitingForPoll = true;
+        } else {
+          if (checklist.waitingForPoll) {
+            checklist.waitingForPoll = false;
+          }
+          checklist.stepResults.push(accumulatedContent);
           setSessions(
             (prev) => prev.map((s) => {
               if (s.id !== checklist.sessionId) return s;
@@ -3235,21 +3207,87 @@ ${result.content}
                   if (m.id !== checklist.messageId || !m.checklistBlock) return m;
                   const updatedItems = m.checklistBlock.items.map((it, idx) => ({
                     ...it,
-                    status: idx === nextStep ? "in_progress" : it.status
+                    status: idx <= stepIndex ? "done" : it.status,
+                    result: idx === stepIndex ? accumulatedContent : it.result
                   }));
                   return {
                     ...m,
-                    checklistBlock: { ...m.checklistBlock, items: updatedItems, currentStep: nextStep }
+                    checklistBlock: { ...m.checklistBlock, items: updatedItems, currentStep: stepIndex + 1 }
                   };
                 })
               };
             })
           );
+          setSessions(
+            (prev) => prev.map((s) => {
+              if (s.id !== capturedSessionId) return s;
+              return {
+                ...s,
+                messages: s.messages.map((m) => {
+                  if (m.id !== assistantMessageId) return m;
+                  return { ...m, hidden: true };
+                })
+              };
+            })
+          );
+          const nextStep = stepIndex + 1;
+          if (nextStep < checklist.items.length) {
+            checklist.currentStep = nextStep;
+            setSessions(
+              (prev) => prev.map((s) => {
+                if (s.id !== checklist.sessionId) return s;
+                return {
+                  ...s,
+                  messages: s.messages.map((m) => {
+                    if (m.id !== checklist.messageId || !m.checklistBlock) return m;
+                    const updatedItems = m.checklistBlock.items.map((it, idx) => ({
+                      ...it,
+                      status: idx === nextStep ? "in_progress" : it.status
+                    }));
+                    return {
+                      ...m,
+                      checklistBlock: { ...m.checklistBlock, items: updatedItems, currentStep: nextStep }
+                    };
+                  })
+                };
+              })
+            );
+            skipNextChecklistParsingRef.current = true;
+            setTimeout(() => {
+              sendMessage(
+                `\uCCB4\uD06C\uB9AC\uC2A4\uD2B8 ${nextStep + 1}/${checklist.items.length}\uB2E8\uACC4\uB97C \uC2E4\uD589\uD558\uC138\uC694: "${checklist.items[nextStep].title}".
+\uC774\uC804 \uB2E8\uACC4 \uACB0\uACFC\uB97C \uCC38\uACE0\uD558\uB418, \uC774 \uB2E8\uACC4\uC5D0\uB9CC \uC9D1\uC911\uD558\uC5EC \uACB0\uACFC\uB97C \uC791\uC131\uD558\uC138\uC694. checklist \uD0DC\uADF8\uB294 \uC0AC\uC6A9\uD558\uC9C0 \uB9C8\uC138\uC694.`,
+                { hiddenUserMessage: true }
+              );
+            }, 100);
+            saveMessagesOnEarlyReturn();
+            setIsLoading(false);
+            abortControllerRef.current = null;
+            return;
+          }
+          const stepSummary = checklist.stepResults.map((r, i) => `### ${i + 1}. ${checklist.items[i].title}
+${r}`).join("\n\n");
+          setSessions(
+            (prev) => prev.map((s) => {
+              if (s.id !== checklist.sessionId) return s;
+              return {
+                ...s,
+                messages: s.messages.map((m) => {
+                  if (m.id !== checklist.messageId || !m.checklistBlock) return m;
+                  return { ...m, checklistBlock: { ...m.checklistBlock, completed: true } };
+                })
+              };
+            })
+          );
+          activeChecklistRef.current = null;
           skipNextChecklistParsingRef.current = true;
           setTimeout(() => {
             sendMessage(
-              `\uCCB4\uD06C\uB9AC\uC2A4\uD2B8 ${nextStep + 1}/${checklist.items.length}\uB2E8\uACC4\uB97C \uC2E4\uD589\uD558\uC138\uC694: "${checklist.items[nextStep].title}".
-\uC774\uC804 \uB2E8\uACC4 \uACB0\uACFC\uB97C \uCC38\uACE0\uD558\uB418, \uC774 \uB2E8\uACC4\uC5D0\uB9CC \uC9D1\uC911\uD558\uC5EC \uACB0\uACFC\uB97C \uC791\uC131\uD558\uC138\uC694. checklist \uD0DC\uADF8\uB294 \uC0AC\uC6A9\uD558\uC9C0 \uB9C8\uC138\uC694.`,
+              `\uBAA8\uB4E0 \uCCB4\uD06C\uB9AC\uC2A4\uD2B8 \uB2E8\uACC4\uAC00 \uC644\uB8CC\uB418\uC5C8\uC2B5\uB2C8\uB2E4. \uC544\uB798\uB294 \uAC01 \uB2E8\uACC4\uBCC4 \uACB0\uACFC\uC785\uB2C8\uB2E4:
+
+${stepSummary}
+
+\uC704 \uACB0\uACFC\uB97C \uC885\uD569\uD558\uC5EC \uCD5C\uC885 \uACB0\uACFC\uBB3C\uC744 \uC644\uC131\uD574\uC8FC\uC138\uC694. checklist \uD0DC\uADF8\uB294 \uC0AC\uC6A9\uD558\uC9C0 \uB9C8\uC138\uC694.`,
               { hiddenUserMessage: true }
             );
           }, 100);
@@ -3258,36 +3296,6 @@ ${result.content}
           abortControllerRef.current = null;
           return;
         }
-        const stepSummary = checklist.stepResults.map((r, i) => `### ${i + 1}. ${checklist.items[i].title}
-${r}`).join("\n\n");
-        setSessions(
-          (prev) => prev.map((s) => {
-            if (s.id !== checklist.sessionId) return s;
-            return {
-              ...s,
-              messages: s.messages.map((m) => {
-                if (m.id !== checklist.messageId || !m.checklistBlock) return m;
-                return { ...m, checklistBlock: { ...m.checklistBlock, completed: true } };
-              })
-            };
-          })
-        );
-        activeChecklistRef.current = null;
-        skipNextChecklistParsingRef.current = true;
-        setTimeout(() => {
-          sendMessage(
-            `\uBAA8\uB4E0 \uCCB4\uD06C\uB9AC\uC2A4\uD2B8 \uB2E8\uACC4\uAC00 \uC644\uB8CC\uB418\uC5C8\uC2B5\uB2C8\uB2E4. \uC544\uB798\uB294 \uAC01 \uB2E8\uACC4\uBCC4 \uACB0\uACFC\uC785\uB2C8\uB2E4:
-
-${stepSummary}
-
-\uC704 \uACB0\uACFC\uB97C \uC885\uD569\uD558\uC5EC \uCD5C\uC885 \uACB0\uACFC\uBB3C\uC744 \uC644\uC131\uD574\uC8FC\uC138\uC694. checklist \uD0DC\uADF8\uB294 \uC0AC\uC6A9\uD558\uC9C0 \uB9C8\uC138\uC694.`,
-            { hiddenUserMessage: true }
-          );
-        }, 100);
-        saveMessagesOnEarlyReturn();
-        setIsLoading(false);
-        abortControllerRef.current = null;
-        return;
       }
       if (!skipNextChecklistParsingRef.current) {
         const { checklistBlock, cleanContent: checklistCleanContent } = parseChecklistFromContent(accumulatedContent);
@@ -3642,6 +3650,32 @@ ${allResults}
       });
       const hasValidResponses = formattedParts.length > 0;
       const isAutoGenerate = responses.every((r) => r.skipped);
+      if (activeChecklistRef.current?.waitingForPoll) {
+        const checklist = activeChecklistRef.current;
+        const stepTitle = checklist.items[checklist.currentStep]?.title || "";
+        setSessions(
+          (prev) => prev.map((s) => {
+            if (s.id !== currentSessionId) return s;
+            return {
+              ...s,
+              messages: s.messages.map((m) => {
+                if (m.id !== messageId) return m;
+                return { ...m, pollBlock: void 0, hidden: true };
+              })
+            };
+          })
+        );
+        skipNextPollParsingRef.current = true;
+        skipNextChecklistParsingRef.current = true;
+        const hiddenPrompt2 = isAutoGenerate ? `\uCCB4\uD06C\uB9AC\uC2A4\uD2B8 \uB2E8\uACC4 "${stepTitle}"\uC744 \uC2E4\uD589\uD558\uC138\uC694. \uC0AC\uC6A9\uC790\uAC00 \uC790\uB3D9 \uC120\uD0DD\uC744 \uC694\uCCAD\uD588\uC73C\uBBC0\uB85C, \uCD5C\uC801\uC758 \uC635\uC158\uC744 \uC120\uD0DD\uD558\uC5EC \uC774 \uB2E8\uACC4\uB97C \uC644\uB8CC\uD574\uC8FC\uC138\uC694. poll \uD0DC\uADF8\uC640 checklist \uD0DC\uADF8\uB294 \uC0AC\uC6A9\uD558\uC9C0 \uB9C8\uC138\uC694.` : `\uCCB4\uD06C\uB9AC\uC2A4\uD2B8 \uB2E8\uACC4 "${stepTitle}"\uC744 \uACC4\uC18D\uD569\uB2C8\uB2E4. \uC0AC\uC6A9\uC790\uC758 \uC120\uD0DD:
+${formattedParts.join("\n")}
+
+\uC704 \uC120\uD0DD\uC744 \uBC18\uC601\uD558\uC5EC \uC774 \uB2E8\uACC4\uC758 \uACB0\uACFC\uB97C \uC644\uC131\uD574\uC8FC\uC138\uC694. poll \uD0DC\uADF8\uC640 checklist \uD0DC\uADF8\uB294 \uC0AC\uC6A9\uD558\uC9C0 \uB9C8\uC138\uC694.`;
+        setTimeout(() => {
+          sendMessage(hiddenPrompt2, { hiddenUserMessage: true });
+        }, 100);
+        return;
+      }
       setSessions(
         (prev) => prev.map((s) => {
           if (s.id !== currentSessionId) return s;