aboutsummaryrefslogtreecommitdiffstats
path: root/dashboard/app/aidb/crud.go
diff options
context:
space:
mode:
Diffstat (limited to 'dashboard/app/aidb/crud.go')
-rw-r--r--dashboard/app/aidb/crud.go7
1 files changed, 1 insertions, 6 deletions
diff --git a/dashboard/app/aidb/crud.go b/dashboard/app/aidb/crud.go
index 4f7e93f6a..872a70ace 100644
--- a/dashboard/app/aidb/crud.go
+++ b/dashboard/app/aidb/crud.go
@@ -129,12 +129,6 @@ func StartJob(ctx context.Context, req *dashapi.AIJobPollReq) (*Job, error) {
job = jobs[0]
}
job.Started = spanner.NullTime{Time: TimeNow(ctx), Valid: true}
- for _, flow := range req.Workflows {
- if job.Workflow == flow.Name {
- job.LLMModel = flow.LLMModel
- break
- }
- }
job.CodeRevision = req.CodeRevision
mut, err := spanner.InsertOrUpdateStruct("Jobs", job)
if err != nil {
@@ -184,6 +178,7 @@ func StoreTrajectorySpan(ctx context.Context, jobID string, span *trajectory.Spa
Nesting: int64(span.Nesting),
Type: string(span.Type),
Name: span.Name,
+ Model: span.Model,
Started: span.Started,
Finished: toNullTime(span.Finished),
Error: toNullString(span.Error),