import { TaskCompletionEvent } from "./events.js";
import { PredictionQuery, PredictionResponse } from "./predictions.js";

export interface BatchEventRequest {
  events: TaskCompletionEvent[];
}

export interface BatchEventResult {
  event_id: string;
  status: "accepted" | "rejected";
  error?: string | null;
}

export interface BatchEventResponse {
  accepted: number;
  rejected: number;
  results: BatchEventResult[];
}

export interface BatchPredictionRequest {
  queries: PredictionQuery[];
}

export interface BatchPredictionResponse {
  results: PredictionResponse[];
}