feat: gen res

web/app/components/share/text-generation/index.tsx

@@ -103,6 +103,20 @@ const TextGeneration: FC<IMainProps> = ({
   const noPendingTask = pendingTaskList.length === 0
   const showTaskList = allTaskList.filter(task => task.status !== TaskStatus.pending)
   const allTaskFinished = allTaskList.every(task => task.status === TaskStatus.completed)
+  const [batchCompletionRes, setBatchCompletionRes, getBatchCompletionRes] = useGetState<Record<string, string>>({})
+  const exportRes = allTaskList.map((task) => {
+    if (!allTaskFinished || allTaskList.length === 0)
+      return {}
+    const batchCompletionResLatest = getBatchCompletionRes()
+    const res: Record<string, string> = {}
+    const { inputs, query } = task.params
+    promptConfig?.prompt_variables.forEach((v) => {
+      res[v.name] = inputs[v.key]
+    })
+    res[t('share.generation.queryTitle')] = query
+    res[t('share.generation.completionResult')] = batchCompletionResLatest[task.id]
+    return res
+  })
   const checkBatchInputs = (data: string[][]) => {
     if (!data || data.length === 0) {
       notify({ type: 'error', message: t('share.generation.errorMsg.empty') })
@@ -228,10 +242,9 @@ const TextGeneration: FC<IMainProps> = ({
     // clear run once task status
     setControlStopResponding(Date.now())
   }
-
-  const handleCompleted = (taskId?: number, isSuccess?: boolean) => {
-    // console.log(taskId, isSuccess)
+  const handleCompleted = (completionRes: string, taskId?: number) => {
     const allTasklistLatest = getLatestTaskList()
+    const batchCompletionResLatest = getBatchCompletionRes()
     const pendingTaskList = allTasklistLatest.filter(task => task.status === TaskStatus.pending)
     const nextPendingTaskId = pendingTaskList[0]?.id
     // console.log(`start: ${allTasklistLatest.map(item => item.status).join(',')}`)
@@ -252,6 +265,12 @@ const TextGeneration: FC<IMainProps> = ({
     })
     // console.log(`end: ${newAllTaskList.map(item => item.status).join(',')}`)
     setAllTaskList(newAllTaskList)
+    if (taskId) {
+      setBatchCompletionRes({
+        ...batchCompletionResLatest,
+        [`${taskId}`]: completionRes,
+      })
+    }
   }
 
   const fetchInitData = async () => {
@@ -343,8 +362,7 @@ const TextGeneration: FC<IMainProps> = ({
             {allTaskFinished && (
               <ResDownload
                 isMobile={isMobile}
-                vars={promptConfig?.prompt_variables || []}
-                values={[['a', 'b', 'c']]}
+                values={exportRes}
               />
             )}
             {!isPC && (

web/app/components/share/text-generation/result/index.tsx

 'use client'
 import type { FC } from 'react'
 import React, { useEffect, useState } from 'react'
-import { useBoolean } from 'ahooks'
+import { useBoolean, useGetState } from 'ahooks'
 import { t } from 'i18next'
 import cn from 'classnames'
 import TextGenerationRes from '@/app/components/app/text-generate/item'
@@ -27,7 +27,7 @@ export type IResultProps = {
   onShowRes: () => void
   handleSaveMessage: (messageId: string) => void
   taskId?: number
-  onCompleted: (taskId?: number, success?: boolean) => void
+  onCompleted: (completionRes: string, taskId?: number, success?: boolean) => void
 }
 
 const Result: FC<IResultProps> = ({
@@ -53,7 +53,7 @@ const Result: FC<IResultProps> = ({
       setResponsingFalse()
   }, [controlStopResponding])
 
-  const [completionRes, setCompletionRes] = useState('')
+  const [completionRes, setCompletionRes, getCompletionRes] = useGetState('')
   const { notify } = Toast
   const isNoData = !completionRes
 
@@ -141,11 +141,11 @@ const Result: FC<IResultProps> = ({
       onCompleted: () => {
         setResponsingFalse()
         setMessageId(tempMessageId)
-        onCompleted(taskId, true)
+        onCompleted(getCompletionRes(), taskId, true)
       },
       onError() {
         setResponsingFalse()
-        onCompleted(taskId, false)
+        onCompleted(getCompletionRes(), taskId, false)
       },
     }, isInstalledApp, installedAppInfo?.id)
   }

web/app/components/share/text-generation/run-batch/res-download/index.tsx

@@ -10,42 +10,26 @@ import { Download02 as DownloadIcon } from '@/app/components/base/icons/src/vend
 import Button from '@/app/components/base/button'
 export type IResDownloadProps = {
   isMobile: boolean
-  vars: { name: string }[]
-  values: string[][]
+  values: Record<string, string>[]
 }
 
 const ResDownload: FC<IResDownloadProps> = ({
   isMobile,
-  vars,
   values,
 }) => {
   const { t } = useTranslation()
   const { CSVDownloader, Type } = useCSVDownloader()
-  const addQueryContentVars = [
-    ...vars,
-    { name: t('share.generation.queryTitle') },
-    { name: 'Result' },
-  ]
-  const headers = (() => {
-    const res: Record<string, string> = {}
-    addQueryContentVars.forEach((item) => {
-      res[item.name] = ''
-    })
-    return res
-  })()
+
   return (
     <CSVDownloader
       className="block cursor-pointer"
       type={Type.Link}
-      filename={'template'}
+      filename={'result'}
       bom={true}
       config={{
         // delimiter: ';',
       }}
-      data={[
-        headers,
-        ...values,
-      ]}
+      data={values}
     >
       <Button className={cn('flex items-center !h-8 space-x-2 bg-white !text-[13px] font-medium', isMobile ? '!p-0 !w-8 justify-center' : '!px-3')}>
         <DownloadIcon className='w-4 h-4 text-[#155EEF]' />

web/i18n/lang/share-app.en.ts

@@ -41,6 +41,7 @@ const translation = {
     },
     title: 'AI Completion',
     queryTitle: 'Query content',
+    completionResult: 'Completion result',
     queryPlaceholder: 'Write your query content...',
     run: 'Execute',
     copy: 'Copy',

web/i18n/lang/share-app.zh.ts

@@ -37,6 +37,7 @@ const translation = {
     },
     title: 'AI 智能书写',
     queryTitle: '查询内容',
+    completionResult: '生成结果',
     queryPlaceholder: '请输入文本内容',
     run: '运行',
     copy: '拷贝',