@dr33m/react-native-litert-lm 0.5.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (168) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +627 -0
  3. package/android/CMakeLists.txt +32 -0
  4. package/android/build.gradle +104 -0
  5. package/android/src/main/AndroidManifest.xml +19 -0
  6. package/android/src/main/java/com/margelo/nitro/dev/litert/litertlm/HybridLiteRTLM.kt +888 -0
  7. package/android/src/main/java/com/margelo/nitro/dev/litert/litertlm/HybridModelStore.kt +189 -0
  8. package/android/src/main/java/com/margelo/nitro/dev/litert/litertlm/LiteRTLMRegistry.kt +32 -0
  9. package/android/src/main/java/com/margelo/nitro/dev/litert/litertlm/MultimodalPartFactories.kt +15 -0
  10. package/android/src/main/java/com/margelo/nitro/dev/litert/litertlm/StreamingCallbackListener.kt +89 -0
  11. package/android/src/main/java/dev/litert/litertlm/LiteRTLMInitProvider.kt +57 -0
  12. package/android/src/main/java/dev/litert/litertlm/LiteRTLMPackage.kt +43 -0
  13. package/android/src/test/java/com/margelo/nitro/core/Promise.kt +46 -0
  14. package/android/src/test/java/com/margelo/nitro/dev/litert/litertlm/HybridLiteRTLMTest.kt +122 -0
  15. package/app.plugin.js +58 -0
  16. package/cpp/cpp-adapter.cpp +21 -0
  17. package/cpp/include/README.md +32 -0
  18. package/cpp/include/litert_lm_engine.h +516 -0
  19. package/cpp/include/stb_image.h +7988 -0
  20. package/ios/HybridLiteRTLM+Execute.swift +287 -0
  21. package/ios/HybridLiteRTLM+Streaming.swift +153 -0
  22. package/ios/HybridLiteRTLM.swift +541 -0
  23. package/ios/HybridModelStore.swift +206 -0
  24. package/ios/MultimodalPart+Factories.swift +22 -0
  25. package/ios/Tests/HybridLiteRTLMTests.swift +187 -0
  26. package/lib/__mocks__/react-native-nitro-modules.d.ts +81 -0
  27. package/lib/__mocks__/react-native-nitro-modules.js +98 -0
  28. package/lib/__tests__/hooks.test.d.ts +1 -0
  29. package/lib/__tests__/hooks.test.js +175 -0
  30. package/lib/__tests__/inferenceRouting.test.d.ts +1 -0
  31. package/lib/__tests__/inferenceRouting.test.js +28 -0
  32. package/lib/__tests__/memoryTracker.test.d.ts +1 -0
  33. package/lib/__tests__/memoryTracker.test.js +74 -0
  34. package/lib/__tests__/modelFactory.test.d.ts +1 -0
  35. package/lib/__tests__/modelFactory.test.js +107 -0
  36. package/lib/__tests__/modelPath.test.d.ts +1 -0
  37. package/lib/__tests__/modelPath.test.js +12 -0
  38. package/lib/__tests__/modelRegistry.test.d.ts +1 -0
  39. package/lib/__tests__/modelRegistry.test.js +61 -0
  40. package/lib/hooks.d.ts +46 -0
  41. package/lib/hooks.js +159 -0
  42. package/lib/index.d.ts +142 -0
  43. package/lib/index.js +184 -0
  44. package/lib/inferenceRouting.d.ts +18 -0
  45. package/lib/inferenceRouting.js +63 -0
  46. package/lib/memoryTracker.d.ts +128 -0
  47. package/lib/memoryTracker.js +155 -0
  48. package/lib/modelFactory.d.ts +24 -0
  49. package/lib/modelFactory.js +128 -0
  50. package/lib/modelPath.d.ts +8 -0
  51. package/lib/modelPath.js +18 -0
  52. package/lib/modelRegistry.d.ts +55 -0
  53. package/lib/modelRegistry.js +82 -0
  54. package/lib/specs/LiteRTLM.nitro.d.ts +380 -0
  55. package/lib/specs/LiteRTLM.nitro.js +2 -0
  56. package/nitrogen/generated/.gitattributes +1 -0
  57. package/nitrogen/generated/android/LiteRTLM+autolinking.cmake +83 -0
  58. package/nitrogen/generated/android/LiteRTLM+autolinking.gradle +27 -0
  59. package/nitrogen/generated/android/LiteRTLMOnLoad.cpp +74 -0
  60. package/nitrogen/generated/android/LiteRTLMOnLoad.hpp +34 -0
  61. package/nitrogen/generated/android/c++/JBackend.hpp +61 -0
  62. package/nitrogen/generated/android/c++/JExecuteResult.hpp +86 -0
  63. package/nitrogen/generated/android/c++/JFunc_void_double.hpp +75 -0
  64. package/nitrogen/generated/android/c++/JFunc_void_std__string_bool.hpp +76 -0
  65. package/nitrogen/generated/android/c++/JGenerationStats.hpp +77 -0
  66. package/nitrogen/generated/android/c++/JHybridLiteRTLMSpec.cpp +371 -0
  67. package/nitrogen/generated/android/c++/JHybridLiteRTLMSpec.hpp +83 -0
  68. package/nitrogen/generated/android/c++/JHybridModelStoreSpec.cpp +101 -0
  69. package/nitrogen/generated/android/c++/JHybridModelStoreSpec.hpp +67 -0
  70. package/nitrogen/generated/android/c++/JLLMConfig.hpp +125 -0
  71. package/nitrogen/generated/android/c++/JMemoryUsage.hpp +69 -0
  72. package/nitrogen/generated/android/c++/JMessage.hpp +63 -0
  73. package/nitrogen/generated/android/c++/JModelFile.hpp +69 -0
  74. package/nitrogen/generated/android/c++/JMultimodalPart.hpp +78 -0
  75. package/nitrogen/generated/android/c++/JPartType.hpp +61 -0
  76. package/nitrogen/generated/android/c++/JRole.hpp +61 -0
  77. package/nitrogen/generated/android/c++/JToolCall.hpp +61 -0
  78. package/nitrogen/generated/android/c++/JToolDefinition.hpp +65 -0
  79. package/nitrogen/generated/android/c++/JToolResponse.hpp +61 -0
  80. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/Backend.kt +24 -0
  81. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/ExecuteResult.kt +61 -0
  82. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/Func_void_double.kt +80 -0
  83. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/Func_void_std__string_bool.kt +80 -0
  84. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/GenerationStats.kt +76 -0
  85. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/HybridLiteRTLMSpec.kt +165 -0
  86. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/HybridModelStoreSpec.kt +76 -0
  87. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/LLMConfig.kt +106 -0
  88. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/LiteRTLMOnLoad.kt +35 -0
  89. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/MemoryUsage.kt +66 -0
  90. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/Message.kt +56 -0
  91. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/ModelFile.kt +66 -0
  92. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/MultimodalPart.kt +71 -0
  93. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/PartType.kt +24 -0
  94. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/Role.kt +24 -0
  95. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/ToolCall.kt +56 -0
  96. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/ToolDefinition.kt +61 -0
  97. package/nitrogen/generated/android/kotlin/com/margelo/nitro/dev/litert/litertlm/ToolResponse.kt +56 -0
  98. package/nitrogen/generated/ios/LiteRTLM+autolinking.rb +62 -0
  99. package/nitrogen/generated/ios/LiteRTLM-Swift-Cxx-Bridge.cpp +98 -0
  100. package/nitrogen/generated/ios/LiteRTLM-Swift-Cxx-Bridge.hpp +585 -0
  101. package/nitrogen/generated/ios/LiteRTLM-Swift-Cxx-Umbrella.hpp +93 -0
  102. package/nitrogen/generated/ios/LiteRTLMAutolinking.mm +41 -0
  103. package/nitrogen/generated/ios/LiteRTLMAutolinking.swift +38 -0
  104. package/nitrogen/generated/ios/c++/HybridLiteRTLMSpecSwift.cpp +11 -0
  105. package/nitrogen/generated/ios/c++/HybridLiteRTLMSpecSwift.hpp +279 -0
  106. package/nitrogen/generated/ios/c++/HybridModelStoreSpecSwift.cpp +11 -0
  107. package/nitrogen/generated/ios/c++/HybridModelStoreSpecSwift.hpp +117 -0
  108. package/nitrogen/generated/ios/swift/Backend.swift +44 -0
  109. package/nitrogen/generated/ios/swift/ExecuteResult.swift +45 -0
  110. package/nitrogen/generated/ios/swift/Func_void.swift +46 -0
  111. package/nitrogen/generated/ios/swift/Func_void_ExecuteResult.swift +46 -0
  112. package/nitrogen/generated/ios/swift/Func_void_double.swift +46 -0
  113. package/nitrogen/generated/ios/swift/Func_void_std__exception_ptr.swift +46 -0
  114. package/nitrogen/generated/ios/swift/Func_void_std__string.swift +46 -0
  115. package/nitrogen/generated/ios/swift/Func_void_std__string_bool.swift +46 -0
  116. package/nitrogen/generated/ios/swift/GenerationStats.swift +54 -0
  117. package/nitrogen/generated/ios/swift/HybridLiteRTLMSpec.swift +75 -0
  118. package/nitrogen/generated/ios/swift/HybridLiteRTLMSpec_cxx.swift +516 -0
  119. package/nitrogen/generated/ios/swift/HybridModelStoreSpec.swift +59 -0
  120. package/nitrogen/generated/ios/swift/HybridModelStoreSpec_cxx.swift +203 -0
  121. package/nitrogen/generated/ios/swift/LLMConfig.swift +239 -0
  122. package/nitrogen/generated/ios/swift/MemoryUsage.swift +44 -0
  123. package/nitrogen/generated/ios/swift/Message.swift +34 -0
  124. package/nitrogen/generated/ios/swift/ModelFile.swift +44 -0
  125. package/nitrogen/generated/ios/swift/MultimodalPart.swift +101 -0
  126. package/nitrogen/generated/ios/swift/PartType.swift +44 -0
  127. package/nitrogen/generated/ios/swift/Role.swift +44 -0
  128. package/nitrogen/generated/ios/swift/ToolCall.swift +34 -0
  129. package/nitrogen/generated/ios/swift/ToolDefinition.swift +39 -0
  130. package/nitrogen/generated/ios/swift/ToolResponse.swift +34 -0
  131. package/nitrogen/generated/shared/c++/Backend.hpp +80 -0
  132. package/nitrogen/generated/shared/c++/ExecuteResult.hpp +94 -0
  133. package/nitrogen/generated/shared/c++/GenerationStats.hpp +103 -0
  134. package/nitrogen/generated/shared/c++/HybridLiteRTLMSpec.cpp +41 -0
  135. package/nitrogen/generated/shared/c++/HybridLiteRTLMSpec.hpp +109 -0
  136. package/nitrogen/generated/shared/c++/HybridModelStoreSpec.cpp +25 -0
  137. package/nitrogen/generated/shared/c++/HybridModelStoreSpec.hpp +71 -0
  138. package/nitrogen/generated/shared/c++/LLMConfig.hpp +134 -0
  139. package/nitrogen/generated/shared/c++/MemoryUsage.hpp +95 -0
  140. package/nitrogen/generated/shared/c++/Message.hpp +89 -0
  141. package/nitrogen/generated/shared/c++/ModelFile.hpp +95 -0
  142. package/nitrogen/generated/shared/c++/MultimodalPart.hpp +103 -0
  143. package/nitrogen/generated/shared/c++/PartType.hpp +80 -0
  144. package/nitrogen/generated/shared/c++/Role.hpp +80 -0
  145. package/nitrogen/generated/shared/c++/ToolCall.hpp +87 -0
  146. package/nitrogen/generated/shared/c++/ToolDefinition.hpp +91 -0
  147. package/nitrogen/generated/shared/c++/ToolResponse.hpp +87 -0
  148. package/package.json +115 -0
  149. package/react-native-litert-lm.podspec +62 -0
  150. package/react-native.config.js +16 -0
  151. package/scripts/download-ios-frameworks.sh +39 -0
  152. package/scripts/framework-source.js +46 -0
  153. package/scripts/postinstall.js +145 -0
  154. package/src/__mocks__/react-native-nitro-modules.ts +132 -0
  155. package/src/__tests__/hooks.test.ts +216 -0
  156. package/src/__tests__/inferenceRouting.test.ts +36 -0
  157. package/src/__tests__/memoryTracker.test.ts +87 -0
  158. package/src/__tests__/modelFactory.test.ts +153 -0
  159. package/src/__tests__/modelPath.test.ts +14 -0
  160. package/src/__tests__/modelRegistry.test.ts +73 -0
  161. package/src/hooks.ts +234 -0
  162. package/src/index.ts +214 -0
  163. package/src/inferenceRouting.ts +80 -0
  164. package/src/memoryTracker.ts +268 -0
  165. package/src/modelFactory.ts +163 -0
  166. package/src/modelPath.ts +16 -0
  167. package/src/modelRegistry.ts +104 -0
  168. package/src/specs/LiteRTLM.nitro.ts +451 -0
@@ -0,0 +1,44 @@
1
+ ///
2
+ /// PartType.swift
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ /**
9
+ * Represents the JS union `PartType`, backed by a C++ enum.
10
+ */
11
+ public typealias PartType = margelo.nitro.litertlm.PartType
12
+
13
+ public extension PartType {
14
+ /**
15
+ * Get a PartType for the given String value, or
16
+ * return `nil` if the given value was invalid/unknown.
17
+ */
18
+ init?(fromString string: String) {
19
+ switch string {
20
+ case "text":
21
+ self = .text
22
+ case "image":
23
+ self = .image
24
+ case "audio":
25
+ self = .audio
26
+ default:
27
+ return nil
28
+ }
29
+ }
30
+
31
+ /**
32
+ * Get the String value this PartType represents.
33
+ */
34
+ var stringValue: String {
35
+ switch self {
36
+ case .text:
37
+ return "text"
38
+ case .image:
39
+ return "image"
40
+ case .audio:
41
+ return "audio"
42
+ }
43
+ }
44
+ }
@@ -0,0 +1,44 @@
1
+ ///
2
+ /// Role.swift
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ /**
9
+ * Represents the JS union `Role`, backed by a C++ enum.
10
+ */
11
+ public typealias Role = margelo.nitro.litertlm.Role
12
+
13
+ public extension Role {
14
+ /**
15
+ * Get a Role for the given String value, or
16
+ * return `nil` if the given value was invalid/unknown.
17
+ */
18
+ init?(fromString string: String) {
19
+ switch string {
20
+ case "user":
21
+ self = .user
22
+ case "model":
23
+ self = .model
24
+ case "system":
25
+ self = .system
26
+ default:
27
+ return nil
28
+ }
29
+ }
30
+
31
+ /**
32
+ * Get the String value this Role represents.
33
+ */
34
+ var stringValue: String {
35
+ switch self {
36
+ case .user:
37
+ return "user"
38
+ case .model:
39
+ return "model"
40
+ case .system:
41
+ return "system"
42
+ }
43
+ }
44
+ }
@@ -0,0 +1,34 @@
1
+ ///
2
+ /// ToolCall.swift
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ import NitroModules
9
+
10
+ /**
11
+ * Represents an instance of `ToolCall`, backed by a C++ struct.
12
+ */
13
+ public typealias ToolCall = margelo.nitro.litertlm.ToolCall
14
+
15
+ public extension ToolCall {
16
+ private typealias bridge = margelo.nitro.litertlm.bridge.swift
17
+
18
+ /**
19
+ * Create a new instance of `ToolCall`.
20
+ */
21
+ init(name: String, argumentsJson: String) {
22
+ self.init(std.string(name), std.string(argumentsJson))
23
+ }
24
+
25
+ @inline(__always)
26
+ var name: String {
27
+ return String(self.__name)
28
+ }
29
+
30
+ @inline(__always)
31
+ var argumentsJson: String {
32
+ return String(self.__argumentsJson)
33
+ }
34
+ }
@@ -0,0 +1,39 @@
1
+ ///
2
+ /// ToolDefinition.swift
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ import NitroModules
9
+
10
+ /**
11
+ * Represents an instance of `ToolDefinition`, backed by a C++ struct.
12
+ */
13
+ public typealias ToolDefinition = margelo.nitro.litertlm.ToolDefinition
14
+
15
+ public extension ToolDefinition {
16
+ private typealias bridge = margelo.nitro.litertlm.bridge.swift
17
+
18
+ /**
19
+ * Create a new instance of `ToolDefinition`.
20
+ */
21
+ init(name: String, description: String, parametersJson: String) {
22
+ self.init(std.string(name), std.string(description), std.string(parametersJson))
23
+ }
24
+
25
+ @inline(__always)
26
+ var name: String {
27
+ return String(self.__name)
28
+ }
29
+
30
+ @inline(__always)
31
+ var description: String {
32
+ return String(self.__description)
33
+ }
34
+
35
+ @inline(__always)
36
+ var parametersJson: String {
37
+ return String(self.__parametersJson)
38
+ }
39
+ }
@@ -0,0 +1,34 @@
1
+ ///
2
+ /// ToolResponse.swift
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ import NitroModules
9
+
10
+ /**
11
+ * Represents an instance of `ToolResponse`, backed by a C++ struct.
12
+ */
13
+ public typealias ToolResponse = margelo.nitro.litertlm.ToolResponse
14
+
15
+ public extension ToolResponse {
16
+ private typealias bridge = margelo.nitro.litertlm.bridge.swift
17
+
18
+ /**
19
+ * Create a new instance of `ToolResponse`.
20
+ */
21
+ init(name: String, responseJson: String) {
22
+ self.init(std.string(name), std.string(responseJson))
23
+ }
24
+
25
+ @inline(__always)
26
+ var name: String {
27
+ return String(self.__name)
28
+ }
29
+
30
+ @inline(__always)
31
+ var responseJson: String {
32
+ return String(self.__responseJson)
33
+ }
34
+ }
@@ -0,0 +1,80 @@
1
+ ///
2
+ /// Backend.hpp
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ #pragma once
9
+
10
+ #if __has_include(<NitroModules/NitroHash.hpp>)
11
+ #include <NitroModules/NitroHash.hpp>
12
+ #else
13
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
14
+ #endif
15
+ #if __has_include(<NitroModules/JSIConverter.hpp>)
16
+ #include <NitroModules/JSIConverter.hpp>
17
+ #else
18
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
19
+ #endif
20
+ #if __has_include(<NitroModules/NitroDefines.hpp>)
21
+ #include <NitroModules/NitroDefines.hpp>
22
+ #else
23
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
24
+ #endif
25
+
26
+ namespace margelo::nitro::litertlm {
27
+
28
+ /**
29
+ * An enum which can be represented as a JavaScript union (Backend).
30
+ */
31
+ enum class Backend {
32
+ CPU SWIFT_NAME(cpu) = 0,
33
+ GPU SWIFT_NAME(gpu) = 1,
34
+ NPU SWIFT_NAME(npu) = 2,
35
+ } CLOSED_ENUM;
36
+
37
+ } // namespace margelo::nitro::litertlm
38
+
39
+ namespace margelo::nitro {
40
+
41
+ // C++ Backend <> JS Backend (union)
42
+ template <>
43
+ struct JSIConverter<margelo::nitro::litertlm::Backend> final {
44
+ static inline margelo::nitro::litertlm::Backend fromJSI(jsi::Runtime& runtime, const jsi::Value& arg) {
45
+ std::string unionValue = JSIConverter<std::string>::fromJSI(runtime, arg);
46
+ switch (hashString(unionValue.c_str(), unionValue.size())) {
47
+ case hashString("cpu"): return margelo::nitro::litertlm::Backend::CPU;
48
+ case hashString("gpu"): return margelo::nitro::litertlm::Backend::GPU;
49
+ case hashString("npu"): return margelo::nitro::litertlm::Backend::NPU;
50
+ default: [[unlikely]]
51
+ throw std::invalid_argument("Cannot convert \"" + unionValue + "\" to enum Backend - invalid value!");
52
+ }
53
+ }
54
+ static inline jsi::Value toJSI(jsi::Runtime& runtime, margelo::nitro::litertlm::Backend arg) {
55
+ switch (arg) {
56
+ case margelo::nitro::litertlm::Backend::CPU: return JSIConverter<std::string>::toJSI(runtime, "cpu");
57
+ case margelo::nitro::litertlm::Backend::GPU: return JSIConverter<std::string>::toJSI(runtime, "gpu");
58
+ case margelo::nitro::litertlm::Backend::NPU: return JSIConverter<std::string>::toJSI(runtime, "npu");
59
+ default: [[unlikely]]
60
+ throw std::invalid_argument("Cannot convert Backend to JS - invalid value: "
61
+ + std::to_string(static_cast<int>(arg)) + "!");
62
+ }
63
+ }
64
+ static inline bool canConvert(jsi::Runtime& runtime, const jsi::Value& value) {
65
+ if (!value.isString()) {
66
+ return false;
67
+ }
68
+ std::string unionValue = JSIConverter<std::string>::fromJSI(runtime, value);
69
+ switch (hashString(unionValue.c_str(), unionValue.size())) {
70
+ case hashString("cpu"):
71
+ case hashString("gpu"):
72
+ case hashString("npu"):
73
+ return true;
74
+ default:
75
+ return false;
76
+ }
77
+ }
78
+ };
79
+
80
+ } // namespace margelo::nitro
@@ -0,0 +1,94 @@
1
+ ///
2
+ /// ExecuteResult.hpp
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ #pragma once
9
+
10
+ #if __has_include(<NitroModules/JSIConverter.hpp>)
11
+ #include <NitroModules/JSIConverter.hpp>
12
+ #else
13
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
14
+ #endif
15
+ #if __has_include(<NitroModules/NitroDefines.hpp>)
16
+ #include <NitroModules/NitroDefines.hpp>
17
+ #else
18
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
19
+ #endif
20
+ #if __has_include(<NitroModules/JSIHelpers.hpp>)
21
+ #include <NitroModules/JSIHelpers.hpp>
22
+ #else
23
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
24
+ #endif
25
+ #if __has_include(<NitroModules/PropNameIDCache.hpp>)
26
+ #include <NitroModules/PropNameIDCache.hpp>
27
+ #else
28
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
29
+ #endif
30
+
31
+ // Forward declaration of `ToolCall` to properly resolve imports.
32
+ namespace margelo::nitro::litertlm { struct ToolCall; }
33
+
34
+ #include <string>
35
+ #include "ToolCall.hpp"
36
+ #include <vector>
37
+
38
+ namespace margelo::nitro::litertlm {
39
+
40
+ /**
41
+ * A struct which can be represented as a JavaScript object (ExecuteResult).
42
+ */
43
+ struct ExecuteResult final {
44
+ public:
45
+ std::string text SWIFT_PRIVATE;
46
+ std::vector<ToolCall> toolCalls SWIFT_PRIVATE;
47
+ std::string thinkingText SWIFT_PRIVATE;
48
+
49
+ public:
50
+ ExecuteResult() = default;
51
+ explicit ExecuteResult(std::string text, std::vector<ToolCall> toolCalls, std::string thinkingText): text(text), toolCalls(toolCalls), thinkingText(thinkingText) {}
52
+
53
+ public:
54
+ friend bool operator==(const ExecuteResult& lhs, const ExecuteResult& rhs) = default;
55
+ };
56
+
57
+ } // namespace margelo::nitro::litertlm
58
+
59
+ namespace margelo::nitro {
60
+
61
+ // C++ ExecuteResult <> JS ExecuteResult (object)
62
+ template <>
63
+ struct JSIConverter<margelo::nitro::litertlm::ExecuteResult> final {
64
+ static inline margelo::nitro::litertlm::ExecuteResult fromJSI(jsi::Runtime& runtime, const jsi::Value& arg) {
65
+ jsi::Object obj = arg.asObject(runtime);
66
+ return margelo::nitro::litertlm::ExecuteResult(
67
+ JSIConverter<std::string>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "text"))),
68
+ JSIConverter<std::vector<margelo::nitro::litertlm::ToolCall>>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "toolCalls"))),
69
+ JSIConverter<std::string>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "thinkingText")))
70
+ );
71
+ }
72
+ static inline jsi::Value toJSI(jsi::Runtime& runtime, const margelo::nitro::litertlm::ExecuteResult& arg) {
73
+ jsi::Object obj(runtime);
74
+ obj.setProperty(runtime, PropNameIDCache::get(runtime, "text"), JSIConverter<std::string>::toJSI(runtime, arg.text));
75
+ obj.setProperty(runtime, PropNameIDCache::get(runtime, "toolCalls"), JSIConverter<std::vector<margelo::nitro::litertlm::ToolCall>>::toJSI(runtime, arg.toolCalls));
76
+ obj.setProperty(runtime, PropNameIDCache::get(runtime, "thinkingText"), JSIConverter<std::string>::toJSI(runtime, arg.thinkingText));
77
+ return obj;
78
+ }
79
+ static inline bool canConvert(jsi::Runtime& runtime, const jsi::Value& value) {
80
+ if (!value.isObject()) {
81
+ return false;
82
+ }
83
+ jsi::Object obj = value.getObject(runtime);
84
+ if (!nitro::isPlainObject(runtime, obj)) {
85
+ return false;
86
+ }
87
+ if (!JSIConverter<std::string>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "text")))) return false;
88
+ if (!JSIConverter<std::vector<margelo::nitro::litertlm::ToolCall>>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "toolCalls")))) return false;
89
+ if (!JSIConverter<std::string>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "thinkingText")))) return false;
90
+ return true;
91
+ }
92
+ };
93
+
94
+ } // namespace margelo::nitro
@@ -0,0 +1,103 @@
1
+ ///
2
+ /// GenerationStats.hpp
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ #pragma once
9
+
10
+ #if __has_include(<NitroModules/JSIConverter.hpp>)
11
+ #include <NitroModules/JSIConverter.hpp>
12
+ #else
13
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
14
+ #endif
15
+ #if __has_include(<NitroModules/NitroDefines.hpp>)
16
+ #include <NitroModules/NitroDefines.hpp>
17
+ #else
18
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
19
+ #endif
20
+ #if __has_include(<NitroModules/JSIHelpers.hpp>)
21
+ #include <NitroModules/JSIHelpers.hpp>
22
+ #else
23
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
24
+ #endif
25
+ #if __has_include(<NitroModules/PropNameIDCache.hpp>)
26
+ #include <NitroModules/PropNameIDCache.hpp>
27
+ #else
28
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
29
+ #endif
30
+
31
+
32
+
33
+
34
+
35
+ namespace margelo::nitro::litertlm {
36
+
37
+ /**
38
+ * A struct which can be represented as a JavaScript object (GenerationStats).
39
+ */
40
+ struct GenerationStats final {
41
+ public:
42
+ double promptTokens SWIFT_PRIVATE;
43
+ double completionTokens SWIFT_PRIVATE;
44
+ double totalTokens SWIFT_PRIVATE;
45
+ double timeToFirstToken SWIFT_PRIVATE;
46
+ double totalTime SWIFT_PRIVATE;
47
+ double tokensPerSecond SWIFT_PRIVATE;
48
+
49
+ public:
50
+ GenerationStats() = default;
51
+ explicit GenerationStats(double promptTokens, double completionTokens, double totalTokens, double timeToFirstToken, double totalTime, double tokensPerSecond): promptTokens(promptTokens), completionTokens(completionTokens), totalTokens(totalTokens), timeToFirstToken(timeToFirstToken), totalTime(totalTime), tokensPerSecond(tokensPerSecond) {}
52
+
53
+ public:
54
+ friend bool operator==(const GenerationStats& lhs, const GenerationStats& rhs) = default;
55
+ };
56
+
57
+ } // namespace margelo::nitro::litertlm
58
+
59
+ namespace margelo::nitro {
60
+
61
+ // C++ GenerationStats <> JS GenerationStats (object)
62
+ template <>
63
+ struct JSIConverter<margelo::nitro::litertlm::GenerationStats> final {
64
+ static inline margelo::nitro::litertlm::GenerationStats fromJSI(jsi::Runtime& runtime, const jsi::Value& arg) {
65
+ jsi::Object obj = arg.asObject(runtime);
66
+ return margelo::nitro::litertlm::GenerationStats(
67
+ JSIConverter<double>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "promptTokens"))),
68
+ JSIConverter<double>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "completionTokens"))),
69
+ JSIConverter<double>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "totalTokens"))),
70
+ JSIConverter<double>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "timeToFirstToken"))),
71
+ JSIConverter<double>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "totalTime"))),
72
+ JSIConverter<double>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "tokensPerSecond")))
73
+ );
74
+ }
75
+ static inline jsi::Value toJSI(jsi::Runtime& runtime, const margelo::nitro::litertlm::GenerationStats& arg) {
76
+ jsi::Object obj(runtime);
77
+ obj.setProperty(runtime, PropNameIDCache::get(runtime, "promptTokens"), JSIConverter<double>::toJSI(runtime, arg.promptTokens));
78
+ obj.setProperty(runtime, PropNameIDCache::get(runtime, "completionTokens"), JSIConverter<double>::toJSI(runtime, arg.completionTokens));
79
+ obj.setProperty(runtime, PropNameIDCache::get(runtime, "totalTokens"), JSIConverter<double>::toJSI(runtime, arg.totalTokens));
80
+ obj.setProperty(runtime, PropNameIDCache::get(runtime, "timeToFirstToken"), JSIConverter<double>::toJSI(runtime, arg.timeToFirstToken));
81
+ obj.setProperty(runtime, PropNameIDCache::get(runtime, "totalTime"), JSIConverter<double>::toJSI(runtime, arg.totalTime));
82
+ obj.setProperty(runtime, PropNameIDCache::get(runtime, "tokensPerSecond"), JSIConverter<double>::toJSI(runtime, arg.tokensPerSecond));
83
+ return obj;
84
+ }
85
+ static inline bool canConvert(jsi::Runtime& runtime, const jsi::Value& value) {
86
+ if (!value.isObject()) {
87
+ return false;
88
+ }
89
+ jsi::Object obj = value.getObject(runtime);
90
+ if (!nitro::isPlainObject(runtime, obj)) {
91
+ return false;
92
+ }
93
+ if (!JSIConverter<double>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "promptTokens")))) return false;
94
+ if (!JSIConverter<double>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "completionTokens")))) return false;
95
+ if (!JSIConverter<double>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "totalTokens")))) return false;
96
+ if (!JSIConverter<double>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "timeToFirstToken")))) return false;
97
+ if (!JSIConverter<double>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "totalTime")))) return false;
98
+ if (!JSIConverter<double>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "tokensPerSecond")))) return false;
99
+ return true;
100
+ }
101
+ };
102
+
103
+ } // namespace margelo::nitro
@@ -0,0 +1,41 @@
1
+ ///
2
+ /// HybridLiteRTLMSpec.cpp
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ #include "HybridLiteRTLMSpec.hpp"
9
+
10
+ namespace margelo::nitro::litertlm {
11
+
12
+ void HybridLiteRTLMSpec::loadHybridMethods() {
13
+ // load base methods/properties
14
+ HybridObject::loadHybridMethods();
15
+ // load custom methods/properties
16
+ registerHybrids(this, [](Prototype& prototype) {
17
+ prototype.registerHybridMethod("loadModel", &HybridLiteRTLMSpec::loadModel);
18
+ prototype.registerHybridMethod("sendMessage", &HybridLiteRTLMSpec::sendMessage);
19
+ prototype.registerHybridMethod("sendMessageWithImage", &HybridLiteRTLMSpec::sendMessageWithImage);
20
+ prototype.registerHybridMethod("sendMessageWithImageAsync", &HybridLiteRTLMSpec::sendMessageWithImageAsync);
21
+ prototype.registerHybridMethod("downloadModel", &HybridLiteRTLMSpec::downloadModel);
22
+ prototype.registerHybridMethod("deleteModel", &HybridLiteRTLMSpec::deleteModel);
23
+ prototype.registerHybridMethod("sendMessageWithAudio", &HybridLiteRTLMSpec::sendMessageWithAudio);
24
+ prototype.registerHybridMethod("sendMessageWithAudioAsync", &HybridLiteRTLMSpec::sendMessageWithAudioAsync);
25
+ prototype.registerHybridMethod("sendMultimodalMessage", &HybridLiteRTLMSpec::sendMultimodalMessage);
26
+ prototype.registerHybridMethod("sendMessageAsync", &HybridLiteRTLMSpec::sendMessageAsync);
27
+ prototype.registerHybridMethod("getHistory", &HybridLiteRTLMSpec::getHistory);
28
+ prototype.registerHybridMethod("resetConversation", &HybridLiteRTLMSpec::resetConversation);
29
+ prototype.registerHybridMethod("isReady", &HybridLiteRTLMSpec::isReady);
30
+ prototype.registerHybridMethod("getStats", &HybridLiteRTLMSpec::getStats);
31
+ prototype.registerHybridMethod("countTokens", &HybridLiteRTLMSpec::countTokens);
32
+ prototype.registerHybridMethod("getMemoryUsage", &HybridLiteRTLMSpec::getMemoryUsage);
33
+ prototype.registerHybridMethod("execute", &HybridLiteRTLMSpec::execute);
34
+ prototype.registerHybridMethod("sendToolResponse", &HybridLiteRTLMSpec::sendToolResponse);
35
+ prototype.registerHybridMethod("getActiveBackend", &HybridLiteRTLMSpec::getActiveBackend);
36
+ prototype.registerHybridMethod("stopGeneration", &HybridLiteRTLMSpec::stopGeneration);
37
+ prototype.registerHybridMethod("close", &HybridLiteRTLMSpec::close);
38
+ });
39
+ }
40
+
41
+ } // namespace margelo::nitro::litertlm
@@ -0,0 +1,109 @@
1
+ ///
2
+ /// HybridLiteRTLMSpec.hpp
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ #pragma once
9
+
10
+ #if __has_include(<NitroModules/HybridObject.hpp>)
11
+ #include <NitroModules/HybridObject.hpp>
12
+ #else
13
+ #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
14
+ #endif
15
+
16
+ // Forward declaration of `LLMConfig` to properly resolve imports.
17
+ namespace margelo::nitro::litertlm { struct LLMConfig; }
18
+ // Forward declaration of `ExecuteResult` to properly resolve imports.
19
+ namespace margelo::nitro::litertlm { struct ExecuteResult; }
20
+ // Forward declaration of `MultimodalPart` to properly resolve imports.
21
+ namespace margelo::nitro::litertlm { struct MultimodalPart; }
22
+ // Forward declaration of `Message` to properly resolve imports.
23
+ namespace margelo::nitro::litertlm { struct Message; }
24
+ // Forward declaration of `GenerationStats` to properly resolve imports.
25
+ namespace margelo::nitro::litertlm { struct GenerationStats; }
26
+ // Forward declaration of `MemoryUsage` to properly resolve imports.
27
+ namespace margelo::nitro::litertlm { struct MemoryUsage; }
28
+ // Forward declaration of `ToolResponse` to properly resolve imports.
29
+ namespace margelo::nitro::litertlm { struct ToolResponse; }
30
+ // Forward declaration of `Backend` to properly resolve imports.
31
+ namespace margelo::nitro::litertlm { enum class Backend; }
32
+
33
+ #include <NitroModules/Promise.hpp>
34
+ #include <string>
35
+ #include "LLMConfig.hpp"
36
+ #include <optional>
37
+ #include "ExecuteResult.hpp"
38
+ #include <functional>
39
+ #include "MultimodalPart.hpp"
40
+ #include <vector>
41
+ #include "Message.hpp"
42
+ #include "GenerationStats.hpp"
43
+ #include "MemoryUsage.hpp"
44
+ #include "ToolResponse.hpp"
45
+ #include "Backend.hpp"
46
+
47
+ namespace margelo::nitro::litertlm {
48
+
49
+ using namespace margelo::nitro;
50
+
51
+ /**
52
+ * An abstract base class for `LiteRTLM`
53
+ * Inherit this class to create instances of `HybridLiteRTLMSpec` in C++.
54
+ * You must explicitly call `HybridObject`'s constructor yourself, because it is virtual.
55
+ * @example
56
+ * ```cpp
57
+ * class HybridLiteRTLM: public HybridLiteRTLMSpec {
58
+ * public:
59
+ * HybridLiteRTLM(...): HybridObject(TAG) { ... }
60
+ * // ...
61
+ * };
62
+ * ```
63
+ */
64
+ class HybridLiteRTLMSpec: public virtual HybridObject {
65
+ public:
66
+ // Constructor
67
+ explicit HybridLiteRTLMSpec(): HybridObject(TAG) { }
68
+
69
+ // Destructor
70
+ ~HybridLiteRTLMSpec() override = default;
71
+
72
+ public:
73
+ // Properties
74
+
75
+
76
+ public:
77
+ // Methods
78
+ virtual std::shared_ptr<Promise<void>> loadModel(const std::string& modelPath, const std::optional<LLMConfig>& config) = 0;
79
+ virtual std::shared_ptr<Promise<ExecuteResult>> sendMessage(const std::string& message) = 0;
80
+ virtual std::shared_ptr<Promise<ExecuteResult>> sendMessageWithImage(const std::string& message, const std::string& imagePath) = 0;
81
+ virtual std::shared_ptr<Promise<void>> sendMessageWithImageAsync(const std::string& message, const std::string& imagePath, const std::function<void(const std::string& /* token */, bool /* done */)>& onToken) = 0;
82
+ virtual std::shared_ptr<Promise<std::string>> downloadModel(const std::string& url, const std::string& fileName, const std::optional<std::function<void(double /* progress */)>>& onProgress) = 0;
83
+ virtual std::shared_ptr<Promise<void>> deleteModel(const std::string& fileName) = 0;
84
+ virtual std::shared_ptr<Promise<ExecuteResult>> sendMessageWithAudio(const std::string& message, const std::string& audioPath) = 0;
85
+ virtual std::shared_ptr<Promise<void>> sendMessageWithAudioAsync(const std::string& message, const std::string& audioPath, const std::function<void(const std::string& /* token */, bool /* done */)>& onToken) = 0;
86
+ virtual std::shared_ptr<Promise<ExecuteResult>> sendMultimodalMessage(const std::vector<MultimodalPart>& parts) = 0;
87
+ virtual std::shared_ptr<Promise<void>> sendMessageAsync(const std::string& message, const std::function<void(const std::string& /* token */, bool /* done */)>& onToken) = 0;
88
+ virtual std::vector<Message> getHistory() = 0;
89
+ virtual void resetConversation() = 0;
90
+ virtual bool isReady() = 0;
91
+ virtual GenerationStats getStats() = 0;
92
+ virtual double countTokens(const std::string& text) = 0;
93
+ virtual MemoryUsage getMemoryUsage() = 0;
94
+ virtual std::shared_ptr<Promise<ExecuteResult>> execute(const std::vector<MultimodalPart>& parts, const std::optional<std::function<void(const std::string& /* token */, bool /* done */)>>& onToken) = 0;
95
+ virtual std::shared_ptr<Promise<ExecuteResult>> sendToolResponse(const std::vector<ToolResponse>& responses, const std::optional<std::function<void(const std::string& /* token */, bool /* done */)>>& onToken) = 0;
96
+ virtual Backend getActiveBackend() = 0;
97
+ virtual void stopGeneration() = 0;
98
+ virtual void close() = 0;
99
+
100
+ protected:
101
+ // Hybrid Setup
102
+ void loadHybridMethods() override;
103
+
104
+ protected:
105
+ // Tag for logging
106
+ static constexpr auto TAG = "LiteRTLM";
107
+ };
108
+
109
+ } // namespace margelo::nitro::litertlm
@@ -0,0 +1,25 @@
1
+ ///
2
+ /// HybridModelStoreSpec.cpp
3
+ /// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
4
+ /// https://github.com/mrousavy/nitro
5
+ /// Copyright © Marc Rousavy @ Margelo
6
+ ///
7
+
8
+ #include "HybridModelStoreSpec.hpp"
9
+
10
+ namespace margelo::nitro::litertlm {
11
+
12
+ void HybridModelStoreSpec::loadHybridMethods() {
13
+ // load base methods/properties
14
+ HybridObject::loadHybridMethods();
15
+ // load custom methods/properties
16
+ registerHybrids(this, [](Prototype& prototype) {
17
+ prototype.registerHybridMethod("isCached", &HybridModelStoreSpec::isCached);
18
+ prototype.registerHybridMethod("getFilePath", &HybridModelStoreSpec::getFilePath);
19
+ prototype.registerHybridMethod("listCachedFiles", &HybridModelStoreSpec::listCachedFiles);
20
+ prototype.registerHybridMethod("deleteFile", &HybridModelStoreSpec::deleteFile);
21
+ prototype.registerHybridMethod("downloadFile", &HybridModelStoreSpec::downloadFile);
22
+ });
23
+ }
24
+
25
+ } // namespace margelo::nitro::litertlm