CreateVectorBackend.ts

import { detectBackend } from "./BackendKind";
import type { VectorBackend } from "./VectorBackend";
import { WasmVectorBackend } from "./WasmVectorBackend";
import { WebGlVectorBackend } from "./WebGLVectorBackend";
import { WebGpuVectorBackend } from "./WebGPUVectorBackend";
import { WebNnVectorBackend } from "./WebNNVectorBackend";

export async function createVectorBackend(
  wasmBytes: ArrayBuffer
): Promise<VectorBackend> {
  const kind = detectBackend();
  if (kind === "webgpu") {
    return WebGpuVectorBackend.create().catch(() =>
      WasmVectorBackend.create(wasmBytes)
    );
  }
  if (kind === "webgl") {
    return Promise.resolve()
      .then(() => WebGlVectorBackend.create())
      .catch(() => WasmVectorBackend.create(wasmBytes));
  }
  if (kind === "webnn") {
    return WebNnVectorBackend.create(wasmBytes).catch(() =>
      WasmVectorBackend.create(wasmBytes)
    );
  }
  return WasmVectorBackend.create(wasmBytes);
}