Spaces:
Sleeping
Sleeping
File size: 4,135 Bytes
03a7eb9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 | /**
* CodeArena API Service
* Connects to the FastAPI backend at localhost:7860 (proxied via Vite).
*
* Real endpoint contracts (from server/app.py):
* POST /reset → { task_id } → { status, observation, info }
* POST /step → { proposed_fix } → { observation, reward, done, info }
* GET /state → → { observation }
*/
const BASE = ''; // proxied through Vite — no prefix needed
// ─── Helpers ────────────────────────────────────────────────────
async function request(url, options = {}) {
const controller = new AbortController();
const timeout = setTimeout(() => controller.abort(), 30_000);
try {
const res = await fetch(`${BASE}${url}`, {
signal: controller.signal,
...options,
});
if (!res.ok) {
const text = await res.text().catch(() => '');
throw new Error(`HTTP ${res.status}: ${text || res.statusText}`);
}
return await res.json();
} catch (err) {
if (err.name === 'AbortError') {
throw new Error(`Request to ${url} timed out after 30s`);
}
throw err;
} finally {
clearTimeout(timeout);
}
}
function post(url, body) {
return request(url, {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(body),
});
}
// ─── Public API ─────────────────────────────────────────────────
/**
* POST /reset
* @param {string} taskId - "easy", "medium", "hard", "auto", or exact ID like "easy-1"
* @returns {{ status, observation: { buggy_code, error_log, test_results, previous_attempts }, info: { task_id, difficulty } }}
*/
export async function resetTask(taskId = 'easy') {
return post('/reset', { task_id: taskId });
}
/**
* POST /step
* @param {string} proposedFix - The code fix to submit
* @returns {{ observation, reward: number, done: boolean, info: { execution_metadata, task_id, reward_components } }}
*/
export async function sendStep(proposedFix) {
return post('/step', { proposed_fix: proposedFix });
}
/**
* GET /state
* @returns {{ observation: { buggy_code, error_log, test_results, previous_attempts } }}
*/
export async function getState() {
return request('/state');
}
/**
* GET / (health check)
* @returns {{ status: "ok", environment: "CodeArena" }}
*/
export async function healthCheck() {
return request('/health');
}
/**
* POST /fix
* Uses built-in pattern fixer + optional Ollama.
* Passes reward + task_id for memory storage and adaptive prompting.
* @param {string} code - Buggy code
* @param {string} errorLog - Error output
* @param {string} ollamaUrl - Ollama server URL
* @param {string} model - Model name
* @param {number} reward - Current reward (for adaptive prompting)
* @param {string} taskId - Task ID (for memory retrieval)
* @returns {{ fixed_code, method, success, explanation, complexity, algo_hint, note? }}
*/
export async function generateFix(code, errorLog = '', ollamaUrl = 'http://localhost:11434', model = 'llama3.2:latest', reward = 0.0, taskId = '') {
return post('/fix', {
code,
error_log: errorLog,
ollama_url: ollamaUrl,
model,
use_ollama: true,
reward,
task_id: taskId,
});
}
/**
* GET /stats
* Returns complexity vs reward stats + episode history.
*/
export async function getStats() {
return request('/stats');
}
/**
* GET /memory
* Returns all stored best solutions from agent memory.
*/
export async function getMemory() {
return request('/memory');
}
/**
* POST /run_raw
* Sandbox mode: executes arbitrary code and returns stdout, stderr, and execution time complexity.
* @param {string} code - The code to execute
* @returns {{ status: "success"|"error", stdout: string, stderr: string, execution_time: number, time_complexity_hint: string, reward: number, reward_components: object, done: boolean }}
*/
export async function runRaw(code) {
return post('/run_raw', { code });
}
|