fix

aws-samples · Jul 30, 2024 · 5461a3a · 5461a3a
1 parent 3e8ebed
commit 5461a3a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/packages/web/src/pages/TranslatePage.tsx b/packages/web/src/pages/TranslatePage.tsx
@@ -104,15 +104,18 @@ const TranslatePage: React.FC = () => {
     loading,
     messages,
     postChat,
+    continueGeneration,
     clear: clearChat,
     updateSystemContextByModel,
+    getStopReason,
   } = useChat(pathname);
   const { setTypingTextInput, typingTextOutput } = useTyping(loading);
   const { modelIds: availableModels } = MODELS;
   const modelId = getModelId();
   const prompter = useMemo(() => {
     return getPrompter(modelId);
   }, [modelId]);
+  const stopReason = getStopReason();
   const [auto, setAuto] = useLocalStorageBoolean('Auto_Translate', true);
   const [audio, setAudioInput] = useState(false); // 音声入力フラグ
 
@@ -340,6 +343,10 @@ const TranslatePage: React.FC = () => {
           </ExpandableField>
 
           <div className="flex justify-end gap-3">
+            {stopReason === 'max_tokens' && (
+              <Button onClick={continueGeneration}>続きを出力</Button>
+            )}
+
             <Button outlined onClick={onClickClear} disabled={disabledExec}>
               クリア
             </Button>