Spaces:
Running
Running
Aakarsh2007
feat(ui): TypeScript types, API client with SSE helpers, createAgentStream, createCompareStream
77ee838 | /// <reference types="vite/client" /> | |
| import type { Observation, StepResult, Task } from './types' | |
| const BASE = (import.meta.env.VITE_API_URL as string | undefined) || '' | |
| async function post<T>(path: string, body: unknown): Promise<T> { | |
| const r = await fetch(`${BASE}${path}`, { | |
| method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify(body), | |
| }) | |
| if (!r.ok) throw new Error(`${r.status} ${await r.text()}`) | |
| return r.json() as Promise<T> | |
| } | |
| async function get<T>(path: string): Promise<T> { | |
| const r = await fetch(`${BASE}${path}`) | |
| if (!r.ok) throw new Error(`${r.status} ${await r.text()}`) | |
| return r.json() as Promise<T> | |
| } | |
| export const api = { | |
| reset: (task_id: string, model = 'manual') => post<Observation>('/reset', { task_id, model }), | |
| step: (action: Record<string, unknown>) => post<StepResult>('/step', action), | |
| state: () => get<Record<string, unknown>>('/state'), | |
| tasks: () => get<{ tasks: Task[] }>('/tasks'), | |
| leaderboard: () => get<{ leaderboard: unknown[] }>('/leaderboard'), | |
| stats: () => get<Record<string, unknown>>('/stats'), | |
| history: () => get<{ history: unknown[] }>('/history'), | |
| replay: (runId: string) => get<Record<string, unknown>>(`/replay/${runId}`), | |
| customScenarios: () => get<{ custom_scenarios: string[] }>('/scenarios/custom'), | |
| createCustomScenario: (body: Record<string, unknown>) => post<Record<string, unknown>>('/scenarios/custom', body), | |
| memory: () => get<{ memory: unknown[] }>('/memory'), | |
| submitFeedback: (body: Record<string, unknown>) => post<Record<string, unknown>>('/feedback', body), | |
| } | |
| // SSE helper for live agent streaming | |
| export function createAgentStream(body: Record<string, unknown>, onEvent: (e: Record<string, unknown>) => void, onDone: () => void) { | |
| const ctrl = new AbortController() | |
| fetch(`${BASE}/agent/run`, { | |
| method: 'POST', headers: { 'Content-Type': 'application/json' }, | |
| body: JSON.stringify(body), signal: ctrl.signal, | |
| }).then(async res => { | |
| const reader = res.body?.getReader() | |
| const decoder = new TextDecoder() | |
| if (!reader) { onDone(); return } | |
| let buf = '' | |
| while (true) { | |
| const { done, value } = await reader.read() | |
| if (done) break | |
| buf += decoder.decode(value, { stream: true }) | |
| const lines = buf.split('\n') | |
| buf = lines.pop() || '' | |
| for (const line of lines) { | |
| if (line.startsWith('data: ')) { | |
| try { onEvent(JSON.parse(line.slice(6))) } catch {} | |
| } | |
| } | |
| } | |
| onDone() | |
| }).catch(() => onDone()) | |
| return () => ctrl.abort() | |
| } | |
| export function createCompareStream(body: Record<string, unknown>, onEvent: (e: Record<string, unknown>) => void, onDone: () => void) { | |
| const ctrl = new AbortController() | |
| fetch(`${BASE}/agent/compare`, { | |
| method: 'POST', headers: { 'Content-Type': 'application/json' }, | |
| body: JSON.stringify(body), signal: ctrl.signal, | |
| }).then(async res => { | |
| const reader = res.body?.getReader() | |
| const decoder = new TextDecoder() | |
| if (!reader) { onDone(); return } | |
| let buf = '' | |
| while (true) { | |
| const { done, value } = await reader.read() | |
| if (done) break | |
| buf += decoder.decode(value, { stream: true }) | |
| const lines = buf.split('\n') | |
| buf = lines.pop() || '' | |
| for (const line of lines) { | |
| if (line.startsWith('data: ')) { | |
| try { onEvent(JSON.parse(line.slice(6))) } catch {} | |
| } | |
| } | |
| } | |
| onDone() | |
| }).catch(() => onDone()) | |
| return () => ctrl.abort() | |
| } | |