syz-agent: extend logging

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)
 		}
 	}