83 lines
1.6 KiB
Go
83 lines
1.6 KiB
Go
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)
|
|
}
|