/**
 * Streaming handler dispatcher
 */

import type { AssistantMessageEventStream } from "@mariozechner/pi-ai";
import type { Context, StreamOptions, VertexModelConfig } from "../types.js";
import { streamGemini } from "./gemini.js";
import { streamMaaS } from "./maas.js";

export function streamVertex(
  model: VertexModelConfig,
  context: Context,
  options?: StreamOptions,
): AssistantMessageEventStream {
  switch (model.endpointType) {
    case "gemini":
      return streamGemini(model, context, options);
    case "maas":
      return streamMaaS(model, context, options);
    default: {
      const exhaustive: never = model.endpointType;
      throw new Error(`Unknown endpoint type: ${String(exhaustive)}`);
    }
  }
}

export { streamGemini, streamMaaS };