defineBidiModel method

Implementation

BidiModel defineBidiModel({
  required String name,
  required BidiActionFn<
    ModelRequest,
    ModelResponse,
    ModelResponseChunk,
    ModelRequest
  >
  fn,
}) {
  final model = BidiModel(
    name: name,
    fn: (input, context) {
      if (context.inputStream == null) {
        throw GenkitException(
          'Bidi model $name called without an input stream',
          status: StatusCodes.INVALID_ARGUMENT,
        );
      }
      return fn(context.inputStream!, context);
    },
  );
  registry.register(model);
  return model;
}