| |
| import type { JobConfig } from '../types'; |
|
|
| export type PipelinePlots = { |
| original_sentiment: string; |
| counterfactual_sentiment: string; |
| }; |
|
|
| export type PipelineResultsDTO = { |
| data_loaded: number; |
| model_loaded: string; |
|
|
| generation_file: string; |
| generation_samples: number; |
|
|
| counterfactual_file: string; |
| counterfactual_added: number; |
| counterfactual_total: number; |
|
|
| sampling_method: string; |
| sentiment_subset_file: string; |
| sentiment_subset_size: number; |
|
|
| cf_sentiment_subset_file: string; |
| cf_sentiment_subset_size: number; |
|
|
| |
| config_used: import('../types').JobConfig; |
| metrics: import('../types').JobMetrics & { |
| finalMeanDiff: number; |
| cfFinalMeanDiff: number; |
| }; |
| plots: PipelinePlots; |
| }; |
|
|
| export type PipelineResponseDTO = { |
| status: 'success' | 'error'; |
| message: string; |
| timestamp: string; |
| results: PipelineResultsDTO; |
| }; |
|
|
| const BASE = import.meta.env.VITE_API_BASE ?? '/api'; |
|
|
| async function runPipeline(config: any) { |
| const r = await fetch(`${BASE}/pipeline`, { |
| method: 'POST', |
| headers: { 'Content-Type': 'application/json' }, |
| body: JSON.stringify({ config }), |
| }); |
| if (!r.ok) { |
| const text = await r.text(); |
| throw new Error(`Pipeline failed (${r.status}): ${text}`); |
| } |
| return r.json(); |
| } |
|
|
| async function checkHealth() { |
| const r = await fetch(`${BASE}/health`); |
| return r.json(); |
| } |
|
|
| function resolvePath(p?: string) { |
| if (!p) return ''; |
| if (p.startsWith('http')) return p; |
| const path = p.startsWith('/') ? p : `/${p}`; |
| return `${BASE}${path}`; |
| } |
|
|
| export const MLBiasAPI = { runPipeline, checkHealth, resolvePath }; |
|
|
