Add a per-user lead agent and make the right pane task-focused

internal/server/lead.go

@@ -0,0 +1,82 @@
+package server
+
+import (
+	"errors"
+	"net/http"
+	"strings"
+
+	"g.fc.run/misc/byre/internal/api"
+)
+
+func (s *Server) ListLeadSessions(w http.ResponseWriter, r *http.Request) {
+	sessions, err := s.agents.LeadSessions(r.Context(), actor(r))
+	if err != nil {
+		writeStoreError(w, err)
+		return
+	}
+	writeJSON(w, http.StatusOK, sessions)
+}
+
+func (s *Server) MessageLead(w http.ResponseWriter, r *http.Request) {
+	req := api.MessageCreate{}
+	if !readJSON(w, r, &req) {
+		return
+	}
+
+	images, ok := s.images(w, r, req.Images)
+	if !ok {
+		return
+	}
+
+	if strings.TrimSpace(req.Text) == "" && len(images) == 0 {
+		writeError(w, http.StatusBadRequest, errors.New("text or images are required"))
+		return
+	}
+
+	u, err := s.store.GetUser(r.Context(), actor(r))
+	if err != nil {
+		writeStoreError(w, err)
+		return
+	}
+
+	model := ""
+	if req.Model != nil {
+		model = *req.Model
+	}
+
+	sess, err := s.agents.MessageLead(r.Context(), u, req.Text, model, images)
+	if err != nil {
+		writeAgentError(w, err)
+		return
+	}
+	writeJSON(w, http.StatusAccepted, sess)
+}
+
+func (s *Server) GetFocus(w http.ResponseWriter, r *http.Request) {
+	f, err := s.store.GetFocus(r.Context(), actor(r))
+	if err != nil {
+		writeStoreError(w, err)
+		return
+	}
+	writeJSON(w, http.StatusOK, f)
+}
+
+func (s *Server) SetFocus(w http.ResponseWriter, r *http.Request) {
+	f := api.Focus{}
+	if !readJSON(w, r, &f) {
+		return
+	}
+
+	err := s.agents.SetFocus(r.Context(), actor(r), f, actor(r))
+	if err != nil {
+		writeAgentError(w, err)
+		return
+	}
+
+	f, err = s.store.GetFocus(r.Context(), actor(r))
+	if err != nil {
+		writeStoreError(w, err)
+		return
+	}
+	writeJSON(w, http.StatusOK, f)
+}