Commit d68c85e4 authored by Joel's avatar Joel

fix: extra column get var problem

parent e4fdc054
...@@ -206,19 +206,21 @@ const TextGeneration: FC<IMainProps> = ({ ...@@ -206,19 +206,21 @@ const TextGeneration: FC<IMainProps> = ({
} }
const payloadData = data.filter(item => !item.every(i => i === '')).slice(1) const payloadData = data.filter(item => !item.every(i => i === '')).slice(1)
const varLen = promptConfig?.prompt_variables.length || 0
setIsCallBatchAPI(true) setIsCallBatchAPI(true)
const allTaskList: Task[] = payloadData.map((item, i) => { const allTaskList: Task[] = payloadData.map((item, i) => {
const inputs: Record<string, string> = {} const inputs: Record<string, string> = {}
item.slice(0, -1).forEach((input, index) => { if (varLen > 0) {
inputs[promptConfig?.prompt_variables[index].key as string] = input item.slice(0, varLen).forEach((input, index) => {
}) inputs[promptConfig?.prompt_variables[index].key as string] = input
})
}
return { return {
id: i + 1, id: i + 1,
status: i < PARALLEL_LIMIT ? TaskStatus.running : TaskStatus.pending, status: i < PARALLEL_LIMIT ? TaskStatus.running : TaskStatus.pending,
params: { params: {
inputs, inputs,
query: item[item.length - 1], query: item[varLen],
}, },
} }
}) })
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment