DBT/clawgo
1
0
Fork 0
mirror of https://github.com/YspCoder/clawgo.git synced 2026-04-14 02:37:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

refine oauth model selection and provider inputs

Browse Source
This commit is contained in:
lpf
2026-03-11 19:48:10 +08:00
parent b3c0f58998
commit 5d74dba0b8
8 changed files with 381 additions and 215 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
pkg/api/server.go
View File

@@ -457,6 +457,7 @@ func (s *Server) Start(ctx context.Context) error {
mux.HandleFunc("/webui/api/provider/oauth/complete", s.handleWebUIProviderOAuthComplete)
mux.HandleFunc("/webui/api/provider/oauth/import", s.handleWebUIProviderOAuthImport)
mux.HandleFunc("/webui/api/provider/oauth/accounts", s.handleWebUIProviderOAuthAccounts)
mux.HandleFunc("/webui/api/provider/models", s.handleWebUIProviderModels)
mux.HandleFunc("/webui/api/provider/runtime", s.handleWebUIProviderRuntime)
mux.HandleFunc("/webui/api/provider/runtime/summary", s.handleWebUIProviderRuntimeSummary)
mux.HandleFunc("/webui/api/whatsapp/status", s.handleWebUIWhatsAppStatus)
@@ -1104,9 +1105,6 @@ func (s *Server) handleWebUIProviderOAuthComplete(w http.ResponseWriter, r *http
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
if len(models) > 0 {
pc.Models = models
}
if session.CredentialFile != "" {
pc.OAuth.CredentialFile = session.CredentialFile
pc.OAuth.CredentialFiles = appendUniqueStrings(pc.OAuth.CredentialFiles, session.CredentialFile)
@@ -1181,9 +1179,6 @@ func (s *Server) handleWebUIProviderOAuthImport(w http.ResponseWriter, r *http.R
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
if len(models) > 0 {
pc.Models = models
}
if session.CredentialFile != "" {
pc.OAuth.CredentialFile = session.CredentialFile
pc.OAuth.CredentialFiles = appendUniqueStrings(pc.OAuth.CredentialFiles, session.CredentialFile)
@@ -1278,6 +1273,49 @@ func (s *Server) handleWebUIProviderOAuthAccounts(w http.ResponseWriter, r *http
}
}
func (s *Server) handleWebUIProviderModels(w http.ResponseWriter, r *http.Request) {
if !s.checkAuth(r) {
http.Error(w, "unauthorized", http.StatusUnauthorized)
return
}
if r.Method != http.MethodPost {
http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
return
}
var body struct {
Provider string `json:"provider"`
Model string `json:"model"`
Models []string `json:"models"`
}
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
http.Error(w, "invalid json", http.StatusBadRequest)
return
}
cfg, pc, err := s.loadProviderConfig(strings.TrimSpace(body.Provider))
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
models := make([]string, 0, len(body.Models)+1)
for _, model := range body.Models {
models = appendUniqueStrings(models, model)
}
models = appendUniqueStrings(models, body.Model)
if len(models) == 0 {
http.Error(w, "model required", http.StatusBadRequest)
return
}
pc.Models = models
if err := s.saveProviderConfig(cfg, body.Provider, pc); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
_ = json.NewEncoder(w).Encode(map[string]interface{}{
"ok": true,
"models": pc.Models,
})
}
func (s *Server) handleWebUIProviderRuntime(w http.ResponseWriter, r *http.Request) {
if !s.checkAuth(r) {
http.Error(w, "unauthorized", http.StatusUnauthorized)