react-native-nitro-mlx 0.2.0 → 0.2.1

package/lib/module/index.js

@@ -2,5 +2,5 @@
 
 export { LLM } from "./llm.js";
 export { ModelManager } from "./modelManager.js";
-export { MLXModel } from "./models.js";
+export { MLXModel, MLXModels, ModelFamily, ModelProvider } from "./models.js";
 //# sourceMappingURL=index.js.map

package/lib/module/index.js.map

@@ -1 +1 @@
-{"version":3,"names":["LLM","ModelManager","MLXModel"],"sourceRoot":"../../src","sources":["index.ts"],"mappings":";;AAAA,SAASA,GAAG,QAAsB,UAAO;AACzC,SAASC,YAAY,QAAQ,mBAAgB;AAC7C,SAASC,QAAQ,QAAQ,aAAU","ignoreList":[]}
+{"version":3,"names":["LLM","ModelManager","MLXModel","MLXModels","ModelFamily","ModelProvider"],"sourceRoot":"../../src","sources":["index.ts"],"mappings":";;AAAA,SAASA,GAAG,QAAsB,UAAO;AACzC,SAASC,YAAY,QAAQ,mBAAgB;AAC7C,SACEC,QAAQ,EACRC,SAAS,EACTC,WAAW,EAEXC,aAAa,QAER,aAAU","ignoreList":[]}

package/lib/module/models.js

@@ -1,5 +1,23 @@
 "use strict";
 
+export let ModelFamily = /*#__PURE__*/function (ModelFamily) {
+  ModelFamily["Llama"] = "Llama";
+  ModelFamily["Qwen"] = "Qwen";
+  ModelFamily["Gemma"] = "Gemma";
+  ModelFamily["Phi"] = "Phi";
+  ModelFamily["SmolLM"] = "SmolLM";
+  ModelFamily["OpenELM"] = "OpenELM";
+  return ModelFamily;
+}({});
+export let ModelProvider = /*#__PURE__*/function (ModelProvider) {
+  ModelProvider["Meta"] = "Meta";
+  ModelProvider["Alibaba"] = "Alibaba";
+  ModelProvider["Google"] = "Google";
+  ModelProvider["Microsoft"] = "Microsoft";
+  ModelProvider["HuggingFace"] = "HuggingFace";
+  ModelProvider["Apple"] = "Apple";
+  return ModelProvider;
+}({});
 export let MLXModel = /*#__PURE__*/function (MLXModel) {
   // Llama 3.2 (Meta) - 1B and 3B variants
   MLXModel["Llama_3_2_1B_Instruct_4bit"] = "mlx-community/Llama-3.2-1B-Instruct-4bit";
@@ -38,4 +56,213 @@ export let MLXModel = /*#__PURE__*/function (MLXModel) {
   MLXModel["OpenELM_3B_8bit"] = "mlx-community/OpenELM-3B-8bit";
   return MLXModel;
 }({});
+export const MLXModels = [{
+  id: MLXModel.Llama_3_2_1B_Instruct_4bit,
+  family: ModelFamily.Llama,
+  provider: ModelProvider.Meta,
+  parameters: '1B',
+  quantization: '4bit',
+  displayName: 'Llama 3.2 1B Instruct (4-bit)',
+  downloadSize: 1407777762
+}, {
+  id: MLXModel.Llama_3_2_1B_Instruct_8bit,
+  family: ModelFamily.Llama,
+  provider: ModelProvider.Meta,
+  parameters: '1B',
+  quantization: '8bit',
+  displayName: 'Llama 3.2 1B Instruct (8-bit)',
+  downloadSize: 1313157436
+}, {
+  id: MLXModel.Llama_3_2_3B_Instruct_4bit,
+  family: ModelFamily.Llama,
+  provider: ModelProvider.Meta,
+  parameters: '3B',
+  quantization: '4bit',
+  displayName: 'Llama 3.2 3B Instruct (4-bit)',
+  downloadSize: 2019397474
+}, {
+  id: MLXModel.Llama_3_2_3B_Instruct_8bit,
+  family: ModelFamily.Llama,
+  provider: ModelProvider.Meta,
+  parameters: '3B',
+  quantization: '8bit',
+  displayName: 'Llama 3.2 3B Instruct (8-bit)',
+  downloadSize: 3413784042
+}, {
+  id: MLXModel.Qwen2_5_0_5B_Instruct_4bit,
+  family: ModelFamily.Qwen,
+  provider: ModelProvider.Alibaba,
+  parameters: '0.5B',
+  quantization: '4bit',
+  displayName: 'Qwen 2.5 0.5B Instruct (4-bit)',
+  downloadSize: 278064920
+}, {
+  id: MLXModel.Qwen2_5_0_5B_Instruct_8bit,
+  family: ModelFamily.Qwen,
+  provider: ModelProvider.Alibaba,
+  parameters: '0.5B',
+  quantization: '8bit',
+  displayName: 'Qwen 2.5 0.5B Instruct (8-bit)',
+  downloadSize: 525045902
+}, {
+  id: MLXModel.Qwen2_5_1_5B_Instruct_4bit,
+  family: ModelFamily.Qwen,
+  provider: ModelProvider.Alibaba,
+  parameters: '1.5B',
+  quantization: '4bit',
+  displayName: 'Qwen 2.5 1.5B Instruct (4-bit)',
+  downloadSize: 868628559
+}, {
+  id: MLXModel.Qwen2_5_1_5B_Instruct_8bit,
+  family: ModelFamily.Qwen,
+  provider: ModelProvider.Alibaba,
+  parameters: '1.5B',
+  quantization: '8bit',
+  displayName: 'Qwen 2.5 1.5B Instruct (8-bit)',
+  downloadSize: 1640414038
+}, {
+  id: MLXModel.Qwen2_5_3B_Instruct_4bit,
+  family: ModelFamily.Qwen,
+  provider: ModelProvider.Alibaba,
+  parameters: '3B',
+  quantization: '4bit',
+  displayName: 'Qwen 2.5 3B Instruct (4-bit)',
+  downloadSize: 1736293090
+}, {
+  id: MLXModel.Qwen2_5_3B_Instruct_8bit,
+  family: ModelFamily.Qwen,
+  provider: ModelProvider.Alibaba,
+  parameters: '3B',
+  quantization: '8bit',
+  displayName: 'Qwen 2.5 3B Instruct (8-bit)',
+  downloadSize: 3279142142
+}, {
+  id: MLXModel.Qwen3_1_7B_4bit,
+  family: ModelFamily.Qwen,
+  provider: ModelProvider.Alibaba,
+  parameters: '1.7B',
+  quantization: '4bit',
+  displayName: 'Qwen 3 1.7B (4-bit)',
+  downloadSize: 979502864
+}, {
+  id: MLXModel.Qwen3_1_7B_8bit,
+  family: ModelFamily.Qwen,
+  provider: ModelProvider.Alibaba,
+  parameters: '1.7B',
+  quantization: '8bit',
+  displayName: 'Qwen 3 1.7B (8-bit)',
+  downloadSize: 1839729195
+}, {
+  id: MLXModel.Gemma_3_1B_IT_4bit,
+  family: ModelFamily.Gemma,
+  provider: ModelProvider.Google,
+  parameters: '1B',
+  quantization: '4bit',
+  displayName: 'Gemma 3 1B IT (4-bit)',
+  downloadSize: 770650946
+}, {
+  id: MLXModel.Gemma_3_1B_IT_8bit,
+  family: ModelFamily.Gemma,
+  provider: ModelProvider.Google,
+  parameters: '1B',
+  quantization: '8bit',
+  displayName: 'Gemma 3 1B IT (8-bit)',
+  downloadSize: 1421522471
+}, {
+  id: MLXModel.Phi_3_5_Mini_Instruct_4bit,
+  family: ModelFamily.Phi,
+  provider: ModelProvider.Microsoft,
+  parameters: '3.8B',
+  quantization: '4bit',
+  displayName: 'Phi 3.5 Mini Instruct (4-bit)',
+  downloadSize: 2150195856
+}, {
+  id: MLXModel.Phi_3_5_Mini_Instruct_8bit,
+  family: ModelFamily.Phi,
+  provider: ModelProvider.Microsoft,
+  parameters: '3.8B',
+  quantization: '8bit',
+  displayName: 'Phi 3.5 Mini Instruct (8-bit)',
+  downloadSize: 4060636056
+}, {
+  id: MLXModel.Phi_4_Mini_Instruct_4bit,
+  family: ModelFamily.Phi,
+  provider: ModelProvider.Microsoft,
+  parameters: '3.8B',
+  quantization: '4bit',
+  displayName: 'Phi 4 Mini Instruct (4-bit)',
+  downloadSize: 2173624891
+}, {
+  id: MLXModel.Phi_4_Mini_Instruct_8bit,
+  family: ModelFamily.Phi,
+  provider: ModelProvider.Microsoft,
+  parameters: '3.8B',
+  quantization: '8bit',
+  displayName: 'Phi 4 Mini Instruct (8-bit)',
+  downloadSize: 4091536167
+}, {
+  id: MLXModel.SmolLM_1_7B_Instruct_4bit,
+  family: ModelFamily.SmolLM,
+  provider: ModelProvider.HuggingFace,
+  parameters: '1.7B',
+  quantization: '4bit',
+  displayName: 'SmolLM 1.7B Instruct (4-bit)',
+  downloadSize: 962855374
+}, {
+  id: MLXModel.SmolLM_1_7B_Instruct_8bit,
+  family: ModelFamily.SmolLM,
+  provider: ModelProvider.HuggingFace,
+  parameters: '1.7B',
+  quantization: '8bit',
+  displayName: 'SmolLM 1.7B Instruct (8-bit)',
+  downloadSize: 1818493993
+}, {
+  id: MLXModel.SmolLM2_1_7B_Instruct_4bit,
+  family: ModelFamily.SmolLM,
+  provider: ModelProvider.HuggingFace,
+  parameters: '1.7B',
+  quantization: '4bit',
+  displayName: 'SmolLM2 1.7B Instruct (4-bit)',
+  downloadSize: 980000000
+}, {
+  id: MLXModel.SmolLM2_1_7B_Instruct_8bit,
+  family: ModelFamily.SmolLM,
+  provider: ModelProvider.HuggingFace,
+  parameters: '1.7B',
+  quantization: '8bit',
+  displayName: 'SmolLM2 1.7B Instruct (8-bit)',
+  downloadSize: 1850000000
+}, {
+  id: MLXModel.OpenELM_1_1B_4bit,
+  family: ModelFamily.OpenELM,
+  provider: ModelProvider.Apple,
+  parameters: '1.1B',
+  quantization: '4bit',
+  displayName: 'OpenELM 1.1B (4-bit)',
+  downloadSize: 608162655
+}, {
+  id: MLXModel.OpenELM_1_1B_8bit,
+  family: ModelFamily.OpenELM,
+  provider: ModelProvider.Apple,
+  parameters: '1.1B',
+  quantization: '8bit',
+  displayName: 'OpenELM 1.1B (8-bit)',
+  downloadSize: 1148048397
+}, {
+  id: MLXModel.OpenELM_3B_4bit,
+  family: ModelFamily.OpenELM,
+  provider: ModelProvider.Apple,
+  parameters: '3B',
+  quantization: '4bit',
+  displayName: 'OpenELM 3B (4-bit)',
+  downloadSize: 1650000000
+}, {
+  id: MLXModel.OpenELM_3B_8bit,
+  family: ModelFamily.OpenELM,
+  provider: ModelProvider.Apple,
+  parameters: '3B',
+  quantization: '8bit',
+  displayName: 'OpenELM 3B (8-bit)',
+  downloadSize: 3100000000
+}];
 //# sourceMappingURL=models.js.map

package/lib/module/models.js.map

@@ -1 +1 @@
-{"version":3,"names":["MLXModel"],"sourceRoot":"../../src","sources":["models.ts"],"mappings":";;AAAA,WAAYA,QAAQ,0BAARA,QAAQ;EAClB;EADUA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAOlB;EAPUA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAelB;EAfUA,QAAQ;EAARA,QAAQ;EAmBlB;EAnBUA,QAAQ;EAARA,QAAQ;EAuBlB;EAvBUA,QAAQ;EAARA,QAAQ;EA2BlB;EA3BUA,QAAQ;EAARA,QAAQ;EA+BlB;EA/BUA,QAAQ;EAARA,QAAQ;EAmClB;EAnCUA,QAAQ;EAARA,QAAQ;EAuClB;EAvCUA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAAA,OAARA,QAAQ;AAAA","ignoreList":[]}
+{"version":3,"names":["ModelFamily","ModelProvider","MLXModel","MLXModels","id","Llama_3_2_1B_Instruct_4bit","family","Llama","provider","Meta","parameters","quantization","displayName","downloadSize","Llama_3_2_1B_Instruct_8bit","Llama_3_2_3B_Instruct_4bit","Llama_3_2_3B_Instruct_8bit","Qwen2_5_0_5B_Instruct_4bit","Qwen","Alibaba","Qwen2_5_0_5B_Instruct_8bit","Qwen2_5_1_5B_Instruct_4bit","Qwen2_5_1_5B_Instruct_8bit","Qwen2_5_3B_Instruct_4bit","Qwen2_5_3B_Instruct_8bit","Qwen3_1_7B_4bit","Qwen3_1_7B_8bit","Gemma_3_1B_IT_4bit","Gemma","Google","Gemma_3_1B_IT_8bit","Phi_3_5_Mini_Instruct_4bit","Phi","Microsoft","Phi_3_5_Mini_Instruct_8bit","Phi_4_Mini_Instruct_4bit","Phi_4_Mini_Instruct_8bit","SmolLM_1_7B_Instruct_4bit","SmolLM","HuggingFace","SmolLM_1_7B_Instruct_8bit","SmolLM2_1_7B_Instruct_4bit","SmolLM2_1_7B_Instruct_8bit","OpenELM_1_1B_4bit","OpenELM","Apple","OpenELM_1_1B_8bit","OpenELM_3B_4bit","OpenELM_3B_8bit"],"sourceRoot":"../../src","sources":["models.ts"],"mappings":";;AAAA,WAAYA,WAAW,0BAAXA,WAAW;EAAXA,WAAW;EAAXA,WAAW;EAAXA,WAAW;EAAXA,WAAW;EAAXA,WAAW;EAAXA,WAAW;EAAA,OAAXA,WAAW;AAAA;AASvB,WAAYC,aAAa,0BAAbA,aAAa;EAAbA,aAAa;EAAbA,aAAa;EAAbA,aAAa;EAAbA,aAAa;EAAbA,aAAa;EAAbA,aAAa;EAAA,OAAbA,aAAa;AAAA;AAqBzB,WAAYC,QAAQ,0BAARA,QAAQ;EAClB;EADUA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAOlB;EAPUA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAelB;EAfUA,QAAQ;EAARA,QAAQ;EAmBlB;EAnBUA,QAAQ;EAARA,QAAQ;EAuBlB;EAvBUA,QAAQ;EAARA,QAAQ;EA2BlB;EA3BUA,QAAQ;EAARA,QAAQ;EA+BlB;EA/BUA,QAAQ;EAARA,QAAQ;EAmClB;EAnCUA,QAAQ;EAARA,QAAQ;EAuClB;EAvCUA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAARA,QAAQ;EAAA,OAARA,QAAQ;AAAA;AA8CpB,OAAO,MAAMC,SAAsB,GAAG,CACpC;EACEC,EAAE,EAAEF,QAAQ,CAACG,0BAA0B;EACvCC,MAAM,EAAEN,WAAW,CAACO,KAAK;EACzBC,QAAQ,EAAEP,aAAa,CAACQ,IAAI;EAC5BC,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,+BAA+B;EAC5CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACY,0BAA0B;EACvCR,MAAM,EAAEN,WAAW,CAACO,KAAK;EACzBC,QAAQ,EAAEP,aAAa,CAACQ,IAAI;EAC5BC,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,+BAA+B;EAC5CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACa,0BAA0B;EACvCT,MAAM,EAAEN,WAAW,CAACO,KAAK;EACzBC,QAAQ,EAAEP,aAAa,CAACQ,IAAI;EAC5BC,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,+BAA+B;EAC5CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACc,0BAA0B;EACvCV,MAAM,EAAEN,WAAW,CAACO,KAAK;EACzBC,QAAQ,EAAEP,aAAa,CAACQ,IAAI;EAC5BC,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,+BAA+B;EAC5CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACe,0BAA0B;EACvCX,MAAM,EAAEN,WAAW,CAACkB,IAAI;EACxBV,QAAQ,EAAEP,aAAa,CAACkB,OAAO;EAC/BT,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,gCAAgC;EAC7CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACkB,0BAA0B;EACvCd,MAAM,EAAEN,WAAW,CAACkB,IAAI;EACxBV,QAAQ,EAAEP,aAAa,CAACkB,OAAO;EAC/BT,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,gCAAgC;EAC7CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACmB,0BAA0B;EACvCf,MAAM,EAAEN,WAAW,CAACkB,IAAI;EACxBV,QAAQ,EAAEP,aAAa,CAACkB,OAAO;EAC/BT,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,gCAAgC;EAC7CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACoB,0BAA0B;EACvChB,MAAM,EAAEN,WAAW,CAACkB,IAAI;EACxBV,QAAQ,EAAEP,aAAa,CAACkB,OAAO;EAC/BT,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,gCAAgC;EAC7CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACqB,wBAAwB;EACrCjB,MAAM,EAAEN,WAAW,CAACkB,IAAI;EACxBV,QAAQ,EAAEP,aAAa,CAACkB,OAAO;EAC/BT,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,8BAA8B;EAC3CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACsB,wBAAwB;EACrClB,MAAM,EAAEN,WAAW,CAACkB,IAAI;EACxBV,QAAQ,EAAEP,aAAa,CAACkB,OAAO;EAC/BT,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,8BAA8B;EAC3CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACuB,eAAe;EAC5BnB,MAAM,EAAEN,WAAW,CAACkB,IAAI;EACxBV,QAAQ,EAAEP,aAAa,CAACkB,OAAO;EAC/BT,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,qBAAqB;EAClCC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACwB,eAAe;EAC5BpB,MAAM,EAAEN,WAAW,CAACkB,IAAI;EACxBV,QAAQ,EAAEP,aAAa,CAACkB,OAAO;EAC/BT,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,qBAAqB;EAClCC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACyB,kBAAkB;EAC/BrB,MAAM,EAAEN,WAAW,CAAC4B,KAAK;EACzBpB,QAAQ,EAAEP,aAAa,CAAC4B,MAAM;EAC9BnB,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,uBAAuB;EACpCC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAAC4B,kBAAkB;EAC/BxB,MAAM,EAAEN,WAAW,CAAC4B,KAAK;EACzBpB,QAAQ,EAAEP,aAAa,CAAC4B,MAAM;EAC9BnB,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,uBAAuB;EACpCC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAAC6B,0BAA0B;EACvCzB,MAAM,EAAEN,WAAW,CAACgC,GAAG;EACvBxB,QAAQ,EAAEP,aAAa,CAACgC,SAAS;EACjCvB,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,+BAA+B;EAC5CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACgC,0BAA0B;EACvC5B,MAAM,EAAEN,WAAW,CAACgC,GAAG;EACvBxB,QAAQ,EAAEP,aAAa,CAACgC,SAAS;EACjCvB,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,+BAA+B;EAC5CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACiC,wBAAwB;EACrC7B,MAAM,EAAEN,WAAW,CAACgC,GAAG;EACvBxB,QAAQ,EAAEP,aAAa,CAACgC,SAAS;EACjCvB,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,6BAA6B;EAC1CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACkC,wBAAwB;EACrC9B,MAAM,EAAEN,WAAW,CAACgC,GAAG;EACvBxB,QAAQ,EAAEP,aAAa,CAACgC,SAAS;EACjCvB,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,6BAA6B;EAC1CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACmC,yBAAyB;EACtC/B,MAAM,EAAEN,WAAW,CAACsC,MAAM;EAC1B9B,QAAQ,EAAEP,aAAa,CAACsC,WAAW;EACnC7B,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,8BAA8B;EAC3CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACsC,yBAAyB;EACtClC,MAAM,EAAEN,WAAW,CAACsC,MAAM;EAC1B9B,QAAQ,EAAEP,aAAa,CAACsC,WAAW;EACnC7B,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,8BAA8B;EAC3CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACuC,0BAA0B;EACvCnC,MAAM,EAAEN,WAAW,CAACsC,MAAM;EAC1B9B,QAAQ,EAAEP,aAAa,CAACsC,WAAW;EACnC7B,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,+BAA+B;EAC5CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACwC,0BAA0B;EACvCpC,MAAM,EAAEN,WAAW,CAACsC,MAAM;EAC1B9B,QAAQ,EAAEP,aAAa,CAACsC,WAAW;EACnC7B,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,+BAA+B;EAC5CC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAACyC,iBAAiB;EAC9BrC,MAAM,EAAEN,WAAW,CAAC4C,OAAO;EAC3BpC,QAAQ,EAAEP,aAAa,CAAC4C,KAAK;EAC7BnC,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,sBAAsB;EACnCC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAAC4C,iBAAiB;EAC9BxC,MAAM,EAAEN,WAAW,CAAC4C,OAAO;EAC3BpC,QAAQ,EAAEP,aAAa,CAAC4C,KAAK;EAC7BnC,UAAU,EAAE,MAAM;EAClBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,sBAAsB;EACnCC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAAC6C,eAAe;EAC5BzC,MAAM,EAAEN,WAAW,CAAC4C,OAAO;EAC3BpC,QAAQ,EAAEP,aAAa,CAAC4C,KAAK;EAC7BnC,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,oBAAoB;EACjCC,YAAY,EAAE;AAChB,CAAC,EACD;EACET,EAAE,EAAEF,QAAQ,CAAC8C,eAAe;EAC5B1C,MAAM,EAAEN,WAAW,CAAC4C,OAAO;EAC3BpC,QAAQ,EAAEP,aAAa,CAAC4C,KAAK;EAC7BnC,UAAU,EAAE,IAAI;EAChBC,YAAY,EAAE,MAAM;EACpBC,WAAW,EAAE,oBAAoB;EACjCC,YAAY,EAAE;AAChB,CAAC,CACF","ignoreList":[]}

package/lib/typescript/src/index.d.ts

@@ -1,6 +1,6 @@
 export { LLM, type Message } from './llm';
 export { ModelManager } from './modelManager';
-export { MLXModel } from './models';
+export { MLXModel, MLXModels, ModelFamily, type ModelInfo, ModelProvider, type ModelQuantization, } from './models';
 export type { GenerationStats, LLM as LLMSpec, LLMLoadOptions } from './specs/LLM.nitro';
 export type { ModelManager as ModelManagerSpec } from './specs/ModelManager.nitro';
 //# sourceMappingURL=index.d.ts.map

package/lib/typescript/src/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,KAAK,OAAO,EAAE,MAAM,OAAO,CAAA;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAC7C,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA;AAEnC,YAAY,EAAE,eAAe,EAAE,GAAG,IAAI,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAA;AACxF,YAAY,EAAE,YAAY,IAAI,gBAAgB,EAAE,MAAM,4BAA4B,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,KAAK,OAAO,EAAE,MAAM,OAAO,CAAA;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAC7C,OAAO,EACL,QAAQ,EACR,SAAS,EACT,WAAW,EACX,KAAK,SAAS,EACd,aAAa,EACb,KAAK,iBAAiB,GACvB,MAAM,UAAU,CAAA;AAEjB,YAAY,EAAE,eAAe,EAAE,GAAG,IAAI,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAA;AACxF,YAAY,EAAE,YAAY,IAAI,gBAAgB,EAAE,MAAM,4BAA4B,CAAA"}

package/lib/typescript/src/models.d.ts

@@ -1,3 +1,29 @@
+export declare enum ModelFamily {
+    Llama = "Llama",
+    Qwen = "Qwen",
+    Gemma = "Gemma",
+    Phi = "Phi",
+    SmolLM = "SmolLM",
+    OpenELM = "OpenELM"
+}
+export declare enum ModelProvider {
+    Meta = "Meta",
+    Alibaba = "Alibaba",
+    Google = "Google",
+    Microsoft = "Microsoft",
+    HuggingFace = "HuggingFace",
+    Apple = "Apple"
+}
+export type ModelQuantization = '4bit' | '8bit';
+export interface ModelInfo {
+    id: MLXModel;
+    family: ModelFamily;
+    provider: ModelProvider;
+    parameters: string;
+    quantization: ModelQuantization;
+    displayName: string;
+    downloadSize: number;
+}
 export declare enum MLXModel {
     Llama_3_2_1B_Instruct_4bit = "mlx-community/Llama-3.2-1B-Instruct-4bit",
     Llama_3_2_1B_Instruct_8bit = "mlx-community/Llama-3.2-1B-Instruct-8bit",
@@ -26,4 +52,5 @@ export declare enum MLXModel {
     OpenELM_3B_4bit = "mlx-community/OpenELM-3B-4bit",
     OpenELM_3B_8bit = "mlx-community/OpenELM-3B-8bit"
 }
+export declare const MLXModels: ModelInfo[];
 //# sourceMappingURL=models.d.ts.map

package/lib/typescript/src/models.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../../src/models.ts"],"names":[],"mappings":"AAAA,oBAAY,QAAQ;IAElB,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IAGvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,wBAAwB,2CAA2C;IACnE,wBAAwB,2CAA2C;IAGnE,eAAe,kCAAkC;IACjD,eAAe,kCAAkC;IAGjD,kBAAkB,qCAAqC;IACvD,kBAAkB,qCAAqC;IAGvD,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IAGvE,wBAAwB,2CAA2C;IACnE,wBAAwB,2CAA2C;IAGnE,yBAAyB,4CAA4C;IACrE,yBAAyB,4CAA4C;IAGrE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IAGvE,iBAAiB,oCAAoC;IACrD,iBAAiB,oCAAoC;IACrD,eAAe,kCAAkC;IACjD,eAAe,kCAAkC;CAClD"}
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../../src/models.ts"],"names":[],"mappings":"AAAA,oBAAY,WAAW;IACrB,KAAK,UAAU;IACf,IAAI,SAAS;IACb,KAAK,UAAU;IACf,GAAG,QAAQ;IACX,MAAM,WAAW;IACjB,OAAO,YAAY;CACpB;AAED,oBAAY,aAAa;IACvB,IAAI,SAAS;IACb,OAAO,YAAY;IACnB,MAAM,WAAW;IACjB,SAAS,cAAc;IACvB,WAAW,gBAAgB;IAC3B,KAAK,UAAU;CAChB;AAED,MAAM,MAAM,iBAAiB,GAAG,MAAM,GAAG,MAAM,CAAA;AAE/C,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,QAAQ,CAAA;IACZ,MAAM,EAAE,WAAW,CAAA;IACnB,QAAQ,EAAE,aAAa,CAAA;IACvB,UAAU,EAAE,MAAM,CAAA;IAClB,YAAY,EAAE,iBAAiB,CAAA;IAC/B,WAAW,EAAE,MAAM,CAAA;IACnB,YAAY,EAAE,MAAM,CAAA;CACrB;AAED,oBAAY,QAAQ;IAElB,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IAGvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IACvE,wBAAwB,2CAA2C;IACnE,wBAAwB,2CAA2C;IAGnE,eAAe,kCAAkC;IACjD,eAAe,kCAAkC;IAGjD,kBAAkB,qCAAqC;IACvD,kBAAkB,qCAAqC;IAGvD,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IAGvE,wBAAwB,2CAA2C;IACnE,wBAAwB,2CAA2C;IAGnE,yBAAyB,4CAA4C;IACrE,yBAAyB,4CAA4C;IAGrE,0BAA0B,6CAA6C;IACvE,0BAA0B,6CAA6C;IAGvE,iBAAiB,oCAAoC;IACrD,iBAAiB,oCAAoC;IACrD,eAAe,kCAAkC;IACjD,eAAe,kCAAkC;CAClD;AAED,eAAO,MAAM,SAAS,EAAE,SAAS,EA2OhC,CAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-native-nitro-mlx",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Nitro module package",
   "main": "./lib/module/index.js",
   "module": "./lib/module/index.js",

package/src/index.ts

@@ -1,6 +1,13 @@
 export { LLM, type Message } from './llm'
 export { ModelManager } from './modelManager'
-export { MLXModel } from './models'
+export {
+  MLXModel,
+  MLXModels,
+  ModelFamily,
+  type ModelInfo,
+  ModelProvider,
+  type ModelQuantization,
+} from './models'
 
 export type { GenerationStats, LLM as LLMSpec, LLMLoadOptions } from './specs/LLM.nitro'
 export type { ModelManager as ModelManagerSpec } from './specs/ModelManager.nitro'

package/src/models.ts

@@ -1,3 +1,33 @@
+export enum ModelFamily {
+  Llama = 'Llama',
+  Qwen = 'Qwen',
+  Gemma = 'Gemma',
+  Phi = 'Phi',
+  SmolLM = 'SmolLM',
+  OpenELM = 'OpenELM',
+}
+
+export enum ModelProvider {
+  Meta = 'Meta',
+  Alibaba = 'Alibaba',
+  Google = 'Google',
+  Microsoft = 'Microsoft',
+  HuggingFace = 'HuggingFace',
+  Apple = 'Apple',
+}
+
+export type ModelQuantization = '4bit' | '8bit'
+
+export interface ModelInfo {
+  id: MLXModel
+  family: ModelFamily
+  provider: ModelProvider
+  parameters: string
+  quantization: ModelQuantization
+  displayName: string
+  downloadSize: number
+}
+
 export enum MLXModel {
   // Llama 3.2 (Meta) - 1B and 3B variants
   Llama_3_2_1B_Instruct_4bit = 'mlx-community/Llama-3.2-1B-Instruct-4bit',
@@ -43,3 +73,240 @@ export enum MLXModel {
   OpenELM_3B_4bit = 'mlx-community/OpenELM-3B-4bit',
   OpenELM_3B_8bit = 'mlx-community/OpenELM-3B-8bit',
 }
+
+export const MLXModels: ModelInfo[] = [
+  {
+    id: MLXModel.Llama_3_2_1B_Instruct_4bit,
+    family: ModelFamily.Llama,
+    provider: ModelProvider.Meta,
+    parameters: '1B',
+    quantization: '4bit',
+    displayName: 'Llama 3.2 1B Instruct (4-bit)',
+    downloadSize: 1407777762,
+  },
+  {
+    id: MLXModel.Llama_3_2_1B_Instruct_8bit,
+    family: ModelFamily.Llama,
+    provider: ModelProvider.Meta,
+    parameters: '1B',
+    quantization: '8bit',
+    displayName: 'Llama 3.2 1B Instruct (8-bit)',
+    downloadSize: 1313157436,
+  },
+  {
+    id: MLXModel.Llama_3_2_3B_Instruct_4bit,
+    family: ModelFamily.Llama,
+    provider: ModelProvider.Meta,
+    parameters: '3B',
+    quantization: '4bit',
+    displayName: 'Llama 3.2 3B Instruct (4-bit)',
+    downloadSize: 2019397474,
+  },
+  {
+    id: MLXModel.Llama_3_2_3B_Instruct_8bit,
+    family: ModelFamily.Llama,
+    provider: ModelProvider.Meta,
+    parameters: '3B',
+    quantization: '8bit',
+    displayName: 'Llama 3.2 3B Instruct (8-bit)',
+    downloadSize: 3413784042,
+  },
+  {
+    id: MLXModel.Qwen2_5_0_5B_Instruct_4bit,
+    family: ModelFamily.Qwen,
+    provider: ModelProvider.Alibaba,
+    parameters: '0.5B',
+    quantization: '4bit',
+    displayName: 'Qwen 2.5 0.5B Instruct (4-bit)',
+    downloadSize: 278064920,
+  },
+  {
+    id: MLXModel.Qwen2_5_0_5B_Instruct_8bit,
+    family: ModelFamily.Qwen,
+    provider: ModelProvider.Alibaba,
+    parameters: '0.5B',
+    quantization: '8bit',
+    displayName: 'Qwen 2.5 0.5B Instruct (8-bit)',
+    downloadSize: 525045902,
+  },
+  {
+    id: MLXModel.Qwen2_5_1_5B_Instruct_4bit,
+    family: ModelFamily.Qwen,
+    provider: ModelProvider.Alibaba,
+    parameters: '1.5B',
+    quantization: '4bit',
+    displayName: 'Qwen 2.5 1.5B Instruct (4-bit)',
+    downloadSize: 868628559,
+  },
+  {
+    id: MLXModel.Qwen2_5_1_5B_Instruct_8bit,
+    family: ModelFamily.Qwen,
+    provider: ModelProvider.Alibaba,
+    parameters: '1.5B',
+    quantization: '8bit',
+    displayName: 'Qwen 2.5 1.5B Instruct (8-bit)',
+    downloadSize: 1640414038,
+  },
+  {
+    id: MLXModel.Qwen2_5_3B_Instruct_4bit,
+    family: ModelFamily.Qwen,
+    provider: ModelProvider.Alibaba,
+    parameters: '3B',
+    quantization: '4bit',
+    displayName: 'Qwen 2.5 3B Instruct (4-bit)',
+    downloadSize: 1736293090,
+  },
+  {
+    id: MLXModel.Qwen2_5_3B_Instruct_8bit,
+    family: ModelFamily.Qwen,
+    provider: ModelProvider.Alibaba,
+    parameters: '3B',
+    quantization: '8bit',
+    displayName: 'Qwen 2.5 3B Instruct (8-bit)',
+    downloadSize: 3279142142,
+  },
+  {
+    id: MLXModel.Qwen3_1_7B_4bit,
+    family: ModelFamily.Qwen,
+    provider: ModelProvider.Alibaba,
+    parameters: '1.7B',
+    quantization: '4bit',
+    displayName: 'Qwen 3 1.7B (4-bit)',
+    downloadSize: 979502864,
+  },
+  {
+    id: MLXModel.Qwen3_1_7B_8bit,
+    family: ModelFamily.Qwen,
+    provider: ModelProvider.Alibaba,
+    parameters: '1.7B',
+    quantization: '8bit',
+    displayName: 'Qwen 3 1.7B (8-bit)',
+    downloadSize: 1839729195,
+  },
+  {
+    id: MLXModel.Gemma_3_1B_IT_4bit,
+    family: ModelFamily.Gemma,
+    provider: ModelProvider.Google,
+    parameters: '1B',
+    quantization: '4bit',
+    displayName: 'Gemma 3 1B IT (4-bit)',
+    downloadSize: 770650946,
+  },
+  {
+    id: MLXModel.Gemma_3_1B_IT_8bit,
+    family: ModelFamily.Gemma,
+    provider: ModelProvider.Google,
+    parameters: '1B',
+    quantization: '8bit',
+    displayName: 'Gemma 3 1B IT (8-bit)',
+    downloadSize: 1421522471,
+  },
+  {
+    id: MLXModel.Phi_3_5_Mini_Instruct_4bit,
+    family: ModelFamily.Phi,
+    provider: ModelProvider.Microsoft,
+    parameters: '3.8B',
+    quantization: '4bit',
+    displayName: 'Phi 3.5 Mini Instruct (4-bit)',
+    downloadSize: 2150195856,
+  },
+  {
+    id: MLXModel.Phi_3_5_Mini_Instruct_8bit,
+    family: ModelFamily.Phi,
+    provider: ModelProvider.Microsoft,
+    parameters: '3.8B',
+    quantization: '8bit',
+    displayName: 'Phi 3.5 Mini Instruct (8-bit)',
+    downloadSize: 4060636056,
+  },
+  {
+    id: MLXModel.Phi_4_Mini_Instruct_4bit,
+    family: ModelFamily.Phi,
+    provider: ModelProvider.Microsoft,
+    parameters: '3.8B',
+    quantization: '4bit',
+    displayName: 'Phi 4 Mini Instruct (4-bit)',
+    downloadSize: 2173624891,
+  },
+  {
+    id: MLXModel.Phi_4_Mini_Instruct_8bit,
+    family: ModelFamily.Phi,
+    provider: ModelProvider.Microsoft,
+    parameters: '3.8B',
+    quantization: '8bit',
+    displayName: 'Phi 4 Mini Instruct (8-bit)',
+    downloadSize: 4091536167,
+  },
+  {
+    id: MLXModel.SmolLM_1_7B_Instruct_4bit,
+    family: ModelFamily.SmolLM,
+    provider: ModelProvider.HuggingFace,
+    parameters: '1.7B',
+    quantization: '4bit',
+    displayName: 'SmolLM 1.7B Instruct (4-bit)',
+    downloadSize: 962855374,
+  },
+  {
+    id: MLXModel.SmolLM_1_7B_Instruct_8bit,
+    family: ModelFamily.SmolLM,
+    provider: ModelProvider.HuggingFace,
+    parameters: '1.7B',
+    quantization: '8bit',
+    displayName: 'SmolLM 1.7B Instruct (8-bit)',
+    downloadSize: 1818493993,
+  },
+  {
+    id: MLXModel.SmolLM2_1_7B_Instruct_4bit,
+    family: ModelFamily.SmolLM,
+    provider: ModelProvider.HuggingFace,
+    parameters: '1.7B',
+    quantization: '4bit',
+    displayName: 'SmolLM2 1.7B Instruct (4-bit)',
+    downloadSize: 980000000,
+  },
+  {
+    id: MLXModel.SmolLM2_1_7B_Instruct_8bit,
+    family: ModelFamily.SmolLM,
+    provider: ModelProvider.HuggingFace,
+    parameters: '1.7B',
+    quantization: '8bit',
+    displayName: 'SmolLM2 1.7B Instruct (8-bit)',
+    downloadSize: 1850000000,
+  },
+  {
+    id: MLXModel.OpenELM_1_1B_4bit,
+    family: ModelFamily.OpenELM,
+    provider: ModelProvider.Apple,
+    parameters: '1.1B',
+    quantization: '4bit',
+    displayName: 'OpenELM 1.1B (4-bit)',
+    downloadSize: 608162655,
+  },
+  {
+    id: MLXModel.OpenELM_1_1B_8bit,
+    family: ModelFamily.OpenELM,
+    provider: ModelProvider.Apple,
+    parameters: '1.1B',
+    quantization: '8bit',
+    displayName: 'OpenELM 1.1B (8-bit)',
+    downloadSize: 1148048397,
+  },
+  {
+    id: MLXModel.OpenELM_3B_4bit,
+    family: ModelFamily.OpenELM,
+    provider: ModelProvider.Apple,
+    parameters: '3B',
+    quantization: '4bit',
+    displayName: 'OpenELM 3B (4-bit)',
+    downloadSize: 1650000000,
+  },
+  {
+    id: MLXModel.OpenELM_3B_8bit,
+    family: ModelFamily.OpenELM,
+    provider: ModelProvider.Apple,
+    parameters: '3B',
+    quantization: '8bit',
+    displayName: 'OpenELM 3B (8-bit)',
+    downloadSize: 3100000000,
+  },
+]