Support configuring effort alongside model
This commit is contained in:
@@ -435,7 +435,11 @@ func (s *Server) CreateTask(w http.ResponseWriter, r *http.Request, projectID st
|
||||
if req.Model != nil {
|
||||
model = *req.Model
|
||||
}
|
||||
t, err := s.agents.StartTask(r.Context(), p, u, title, prompt, model, r.Header.Get("Byre-Client"), images)
|
||||
effort := ""
|
||||
if req.Effort != nil {
|
||||
effort = *req.Effort
|
||||
}
|
||||
t, err := s.agents.StartTask(r.Context(), p, u, title, prompt, model, effort, r.Header.Get("Byre-Client"), images)
|
||||
if err != nil {
|
||||
writeAgentError(w, err)
|
||||
return
|
||||
@@ -485,8 +489,12 @@ func (s *Server) StartTodoItem(w http.ResponseWriter, r *http.Request, projectID
|
||||
if req.Model != nil {
|
||||
model = *req.Model
|
||||
}
|
||||
effort := ""
|
||||
if req.Effort != nil {
|
||||
effort = *req.Effort
|
||||
}
|
||||
|
||||
tasks, err := s.agents.StartTodo(r.Context(), p, u, req.Title, model, u.Login, "", req.Discuss != nil && *req.Discuss)
|
||||
tasks, err := s.agents.StartTodo(r.Context(), p, u, req.Title, model, effort, u.Login, "", req.Discuss != nil && *req.Discuss)
|
||||
if err != nil {
|
||||
writeAgentError(w, err)
|
||||
return
|
||||
@@ -863,6 +871,10 @@ func (s *Server) ContinueSession(w http.ResponseWriter, r *http.Request, session
|
||||
if req.Model != nil {
|
||||
model = *req.Model
|
||||
}
|
||||
effort := ""
|
||||
if req.Effort != nil {
|
||||
effort = *req.Effort
|
||||
}
|
||||
sendID := ""
|
||||
if req.SendId != nil {
|
||||
sendID = *req.SendId
|
||||
@@ -889,7 +901,7 @@ func (s *Server) ContinueSession(w http.ResponseWriter, r *http.Request, session
|
||||
return
|
||||
}
|
||||
|
||||
task, err := s.agents.Continue(r.Context(), src, t, p, u, req.Text, model, sendID, r.Header.Get("Byre-Client"), images...)
|
||||
task, err := s.agents.Continue(r.Context(), src, t, p, u, req.Text, model, effort, sendID, r.Header.Get("Byre-Client"), images...)
|
||||
if err != nil {
|
||||
writeAgentError(w, err)
|
||||
return
|
||||
@@ -944,12 +956,16 @@ func (s *Server) CreateSession(w http.ResponseWriter, r *http.Request, taskID st
|
||||
if req.Model != nil {
|
||||
model = *req.Model
|
||||
}
|
||||
effort := ""
|
||||
if req.Effort != nil {
|
||||
effort = *req.Effort
|
||||
}
|
||||
sendID := ""
|
||||
if req.SendId != nil {
|
||||
sendID = *req.SendId
|
||||
}
|
||||
|
||||
sess, err := s.agents.Create(r.Context(), t, u, req.Prompt, model, sendID, r.Header.Get("Byre-Client"), images...)
|
||||
sess, err := s.agents.Create(r.Context(), t, u, req.Prompt, model, effort, sendID, r.Header.Get("Byre-Client"), images...)
|
||||
if err != nil {
|
||||
writeAgentError(w, err)
|
||||
return
|
||||
@@ -986,12 +1002,16 @@ func (s *Server) SendMessage(w http.ResponseWriter, r *http.Request, sessionID s
|
||||
if req.Model != nil {
|
||||
model = *req.Model
|
||||
}
|
||||
effort := ""
|
||||
if req.Effort != nil {
|
||||
effort = *req.Effort
|
||||
}
|
||||
sendID := ""
|
||||
if req.SendId != nil {
|
||||
sendID = *req.SendId
|
||||
}
|
||||
|
||||
sess, _, err := s.agents.Message(r.Context(), sessionID, actor(r), req.Text, model, sendID, r.Header.Get("Byre-Client"), images...)
|
||||
sess, _, err := s.agents.Message(r.Context(), sessionID, actor(r), req.Text, model, effort, sendID, r.Header.Get("Byre-Client"), images...)
|
||||
if err != nil {
|
||||
writeAgentError(w, err)
|
||||
return
|
||||
@@ -1017,7 +1037,11 @@ func (s *Server) CompactSession(w http.ResponseWriter, r *http.Request, sessionI
|
||||
if req.Model != nil {
|
||||
model = *req.Model
|
||||
}
|
||||
sess, err := s.agents.Compact(r.Context(), sessionID, model)
|
||||
effort := ""
|
||||
if req.Effort != nil {
|
||||
effort = *req.Effort
|
||||
}
|
||||
sess, err := s.agents.Compact(r.Context(), sessionID, model, effort)
|
||||
if err != nil {
|
||||
writeAgentError(w, err)
|
||||
return
|
||||
|
||||
Reference in New Issue
Block a user