Add skill sync action

This commit is contained in:
wuyang6
2026-05-11 09:59:34 +08:00
parent 010dc1a88d
commit b4bd7698a4
3 changed files with 162 additions and 0 deletions
+80
View File
@@ -13,6 +13,7 @@ import queue
import re
import shutil
import signal
import subprocess
import threading
import time
import venv
@@ -632,6 +633,69 @@ class AgentState:
if key.startswith(account_prefix):
self._agents.pop(key, None)
def _clear_agents_for_account(self, account_id: str | None) -> None:
account_prefix = f'{self._account_key(account_id)}:'
for key in list(self._agents):
if key.startswith(account_prefix):
self._agents.pop(key, None)
def sync_skills_from_git(self, account_id: str | None) -> dict[str, Any]:
"""Pull the current git branch and refresh account-scoped skill state."""
with self._lock:
config = self._config_for(account_id)
repo = config.cwd
if not (repo / '.git').exists():
raise ValueError(f'当前工作目录不是 git 仓库:{repo}')
def run_git(args: list[str], *, timeout: int = 60) -> subprocess.CompletedProcess[str]:
env = {
**os.environ,
'GIT_TERMINAL_PROMPT': '0',
}
proc = subprocess.run(
['git', *args],
cwd=repo,
capture_output=True,
text=True,
timeout=timeout,
env=env,
)
if proc.returncode != 0:
detail = (proc.stderr or proc.stdout or '').strip()
raise ValueError(f'git {" ".join(args)} 失败:{detail}')
return proc
dirty = run_git(['status', '--porcelain', '--untracked-files=no']).stdout.strip()
if dirty:
raise ValueError('工作区存在未提交的 tracked 改动,已停止同步 Skill。')
branch = run_git(['branch', '--show-current']).stdout.strip() or 'main'
before = run_git(['rev-parse', 'HEAD']).stdout.strip()
run_git(['fetch', 'origin'], timeout=120)
pull = run_git(['pull', '--ff-only', 'origin', branch], timeout=120)
after = run_git(['rev-parse', 'HEAD']).stdout.strip()
changed_files: list[str] = []
if before != after:
changed_files = [
line.strip()
for line in run_git(
['diff', '--name-only', f'{before}..{after}'],
timeout=60,
).stdout.splitlines()
if line.strip()
]
self._clear_agents_for_account(account_id)
skills = get_bundled_skills(config.cwd)
return {
'branch': branch,
'before': before,
'after': after,
'updated': before != after,
'changed_files': changed_files,
'skill_count': sum(1 for skill in skills if skill.user_invocable),
'message': pull.stdout.strip() or pull.stderr.strip(),
}
def _build_agent(self, account_id: str | None = None) -> LocalCodingAgent:
config = self._config_for(account_id)
paths = self.account_paths(account_id)
@@ -797,6 +861,10 @@ class SkillPreferenceUpdate(BaseModel):
account_id: str | None = None
class SkillSyncRequest(BaseModel):
account_id: str | None = None
class SessionTitleUpdate(BaseModel):
title: str = Field(min_length=1, max_length=80)
@@ -879,6 +947,18 @@ def create_app(state: AgentState) -> FastAPI:
raise HTTPException(status_code=400, detail=str(exc))
return await list_skills(payload.account_id)
@app.post('/api/skills/sync')
async def sync_skills(payload: SkillSyncRequest) -> dict[str, Any]:
try:
result = await asyncio.to_thread(
state.sync_skills_from_git,
payload.account_id,
)
except (ValueError, subprocess.TimeoutExpired) as exc:
raise HTTPException(status_code=400, detail=str(exc))
result['skills'] = await list_skills(payload.account_id)
return result
@app.get('/api/models')
async def list_models_get(account_id: str | None = None) -> dict[str, Any]:
config = state.config_for(account_id)
+25
View File
@@ -47,3 +47,28 @@ export async function PATCH(request: Request) {
},
});
}
export async function POST() {
const account = await getCurrentAccount();
if (!account)
return Response.json({ error: "unauthorized" }, { status: 401 });
const response = await fetch(`${CLAW_API_URL}/api/skills/sync`, {
method: "POST",
headers: {
"content-type": "application/json",
},
body: JSON.stringify({
account_id: account.id,
}),
cache: "no-store",
});
const payload = await response.text();
return new Response(payload, {
status: response.status,
headers: {
"content-type":
response.headers.get("content-type") ?? "application/json",
},
});
}
@@ -76,6 +76,17 @@ type SkillSummary = {
configurable?: boolean;
};
type SkillSyncResponse = {
skills?: SkillSummary[];
updated?: boolean;
before?: string;
after?: string;
changed_files?: string[];
message?: string;
error?: string;
detail?: string;
};
type SlashCommandSummary = {
names: string[];
primary: string;
@@ -1099,6 +1110,7 @@ function SkillInsertDialog() {
const [skills, setSkills] = useState<SkillSummary[]>([]);
const [status, setStatus] = useState("点击后读取当前后端 Skill 列表。");
const [updatingSkill, setUpdatingSkill] = useState<string | null>(null);
const [syncingSkills, setSyncingSkills] = useState(false);
async function loadSkills() {
setStatus("正在读取 Skill 列表...");
@@ -1119,6 +1131,33 @@ function SkillInsertDialog() {
}
}
async function syncSkills() {
setSyncingSkills(true);
setStatus("正在从 git 同步 Skill...");
try {
const response = await fetch("/api/claw/skills", {
method: "POST",
cache: "no-store",
});
const payload = (await response.json()) as SkillSyncResponse;
if (!response.ok || !Array.isArray(payload.skills)) {
throw new Error(payload.detail ?? payload.error ?? "同步 Skill 失败");
}
setSkills(payload.skills);
const changedCount = payload.changed_files?.length ?? 0;
if (payload.updated) {
const shortHash = payload.after ? payload.after.slice(0, 7) : "最新提交";
setStatus(`Skill 已更新到 ${shortHash},变更 ${changedCount} 个文件。`);
} else {
setStatus("Skill 已是最新。");
}
} catch (err) {
setStatus(err instanceof Error ? err.message : "同步 Skill 失败");
} finally {
setSyncingSkills(false);
}
}
async function setSkillEnabled(skill: SkillSummary, enabled: boolean) {
if (skill.configurable === false) return;
const previous = skills;
@@ -1182,6 +1221,24 @@ function SkillInsertDialog() {
Skill
</DialogDescription>
</DialogHeader>
<div className="flex items-center justify-between gap-3">
<Button
type="button"
variant="outline"
size="sm"
className="h-8 gap-2"
disabled={syncingSkills}
onClick={syncSkills}
>
<RefreshCwIcon
className={cn("size-3.5", syncingSkills && "animate-spin")}
/>
Skill
</Button>
<p className="text-muted-foreground text-xs">
git skill
</p>
</div>
{status ? (
<p className="text-muted-foreground text-sm">{status}</p>
) : null}