diff options
| -rw-r--r-- | syz-agent/agent.go | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/syz-agent/agent.go b/syz-agent/agent.go index 301ccec11..6bccb0564 100644 --- a/syz-agent/agent.go +++ b/syz-agent/agent.go @@ -183,6 +183,7 @@ func (s *Server) poll(ctx context.Context) (bool, error) { Name: flow.Name, }) } + log.Logf(0, "querying jobs for %v", req.Workflows) resp, err := s.dash.AIJobPoll(req) if err != nil { return false, err @@ -190,6 +191,8 @@ func (s *Server) poll(ctx context.Context) (bool, error) { if resp.ID == "" { return false, nil } + log.Logf(0, "starting job %v %v", resp.Workflow, resp.ID) + defer log.Logf(0, "finished job %v %v", resp.Workflow, resp.ID) doneReq := &dashapi.AIJobDoneReq{ ID: resp.ID, } @@ -208,10 +211,14 @@ func (s *Server) poll(ctx context.Context) (bool, error) { // the dashboard at all. For the dashboard it will look like // the server has crashed while executing the job, and it should // eventually retry it on common grounds. - s.overQuotaModels[model] = time.Now() + now := time.Now() + s.overQuotaModels[model] = now + log.Logf(0, "model %v is over daily quota until %v", + model, aflow.QuotaResetTime(now)) return true, nil } } + log.Logf(0, "done executing job %v %v", resp.Workflow, resp.ID) if err := s.dash.AIJobDone(doneReq); err != nil { return false, err } @@ -261,6 +268,7 @@ func (s *Server) modelOverQuota(flow *aflow.Flow) bool { func (s *Server) resetModelQuota() { for model, when := range s.overQuotaModels { if aflow.QuotaResetTime(when).After(time.Now()) { + log.Logf(0, "model %v daily quota is replenished", model) delete(s.overQuotaModels, model) } } |