kandev/apps/web/components/settings/utility-agents-section.tsx

"use client";

import { useCallback, useEffect, useMemo, useState } from "react";
import {
  listUtilityAgents,
  listInferenceAgents,
  updateUtilityAgent,
  deleteUtilityAgent,
  type UtilityAgent,
  type InferenceAgent,
} from "@/lib/api/domains/utility-api";
import { fetchUserSettings, updateUserSettings } from "@/lib/api/domains/settings-api";
import { UtilityAgentDialog } from "@/components/settings/utility-agent-dialog";
import {
  DefaultModelSection,
  PerActionOverridesSection,
  CustomAgentsSection,
  USE_DEFAULT,
} from "@/components/settings/utility-sections";
import { useInferenceAgents } from "@/components/settings/use-inference-agents";

function buildAllModels(inferenceAgents: InferenceAgent[]) {
  return inferenceAgents.flatMap((ia) =>
    (ia.models ?? []).map((m) => ({
      value: `${ia.id}|${m.id}`,
      label: `${ia.display_name} / ${m.name}`,
      agentName: ia.display_name,
      modelName: m.name,
    })),
  );
}

async function handleBuiltinChange(
  agent: UtilityAgent,
  value: string,
  setAgents: React.Dispatch<React.SetStateAction<UtilityAgent[]>>,
) {
  const isDefault = value === USE_DEFAULT;
  const [agentId, model] = isDefault ? ["", ""] : value.split("|");
  await updateUtilityAgent(agent.id, { agent_id: agentId, model, enabled: true });
  setAgents((prev) =>
    prev.map((a) => (a.id === agent.id ? { ...a, agent_id: agentId, model } : a)),
  );
}

export function UtilityAgentsSection() {
  const [agents, setAgents] = useState<UtilityAgent[]>([]);
  const { inferenceAgents, setInferenceAgents, refreshAgent } = useInferenceAgents();
  const [defaultAgentId, setDefaultAgentId] = useState("");
  const [defaultModel, setDefaultModel] = useState("");
  const [loading, setLoading] = useState(true);
  const [dialogOpen, setDialogOpen] = useState(false);
  const [editingAgent, setEditingAgent] = useState<UtilityAgent | null>(null);

  const builtins = useMemo(() => agents.filter((a) => a.builtin), [agents]);
  const customAgents = useMemo(() => agents.filter((a) => !a.builtin), [agents]);
  const allModels = useMemo(() => buildAllModels(inferenceAgents), [inferenceAgents]);

  const fetchData = useCallback(async () => {
    try {
      const [agentsRes, inferenceRes, settingsRes] = await Promise.all([
        listUtilityAgents({ cache: "no-store" }),
        listInferenceAgents(),
        fetchUserSettings({ cache: "no-store" }),
      ]);
      setAgents(agentsRes.agents);
      setInferenceAgents(inferenceRes.agents);
      setDefaultAgentId(settingsRes.settings.default_utility_agent_id || "");
      setDefaultModel(settingsRes.settings.default_utility_model || "");
    } catch {
      setAgents([]);
      setInferenceAgents([]);
    } finally {
      setLoading(false);
    }
  }, [setInferenceAgents]);

  const handleDefaultChange = async (agentId: string, model: string) => {
    const prevAgentId = defaultAgentId;
    const prevModel = defaultModel;
    setDefaultAgentId(agentId);
    setDefaultModel(model);
    try {
      await updateUserSettings({ default_utility_agent_id: agentId, default_utility_model: model });
    } catch {
      setDefaultAgentId(prevAgentId);
      setDefaultModel(prevModel);
    }
  };

  const openEditDialog = (agent: UtilityAgent | null) => {
    setEditingAgent(agent);
    setDialogOpen(true);
  };

  const closeDialog = () => {
    setDialogOpen(false);
    setEditingAgent(null);
    fetchData();
  };

  useEffect(() => {
    fetchData();
  }, [fetchData]);

  if (loading) return null;

  return (
    <>
      <div className="space-y-4">
        <div>
          <h2 className="text-2xl font-bold">Utility Agents</h2>
          <p className="text-sm text-muted-foreground">
            One-shot AI helpers for commits, PRs, and prompts.
          </p>
        </div>
        <DefaultModelSection
          inferenceAgents={inferenceAgents}
          defaultAgentId={defaultAgentId}
          defaultModel={defaultModel}
          onDefaultChange={handleDefaultChange}
          onRefreshAgent={refreshAgent}
        />
        <PerActionOverridesSection
          builtins={builtins}
          allModels={allModels}
          defaultModel={defaultModel}
          onModelChange={(agent, value) => handleBuiltinChange(agent, value, setAgents)}
          onEdit={openEditDialog}
        />
        <CustomAgentsSection
          agents={customAgents}
          onAdd={() => openEditDialog(null)}
          onEdit={openEditDialog}
          onDelete={async (agent) => {
            try {
              await deleteUtilityAgent(agent.id);
              setAgents((prev) => prev.filter((a) => a.id !== agent.id));
            } catch {
              // Error already logged by API layer
            }
          }}
        />
      </div>
      <UtilityAgentDialog
        open={dialogOpen}
        onOpenChange={setDialogOpen}
        agent={editingAgent}
        onSuccess={closeDialog}
      />
    </>
  );
}