react-native-executorch/src/Error.ts

An easy way to run AI models in react native with ExecuTorch

export enum ETError {
  // React-native-ExecuTorch errors
  UndefinedError = 0x65,
  ModuleNotLoaded = 0x66,
  FileWriteFailed = 0x67,
  ModelGenerating = 0x68,
  LanguageNotSupported = 0x69,
  InvalidModelSource = 0xff,

  // SpeechToText errors
  MultilingualConfiguration = 0xa0,
  MissingDataChunk = 0xa1,
  StreamingNotStarted = 0xa2,

  // ExecuTorch mapped errors
  // Based on: https://github.com/pytorch/executorch/blob/main/runtime/core/error.h
  // System errors
  Ok = 0x00,
  Internal = 0x01,
  InvalidState = 0x02,
  EndOfMethod = 0x03,

  // Logical errors
  NotSupported = 0x10,
  NotImplemented = 0x11,
  InvalidArgument = 0x12,
  InvalidType = 0x13,
  OperatorMissing = 0x14,

  // Resource errors
  NotFound = 0x20,
  MemoryAllocationFailed = 0x21,
  AccessFailed = 0x22,
  InvalidProgram = 0x23,

  // Delegate errors
  DelegateInvalidCompatibility = 0x30,
  DelegateMemoryAllocationFailed = 0x31,
  DelegateInvalidHandle = 0x32,
}

export const getError = (e: unknown | ETError | Error): string => {
  if (typeof e === 'number') {
    return ETError[e] ?? ETError[ETError.UndefinedError];
  }

  // try to extract number from message (can contain false positives)
  const error = e as Error;
  const errorCode = parseInt(error.message, 10);

  if (Number.isNaN(errorCode)) {
    return error.message;
  }

  return ETError[errorCode] ?? ETError[ETError.UndefinedError];
};