同步

server/task_resolution.go

@@ -0,0 +1,123 @@
+package server
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/Autumn-27/artex/db"
+)
+
+type taskLLMResolution struct {
+	ProfileID *int64 `json:"profile_id,omitempty"`
+	Name      string `json:"name"`
+	Format    string `json:"format"`
+	Model     string `json:"model"`
+	Source    string `json:"source"`
+	Available bool   `json:"available"`
+	Reason    string `json:"reason,omitempty"`
+}
+
+func (s *Server) resolutionFromProfile(p *db.LLMProfile, source string) taskLLMResolution {
+	if p == nil {
+		return taskLLMResolution{Source: source, Reason: "LLM 配置不存在"}
+	}
+	id := p.ID
+	result := taskLLMResolution{
+		ProfileID: &id,
+		Name:      p.Name,
+		Format:    p.Format,
+		Model:     p.Model,
+		Source:    source,
+	}
+	if p.APIKey == "" {
+		result.Reason = "LLM 配置未设置 API Key"
+		return result
+	}
+	if _, _, ok := s.providerForProfile(p.ID); !ok {
+		result.Reason = "LLM 配置格式或参数无效"
+		return result
+	}
+	result.Available = true
+	return result
+}
+
+// resolveTaskRoleLLM mirrors taskLLMRuntime.current without creating a provider:
+// role Agent binding -> explicit task chain -> active/global environment config.
+// Database failures are returned rather than represented as an unavailable model.
+func (s *Server) resolveTaskRoleLLM(t *Task, agentKey string) (taskLLMResolution, error) {
+	if s == nil || s.m == nil || s.m.pg == nil {
+		return taskLLMResolution{}, fmt.Errorf("database unavailable")
+	}
+	a, err := s.m.pg.GetAgentByKey(agentKey)
+	if err != nil {
+		return taskLLMResolution{}, fmt.Errorf("load agent %q: %w", agentKey, err)
+	}
+	if a != nil && a.LLMProfileID != nil {
+		p, err := s.m.pg.ProfileByID(*a.LLMProfileID)
+		if err != nil {
+			return taskLLMResolution{}, fmt.Errorf("load agent LLM profile: %w", err)
+		}
+		if p != nil {
+			resolved := s.resolutionFromProfile(p, "agent_binding")
+			if resolved.Available {
+				return resolved, nil
+			}
+		}
+	}
+	state := t.llmStateSnapshot()
+	if len(state.ProfileIDs) > 0 {
+		if state.ActiveID == nil {
+			return taskLLMResolution{Source: "task_chain", Reason: "任务 LLM 配置链额度已耗尽"}, nil
+		}
+		p, err := s.m.pg.ProfileByID(*state.ActiveID)
+		if err != nil {
+			return taskLLMResolution{}, fmt.Errorf("load task LLM profile: %w", err)
+		}
+		return s.resolutionFromProfile(p, "task_chain"), nil
+	}
+	p, err := s.m.pg.ActiveProfile()
+	if err != nil {
+		return taskLLMResolution{}, fmt.Errorf("load global LLM profile: %w", err)
+	}
+	if p != nil {
+		resolved := s.resolutionFromProfile(p, "global_profile")
+		if resolved.Available {
+			return resolved, nil
+		}
+	}
+	s.cfgMu.Lock()
+	on, name, cfg := s.llmOn, s.llmProf, s.llmCfg
+	s.cfgMu.Unlock()
+	if on {
+		if name == "" {
+			name = "全局配置"
+		}
+		return taskLLMResolution{
+			Name: name, Format: cfg.Provider(), Model: cfg.Model,
+			Source: "environment", Available: true,
+		}, nil
+	}
+	return taskLLMResolution{Source: "global", Reason: "没有可用的 LLM 配置"}, nil
+}
+
+func (s *Server) taskLLMResolutionHandler(w http.ResponseWriter, r *http.Request) {
+	t, ok := s.m.Task(r.PathValue("id"))
+	if !ok {
+		writeErr(w, 404, "task not found")
+		return
+	}
+	roles := map[string]taskLLMResolution{}
+	for responseKey, agentKey := range map[string]string{
+		"mainagent": "mainagent",
+		"planner":   "planner",
+		"worker":    "worker",
+	} {
+		resolved, err := s.resolveTaskRoleLLM(t, agentKey)
+		if err != nil {
+			writeErr(w, http.StatusInternalServerError, err.Error())
+			return
+		}
+		roles[responseKey] = resolved
+	}
+	writeJSON(w, http.StatusOK, roles)
+}