diff --git a/src/components/AssistantFormOutputs.vue b/src/components/AssistantFormOutputs.vue index 4a997af2..00bad9df 100644 --- a/src/components/AssistantFormOutputs.vue +++ b/src/components/AssistantFormOutputs.vue @@ -21,8 +21,10 @@ :optional-shape-options="selectedTaskType.optionalOutputShapeEnumValues ?? null" :values="outputs" :show-advanced="showAdvanced" + :can-improve-output="canImproveOutput" @update:values="$emit('update:outputs', $event)" - @update:show-advanced="$emit('update:show-advanced', $event)" /> + @update:show-advanced="$emit('update:show-advanced', $event)" + @improve="$emit('improve', $event)" /> @@ -70,11 +72,16 @@ export default { type: Boolean, default: false, }, + canImproveOutput: { + type: Boolean, + default: false, + }, }, emits: [ 'update:outputs', 'update:show-advanced', + 'improve', ], computed: { diff --git a/src/components/AssistantTextProcessingForm.vue b/src/components/AssistantTextProcessingForm.vue index c3006e8b..667c26e7 100644 --- a/src/components/AssistantTextProcessingForm.vue +++ b/src/components/AssistantTextProcessingForm.vue @@ -123,7 +123,9 @@ v-model:show-advanced="showAdvanced" v-model:outputs="myOutputs" :inputs="myInputs" - :selected-task-type="selectedTaskType" /> + :selected-task-type="selectedTaskType" + :can-improve-output="canImproveOutput" + @improve="onImprove" />
@@ -215,6 +217,7 @@ import { saveLastSelectedTaskType } from '../assistant.js' const TEXT2TEXT_TASK_TYPE_ID = 'core:text2text' const CHAT_TASK_TYPE_ID = 'chatty-llm' +const TEXT2TEXT_IMPROVE_TASK_TYPE_ID = 'core:text2text:improve' export default { name: 'AssistantTextProcessingForm', @@ -460,6 +463,20 @@ export default { actionButtonsToShow() { return this.hasOutput ? this.actionButtons : [] }, + improveTaskType() { + const taskType = this.taskTypes.find(tt => tt.id === TEXT2TEXT_IMPROVE_TASK_TYPE_ID) + if (taskType === undefined) { + return null + } + if (this.taskTypeIdList !== null && !this.taskTypeIdList.includes(TEXT2TEXT_IMPROVE_TASK_TYPE_ID)) { + return null + } + return taskType + }, + canImproveOutput() { + return this.hasOutput + && this.improveTaskType !== null + }, showRunningEmptyContent() { return this.showSyncTaskRunning && this.myOutputs === null }, @@ -484,6 +501,22 @@ export default { console.debug('[assistant] form\'s myoutputs', this.myOutputs) }, methods: { + onImprove(content) { + if (!this.canImproveOutput) { + return + } + this.$emit('new-task') + this.mySelectedTaskTypeId = TEXT2TEXT_IMPROVE_TASK_TYPE_ID + this.$nextTick(() => { + this.$refs?.assistantFormInputs?.setDefaultValues(true) + this.$nextTick(() => { + this.myInputs = { + ...this.myInputs, + input: content, + } + }) + }) + }, // Parse the file if a fileId is passed as initial value to a text field parseTextFileInputs(taskType) { if (taskType === undefined || taskType === null) { diff --git a/src/components/fields/TaskTypeField.vue b/src/components/fields/TaskTypeField.vue index 30b75b0e..4b6a58ce 100644 --- a/src/components/fields/TaskTypeField.vue +++ b/src/components/fields/TaskTypeField.vue @@ -10,8 +10,10 @@ :field="field" :options="options ?? undefined" :is-output="isOutput" + :can-improve-output="isTextField ? canImproveOutput : false" @submit="$emit('submit', $event)" - @update:value="$emit('update:value', $event)" /> + @update:value="$emit('update:value', $event)" + @improve="$emit('improve', $event)" />