dailz/hpc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(service): pass work directory to Slurm job submission

Browse Source 
Add WorkDir to SubmitJobRequest and pass it as CurrentWorkingDirectory to Slurm REST API. Fixes Slurm 500 error when working directory is not specified.

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent)

Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
dailz
2026-04-13 17:12:28 +08:00
parent 328691adff
commit 32f5792b68
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
internal/model/job.go
View File

@@ -10,6 +10,7 @@ type SubmitJobRequest struct {
TimeLimit string `json:"time_limit,omitempty"` // 运行时间限制 (分钟) TimeLimit string `json:"time_limit,omitempty"` // 运行时间限制 (分钟)
JobName string `json:"job_name,omitempty"` // 作业名称 JobName string `json:"job_name,omitempty"` // 作业名称
Environment map[string]string `json:"environment,omitempty"` // 环境变量键值对 Environment map[string]string `json:"environment,omitempty"` // 环境变量键值对
WorkDir string `json:"work_dir,omitempty"` // 作业工作目录
} }
// JobResponse is the API response for a job. // JobResponse is the API response for a job.

3
internal/service/job_service.go
View File

@@ -32,6 +32,9 @@ func (s *JobService) SubmitJob(ctx context.Context, req *model.SubmitJobRequest)
Qos: strToPtrOrNil(req.QOS), Qos: strToPtrOrNil(req.QOS),
Name: strToPtrOrNil(req.JobName), Name: strToPtrOrNil(req.JobName),
} }
if req.WorkDir != "" {
jobDesc.CurrentWorkingDirectory = &req.WorkDir
}
if req.CPUs > 0 { if req.CPUs > 0 {
jobDesc.MinimumCpus = slurm.Ptr(req.CPUs) jobDesc.MinimumCpus = slurm.Ptr(req.CPUs)
} }