diff --git a/agent/app/service/ai.go b/agent/app/service/ai.go index d267202b7..f2ece5d5c 100644 --- a/agent/app/service/ai.go +++ b/agent/app/service/ai.go @@ -164,7 +164,18 @@ func (u *AIToolService) Recreate(req dto.OllamaModelName) error { taskItem.AddSubTask(i18n.GetWithName("OllamaModelPull", req.Name), func(t *task.Task) error { return cmd.ExecShellWithTask(taskItem, time.Hour, "docker", "exec", containerName, "ollama", "pull", req.Name) }, nil) - _ = taskItem.Execute() + taskItem.AddSubTask(i18n.GetWithName("OllamaModelSize", req.Name), func(t *task.Task) error { + itemSize, err := loadModelSize(modelInfo.Name, containerName) + if len(itemSize) != 0 { + _ = aiRepo.Update(modelInfo.ID, map[string]interface{}{"status": constant.StatusSuccess, "size": itemSize}) + } else { + _ = aiRepo.Update(modelInfo.ID, map[string]interface{}{"status": constant.StatusFailed, "message": err.Error()}) + } + return nil + }, nil) + if err := taskItem.Execute(); err != nil { + _ = aiRepo.Update(modelInfo.ID, map[string]interface{}{"status": constant.StatusFailed, "message": err.Error()}) + } }() return nil } diff --git a/frontend/src/views/ai/model/del/index.vue b/frontend/src/views/ai/model/del/index.vue index b804e902f..2471b80af 100644 --- a/frontend/src/views/ai/model/del/index.vue +++ b/frontend/src/views/ai/model/del/index.vue @@ -12,7 +12,7 @@ :indeterminate="isIndeterminate" @change="handleCheckAllChange" > - {{ $t('setting.all') }} + {{ $t('commons.table.all') }} @@ -55,9 +55,10 @@ interface DialogProps { list: Array; } const acceptParams = (props: DialogProps): void => { + checkAll.value = false; + checkedItems.value = []; list.value = props.list; checkAll.value = true; - handleCheckAllChange(true); open.value = true; }; diff --git a/frontend/src/views/ai/model/terminal/index.vue b/frontend/src/views/ai/model/terminal/index.vue index 7441684cb..926ff54d8 100644 --- a/frontend/src/views/ai/model/terminal/index.vue +++ b/frontend/src/views/ai/model/terminal/index.vue @@ -42,7 +42,7 @@ const acceptParams = async (params: DialogProps): Promise => { const initTerm = () => { nextTick(() => { terminalRef.value.acceptParams({ - endpoint: '/api/v2/ai/containers/exec', + endpoint: '/api/v2/containers/exec', args: `source=ollama&name=${itemName.value}`, error: '', initCmd: '',