diff options
Diffstat (limited to 'internal/api/server.go')
| -rw-r--r-- | internal/api/server.go | 39 |
1 files changed, 30 insertions, 9 deletions
diff --git a/internal/api/server.go b/internal/api/server.go index bac98b6..5758347 100644 --- a/internal/api/server.go +++ b/internal/api/server.go @@ -112,31 +112,52 @@ func (s *Server) BroadcastQuestion(taskID, toolUseID string, questionData json.R func (s *Server) handleAnswerQuestion(w http.ResponseWriter, r *http.Request) { taskID := r.PathValue("id") - if _, err := s.store.GetTask(taskID); err != nil { + tk, err := s.store.GetTask(taskID) + if err != nil { writeJSON(w, http.StatusNotFound, map[string]string{"error": "task not found"}) return } + if tk.State != task.StateBlocked { + writeJSON(w, http.StatusConflict, map[string]string{"error": "task is not blocked"}) + return + } var input struct { - QuestionID string `json:"question_id"` - Answer string `json:"answer"` + Answer string `json:"answer"` } if err := json.NewDecoder(r.Body).Decode(&input); err != nil { writeJSON(w, http.StatusBadRequest, map[string]string{"error": "invalid JSON: " + err.Error()}) return } - if input.QuestionID == "" { - writeJSON(w, http.StatusBadRequest, map[string]string{"error": "question_id is required"}) + if input.Answer == "" { + writeJSON(w, http.StatusBadRequest, map[string]string{"error": "answer is required"}) + return + } + + // Look up the session ID from the most recent execution. + latest, err := s.store.GetLatestExecution(taskID) + if err != nil || latest.SessionID == "" { + writeJSON(w, http.StatusInternalServerError, map[string]string{"error": "no resumable session found"}) return } - ok := s.pool.Questions.Answer(input.QuestionID, input.Answer) - if !ok { - writeJSON(w, http.StatusNotFound, map[string]string{"error": "no pending question with that ID"}) + // Clear the question and transition to QUEUED. + s.store.UpdateTaskQuestion(taskID, "") + s.store.UpdateTaskState(taskID, task.StateQueued) + + // Submit a resume execution. + resumeExec := &storage.Execution{ + ID: uuid.New().String(), + TaskID: taskID, + ResumeSessionID: latest.SessionID, + ResumeAnswer: input.Answer, + } + if err := s.pool.SubmitResume(r.Context(), tk, resumeExec); err != nil { + writeJSON(w, http.StatusServiceUnavailable, map[string]string{"error": err.Error()}) return } - writeJSON(w, http.StatusOK, map[string]string{"status": "delivered"}) + writeJSON(w, http.StatusOK, map[string]string{"status": "queued"}) } func (s *Server) handleHealth(w http.ResponseWriter, r *http.Request) { |