"use strict"; /** * agent-session-runner.ts * * Streaming variant of runAgent wired to a VIBN agent_sessions row. * After every LLM turn + tool call, it PATCHes the session in the VIBN DB * so the frontend can poll (and later WebSocket) the live output. * * Key differences from runAgent: * - Accepts an `emit` callback instead of updating job-store * - Accepts an `isStopped` check so the frontend can cancel mid-run * - Tracks which files were written/modified for the changed_files panel * - Calls vibn-frontend's PATCH /api/projects/[id]/agent/sessions/[sid] */ Object.defineProperty(exports, "__esModule", { value: true }); exports.runSessionAgent = runSessionAgent; const child_process_1 = require("child_process"); const vibn_chat_model_1 = require("./llm/vibn-chat-model"); const tools_1 = require("./tools"); const loader_1 = require("./prompts/loader"); const MAX_TURNS = 60; // ── VIBN DB bridge ──────────────────────────────────────────────────────────── async function patchSession(opts, payload) { const url = `${opts.vibnApiUrl}/api/projects/${opts.projectId}/agent/sessions/${opts.sessionId}`; try { await fetch(url, { method: "PATCH", headers: { "Content-Type": "application/json", "x-agent-runner-secret": process.env.AGENT_RUNNER_SECRET ?? "", }, body: JSON.stringify(payload), }); } catch (err) { // Log but don't crash — output will be lost for this line but loop continues console.warn("[session-runner] PATCH failed:", err instanceof Error ? err.message : err); } } function now() { return new Date().toISOString(); } // ── File change tracking ────────────────────────────────────────────────────── const FILE_WRITE_TOOLS = new Set([ "write_file", "replace_in_file", "create_file", ]); function extractChangedFile(toolName, args, workspaceRoot, appPath) { if (!FILE_WRITE_TOOLS.has(toolName)) return null; const rawPath = String(args.path ?? args.file_path ?? ""); if (!rawPath) return null; // Make path relative to appPath for display const fullPrefix = `${workspaceRoot}/${appPath}/`; const appPrefix = `${appPath}/`; let displayPath = rawPath.replace(fullPrefix, "").replace(appPrefix, ""); const fileStatus = toolName === "write_file" ? "added" : "modified"; return { path: displayPath, status: fileStatus }; } // ── Auto-commit helper ──────────────────────────────────────────────────────── async function autoCommitAndDeploy(opts, task, emit) { const repoRoot = opts.repoRoot; if (!repoRoot || !opts.giteaRepo) { await emit({ ts: now(), type: "info", text: "Auto-approve skipped — no repo root available.", }); return; } const gitOpts = { cwd: repoRoot, stdio: "pipe" }; const giteaApiUrl = process.env.GITEA_API_URL || ""; const giteaUsername = process.env.GITEA_USERNAME || "agent"; const giteaToken = process.env.GITEA_API_TOKEN || ""; try { try { (0, child_process_1.execSync)('git config user.email "agent@vibnai.com"', gitOpts); (0, child_process_1.execSync)('git config user.name "VIBN Agent"', gitOpts); } catch { /* already set */ } (0, child_process_1.execSync)("git add -A", gitOpts); const status = (0, child_process_1.execSync)("git status --porcelain", gitOpts) .toString() .trim(); if (!status) { await emit({ ts: now(), type: "info", text: "✓ No file changes to commit.", }); await patchSession(opts, { status: "approved" }); return; } const commitMsg = `agent: ${task.slice(0, 72)}`; const msgFile = require("path").join(opts.repoRoot || process.cwd(), ".git", "COMMIT_EDITMSG"); require("fs").writeFileSync(msgFile, commitMsg, "utf8"); (0, child_process_1.execSync)("git commit -F .git/COMMIT_EDITMSG", gitOpts); try { require("fs").unlinkSync(msgFile); } catch { } await emit({ ts: now(), type: "info", text: `✓ Committed: "${commitMsg}"`, }); const authedUrl = `${giteaApiUrl}/${opts.giteaRepo}.git`.replace("https://", `https://${giteaUsername}:${giteaToken}@`); (0, child_process_1.execSync)(`git push "${authedUrl}" HEAD:main`, gitOpts); await emit({ ts: now(), type: "info", text: "✓ Pushed to Gitea." }); // Optional Coolify deploy let deployed = false; if (opts.coolifyApiUrl && opts.coolifyApiToken && opts.coolifyAppUuid) { try { const deployRes = await fetch(`${opts.coolifyApiUrl}/api/v1/applications/${opts.coolifyAppUuid}/start`, { method: "POST", headers: { Authorization: `Bearer ${opts.coolifyApiToken}` }, }); deployed = deployRes.ok; if (deployed) await emit({ ts: now(), type: "info", text: "✓ Deployment triggered.", }); } catch { /* best-effort */ } } await patchSession(opts, { status: "approved", outputLine: { ts: now(), type: "done", text: `✓ Auto-committed & ${deployed ? "deployed" : "pushed"}. No approval needed.`, }, }); } catch (err) { const msg = err instanceof Error ? err.message : String(err); await emit({ ts: now(), type: "error", text: `Auto-commit failed: ${msg}`, }); // Fall back to done so user can manually approve await patchSession(opts, { status: "done" }); } } // ── Main streaming execution loop ───────────────────────────────────────────── async function runSessionAgent(config, task, ctx, opts) { const systemPrompt = (0, loader_1.resolvePrompt)(config.promptId); const emit = async (line) => { console.log(`[session ${opts.sessionId}] ${line.type}: ${line.text}`); await patchSession(opts, { outputLine: line }); }; await emit({ ts: now(), type: "info", text: `Agent starting working in ${opts.appPath}`, }); // Scope the system prompt to the specific app within the monorepo const basePrompt = (0, loader_1.resolvePrompt)(config.promptId); const scopedPrompt = `${basePrompt} ## Active context You are working inside the monorepo directory: ${opts.appPath} All file paths you use should be relative to this directory unless otherwise specified. When running commands, always cd into ${opts.appPath} first unless already there. Do NOT run git commit or git push — the platform handles committing after you finish. `; const history = [{ role: "user", content: task }]; let turn = 0; let toolCallsSinceText = 0; let roundsSinceText = 0; const toolFingerprints = []; let loopBreakReason = null; function fingerprintToolCall(tc) { if (tc.name === "shell_exec") { const cmd = String(tc.args?.command ?? "").trim(); const verb = cmd.split("&&").map(s => s.trim()).find(s => !s.startsWith("cd "))?.split(/\s+/)[0] ?? "shell"; return `shell_exec:${verb}`; } if (tc.name === "fs_write" || tc.name === "fs_edit" || tc.name === "fs_read") { return `${tc.name}:${tc.args?.path}`; } return `${tc.name}:${Object.values(tc.args ?? {})[0]}`; } while (turn < 30) { if (opts.isStopped()) { await emit({ ts: now(), type: "info", text: "Stopped by user." }); await patchSession(opts, { status: "stopped" }); return; } turn++; const isSilent = roundsSinceText >= 15 || toolCallsSinceText >= 20; const extraSystem = isSilent ? "\n\n[STATUS NUDGE] You have run " + `${toolCallsSinceText} tool call(s) over ${roundsSinceText} round(s) ` + "without sending the user any text. Before any more tool calls, " + "send ONE short sentence describing what you are currently working " + "on and why. The user is staring at silent tool pills." : ""; let resp; try { resp = await (0, vibn_chat_model_1.callVibnChat)({ systemPrompt: scopedPrompt + extraSystem, messages: history, tools: config.tools, temperature: 0.2 }); } catch (err) { const msg = err instanceof Error ? err.message : String(err); await emit({ ts: now(), type: "error", text: `LLM error: ${msg}` }); await patchSession(opts, { status: "failed", error: msg }); return; } if (resp.error) { await emit({ ts: now(), type: "error", text: `LLM error: ${resp.error}` }); await patchSession(opts, { status: "failed", error: resp.error }); return; } if (resp.text) { await emit({ ts: now(), type: "info", text: resp.text }); roundsSinceText = 0; toolCallsSinceText = 0; } else if (resp.toolCalls.length) { roundsSinceText++; toolCallsSinceText += resp.toolCalls.length; } if (!resp.toolCalls.length) { await patchSession(opts, { status: "completed" }); return; } for (const tc of resp.toolCalls) { toolFingerprints.push(fingerprintToolCall(tc)); } const window = toolFingerprints.slice(-10); const counts = new Map(); for (const fp of window) counts.set(fp, (counts.get(fp) ?? 0) + 1); let maxRepeats = 0; let repeatedCmd = ""; for (const [fp, n] of counts.entries()) { if (n > maxRepeats) { maxRepeats = n; repeatedCmd = fp.split("|")[0]; } } if (maxRepeats >= 6) { loopBreakReason = `Repeated ${repeatedCmd} ${maxRepeats}× in last 10 calls`; break; } history.push({ role: "assistant", content: resp.text, toolCalls: resp.toolCalls }); for (const tc of resp.toolCalls) { await emit({ ts: now(), type: "step", text: `Running ${tc.name}...` }); let result; try { result = await (0, tools_1.executeTool)(tc.name, tc.args, ctx); } catch (err) { result = { error: err instanceof Error ? err.message : String(err) }; } const resultStr = typeof result === "string" ? result : JSON.stringify(result, null, 2); history.push({ role: "tool", content: resultStr, toolCallId: tc.id, toolName: tc.name }); } } if (loopBreakReason) { await emit({ ts: now(), type: "error", text: `Loop broken: ${loopBreakReason}` }); await patchSession(opts, { status: "failed", error: loopBreakReason }); } else { await patchSession(opts, { status: "failed", error: "Max turns reached" }); } }