diff --git a/agent/app/service/tensorrt_llm.go b/agent/app/service/tensorrt_llm.go index 703d6526e..e944d83ac 100644 --- a/agent/app/service/tensorrt_llm.go +++ b/agent/app/service/tensorrt_llm.go @@ -88,31 +88,25 @@ func (t TensorRTLLMService) Page(req request.TensorRTLLMSearch) response.TensorR } composeByte, err := files.NewFileOp().GetContent(path.Join(global.Dir.TensorRTLLMDir, item.Name, "docker-compose.yml")) - if err != nil { - continue - } - serverDTO.Environments, err = getDockerComposeEnvironments(composeByte) - if err != nil { - continue + if err == nil { + serverDTO.Environments, _ = getDockerComposeEnvironments(composeByte) } volumes, err := getDockerComposeVolumes(composeByte) - if err != nil { - continue - } - - var defaultVolumes = map[string]string{ - "${MODEL_PATH}": "${MODEL_PATH}", - } - for _, volume := range volumes { - exist := false - for key, value := range defaultVolumes { - if key == volume.Source && value == volume.Target { - exist = true - break - } + if err == nil { + var defaultVolumes = map[string]string{ + "${MODEL_PATH}": "${MODEL_PATH}", } - if !exist { - serverDTO.Volumes = append(serverDTO.Volumes, volume) + for _, volume := range volumes { + exist := false + for key, value := range defaultVolumes { + if key == volume.Source && value == volume.Target { + exist = true + break + } + } + if !exist { + serverDTO.Volumes = append(serverDTO.Volumes, volume) + } } } items = append(items, serverDTO) diff --git a/frontend/src/views/ai/model/tensorrt/operate/index.vue b/frontend/src/views/ai/model/tensorrt/operate/index.vue index a5fa021df..661c301a5 100644 --- a/frontend/src/views/ai/model/tensorrt/operate/index.vue +++ b/frontend/src/views/ai/model/tensorrt/operate/index.vue @@ -110,6 +110,7 @@ const openEdit = (rowData: any): void => { if (tensorRTLLM.value.exposedPorts == null) { tensorRTLLM.value.exposedPorts = []; } + tensorRTLLM.value.command = rowData.command.slice(1, -1).replace(/^'|'$/g, '').replace(/\\"/g, '"'); if (tensorRTLLM.value.extraHosts == null) { tensorRTLLM.value.extraHosts = []; }