150 lines
11 KiB
TypeScript
150 lines
11 KiB
TypeScript
import { useRef, useState } from "react";
|
||
import { useConsole } from "../context";
|
||
import { Card, Bar } from "../components/ui";
|
||
import { gatherAttachments } from "../api/client";
|
||
import { chvAttachmentBlocked, chvAvailability, chvAvailabilityText, isChvExecutor } from "../api/chv";
|
||
import { EFFORTS } from "../api/types";
|
||
|
||
export default function DispatchSection() {
|
||
const ctx = useConsole();
|
||
const [prompt, setPrompt] = useState("");
|
||
const [coordParts, setCoordParts] = useState("");
|
||
const [goalMode, setGoalMode] = useState(false);
|
||
const [goalInvincible, setGoalInvincible] = useState(false);
|
||
const [planMode, setPlanMode] = useState(false);
|
||
const [executor, setExecutor] = useState("docker");
|
||
const [model, setModel] = useState("");
|
||
const [modelCustom, setModelCustom] = useState("");
|
||
const [effort, setEffort] = useState("");
|
||
const [approval, setApproval] = useState("");
|
||
const [childModel, setChildModel] = useState("");
|
||
const [childModelCustom, setChildModelCustom] = useState("");
|
||
const [childEffort, setChildEffort] = useState("");
|
||
const filesRef = useRef<HTMLInputElement>(null);
|
||
const dirRef = useRef<HTMLInputElement>(null);
|
||
const imgRef = useRef<HTMLInputElement>(null);
|
||
const [hasAttachments, setHasAttachments] = useState(false);
|
||
const [msg, setMsg] = useState("");
|
||
const chvAvail = chvAvailability(ctx.workers, ctx.workersSummary, ctx.workersError);
|
||
const isChv = isChvExecutor(executor);
|
||
function refreshAttachments() {
|
||
setHasAttachments(
|
||
!!(filesRef.current?.files?.length || dirRef.current?.files?.length || imgRef.current?.files?.length),
|
||
);
|
||
}
|
||
|
||
async function submit() {
|
||
setMsg("");
|
||
if (!ctx.projectId) { setMsg("先选项目"); return; }
|
||
if (!prompt.trim()) { setMsg("prompt 必填"); return; }
|
||
const body: any = { project_id: ctx.projectId, prompt: prompt.trim() };
|
||
if (ctx.envSelect) body.environment_id = ctx.envSelect;
|
||
if (executor) body.executor = executor;
|
||
const m = modelCustom.trim() || model;
|
||
if (m) body.model = m;
|
||
if (effort) body.effort = effort;
|
||
const cm = childModelCustom.trim() || childModel;
|
||
if (cm) body.child_model = cm;
|
||
if (childEffort) body.child_effort = childEffort;
|
||
if (approval) {
|
||
// 选了完全访问也不再弹确认框：下拉里选 full 本身就是
|
||
// 显式选择，提交时直接送 approval；计划模式与 full
|
||
// 互斥仍在下面拦截。
|
||
body.approval = approval;
|
||
}
|
||
if (planMode) {
|
||
if (approval === "full") { setMsg("计划模式不能与完全访问同用"); return; }
|
||
if (goalMode || goalInvincible) { setMsg("计划模式不能与 goal 模式同用"); return; }
|
||
body.plan = 1;
|
||
}
|
||
// 无敌版 = goal 模式 + 任务不限时：勾了它就同时发 goal_mode 与
|
||
// no_time_limit（服务端已支持该字段）；与普通 goal 勾选不冲突，
|
||
// 无敌版优先，单独勾普通 goal 时行为不变。
|
||
if (goalInvincible) { body.goal_mode = 1; body.no_time_limit = 1; }
|
||
else if (goalMode) body.goal_mode = 1;
|
||
const coordRaw = coordParts.trim();
|
||
if (coordRaw) {
|
||
try { body.coordinator_parts = JSON.parse(coordRaw); } catch { setMsg("协调员子任务 JSON 解析失败"); return; }
|
||
if (!Array.isArray(body.coordinator_parts)) { setMsg("协调员子任务必须是数组"); return; }
|
||
if (goalMode || goalInvincible || planMode) { setMsg("协调员任务不能与 goal/计划模式同用：协调员任务本身不跑模型回合，请取消 goal/计划勾选，或清空协调员 JSON"); return; }
|
||
}
|
||
const atts = gatherAttachments(filesRef.current?.files || null, dirRef.current?.files || null, imgRef.current?.files || null);
|
||
if (chvAttachmentBlocked(executor, atts.length > 0)) { setMsg("cloud-hypervisor 任务不支持附件：请先清空附件再提交（后端对带附件的 CHV 任务会 400）"); return; }
|
||
try {
|
||
const result = await ctx.dispatch(body, atts);
|
||
setMsg(result);
|
||
setPrompt(""); setCoordParts("");
|
||
if (filesRef.current) filesRef.current.value = "";
|
||
if (dirRef.current) dirRef.current.value = "";
|
||
if (imgRef.current) imgRef.current.value = "";
|
||
setHasAttachments(false);
|
||
} catch (e: any) { setMsg("提交失败：" + (e?.message || String(e))); }
|
||
}
|
||
|
||
return (
|
||
<Card id="nav-newtask" className="composer" title="新建任务">
|
||
<Bar>
|
||
<select aria-label="环境" value={ctx.envSelect} onChange={(e) => ctx.setEnvSelect(e.target.value)}>
|
||
<option value="">环境：不指定（走项目仓库）</option>
|
||
{ctx.envs.map((e) => <option key={e.id} value={e.id} disabled={e.status !== "ready"}>{e.name}（{e.status}）</option>)}
|
||
</select>
|
||
<select aria-label="执行器" value={executor} onChange={(e) => setExecutor(e.target.value)}>
|
||
<option value="docker">docker（默认，多用户下隔离最强）</option><option value="local">local</option><option value="cloud">cloud</option><option value="cloud-hypervisor">cloud-hypervisor（实验，默认关闭）</option>
|
||
</select>
|
||
<label><input type="checkbox" checked={goalMode} onChange={(e) => setGoalMode(e.target.checked)} /> goal 模式</label>
|
||
<label><input type="checkbox" checked={goalInvincible} onChange={(e) => setGoalInvincible(e.target.checked)} /> goal 模式无敌版</label>
|
||
<label><input type="checkbox" checked={planMode} onChange={(e) => setPlanMode(e.target.checked)} /> 计划模式（只读）</label>
|
||
</Bar>
|
||
{isChv ? <div className="muted">实验执行器，默认关闭：需后端已开启 CHV worker 且宿主有 KVM，否则任务无人认领、一直排队；带附件时建任务即被后端拒绝（错误文案原样显示在下方）。前端只传 executor 值，不绕过后端白名单；默认仍是 docker。</div> : null}
|
||
{isChv ? (
|
||
<div className="muted" aria-live="polite">
|
||
{chvAvailabilityText(chvAvail)}
|
||
{" "}<button type="button" onClick={ctx.refreshWorkers}>刷新 CHV 可用性</button>
|
||
</div>
|
||
) : null}
|
||
<textarea className="prompt stacked" placeholder="任务 prompt（必填）" value={prompt} onChange={(e) => setPrompt(e.target.value)} />
|
||
<textarea className="coord stacked" placeholder="协调员子任务 JSON（可选），留空为普通任务" value={coordParts} onChange={(e) => setCoordParts(e.target.value)} />
|
||
<div className="muted">协调员格式：[{"{"}"name":"后端","prompt":"做后端"{"}"},{"{"}"name":"前端","prompt":"做前端"{"}"}]；提交后 worker 并行建子任务、等齐、合流汇总。不能与 goal/计划模式同用。协调员与 goal/计划互斥：勾了 goal 或计划模式时不要填协调员 JSON，提交前会校验并拦截。子任务费用记父任务创建人账；父任务创建时按分工数预留配额，余额不足时整个父任务被挡下。</div>
|
||
<Bar>
|
||
<select aria-label="模型" value={model} onChange={(e) => setModel(e.target.value)}>
|
||
<option value="">模型：默认（跟随配置）</option>
|
||
{ctx.models.map((m) => <option key={m} value={m}>{m}</option>)}
|
||
</select>
|
||
{ctx.modelCustomVisible ? <input type="text" placeholder="手动输入模型（中转列表取不到时用）" value={modelCustom} onChange={(e) => setModelCustom(e.target.value)} /> : null}
|
||
<select aria-label="推理强度" value={effort} onChange={(e) => setEffort(e.target.value)}>
|
||
<option value="">推理强度：默认（跟随配置）</option>
|
||
{EFFORTS.map((x) => <option key={x} value={x}>{x}</option>)}
|
||
</select>
|
||
<select aria-label="权限" value={approval} onChange={(e) => setApproval(e.target.value)}>
|
||
<option value="">权限：默认（跟随配置）</option><option value="ask">请求批准</option><option value="auto">帮我批准</option><option value="full">完全访问</option>
|
||
</select>
|
||
</Bar>
|
||
<Bar>
|
||
<select aria-label="子任务模型" value={childModel} onChange={(e) => setChildModel(e.target.value)}>
|
||
<option value="">子任务模型：跟随本任务模型</option>
|
||
{ctx.models.map((m) => <option key={m} value={m}>{m}</option>)}
|
||
</select>
|
||
<input type="text" placeholder="子任务模型手动输入（中转列表里没有时填，如 gpt-6-luna；留空=跟随本任务模型）" value={childModelCustom} onChange={(e) => setChildModelCustom(e.target.value)} />
|
||
<select aria-label="子任务强度" value={childEffort} onChange={(e) => setChildEffort(e.target.value)}>
|
||
<option value="">子任务强度：跟随本任务强度</option>
|
||
{EFFORTS.map((x) => <option key={x} value={x}>{x}</option>)}
|
||
</select>
|
||
</Bar>
|
||
<div className="muted">子任务模型/强度是派生默认：本任务用 marker task-create 或协调员分工派生的子任务，未显式写模型/强度时用它；块/分工里显式写了则覆盖它。也可以不在这里填、直接在 prompt 里写“子任务用 gpt-6-luna max”，建任务时服务端会解析成同一对字段（任务详情里可核对）。</div>
|
||
{goalMode ? <div className="muted">goal 模式：输入框内容即目标，不需要专用 prompt 与验收命令；不限轮次，模型确认达成后在回复末尾输出“目标已达成”即结束，未达成会一直运行，可用取消停止。</div> : null}
|
||
{goalInvincible ? <div className="muted">goal 模式无敌版：goal 模式 + 任务不限时（去掉 worker/docker 总超时；单轮 600 秒护栏还在，不限的是任务总时长与总轮次）。服务端默认的自动续跑与断点镜像同时生效。</div> : null}
|
||
<Bar>
|
||
<span className="muted">附件（放进任务工作区，图片与普通文件不区分）：</span>
|
||
<input ref={filesRef} type="file" multiple title="文件" aria-label="附件文件" disabled={isChv} onChange={refreshAttachments} />
|
||
<input ref={dirRef} type="file" multiple title="文件夹" aria-label="附件文件夹" disabled={isChv} onChange={refreshAttachments} {...({ webkitdirectory: "" } as any)} />
|
||
<input ref={imgRef} type="file" accept="image/*" multiple title="图片" aria-label="附件图片" disabled={isChv} onChange={refreshAttachments} />
|
||
</Bar>
|
||
{isChv ? <div className="muted">cloud-hypervisor 任务不支持附件：附件输入已禁用（后端对带附件的 CHV 任务会 400，错误文案原样显示在下方；切换回其他执行器可恢复附件输入）。{hasAttachments ? <>当前仍有已选附件，请先清空再提交。 <button type="button" onClick={() => { if (filesRef.current) filesRef.current.value = ""; if (dirRef.current) dirRef.current.value = ""; if (imgRef.current) imgRef.current.value = ""; setHasAttachments(false); }}>清空附件</button></> : ""}</div> : null}
|
||
<Bar>
|
||
<button className="seal" onClick={submit}>提交任务</button>
|
||
<span className="muted" role="status" aria-live="polite">{msg || ctx.dispatchMsg}</span>
|
||
</Bar>
|
||
</Card>
|
||
);
|
||
}
|