feat: integration with agents

This commit is contained in:
Gab
2026-03-27 08:42:09 +11:00
parent cf023340a6
commit b9ced47bf8
11 changed files with 1341 additions and 223 deletions

View File

@@ -1,63 +1,73 @@
#!/usr/bin/env node
import { spawn } from 'child_process';
import { existsSync, mkdirSync, writeFileSync, readFileSync } from 'fs';
import { join, dirname } from 'path';
import { homedir } from 'os';
import * as readline from 'readline';
import { fileURLToPath } from 'url';
import { spawn } from "child_process"
import { existsSync, mkdirSync, writeFileSync, readFileSync } from "fs"
import { join, dirname } from "path"
import { homedir } from "os"
import * as readline from "readline"
import { fileURLToPath } from "url"
const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);
const __filename = fileURLToPath(import.meta.url)
const __dirname = dirname(__filename)
const TFCODE_DIR = join(homedir(), '.tfcode');
const TOOLS_FILE = join(TFCODE_DIR, 'tools.json');
const CREDENTIALS_FILE = join(TFCODE_DIR, 'credentials.json');
const CONFIG_FILE = join(TFCODE_DIR, 'config.json');
const TFCODE_DIR = join(homedir(), ".tfcode")
const TOOLS_FILE = join(TFCODE_DIR, "tools.json")
const CREDENTIALS_FILE = join(TFCODE_DIR, "credentials.json")
const CONFIG_FILE = join(TFCODE_DIR, "config.json")
const COLORS = {
reset: '\x1b[0m',
bold: '\x1b[1m',
green: '\x1b[32m',
red: '\x1b[31m',
cyan: '\x1b[36m',
dim: '\x1b[90m',
yellow: '\x1b[33m',
magenta: '\x1b[35m'
};
reset: "\x1b[0m",
bold: "\x1b[1m",
green: "\x1b[32m",
red: "\x1b[31m",
cyan: "\x1b[36m",
dim: "\x1b[90m",
yellow: "\x1b[33m",
magenta: "\x1b[35m",
}
function log(msg) { console.log(msg); }
function success(msg) { console.log(`${COLORS.green}${COLORS.reset} ${msg}`); }
function error(msg) { console.error(`${COLORS.red}${COLORS.reset} ${msg}`); }
function info(msg) { console.log(`${COLORS.cyan}${COLORS.reset} ${msg}`); }
function log(msg) {
console.log(msg)
}
function success(msg) {
console.log(`${COLORS.green}${COLORS.reset} ${msg}`)
}
function error(msg) {
console.error(`${COLORS.red}${COLORS.reset} ${msg}`)
}
function info(msg) {
console.log(`${COLORS.cyan}${COLORS.reset} ${msg}`)
}
function ensureConfigDir() {
if (!existsSync(TFCODE_DIR)) mkdirSync(TFCODE_DIR, { recursive: true });
if (!existsSync(TFCODE_DIR)) mkdirSync(TFCODE_DIR, { recursive: true })
}
function loadConfig() {
const envConfig = {
workspace_id: process.env.TF_WORKSPACE_ID,
api_key: process.env.TF_API_KEY,
region: process.env.TF_REGION
};
if (envConfig.workspace_id && envConfig.api_key) return envConfig;
if (existsSync(CONFIG_FILE)) {
try { return JSON.parse(readFileSync(CONFIG_FILE, 'utf-8')); } catch {}
region: process.env.TF_REGION,
}
return null;
if (envConfig.workspace_id && envConfig.api_key) return envConfig
if (existsSync(CONFIG_FILE)) {
try {
return JSON.parse(readFileSync(CONFIG_FILE, "utf-8"))
} catch {}
}
return null
}
function saveConfig(config) {
ensureConfigDir();
writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
ensureConfigDir()
writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2))
}
function runPythonSync(method, config = null) {
const wsId = config?.workspace_id || process.env.TF_WORKSPACE_ID || '';
const apiKey = config?.api_key || process.env.TF_API_KEY || '';
const region = config?.region || process.env.TF_REGION || 'au';
const wsId = config?.workspace_id || process.env.TF_WORKSPACE_ID || ""
const apiKey = config?.api_key || process.env.TF_API_KEY || ""
const region = config?.region || process.env.TF_REGION || "au"
const pythonCode = `
import json, sys, os
try:
@@ -77,222 +87,430 @@ try:
elif method == "sync":
config = load_config()
result = sync_tools(config)
tools_data = [{"id": t.id, "name": t.name, "description": t.description, "tool_type": t.tool_type.value, "request_type": t.request_type.value if t.request_type else None, "url": t.url, "auth_via": t.auth_via} for t in result.tools]
tools_data = [{"id": t.id, "name": t.name, "description": t.description, "tool_type": t.tool_type.value, "request_type": t.request_type.value if t.request_type else None, "url": t.url, "auth_via": t.auth_via, "interpolation_string": t.interpolation_string, "goals": t.goals, "temperature": t.temperature, "max_tokens": t.max_tokens, "llm_base_model": t.llm_base_model, "llm_provider": t.llm_provider} for t in result.tools]
print(json.dumps({"success": result.success, "tools": tools_data, "by_type": result.by_type, "error": result.error}))
except Exception as e:
print(json.dumps({"success": False, "error": str(e)}))
`;
`
return new Promise((resolve, reject) => {
const proc = spawn(process.env.TFCODE_PYTHON_PATH || 'python3', ['-c', pythonCode], { env: { ...process.env } });
let stdout = '', stderr = '';
proc.stdout.on('data', (d) => stdout += d);
proc.stderr.on('data', (d) => stderr += d);
proc.on('close', (code) => {
if (code !== 0 && !stdout) reject(new Error(`Python failed: ${stderr}`));
else try { resolve(JSON.parse(stdout.trim())); } catch (e) { reject(new Error(`Parse error: ${stdout}`)); }
});
proc.on('error', reject);
});
const proc = spawn(process.env.TFCODE_PYTHON_PATH || "python3", ["-c", pythonCode], { env: { ...process.env } })
let stdout = "",
stderr = ""
proc.stdout.on("data", (d) => (stdout += d))
proc.stderr.on("data", (d) => (stderr += d))
proc.on("close", (code) => {
if (code !== 0 && !stdout) reject(new Error(`Python failed: ${stderr}`))
else
try {
resolve(JSON.parse(stdout.trim()))
} catch (e) {
reject(new Error(`Parse error: ${stdout}`))
}
})
proc.on("error", reject)
})
}
function loadCachedTools() {
if (!existsSync(TOOLS_FILE)) return null;
try { return JSON.parse(readFileSync(TOOLS_FILE, 'utf-8')); } catch { return null; }
if (!existsSync(TOOLS_FILE)) return null
try {
return JSON.parse(readFileSync(TOOLS_FILE, "utf-8"))
} catch {
return null
}
}
function saveToolsCache(tools) {
ensureConfigDir();
writeFileSync(TOOLS_FILE, JSON.stringify(tools, null, 2));
ensureConfigDir()
writeFileSync(TOOLS_FILE, JSON.stringify(tools, null, 2))
}
async function question(prompt) {
return new Promise((resolve) => {
const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
rl.question(prompt, (answer) => { rl.close(); resolve(answer.trim()); });
});
const rl = readline.createInterface({ input: process.stdin, output: process.stdout })
rl.question(prompt, (answer) => {
rl.close()
resolve(answer.trim())
})
})
}
async function select(prompt, options) {
log('');
log(prompt);
log('');
options.forEach((opt, i) => log(` ${COLORS.cyan}${i + 1}.${COLORS.reset} ${opt}`));
log('');
const answer = await question('Select (1-' + options.length + '): ');
const idx = parseInt(answer) - 1;
return idx >= 0 && idx < options.length ? idx : 0;
log("")
log(prompt)
log("")
options.forEach((opt, i) => log(` ${COLORS.cyan}${i + 1}.${COLORS.reset} ${opt}`))
log("")
const answer = await question("Select (1-" + options.length + "): ")
const idx = parseInt(answer) - 1
return idx >= 0 && idx < options.length ? idx : 0
}
async function interactiveSetup() {
log('');
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
log(`${COLORS.bold}${COLORS.magenta} tfcode Setup${COLORS.reset}`);
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
log('');
log('This will guide you through setting up your ToothFairyAI credentials.');
log('');
log(`${COLORS.dim}You can find your credentials at:${COLORS.reset}`);
log(`${COLORS.dim} https://app.toothfairyai.com → Settings → API Keys${COLORS.reset}`);
log('');
log(`${COLORS.bold}Step 1: Workspace ID${COLORS.reset}`);
log(`${COLORS.dim}This is your workspace UUID${COLORS.reset}`);
log('');
const workspaceId = await question('Enter your Workspace ID: ');
if (!workspaceId) { error('Workspace ID is required'); process.exit(1); }
log('');
log(`${COLORS.bold}Step 2: API Key${COLORS.reset}`);
log(`${COLORS.dim}Paste or type your API key${COLORS.reset}`);
log('');
const apiKey = await question('Enter your API Key: ');
if (!apiKey) { error('API Key is required'); process.exit(1); }
log('');
log(`${COLORS.bold}Step 3: Region${COLORS.reset}`);
const regions = ['dev (Development)', 'au (Australia)', 'eu (Europe)', 'us (United States)'];
const regionIdx = await select('Select your region:', regions);
const region = ['dev', 'au', 'eu', 'us'][regionIdx];
log('');
log(`${COLORS.bold}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
log('');
log(`${COLORS.bold}Summary:${COLORS.reset}`);
log(` Workspace ID: ${workspaceId}`);
log(` API Key: ***${apiKey.slice(-4)}`);
log(` Region: ${region}`);
log('');
const confirm = await question('Save these credentials? (Y/n): ');
if (confirm.toLowerCase() === 'n' || confirm.toLowerCase() === 'no') { log('Setup cancelled.'); return; }
const config = { workspace_id: workspaceId, api_key: apiKey, region };
saveConfig(config);
success('Credentials saved to ~/.tfcode/config.json');
log('');
const testNow = await question('Validate credentials now? (Y/n): ');
if (testNow.toLowerCase() === 'n' || testNow.toLowerCase() === 'no') return;
log('');
info('Validating credentials...');
log('');
log("")
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.magenta} tfcode Setup${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log("")
log("This will guide you through setting up your ToothFairyAI credentials.")
log("")
log(`${COLORS.dim}You can find your credentials at:${COLORS.reset}`)
log(`${COLORS.dim} https://app.toothfairyai.com → Settings → API Keys${COLORS.reset}`)
log("")
log(`${COLORS.bold}Step 1: Workspace ID${COLORS.reset}`)
log(`${COLORS.dim}This is your workspace UUID${COLORS.reset}`)
log("")
const workspaceId = await question("Enter your Workspace ID: ")
if (!workspaceId) {
error("Workspace ID is required")
process.exit(1)
}
log("")
log(`${COLORS.bold}Step 2: API Key${COLORS.reset}`)
log(`${COLORS.dim}Paste or type your API key${COLORS.reset}`)
log("")
const apiKey = await question("Enter your API Key: ")
if (!apiKey) {
error("API Key is required")
process.exit(1)
}
log("")
log(`${COLORS.bold}Step 3: Region${COLORS.reset}`)
const regions = ["dev (Development)", "au (Australia)", "eu (Europe)", "us (United States)"]
const regionIdx = await select("Select your region:", regions)
const region = ["dev", "au", "eu", "us"][regionIdx]
log("")
log(`${COLORS.bold}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log("")
log(`${COLORS.bold}Summary:${COLORS.reset}`)
log(` Workspace ID: ${workspaceId}`)
log(` API Key: ***${apiKey.slice(-4)}`)
log(` Region: ${region}`)
log("")
const confirm = await question("Save these credentials? (Y/n): ")
if (confirm.toLowerCase() === "n" || confirm.toLowerCase() === "no") {
log("Setup cancelled.")
return
}
const config = { workspace_id: workspaceId, api_key: apiKey, region }
saveConfig(config)
success("Credentials saved to ~/.tfcode/config.json")
log("")
const testNow = await question("Validate credentials now? (Y/n): ")
if (testNow.toLowerCase() === "n" || testNow.toLowerCase() === "no") return
log("")
info("Validating credentials...")
log("")
try {
const result = await runPythonSync('validate', config);
const result = await runPythonSync("validate", config)
if (result.success) {
success('Credentials valid!');
log(` API URL: ${result.base_url}`);
log(` Workspace ID: ${result.workspace_id}`);
log('');
const syncNow = await question('Sync tools now? (Y/n): ');
if (syncNow.toLowerCase() === 'n' || syncNow.toLowerCase() === 'no') return;
log('');
info('Syncing tools...');
log('');
const syncResult = await runPythonSync('sync', config);
success("Credentials valid!")
log(` API URL: ${result.base_url}`)
log(` Workspace ID: ${result.workspace_id}`)
log("")
const syncNow = await question("Sync tools now? (Y/n): ")
if (syncNow.toLowerCase() === "n" || syncNow.toLowerCase() === "no") return
log("")
info("Syncing tools...")
log("")
const syncResult = await runPythonSync("sync", config)
if (syncResult.success) {
saveToolsCache(syncResult);
success(`Synced ${syncResult.tools.length} tools`);
saveToolsCache(syncResult)
success(`Synced ${syncResult.tools.length} tools`)
if (syncResult.by_type && Object.keys(syncResult.by_type).length > 0) {
log('');
log('By type:');
for (const [type, count] of Object.entries(syncResult.by_type)) log(` ${type}: ${count}`);
log("")
log("By type:")
for (const [type, count] of Object.entries(syncResult.by_type)) log(` ${type}: ${count}`)
}
} else {
error(`Sync failed: ${syncResult.error}`);
error(`Sync failed: ${syncResult.error}`)
}
} else {
error(`Validation failed: ${result.error}`);
error(`Validation failed: ${result.error}`)
}
} catch (e) {
error(`Failed: ${e.message}`);
error(`Failed: ${e.message}`)
}
log('');
log(`${COLORS.bold}${COLORS.green}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
log(`${COLORS.bold}${COLORS.green} Setup Complete!${COLORS.reset}`);
log(`${COLORS.bold}${COLORS.green}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
log('');
log('Commands:');
log(` ${COLORS.cyan}tfcode validate${COLORS.reset} Check credentials`);
log(` ${COLORS.cyan}tfcode sync${COLORS.reset} Sync tools`);
log(` ${COLORS.cyan}tfcode tools list${COLORS.reset} List tools`);
log('');
log("")
log(`${COLORS.bold}${COLORS.green}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.green} Setup Complete!${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.green}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log("")
log("Commands:")
log(` ${COLORS.cyan}tfcode validate${COLORS.reset} Check credentials`)
log(` ${COLORS.cyan}tfcode sync${COLORS.reset} Sync tools`)
log(` ${COLORS.cyan}tfcode tools list${COLORS.reset} List tools`)
log("")
}
function showHelp() {
log('');
log(`${COLORS.bold}tfcode${COLORS.reset} - ToothFairyAI's AI coding agent`);
log('');
log('Commands:');
log(` ${COLORS.cyan}tfcode setup${COLORS.reset} Interactive credential setup`);
log(` ${COLORS.cyan}tfcode validate${COLORS.reset} Test credentials`);
log(` ${COLORS.cyan}tfcode sync${COLORS.reset} Sync tools from workspace`);
log(` ${COLORS.cyan}tfcode tools list${COLORS.reset} List synced tools`);
log(` ${COLORS.cyan}tfcode --help${COLORS.reset} Show this help`);
log(` ${COLORS.cyan}tfcode --version${COLORS.reset} Show version`);
log('');
log(`${COLORS.dim}For full TUI, run from source:${COLORS.reset}`);
log(`${COLORS.dim} bun run packages/tfcode/src/index.ts${COLORS.reset}`);
log('');
log("")
log(`${COLORS.bold}tfcode${COLORS.reset} - ToothFairyAI's AI coding agent`)
log("")
log("Commands:")
log(` ${COLORS.cyan}tfcode setup${COLORS.reset} Interactive credential setup`)
log(` ${COLORS.cyan}tfcode validate${COLORS.reset} Test credentials`)
log(` ${COLORS.cyan}tfcode sync${COLORS.reset} Sync tools from workspace`)
log(` ${COLORS.cyan}tfcode tools list${COLORS.reset} List synced tools`)
log(` ${COLORS.cyan}tfcode test-agent <id>${COLORS.reset} Test agent prompt injection`)
log(` ${COLORS.cyan}tfcode debug${COLORS.reset} Show debug info`)
log(` ${COLORS.cyan}tfcode --help${COLORS.reset} Show this help`)
log(` ${COLORS.cyan}tfcode --version${COLORS.reset} Show version`)
log("")
log(`${COLORS.dim}For full TUI, run from source:${COLORS.reset}`)
log(`${COLORS.dim} bun run packages/tfcode/src/index.ts${COLORS.reset}`)
log("")
}
const args = process.argv.slice(2);
const command = args[0];
if (args.includes('--help') || args.includes('-h')) {
showHelp();
} else if (args.includes('--version') || args.includes('-v')) {
log('tfcode v1.0.0-beta.9');
} else if (command === 'setup') {
interactiveSetup();
} else if (command === 'validate') {
(async () => {
const config = loadConfig();
if (!config) { error('No credentials. Run: tfcode setup'); process.exit(1); }
info('Validating...');
try {
const result = await runPythonSync('validate', config);
if (result.success) { success('Credentials valid'); log(` API URL: ${result.base_url}`); }
else { error(`Failed: ${result.error}`); process.exit(1); }
} catch (e) { error(`Failed: ${e.message}`); process.exit(1); }
})();
} else if (command === 'sync') {
(async () => {
const config = loadConfig();
if (!config) { error('No credentials. Run: tfcode setup'); process.exit(1); }
info('Syncing tools...');
try {
const result = await runPythonSync('sync', config);
if (result.success) {
saveToolsCache(result);
success(`Synced ${result.tools.length} tools`);
if (result.by_type) { log(''); log('By type:'); for (const [t, c] of Object.entries(result.by_type)) log(` ${t}: ${c}`); }
} else { error(`Failed: ${result.error}`); process.exit(1); }
} catch (e) { error(`Failed: ${e.message}`); process.exit(1); }
})();
} else if (command === 'tools' && args[1] === 'list') {
const cached = loadCachedTools();
if (!cached?.success) { error('No tools. Run: tfcode sync'); process.exit(1); }
let tools = cached.tools;
if (args[3] === '--type' && args[4]) tools = tools.filter(t => t.tool_type === args[4]);
log(`\n${tools.length} tool(s):\n`);
for (const t of tools) {
log(` ${COLORS.cyan}${t.name}${COLORS.reset}`);
log(` Type: ${t.tool_type}`);
if (t.description) log(` ${COLORS.dim}${t.description.slice(0, 60)}${COLORS.reset}`);
log(` Auth: ${t.auth_via}\n`);
function showDebugInfo() {
log("")
log(`${COLORS.bold}Debug Information${COLORS.reset}`)
log("")
log(` ${COLORS.bold}TFCODE_DIR:${COLORS.reset} ${TFCODE_DIR}`)
log(` ${COLORS.bold}TOOLS_FILE:${COLORS.reset} ${TOOLS_FILE}`)
log(` ${COLORS.bold}CONFIG_FILE:${COLORS.reset} ${CONFIG_FILE}`)
log(` ${COLORS.bold}CREDENTIALS_FILE:${COLORS.reset} ${CREDENTIALS_FILE}`)
log("")
log(` ${COLORS.bold}tools.json exists:${COLORS.reset} ${existsSync(TOOLS_FILE)}`)
if (existsSync(TOOLS_FILE)) {
const tools = loadCachedTools()
log(` ${COLORS.bold}tools.json valid:${COLORS.reset} ${tools?.success ?? false}`)
log(` ${COLORS.bold}tools count:${COLORS.reset} ${tools?.tools?.length ?? 0}`)
const coderAgents = tools?.tools?.filter((t) => t.tool_type === "coder_agent") ?? []
log(` ${COLORS.bold}coder_agent count:${COLORS.reset} ${coderAgents.length}`)
if (coderAgents.length > 0) {
log("")
log(` ${COLORS.bold}Coder Agents:${COLORS.reset}`)
coderAgents.forEach((a) => {
log(` - ${a.name} (id: ${a.id})`)
log(` interpolation_string: ${a.interpolation_string ? "YES" : "NO"}`)
log(` goals: ${a.goals ? "YES" : "NO"}`)
log(` llm_provider: ${a.llm_provider ?? "(null)"}`)
log(` llm_base_model: ${a.llm_base_model ?? "(null)"}`)
})
}
}
log("")
}
function testAgentPrompt(agentId) {
const tools = loadCachedTools()
if (!tools?.success) {
error("No tools. Run: tfcode sync")
process.exit(1)
}
const agent = tools.tools.find((t) => t.id === agentId || t.name === agentId)
if (!agent) {
error(`Agent not found: ${agentId}`)
log("")
log("Available coder agents:")
tools.tools
.filter((t) => t.tool_type === "coder_agent")
.forEach((t) => {
log(` ${COLORS.cyan}${t.id}${COLORS.reset} - ${t.name}`)
})
process.exit(1)
}
log("")
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.magenta} Agent Data from tools.json${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log("")
log(` ${COLORS.bold}id:${COLORS.reset} ${agent.id}`)
log(` ${COLORS.bold}name:${COLORS.reset} ${agent.name}`)
log(` ${COLORS.bold}description:${COLORS.reset} ${agent.description || "(none)"}`)
log(` ${COLORS.bold}tool_type:${COLORS.reset} ${agent.tool_type}`)
log(` ${COLORS.bold}auth_via:${COLORS.reset} ${agent.auth_via}`)
log("")
log(` ${COLORS.bold}interpolation_string:${COLORS.reset}`)
if (agent.interpolation_string) {
log(` ${agent.interpolation_string.substring(0, 200)}${agent.interpolation_string.length > 200 ? "..." : ""}`)
} else {
log(` ${COLORS.dim}(none)${COLORS.reset}`)
}
log("")
log(` ${COLORS.bold}goals:${COLORS.reset}`)
if (agent.goals) {
log(` ${agent.goals.substring(0, 200)}${agent.goals.length > 200 ? "..." : ""}`)
} else {
log(` ${COLORS.dim}(none)${COLORS.reset}`)
}
log("")
log(` ${COLORS.bold}temperature:${COLORS.reset} ${agent.temperature ?? "(none)"}`)
log(` ${COLORS.bold}max_tokens:${COLORS.reset} ${agent.max_tokens ?? "(none)"}`)
log(` ${COLORS.bold}llm_base_model:${COLORS.reset} ${agent.llm_base_model ?? "(none)"}`)
log(` ${COLORS.bold}llm_provider:${COLORS.reset} ${agent.llm_provider ?? "(none)"}`)
log("")
// Build highlighted instructions
const isTFProvider = !agent.llm_provider || agent.llm_provider === "toothfairyai" || agent.llm_provider === "tf"
const hasPrompt = agent.interpolation_string && agent.interpolation_string.trim().length > 0
const hasGoals = agent.goals && agent.goals.trim().length > 0
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.magenta} Model Mapping${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log("")
log(` ${COLORS.bold}isTFProvider:${COLORS.reset} ${isTFProvider}`)
log(
` ${COLORS.bold}mapped model:${COLORS.reset} ${isTFProvider && agent.llm_base_model ? `toothfairyai/${agent.llm_base_model}` : "(no mapping)"}`,
)
log("")
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.magenta} Highlighted Instructions Preview${COLORS.reset}`)
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`)
log("")
if (!hasPrompt && !hasGoals) {
log(
` ${COLORS.dim}(No interpolation_string or goals - no highlighted instructions will be generated)${COLORS.reset}`,
)
} else {
log("")
log("═══════════════════════════════════════════════════════════════════════════════")
log("⚠️ ULTRA IMPORTANT - AGENT CONFIGURATION ⚠️")
log("═══════════════════════════════════════════════════════════════════════════════")
log("")
log(`You are acting as the agent: "${agent.name}"`)
if (agent.description) {
log(`Description: ${agent.description}`)
}
log("")
log("The following instructions and goals are MANDATORY and MUST be followed")
log("with the HIGHEST PRIORITY. These override any conflicting default behaviors.")
log("═══════════════════════════════════════════════════════════════════════════════")
if (hasPrompt) {
log("")
log("┌─────────────────────────────────────────────────────────────────────────────┐")
log(`│ 🎯 AGENT "${agent.name}" INSTRUCTIONS (CRITICAL - MUST FOLLOW) │`)
log("└─────────────────────────────────────────────────────────────────────────────┘")
log("")
log(agent.interpolation_string)
}
if (hasGoals) {
log("")
log("┌─────────────────────────────────────────────────────────────────────────────┐")
log(`│ 🎯 AGENT "${agent.name}" GOALS (CRITICAL - MUST ACHIEVE) │`)
log("└─────────────────────────────────────────────────────────────────────────────┘")
log("")
log(agent.goals)
}
log("")
log("═══════════════════════════════════════════════════════════════════════════════")
log(`⚠️ END OF ULTRA IMPORTANT AGENT "${agent.name}" CONFIGURATION ⚠️`)
log("═══════════════════════════════════════════════════════════════════════════════")
}
log("")
}
const args = process.argv.slice(2)
const command = args[0]
if (args.includes("--help") || args.includes("-h")) {
showHelp()
} else if (args.includes("--version") || args.includes("-v")) {
log("tfcode v1.0.0-beta.9")
} else if (command === "setup") {
interactiveSetup()
} else if (command === "validate") {
;(async () => {
const config = loadConfig()
if (!config) {
error("No credentials. Run: tfcode setup")
process.exit(1)
}
info("Validating...")
try {
const result = await runPythonSync("validate", config)
if (result.success) {
success("Credentials valid")
log(` API URL: ${result.base_url}`)
} else {
error(`Failed: ${result.error}`)
process.exit(1)
}
} catch (e) {
error(`Failed: ${e.message}`)
process.exit(1)
}
})()
} else if (command === "sync") {
;(async () => {
const config = loadConfig()
if (!config) {
error("No credentials. Run: tfcode setup")
process.exit(1)
}
info("Syncing tools...")
try {
const result = await runPythonSync("sync", config)
if (result.success) {
saveToolsCache(result)
success(`Synced ${result.tools.length} tools`)
if (result.by_type) {
log("")
log("By type:")
for (const [t, c] of Object.entries(result.by_type)) log(` ${t}: ${c}`)
}
} else {
error(`Failed: ${result.error}`)
process.exit(1)
}
} catch (e) {
error(`Failed: ${e.message}`)
process.exit(1)
}
})()
} else if (command === "tools" && args[1] === "list") {
const cached = loadCachedTools()
if (!cached?.success) {
error("No tools. Run: tfcode sync")
process.exit(1)
}
let tools = cached.tools
if (args[2] === "--type" && args[3]) tools = tools.filter((t) => t.tool_type === args[3])
log(`\n${tools.length} tool(s):\n`)
for (const t of tools) {
log(` ${COLORS.cyan}${t.name}${COLORS.reset}`)
log(` Type: ${t.tool_type}`)
if (t.description) log(` ${COLORS.dim}${t.description.slice(0, 60)}${COLORS.reset}`)
log(` Auth: ${t.auth_via}\n`)
}
} else if (command === "test-agent") {
const agentId = args[1]
if (!agentId) {
error("Usage: tfcode test-agent <agent-id-or-name>")
process.exit(1)
}
testAgentPrompt(agentId)
} else if (command === "debug") {
showDebugInfo()
} else if (!command) {
// Show help instead of trying TUI (TUI requires full build)
showHelp();
showHelp()
} else {
error(`Unknown command: ${command}`);
showHelp();
process.exit(1);
}
error(`Unknown command: ${command}`)
showHelp()
process.exit(1)
}

View File

@@ -8,6 +8,7 @@ import { Instance } from "../project/instance"
import { Truncate } from "../tool/truncate"
import { Auth } from "../auth"
import { ProviderTransform } from "../provider/transform"
import os from "os"
import PROMPT_GENERATE from "./generate.txt"
import PROMPT_COMPACTION from "./prompt/compaction.txt"
@@ -41,6 +42,7 @@ export namespace Agent {
.optional(),
variant: z.string().optional(),
prompt: z.string().optional(),
goals: z.string().optional(),
options: z.record(z.string(), z.any()),
steps: z.number().int().positive().optional(),
})
@@ -263,7 +265,8 @@ export namespace Agent {
}
async function loadTFCoderAgents(): Promise<Info[]> {
const toolsPath = path.join(Global.Path.data, ".tfcode", "tools.json")
// tools.json is synced to ~/.tfcode/tools.json by the CLI
const toolsPath = path.join(os.homedir(), ".tfcode", "tools.json")
try {
const content = await Bun.file(toolsPath).text()
const data = JSON.parse(content)
@@ -271,19 +274,33 @@ export namespace Agent {
return data.tools
.filter((t: any) => t.tool_type === "coder_agent")
.map(
(t: any): Info => ({
.map((t: any): Info => {
// Map model for ToothFairyAI providers only
// Only map when llm_provider is None, "toothfairyai", or "tf"
const isTFProvider = !t.llm_provider || t.llm_provider === "toothfairyai" || t.llm_provider === "tf"
const model =
isTFProvider && t.llm_base_model
? { modelID: t.llm_base_model as ModelID, providerID: "toothfairyai" as ProviderID }
: undefined
return {
name: t.name,
description: t.description,
mode: "primary" as const,
permission: Permission.fromConfig({ "*": "allow" }),
native: false,
prompt: t.interpolation_string,
goals: t.goals,
temperature: t.temperature,
model,
options: {
tf_agent_id: t.id,
tf_auth_via: t.auth_via,
tf_max_tokens: t.max_tokens,
},
}),
)
}
})
} catch {
return []
}

View File

@@ -68,10 +68,16 @@ export namespace LLM {
const isOpenaiOauth = provider.id === "openai" && auth?.type === "oauth"
const system: string[] = []
// Build highlighted agent instructions for ToothFairyAI agents
const tfHighlightedInstructions = buildTFAgentInstructions(input.agent)
system.push(
[
// use agent prompt otherwise provider prompt
...(input.agent.prompt ? [input.agent.prompt] : SystemPrompt.provider(input.model)),
// highlighted TF agent instructions (if any)
...(tfHighlightedInstructions ? [tfHighlightedInstructions] : []),
// any custom prompt passed into this call
...input.system,
// any custom prompt from last user message
@@ -168,7 +174,7 @@ export namespace LLM {
const maxOutputTokens =
isOpenaiOauth || provider.id.includes("github-copilot")
? undefined
: ProviderTransform.maxOutputTokens(input.model)
: (input.agent.options?.tf_max_tokens ?? ProviderTransform.maxOutputTokens(input.model))
const tools = await resolveTools(input)
@@ -320,4 +326,63 @@ export namespace LLM {
}
return false
}
/**
* Build highlighted instructions for ToothFairyAI agents.
* When a TF agent is selected, its instructions and goals are highlighted
* as ULTRA IMPORTANT to ensure they are followed with highest priority.
*/
function buildTFAgentInstructions(agent: Agent.Info): string | null {
// Only highlight for ToothFairyAI agents (non-native with tf_agent_id)
const isTFAgent = !agent.native && agent.options?.tf_agent_id
if (!isTFAgent) return null
const parts: string[] = []
// Check if agent has custom prompt (interpolation_string) or goals
const hasPrompt = agent.prompt && agent.prompt.trim().length > 0
const hasGoals = agent.goals && agent.goals.trim().length > 0
if (!hasPrompt && !hasGoals) return null
parts.push("")
parts.push("═══════════════════════════════════════════════════════════════════════════════")
parts.push("⚠️ ULTRA IMPORTANT - AGENT CONFIGURATION ⚠️")
parts.push("═══════════════════════════════════════════════════════════════════════════════")
parts.push("")
parts.push(`You are acting as the agent: "${agent.name}"`)
if (agent.description) {
parts.push(`Description: ${agent.description}`)
}
parts.push("")
parts.push("The following instructions and goals are MANDATORY and MUST be followed")
parts.push("with the HIGHEST PRIORITY. These override any conflicting default behaviors.")
parts.push("═══════════════════════════════════════════════════════════════════════════════")
if (hasPrompt) {
parts.push("")
parts.push("┌─────────────────────────────────────────────────────────────────────────────┐")
parts.push(`│ 🎯 AGENT "${agent.name}" INSTRUCTIONS (CRITICAL - MUST FOLLOW) │`)
parts.push("└─────────────────────────────────────────────────────────────────────────────┘")
parts.push("")
parts.push(agent.prompt!)
}
if (hasGoals) {
parts.push("")
parts.push("┌─────────────────────────────────────────────────────────────────────────────┐")
parts.push(`│ 🎯 AGENT "${agent.name}" GOALS (CRITICAL - MUST ACHIEVE) │`)
parts.push("└─────────────────────────────────────────────────────────────────────────────┘")
parts.push("")
parts.push(agent.goals!)
}
parts.push("")
parts.push("═══════════════════════════════════════════════════════════════════════════════")
parts.push(`⚠️ END OF ULTRA IMPORTANT AGENT "${agent.name}" CONFIGURATION ⚠️`)
parts.push("═══════════════════════════════════════════════════════════════════════════════")
parts.push("")
return parts.join("\n")
}
}

View File

@@ -0,0 +1,492 @@
import { afterAll, beforeAll, beforeEach, afterEach, test, expect, describe } from "bun:test"
import path from "path"
import fs from "fs"
import { tmpdir } from "../fixture/fixture"
import { Instance } from "../../src/project/instance"
import { Agent } from "../../src/agent/agent"
import { LLM } from "../../src/session/llm"
import { Provider } from "../../src/provider/provider"
import { Global } from "../../src/global"
import { Filesystem } from "../../src/util/filesystem"
import { ModelsDev } from "../../src/provider/models"
import { ProviderID, ModelID } from "../../src/provider/schema"
import { SessionID, MessageID } from "../../src/session/schema"
import type { MessageV2 } from "../../src/session/message-v2"
// Server for capturing LLM requests
const state = {
server: null as ReturnType<typeof Bun.serve> | null,
queue: [] as Array<{ path: string; response: Response; resolve: (value: any) => void }>,
}
function deferred<T>() {
const result = {} as { promise: Promise<T>; resolve: (value: T) => void }
result.promise = new Promise((resolve) => {
result.resolve = resolve
})
return result
}
function waitRequest(pathname: string, response: Response) {
const pending = deferred<{ url: URL; headers: Headers; body: Record<string, unknown> }>()
state.queue.push({ path: pathname, response, resolve: pending.resolve })
return pending.promise
}
function createChatStream(text: string) {
const payload =
[
`data: ${JSON.stringify({
id: "chatcmpl-1",
object: "chat.completion.chunk",
choices: [{ delta: { role: "assistant" } }],
})}`,
`data: ${JSON.stringify({
id: "chatcmpl-1",
object: "chat.completion.chunk",
choices: [{ delta: { content: text } }],
})}`,
`data: ${JSON.stringify({
id: "chatcmpl-1",
object: "chat.completion.chunk",
choices: [{ delta: {}, finish_reason: "stop" }],
})}`,
"data: [DONE]",
].join("\n\n") + "\n\n"
return new ReadableStream({
start(controller) {
controller.enqueue(new TextEncoder().encode(payload))
controller.close()
},
})
}
async function loadFixture(providerID: string, modelID: string) {
const fixturePath = path.join(import.meta.dir, "../tool/fixtures/models-api.json")
const data = await Filesystem.readJson<Record<string, ModelsDev.Provider>>(fixturePath)
const provider = data[providerID]
if (!provider) throw new Error(`Missing provider in fixture: ${providerID}`)
const model = provider.models[modelID]
if (!model) throw new Error(`Missing model in fixture: ${modelID}`)
return { provider, model }
}
// Test the flow from tools.json -> Agent.Info -> highlighted instructions
describe("ToothFairyAI Agent Loading", () => {
let originalDataPath: string
beforeEach(async () => {
originalDataPath = Global.Path.data
const testDataDir = path.join(path.dirname(originalDataPath), "tf-agent-test-data")
;(Global.Path as { data: string }).data = testDataDir
await fs.promises.mkdir(path.join(testDataDir, ".tfcode"), { recursive: true })
})
afterEach(async () => {
await Instance.disposeAll()
;(Global.Path as { data: string }).data = originalDataPath
})
describe("loadTFCoderAgents", () => {
test("parses tools.json with full agent data", async () => {
const toolsData = {
success: true,
tools: [
{
id: "coder-agent-1",
name: "Code Reviewer",
description: "Reviews code for quality",
tool_type: "coder_agent",
request_type: null,
url: null,
auth_via: "tf_agent",
interpolation_string: "You are a code reviewer. Review code thoroughly.",
goals: "Identify bugs. Suggest improvements.",
temperature: 0.3,
max_tokens: 4096,
llm_base_model: "claude-3-5-sonnet",
llm_provider: "toothfairyai",
},
{
id: "coder-agent-2",
name: "Test Writer",
description: "Writes tests",
tool_type: "coder_agent",
request_type: null,
url: null,
auth_via: "tf_agent",
interpolation_string: "You are a test writer.",
goals: "Write comprehensive tests.",
temperature: 0.5,
max_tokens: null,
llm_base_model: "gpt-4",
llm_provider: null,
},
],
by_type: { coder_agent: 2 },
}
const toolsPath = path.join(Global.Path.data, ".tfcode", "tools.json")
await fs.promises.writeFile(toolsPath, JSON.stringify(toolsData, null, 2))
await using tmp = await tmpdir()
await Instance.provide({
directory: tmp.path,
fn: async () => {
const agents = await Agent.list()
const codeReviewer = agents.find((a) => a.name === "Code Reviewer")
const testWriter = agents.find((a) => a.name === "Test Writer")
expect(codeReviewer).toBeDefined()
expect(codeReviewer?.description).toBe("Reviews code for quality")
expect(codeReviewer?.prompt).toBe("You are a code reviewer. Review code thoroughly.")
expect(codeReviewer?.goals).toBe("Identify bugs. Suggest improvements.")
expect(codeReviewer?.temperature).toBe(0.3)
expect(codeReviewer?.native).toBe(false)
expect(codeReviewer?.options?.tf_agent_id).toBe("coder-agent-1")
expect(codeReviewer?.options?.tf_auth_via).toBe("tf_agent")
expect(codeReviewer?.options?.tf_max_tokens).toBe(4096)
expect(String(codeReviewer?.model?.providerID)).toBe("toothfairyai")
expect(String(codeReviewer?.model?.modelID)).toBe("claude-3-5-sonnet")
expect(testWriter).toBeDefined()
expect(String(testWriter?.model?.providerID)).toBe("toothfairyai")
expect(String(testWriter?.model?.modelID)).toBe("gpt-4")
},
})
})
test("maps tf provider to toothfairyai", async () => {
const toolsData = {
success: true,
tools: [
{
id: "tf-provider-agent",
name: "TF Provider Agent",
description: "Test",
tool_type: "coder_agent",
interpolation_string: "Test",
goals: "Test",
temperature: 0.7,
max_tokens: 2048,
llm_base_model: "test-model",
llm_provider: "tf",
},
],
}
const toolsPath = path.join(Global.Path.data, ".tfcode", "tools.json")
await fs.promises.writeFile(toolsPath, JSON.stringify(toolsData))
await using tmp = await tmpdir()
await Instance.provide({
directory: tmp.path,
fn: async () => {
const agent = await Agent.get("TF Provider Agent")
expect(String(agent?.model?.providerID)).toBe("toothfairyai")
expect(String(agent?.model?.modelID)).toBe("test-model")
},
})
})
test("does not map external providers", async () => {
const toolsData = {
success: true,
tools: [
{
id: "external-agent",
name: "External Agent",
description: "Test",
tool_type: "coder_agent",
interpolation_string: "Test",
goals: "Test",
temperature: 0.7,
max_tokens: 2048,
llm_base_model: "claude-3-5-sonnet",
llm_provider: "anthropic",
},
],
}
const toolsPath = path.join(Global.Path.data, ".tfcode", "tools.json")
await fs.promises.writeFile(toolsPath, JSON.stringify(toolsData))
await using tmp = await tmpdir()
await Instance.provide({
directory: tmp.path,
fn: async () => {
const agent = await Agent.get("External Agent")
expect(agent?.model).toBeUndefined()
},
})
})
test("handles agent without interpolation_string or goals", async () => {
const toolsData = {
success: true,
tools: [
{
id: "minimal-agent",
name: "Minimal Agent",
description: "Test",
tool_type: "coder_agent",
interpolation_string: null,
goals: null,
temperature: null,
max_tokens: null,
llm_base_model: null,
llm_provider: null,
},
],
}
const toolsPath = path.join(Global.Path.data, ".tfcode", "tools.json")
await fs.promises.writeFile(toolsPath, JSON.stringify(toolsData))
await using tmp = await tmpdir()
await Instance.provide({
directory: tmp.path,
fn: async () => {
const agent = await Agent.get("Minimal Agent")
expect(agent).toBeDefined()
expect(agent?.prompt).toBeNull()
expect(agent?.goals).toBeNull()
expect(agent?.model).toBeUndefined()
},
})
})
})
})
// Separate describe block for LLM stream tests
describe("ToothFairyAI Agent Instructions in LLM Stream", () => {
let originalDataPath: string
beforeAll(() => {
state.server = Bun.serve({
port: 0,
async fetch(req) {
const next = state.queue.shift()
if (!next) return new Response("unexpected request", { status: 500 })
const url = new URL(req.url)
const body = (await req.json()) as Record<string, unknown>
next.resolve({ url, headers: req.headers, body })
if (!url.pathname.endsWith(next.path)) return new Response("not found", { status: 404 })
return next.response
},
})
})
afterAll(() => {
state.server?.stop()
})
beforeEach(async () => {
state.queue.length = 0
originalDataPath = Global.Path.data
const testDataDir = path.join(path.dirname(originalDataPath), "tf-agent-test-data")
;(Global.Path as { data: string }).data = testDataDir
await fs.promises.mkdir(path.join(testDataDir, ".tfcode"), { recursive: true })
})
afterEach(async () => {
await Instance.disposeAll()
;(Global.Path as { data: string }).data = originalDataPath
})
test("includes highlighted TF agent instructions in system prompt", async () => {
const server = state.server
if (!server) throw new Error("Server not initialized")
const providerID = "alibaba"
const modelID = "qwen-plus"
const fixture = await loadFixture(providerID, modelID)
// Setup TF agent with this model
const toolsData = {
success: true,
tools: [
{
id: "code-reviewer-123",
name: "Code Reviewer",
description: "Reviews code for quality and best practices",
tool_type: "coder_agent",
auth_via: "tf_agent",
interpolation_string: "You are a code reviewer. Always check for bugs, security issues, and suggest improvements.",
goals: "Review all code thoroughly. Provide actionable feedback. Ensure code quality standards.",
temperature: 0.3,
max_tokens: 4096,
llm_base_model: modelID,
llm_provider: providerID,
},
],
}
const toolsPath = path.join(Global.Path.data, ".tfcode", "tools.json")
await fs.promises.writeFile(toolsPath, JSON.stringify(toolsData, null, 2))
const request = waitRequest(
"/chat/completions",
new Response(createChatStream("I'll review your code."), {
status: 200,
headers: { "Content-Type": "text/event-stream" },
}),
)
await using tmp = await tmpdir({
init: async (dir) => {
await Bun.write(
path.join(dir, "opencode.json"),
JSON.stringify({
$schema: "https://opencode.ai/config.json",
enabled_providers: [providerID],
provider: {
[providerID]: {
options: {
apiKey: "test-key",
baseURL: `${server.url.origin}/v1`,
},
},
},
}),
)
},
})
await Instance.provide({
directory: tmp.path,
fn: async () => {
const agent = await Agent.get("Code Reviewer")
expect(agent).toBeDefined()
const resolved = await Provider.getModel(ProviderID.make(providerID), ModelID.make(modelID))
const sessionID = SessionID.make("test-session")
const user: MessageV2.User = {
id: MessageID.make("user-1"),
sessionID,
role: "user",
time: { created: Date.now() },
agent: "Code Reviewer",
model: { providerID: ProviderID.make(providerID), modelID: ModelID.make(modelID) },
}
const stream = await LLM.stream({
user,
sessionID,
model: resolved,
agent: agent!,
system: [],
abort: new AbortController().signal,
messages: [{ role: "user", content: "Hello" }],
tools: {},
})
for await (const _ of stream.fullStream) {}
const capture = await request
const body = capture.body
const messages = body.messages as Array<{ role: string; content: string }>
const systemMessage = messages.find((m) => m.role === "system")
expect(systemMessage).toBeDefined()
const systemContent = systemMessage!.content
expect(systemContent).toContain("ULTRA IMPORTANT - AGENT CONFIGURATION")
expect(systemContent).toContain('You are acting as the agent: "Code Reviewer"')
expect(systemContent).toContain("Reviews code for quality and best practices")
expect(systemContent).toContain("AGENT \"Code Reviewer\" INSTRUCTIONS")
expect(systemContent).toContain("You are a code reviewer. Always check for bugs, security issues, and suggest improvements.")
expect(systemContent).toContain("AGENT \"Code Reviewer\" GOALS")
expect(systemContent).toContain("Review all code thoroughly. Provide actionable feedback. Ensure code quality standards.")
},
})
})
test("does NOT include highlighted instructions for native agents", async () => {
const server = state.server
if (!server) throw new Error("Server not initialized")
const providerID = "alibaba"
const modelID = "qwen-plus"
const fixture = await loadFixture(providerID, modelID)
const request = waitRequest(
"/chat/completions",
new Response(createChatStream("Hello"), {
status: 200,
headers: { "Content-Type": "text/event-stream" },
}),
)
await using tmp = await tmpdir({
init: async (dir) => {
await Bun.write(
path.join(dir, "opencode.json"),
JSON.stringify({
$schema: "https://opencode.ai/config.json",
enabled_providers: [providerID],
provider: {
[providerID]: {
options: {
apiKey: "test-key",
baseURL: `${server.url.origin}/v1`,
},
},
},
}),
)
},
})
await Instance.provide({
directory: tmp.path,
fn: async () => {
const agent = await Agent.get("build")
expect(agent).toBeDefined()
expect(agent?.native).toBe(true)
const resolved = await Provider.getModel(ProviderID.make(providerID), ModelID.make(modelID))
const sessionID = SessionID.make("test-session")
const user: MessageV2.User = {
id: MessageID.make("user-1"),
sessionID,
role: "user",
time: { created: Date.now() },
agent: "build",
model: { providerID: ProviderID.make(providerID), modelID: ModelID.make(modelID) },
}
const stream = await LLM.stream({
user,
sessionID,
model: resolved,
agent: agent!,
system: [],
abort: new AbortController().signal,
messages: [{ role: "user", content: "Hello" }],
tools: {},
})
for await (const _ of stream.fullStream) {}
const capture = await request
const body = capture.body
const messages = body.messages as Array<{ role: string; content: string }>
const systemMessage = messages.find((m) => m.role === "system")
expect(systemMessage).toBeDefined()
const systemContent = systemMessage!.content
expect(systemContent).not.toContain("ULTRA IMPORTANT - AGENT CONFIGURATION")
expect(systemContent).not.toContain("You are acting as the agent:")
},
})
})
})