node-llama-cpp 3.0.0-beta.4 → 3.0.0-beta.6

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 (229) hide show
  1. package/dist/{utils/getBin.d.ts → bindings/AddonTypes.d.ts} +2 -2
  2. package/dist/bindings/AddonTypes.js +2 -0
  3. package/dist/bindings/AddonTypes.js.map +1 -0
  4. package/dist/bindings/Llama.d.ts +21 -0
  5. package/dist/bindings/Llama.js +213 -0
  6. package/dist/bindings/Llama.js.map +1 -0
  7. package/dist/bindings/getLlama.d.ts +83 -0
  8. package/dist/bindings/getLlama.js +216 -0
  9. package/dist/bindings/getLlama.js.map +1 -0
  10. package/dist/bindings/types.d.ts +33 -0
  11. package/dist/bindings/types.js +30 -0
  12. package/dist/bindings/types.js.map +1 -0
  13. package/dist/bindings/utils/NoBinaryFoundError.d.ts +2 -0
  14. package/dist/bindings/utils/NoBinaryFoundError.js +7 -0
  15. package/dist/bindings/utils/NoBinaryFoundError.js.map +1 -0
  16. package/dist/{utils → bindings/utils}/binariesGithubRelease.js +1 -1
  17. package/dist/bindings/utils/binariesGithubRelease.js.map +1 -0
  18. package/dist/bindings/utils/clearAllLocalBuilds.d.ts +1 -0
  19. package/dist/bindings/utils/clearAllLocalBuilds.js +47 -0
  20. package/dist/bindings/utils/clearAllLocalBuilds.js.map +1 -0
  21. package/dist/bindings/utils/cloneLlamaCppRepo.d.ts +11 -0
  22. package/dist/bindings/utils/cloneLlamaCppRepo.js +151 -0
  23. package/dist/bindings/utils/cloneLlamaCppRepo.js.map +1 -0
  24. package/dist/bindings/utils/compileLLamaCpp.d.ts +12 -0
  25. package/dist/bindings/utils/compileLLamaCpp.js +157 -0
  26. package/dist/bindings/utils/compileLLamaCpp.js.map +1 -0
  27. package/dist/bindings/utils/getBuildFolderNameForBuildOptions.d.ts +5 -0
  28. package/dist/bindings/utils/getBuildFolderNameForBuildOptions.js +85 -0
  29. package/dist/bindings/utils/getBuildFolderNameForBuildOptions.js.map +1 -0
  30. package/dist/bindings/utils/getCanUsePrebuiltBinaries.d.ts +1 -0
  31. package/dist/bindings/utils/getCanUsePrebuiltBinaries.js +8 -0
  32. package/dist/bindings/utils/getCanUsePrebuiltBinaries.js.map +1 -0
  33. package/dist/bindings/utils/getExampleUsageCodeOfGetLlama.d.ts +2 -0
  34. package/dist/bindings/utils/getExampleUsageCodeOfGetLlama.js +21 -0
  35. package/dist/bindings/utils/getExampleUsageCodeOfGetLlama.js.map +1 -0
  36. package/dist/bindings/utils/getPlatform.d.ts +2 -0
  37. package/dist/bindings/utils/getPlatform.js +15 -0
  38. package/dist/bindings/utils/getPlatform.js.map +1 -0
  39. package/dist/bindings/utils/lastBuildInfo.d.ts +6 -0
  40. package/dist/bindings/utils/lastBuildInfo.js +17 -0
  41. package/dist/bindings/utils/lastBuildInfo.js.map +1 -0
  42. package/dist/bindings/utils/logBinaryUsageExampleToConsole.d.ts +2 -0
  43. package/dist/bindings/utils/logBinaryUsageExampleToConsole.js +28 -0
  44. package/dist/bindings/utils/logBinaryUsageExampleToConsole.js.map +1 -0
  45. package/dist/bindings/utils/resolveCustomCmakeOptions.d.ts +1 -0
  46. package/dist/bindings/utils/resolveCustomCmakeOptions.js +43 -0
  47. package/dist/bindings/utils/resolveCustomCmakeOptions.js.map +1 -0
  48. package/dist/chatWrappers/ChatMLChatWrapper.d.ts +5 -0
  49. package/dist/chatWrappers/ChatMLChatWrapper.js.map +1 -1
  50. package/dist/chatWrappers/resolveChatWrapperBasedOnModel.d.ts +1 -1
  51. package/dist/chatWrappers/resolveChatWrapperBasedOnModel.js.map +1 -1
  52. package/dist/cli/cli.js +2 -0
  53. package/dist/cli/cli.js.map +1 -1
  54. package/dist/cli/commands/BuildCommand.d.ts +2 -1
  55. package/dist/cli/commands/BuildCommand.js +43 -10
  56. package/dist/cli/commands/BuildCommand.js.map +1 -1
  57. package/dist/cli/commands/ChatCommand.d.ts +6 -2
  58. package/dist/cli/commands/ChatCommand.js +69 -15
  59. package/dist/cli/commands/ChatCommand.js.map +1 -1
  60. package/dist/cli/commands/ClearCommand.js +4 -6
  61. package/dist/cli/commands/ClearCommand.js.map +1 -1
  62. package/dist/cli/commands/DownloadCommand.d.ts +2 -1
  63. package/dist/cli/commands/DownloadCommand.js +47 -40
  64. package/dist/cli/commands/DownloadCommand.js.map +1 -1
  65. package/dist/cli/commands/OnPostInstallCommand.js +4 -10
  66. package/dist/cli/commands/OnPostInstallCommand.js.map +1 -1
  67. package/dist/config.d.ts +9 -3
  68. package/dist/config.js +14 -6
  69. package/dist/config.js.map +1 -1
  70. package/dist/{llamaEvaluator → evaluator}/LlamaChat/LlamaChat.js +5 -4
  71. package/dist/evaluator/LlamaChat/LlamaChat.js.map +1 -0
  72. package/dist/{llamaEvaluator → evaluator}/LlamaChat/utils/FunctionCallGrammar.d.ts +2 -1
  73. package/dist/{llamaEvaluator → evaluator}/LlamaChat/utils/FunctionCallGrammar.js +2 -1
  74. package/dist/evaluator/LlamaChat/utils/FunctionCallGrammar.js.map +1 -0
  75. package/dist/{llamaEvaluator → evaluator}/LlamaChat/utils/contextShiftStrategies/eraseFirstResponseAndKeepFirstSystemChatContextShiftStrategy.js +18 -0
  76. package/dist/evaluator/LlamaChat/utils/contextShiftStrategies/eraseFirstResponseAndKeepFirstSystemChatContextShiftStrategy.js.map +1 -0
  77. package/dist/{llamaEvaluator → evaluator}/LlamaChatSession/LlamaChatSession.js +2 -0
  78. package/dist/evaluator/LlamaChatSession/LlamaChatSession.js.map +1 -0
  79. package/dist/evaluator/LlamaChatSession/utils/defineChatSessionFunction.js.map +1 -0
  80. package/dist/{llamaEvaluator → evaluator}/LlamaContext/LlamaContext.js +5 -2
  81. package/dist/evaluator/LlamaContext/LlamaContext.js.map +1 -0
  82. package/dist/evaluator/LlamaContext/types.js.map +1 -0
  83. package/dist/evaluator/LlamaContext/utils/batchItemsPrioritizingStrategies/firstInFirstOutStrategy.js.map +1 -0
  84. package/dist/evaluator/LlamaContext/utils/batchItemsPrioritizingStrategies/maximumParallelismStrategy.js.map +1 -0
  85. package/dist/evaluator/LlamaContext/utils/resolveBatchItemsPrioritizingStrategy.js.map +1 -0
  86. package/dist/{llamaEvaluator → evaluator}/LlamaEmbeddingContext.d.ts +2 -0
  87. package/dist/{llamaEvaluator → evaluator}/LlamaEmbeddingContext.js +13 -8
  88. package/dist/evaluator/LlamaEmbeddingContext.js.map +1 -0
  89. package/dist/{llamaEvaluator → evaluator}/LlamaGrammar.d.ts +4 -2
  90. package/dist/{llamaEvaluator → evaluator}/LlamaGrammar.js +8 -7
  91. package/dist/evaluator/LlamaGrammar.js.map +1 -0
  92. package/dist/{llamaEvaluator → evaluator}/LlamaGrammarEvaluationState.js +4 -4
  93. package/dist/evaluator/LlamaGrammarEvaluationState.js.map +1 -0
  94. package/dist/{llamaEvaluator → evaluator}/LlamaJsonSchemaGrammar.d.ts +2 -1
  95. package/dist/{llamaEvaluator → evaluator}/LlamaJsonSchemaGrammar.js +2 -1
  96. package/dist/evaluator/LlamaJsonSchemaGrammar.js.map +1 -0
  97. package/dist/{llamaEvaluator → evaluator}/LlamaModel.d.ts +4 -3
  98. package/dist/{llamaEvaluator → evaluator}/LlamaModel.js +4 -6
  99. package/dist/evaluator/LlamaModel.js.map +1 -0
  100. package/dist/index.d.ts +16 -12
  101. package/dist/index.js +15 -11
  102. package/dist/index.js.map +1 -1
  103. package/dist/state.d.ts +2 -0
  104. package/dist/state.js +7 -0
  105. package/dist/state.js.map +1 -1
  106. package/dist/utils/cmake.js +16 -11
  107. package/dist/utils/cmake.js.map +1 -1
  108. package/dist/utils/getConsoleLogPrefix.d.ts +1 -0
  109. package/dist/utils/getConsoleLogPrefix.js +9 -0
  110. package/dist/utils/getConsoleLogPrefix.js.map +1 -0
  111. package/dist/utils/getGrammarsFolder.d.ts +2 -1
  112. package/dist/utils/getGrammarsFolder.js +8 -7
  113. package/dist/utils/getGrammarsFolder.js.map +1 -1
  114. package/dist/utils/getModuleVersion.d.ts +1 -0
  115. package/dist/utils/getModuleVersion.js +13 -0
  116. package/dist/utils/getModuleVersion.js.map +1 -0
  117. package/dist/utils/gitReleaseBundles.js +6 -5
  118. package/dist/utils/gitReleaseBundles.js.map +1 -1
  119. package/dist/utils/hashString.d.ts +1 -0
  120. package/dist/utils/hashString.js +8 -0
  121. package/dist/utils/hashString.js.map +1 -0
  122. package/dist/utils/isLockfileActive.d.ts +4 -0
  123. package/dist/utils/isLockfileActive.js +12 -0
  124. package/dist/utils/isLockfileActive.js.map +1 -0
  125. package/dist/utils/parseModelTypeDescription.d.ts +1 -1
  126. package/dist/utils/prettyPrintObject.d.ts +1 -0
  127. package/dist/utils/prettyPrintObject.js +38 -0
  128. package/dist/utils/prettyPrintObject.js.map +1 -0
  129. package/dist/utils/removeNullFields.d.ts +1 -0
  130. package/dist/utils/removeNullFields.js +8 -0
  131. package/dist/utils/removeNullFields.js.map +1 -1
  132. package/dist/utils/resolveChatWrapper.d.ts +1 -1
  133. package/dist/utils/resolveGithubRelease.d.ts +2 -0
  134. package/dist/utils/resolveGithubRelease.js +36 -0
  135. package/dist/utils/resolveGithubRelease.js.map +1 -0
  136. package/dist/utils/spawnCommand.d.ts +1 -1
  137. package/dist/utils/spawnCommand.js +4 -2
  138. package/dist/utils/spawnCommand.js.map +1 -1
  139. package/dist/utils/tokenizeInput.d.ts +3 -0
  140. package/dist/utils/tokenizeInput.js +9 -0
  141. package/dist/utils/tokenizeInput.js.map +1 -0
  142. package/dist/utils/waitForLockfileRelease.d.ts +5 -0
  143. package/dist/utils/waitForLockfileRelease.js +20 -0
  144. package/dist/utils/waitForLockfileRelease.js.map +1 -0
  145. package/dist/utils/withLockfile.d.ts +7 -0
  146. package/dist/utils/withLockfile.js +44 -0
  147. package/dist/utils/withLockfile.js.map +1 -0
  148. package/dist/utils/withStatusLogs.d.ts +2 -1
  149. package/dist/utils/withStatusLogs.js +11 -8
  150. package/dist/utils/withStatusLogs.js.map +1 -1
  151. package/llama/addon.cpp +102 -0
  152. package/llama/binariesGithubRelease.json +1 -1
  153. package/llama/gitRelease.bundle +0 -0
  154. package/llama/llama.cpp.info.json +4 -0
  155. package/llamaBins/linux-arm64/.buildMetadata.json +1 -0
  156. package/llamaBins/linux-arm64/llama-addon.node +0 -0
  157. package/llamaBins/linux-armv7l/.buildMetadata.json +1 -0
  158. package/llamaBins/linux-armv7l/llama-addon.node +0 -0
  159. package/llamaBins/linux-x64/.buildMetadata.json +1 -0
  160. package/llamaBins/linux-x64/llama-addon.node +0 -0
  161. package/llamaBins/linux-x64-cuda/.buildMetadata.json +1 -0
  162. package/llamaBins/linux-x64-cuda/llama-addon.node +0 -0
  163. package/llamaBins/mac-arm64/.buildMetadata.json +1 -0
  164. package/llamaBins/mac-arm64/llama-addon.node +0 -0
  165. package/llamaBins/mac-x64-metal/.buildMetadata.json +1 -0
  166. package/llamaBins/mac-x64-metal/ggml-metal.metal +6119 -0
  167. package/llamaBins/mac-x64-metal/llama-addon.node +0 -0
  168. package/llamaBins/win-x64/.buildMetadata.json +1 -0
  169. package/llamaBins/win-x64/llama-addon.exp +0 -0
  170. package/llamaBins/win-x64/llama-addon.node +0 -0
  171. package/llamaBins/win-x64-cuda/.buildMetadata.json +1 -0
  172. package/llamaBins/win-x64-cuda/llama-addon.exp +0 -0
  173. package/llamaBins/win-x64-cuda/llama-addon.lib +0 -0
  174. package/llamaBins/win-x64-cuda/llama-addon.node +0 -0
  175. package/package.json +10 -4
  176. package/dist/llamaEvaluator/LlamaBins.d.ts +0 -18
  177. package/dist/llamaEvaluator/LlamaBins.js +0 -5
  178. package/dist/llamaEvaluator/LlamaBins.js.map +0 -1
  179. package/dist/llamaEvaluator/LlamaChat/LlamaChat.js.map +0 -1
  180. package/dist/llamaEvaluator/LlamaChat/utils/FunctionCallGrammar.js.map +0 -1
  181. package/dist/llamaEvaluator/LlamaChat/utils/contextShiftStrategies/eraseFirstResponseAndKeepFirstSystemChatContextShiftStrategy.js.map +0 -1
  182. package/dist/llamaEvaluator/LlamaChatSession/LlamaChatSession.js.map +0 -1
  183. package/dist/llamaEvaluator/LlamaChatSession/utils/defineChatSessionFunction.js.map +0 -1
  184. package/dist/llamaEvaluator/LlamaContext/LlamaContext.js.map +0 -1
  185. package/dist/llamaEvaluator/LlamaContext/types.js.map +0 -1
  186. package/dist/llamaEvaluator/LlamaContext/utils/batchItemsPrioritizingStrategies/firstInFirstOutStrategy.js.map +0 -1
  187. package/dist/llamaEvaluator/LlamaContext/utils/batchItemsPrioritizingStrategies/maximumParallelismStrategy.js.map +0 -1
  188. package/dist/llamaEvaluator/LlamaContext/utils/resolveBatchItemsPrioritizingStrategy.js.map +0 -1
  189. package/dist/llamaEvaluator/LlamaEmbeddingContext.js.map +0 -1
  190. package/dist/llamaEvaluator/LlamaGrammar.js.map +0 -1
  191. package/dist/llamaEvaluator/LlamaGrammarEvaluationState.js.map +0 -1
  192. package/dist/llamaEvaluator/LlamaJsonSchemaGrammar.js.map +0 -1
  193. package/dist/llamaEvaluator/LlamaModel.js.map +0 -1
  194. package/dist/utils/binariesGithubRelease.js.map +0 -1
  195. package/dist/utils/clearLlamaBuild.d.ts +0 -1
  196. package/dist/utils/clearLlamaBuild.js +0 -12
  197. package/dist/utils/clearLlamaBuild.js.map +0 -1
  198. package/dist/utils/cloneLlamaCppRepo.d.ts +0 -2
  199. package/dist/utils/cloneLlamaCppRepo.js +0 -102
  200. package/dist/utils/cloneLlamaCppRepo.js.map +0 -1
  201. package/dist/utils/compileLLamaCpp.d.ts +0 -8
  202. package/dist/utils/compileLLamaCpp.js +0 -132
  203. package/dist/utils/compileLLamaCpp.js.map +0 -1
  204. package/dist/utils/getBin.js +0 -78
  205. package/dist/utils/getBin.js.map +0 -1
  206. package/dist/utils/getReleaseInfo.d.ts +0 -7
  207. package/dist/utils/getReleaseInfo.js +0 -30
  208. package/dist/utils/getReleaseInfo.js.map +0 -1
  209. package/dist/utils/usedBinFlag.d.ts +0 -6
  210. package/dist/utils/usedBinFlag.js +0 -15
  211. package/dist/utils/usedBinFlag.js.map +0 -1
  212. package/llama/usedBin.json +0 -3
  213. package/llamaBins/mac-x64/llama-addon.node +0 -0
  214. /package/dist/{utils → bindings/utils}/binariesGithubRelease.d.ts +0 -0
  215. /package/dist/{llamaEvaluator → evaluator}/LlamaChat/LlamaChat.d.ts +0 -0
  216. /package/dist/{llamaEvaluator → evaluator}/LlamaChat/utils/contextShiftStrategies/eraseFirstResponseAndKeepFirstSystemChatContextShiftStrategy.d.ts +0 -0
  217. /package/dist/{llamaEvaluator → evaluator}/LlamaChatSession/LlamaChatSession.d.ts +0 -0
  218. /package/dist/{llamaEvaluator → evaluator}/LlamaChatSession/utils/defineChatSessionFunction.d.ts +0 -0
  219. /package/dist/{llamaEvaluator → evaluator}/LlamaChatSession/utils/defineChatSessionFunction.js +0 -0
  220. /package/dist/{llamaEvaluator → evaluator}/LlamaContext/LlamaContext.d.ts +0 -0
  221. /package/dist/{llamaEvaluator → evaluator}/LlamaContext/types.d.ts +0 -0
  222. /package/dist/{llamaEvaluator → evaluator}/LlamaContext/types.js +0 -0
  223. /package/dist/{llamaEvaluator → evaluator}/LlamaContext/utils/batchItemsPrioritizingStrategies/firstInFirstOutStrategy.d.ts +0 -0
  224. /package/dist/{llamaEvaluator → evaluator}/LlamaContext/utils/batchItemsPrioritizingStrategies/firstInFirstOutStrategy.js +0 -0
  225. /package/dist/{llamaEvaluator → evaluator}/LlamaContext/utils/batchItemsPrioritizingStrategies/maximumParallelismStrategy.d.ts +0 -0
  226. /package/dist/{llamaEvaluator → evaluator}/LlamaContext/utils/batchItemsPrioritizingStrategies/maximumParallelismStrategy.js +0 -0
  227. /package/dist/{llamaEvaluator → evaluator}/LlamaContext/utils/resolveBatchItemsPrioritizingStrategy.d.ts +0 -0
  228. /package/dist/{llamaEvaluator → evaluator}/LlamaContext/utils/resolveBatchItemsPrioritizingStrategy.js +0 -0
  229. /package/dist/{llamaEvaluator → evaluator}/LlamaGrammarEvaluationState.d.ts +0 -0
@@ -8,7 +8,8 @@ type DownloadCommandArgs = {
8
8
  cuda?: boolean;
9
9
  skipBuild?: boolean;
10
10
  noBundle?: boolean;
11
+ noUsageExample?: boolean;
11
12
  };
12
13
  export declare const DownloadCommand: CommandModule<object, DownloadCommandArgs>;
13
- export declare function DownloadLlamaCppCommand({ repo, release, arch, nodeTarget, metal, cuda, skipBuild, noBundle, updateBinariesReleaseMetadataAndSaveGitBundle }: DownloadCommandArgs): Promise<void>;
14
+ export declare function DownloadLlamaCppCommand({ repo, release, arch, nodeTarget, metal, cuda, skipBuild, noBundle, noUsageExample, updateBinariesReleaseMetadataAndSaveGitBundle }: DownloadCommandArgs): Promise<void>;
14
15
  export {};
@@ -1,17 +1,20 @@
1
1
  import process from "process";
2
- import { Octokit } from "octokit";
3
2
  import fs from "fs-extra";
4
3
  import chalk from "chalk";
5
- import { defaultLlamaCppCudaSupport, defaultLlamaCppGitHubRepo, defaultLlamaCppMetalSupport, defaultLlamaCppRelease, isCI, llamaCppDirectory, llamaCppDirectoryTagFilePath } from "../../config.js";
6
- import { compileLlamaCpp } from "../../utils/compileLLamaCpp.js";
4
+ import { defaultLlamaCppCudaSupport, defaultLlamaCppGitHubRepo, defaultLlamaCppMetalSupport, defaultLlamaCppRelease, isCI, llamaCppDirectory, llamaCppDirectoryInfoFilePath } from "../../config.js";
5
+ import { compileLlamaCpp } from "../../bindings/utils/compileLLamaCpp.js";
7
6
  import withOra from "../../utils/withOra.js";
8
7
  import { clearTempFolder } from "../../utils/clearTempFolder.js";
9
- import { setBinariesGithubRelease } from "../../utils/binariesGithubRelease.js";
8
+ import { setBinariesGithubRelease } from "../../bindings/utils/binariesGithubRelease.js";
10
9
  import { downloadCmakeIfNeeded } from "../../utils/cmake.js";
11
10
  import withStatusLogs from "../../utils/withStatusLogs.js";
12
11
  import { getIsInDocumentationMode } from "../../state.js";
13
12
  import { getGitBundlePathForRelease, unshallowAndSquashCurrentRepoAndSaveItAsReleaseBundle } from "../../utils/gitReleaseBundles.js";
14
- import { cloneLlamaCppRepo } from "../../utils/cloneLlamaCppRepo.js";
13
+ import { cloneLlamaCppRepo } from "../../bindings/utils/cloneLlamaCppRepo.js";
14
+ import { getPlatform } from "../../bindings/utils/getPlatform.js";
15
+ import { resolveCustomCmakeOptions } from "../../bindings/utils/resolveCustomCmakeOptions.js";
16
+ import { logBinaryUsageExampleToConsole } from "../../bindings/utils/logBinaryUsageExampleToConsole.js";
17
+ import { resolveGithubRelease } from "../../utils/resolveGithubRelease.js";
15
18
  export const DownloadCommand = {
16
19
  command: "download",
17
20
  describe: "Download a release of llama.cpp and compile it",
@@ -60,6 +63,12 @@ export const DownloadCommand = {
60
63
  type: "boolean",
61
64
  default: false,
62
65
  description: "Download a llama.cpp release only from GitHub, even if a local git bundle exists for the release"
66
+ })
67
+ .option("noUsageExample", {
68
+ alias: "nu",
69
+ type: "boolean",
70
+ default: false,
71
+ description: "Don't print code usage example after building"
63
72
  })
64
73
  .option("updateBinariesReleaseMetadataAndSaveGitBundle", {
65
74
  type: "boolean",
@@ -70,14 +79,16 @@ export const DownloadCommand = {
70
79
  },
71
80
  handler: DownloadLlamaCppCommand
72
81
  };
73
- export async function DownloadLlamaCppCommand({ repo = defaultLlamaCppGitHubRepo, release = defaultLlamaCppRelease, arch = undefined, nodeTarget = undefined, metal = defaultLlamaCppMetalSupport, cuda = defaultLlamaCppCudaSupport, skipBuild = false, noBundle = false, updateBinariesReleaseMetadataAndSaveGitBundle = false }) {
82
+ // TODO: clear the builds folder
83
+ export async function DownloadLlamaCppCommand({ repo = defaultLlamaCppGitHubRepo, release = defaultLlamaCppRelease, arch = undefined, nodeTarget = undefined, metal = defaultLlamaCppMetalSupport, cuda = defaultLlamaCppCudaSupport, skipBuild = false, noBundle = false, noUsageExample = false, updateBinariesReleaseMetadataAndSaveGitBundle = false }) {
74
84
  const useBundle = noBundle != true;
75
- const octokit = new Octokit();
85
+ const platform = getPlatform();
86
+ const customCmakeOptions = resolveCustomCmakeOptions();
76
87
  const [githubOwner, githubRepo] = repo.split("/");
77
88
  console.log(`${chalk.yellow("Repo:")} ${repo}`);
78
89
  console.log(`${chalk.yellow("Release:")} ${release}`);
79
90
  if (!skipBuild) {
80
- if (metal && process.platform === "darwin") {
91
+ if (metal && platform === "mac") {
81
92
  console.log(`${chalk.yellow("Metal:")} enabled`);
82
93
  }
83
94
  if (cuda) {
@@ -94,32 +105,7 @@ export async function DownloadLlamaCppCommand({ repo = defaultLlamaCppGitHubRepo
94
105
  success: chalk.blue("Fetched llama.cpp info"),
95
106
  fail: chalk.blue("Failed to fetch llama.cpp info")
96
107
  }, async () => {
97
- let githubRelease = null;
98
- try {
99
- if (release === "latest") {
100
- githubRelease = await octokit.rest.repos.getLatestRelease({
101
- owner: githubOwner,
102
- repo: githubRepo
103
- });
104
- }
105
- else {
106
- githubRelease = await octokit.rest.repos.getReleaseByTag({
107
- owner: githubOwner,
108
- repo: githubRepo,
109
- tag: release
110
- });
111
- }
112
- }
113
- catch (err) {
114
- console.error("Failed to fetch llama.cpp release info", err);
115
- }
116
- if (githubRelease == null) {
117
- throw new Error(`Failed to find release "${release}" of "${repo}"`);
118
- }
119
- if (githubRelease.data.tag_name == null) {
120
- throw new Error(`Failed to find tag of release "${release}" of "${repo}"`);
121
- }
122
- githubReleaseTag = githubRelease.data.tag_name;
108
+ githubReleaseTag = await resolveGithubRelease(githubOwner, githubRepo, release);
123
109
  });
124
110
  await clearTempFolder();
125
111
  await withOra({
@@ -128,10 +114,26 @@ export async function DownloadLlamaCppCommand({ repo = defaultLlamaCppGitHubRepo
128
114
  fail: chalk.blue("Failed to remove existing llama.cpp directory")
129
115
  }, async () => {
130
116
  await fs.remove(llamaCppDirectory);
131
- await fs.remove(llamaCppDirectoryTagFilePath);
117
+ await fs.remove(llamaCppDirectoryInfoFilePath);
132
118
  });
133
119
  console.log(chalk.blue("Cloning llama.cpp"));
134
120
  await cloneLlamaCppRepo(githubOwner, githubRepo, githubReleaseTag, useBundle);
121
+ const buildOptions = {
122
+ customCmakeOptions,
123
+ progressLogs: true,
124
+ platform,
125
+ arch: arch
126
+ ? arch
127
+ : process.arch,
128
+ computeLayers: {
129
+ metal,
130
+ cuda
131
+ },
132
+ llamaCpp: {
133
+ repo,
134
+ release: githubReleaseTag
135
+ }
136
+ };
135
137
  if (!skipBuild) {
136
138
  await downloadCmakeIfNeeded(true);
137
139
  await withStatusLogs({
@@ -139,12 +141,12 @@ export async function DownloadLlamaCppCommand({ repo = defaultLlamaCppGitHubRepo
139
141
  success: chalk.blue("Compiled llama.cpp"),
140
142
  fail: chalk.blue("Failed to compile llama.cpp")
141
143
  }, async () => {
142
- await compileLlamaCpp({
143
- arch: arch ? arch : undefined,
144
+ await compileLlamaCpp(buildOptions, {
144
145
  nodeTarget: nodeTarget ? nodeTarget : undefined,
145
- setUsedBinFlag: true,
146
- metal,
147
- cuda
146
+ updateLastBuildInfo: true,
147
+ downloadCmakeIfNeeded: false,
148
+ ensureLlamaCppRepoIsCloned: false,
149
+ includeBuildOptionsInBinaryFolderName: true
148
150
  });
149
151
  });
150
152
  }
@@ -152,6 +154,11 @@ export async function DownloadLlamaCppCommand({ repo = defaultLlamaCppGitHubRepo
152
154
  await setBinariesGithubRelease(githubReleaseTag);
153
155
  await unshallowAndSquashCurrentRepoAndSaveItAsReleaseBundle();
154
156
  }
157
+ if (!noUsageExample) {
158
+ console.log();
159
+ console.log();
160
+ logBinaryUsageExampleToConsole(buildOptions, true);
161
+ }
155
162
  console.log();
156
163
  console.log();
157
164
  console.log(`${chalk.yellow("Repo:")} ${repo}`);
@@ -1 +1 @@
1
- {"version":3,"file":"DownloadCommand.js","sourceRoot":"","sources":["../../../src/cli/commands/DownloadCommand.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,SAAS,CAAC;AAE9B,OAAO,EAAC,OAAO,EAAC,MAAM,SAAS,CAAC;AAChC,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EACH,0BAA0B,EAAE,yBAAyB,EAAE,2BAA2B,EAAE,sBAAsB,EAAE,IAAI,EAChH,iBAAiB,EAAE,4BAA4B,EAClD,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAC,eAAe,EAAC,MAAM,gCAAgC,CAAC;AAC/D,OAAO,OAAO,MAAM,wBAAwB,CAAC;AAC7C,OAAO,EAAC,eAAe,EAAC,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAC,wBAAwB,EAAC,MAAM,sCAAsC,CAAC;AAC9E,OAAO,EAAC,qBAAqB,EAAC,MAAM,sBAAsB,CAAC;AAC3D,OAAO,cAAc,MAAM,+BAA+B,CAAC;AAC3D,OAAO,EAAC,wBAAwB,EAAC,MAAM,gBAAgB,CAAC;AACxD,OAAO,EACH,0BAA0B,EAC1B,qDAAqD,EACxD,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAC,iBAAiB,EAAC,MAAM,kCAAkC,CAAC;AAgBnE,MAAM,CAAC,MAAM,eAAe,GAA+C;IACvE,OAAO,EAAE,UAAU;IACnB,QAAQ,EAAE,gDAAgD;IAC1D,OAAO,CAAC,KAAK;QACT,MAAM,qBAAqB,GAAG,wBAAwB,EAAE,CAAC;QAEzD,OAAO,KAAK;aACP,MAAM,CAAC,MAAM,EAAE;YACZ,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,yBAAyB;YAClC,WAAW,EAAE,iIAAiI;SACjJ,CAAC;aACD,MAAM,CAAC,SAAS,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,sBAAsB;YAC3E,WAAW,EAAE,0KAA0K;SAC1L,CAAC;aACD,MAAM,CAAC,MAAM,EAAE;YACZ,KAAK,EAAE,GAAG;YACV,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,2CAA2C;SAC3D,CAAC;aACD,MAAM,CAAC,YAAY,EAAE;YAClB,KAAK,EAAE,GAAG;YACV,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,gEAAgE;SAChF,CAAC;aACD,MAAM,CAAC,OAAO,EAAE;YACb,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,2BAA2B,IAAI,qBAAqB;YAC7D,MAAM,EAAE,OAAO,CAAC,QAAQ,KAAK,QAAQ,IAAI,CAAC,qBAAqB;YAC/D,WAAW,EAAE,2KAA2K;SAC3L,CAAC;aACD,MAAM,CAAC,MAAM,EAAE;YACZ,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,0BAA0B;YACnC,WAAW,EAAE,uGAAuG;SACvH,CAAC;aACD,MAAM,CAAC,WAAW,EAAE;YACjB,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,8CAA8C;SAC9D,CAAC;aACD,MAAM,CAAC,UAAU,EAAE;YAChB,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,kGAAkG;SAClH,CAAC;aACD,MAAM,CAAC,+CAA+C,EAAE;YACrD,IAAI,EAAE,SAAS;YACf,MAAM,EAAE,IAAI;YACZ,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,8FAA8F;SAC9G,CAAC,CAAC;IACX,CAAC;IACD,OAAO,EAAE,uBAAuB;CACnC,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,EAC1C,IAAI,GAAG,yBAAyB,EAChC,OAAO,GAAG,sBAAsB,EAChC,IAAI,GAAG,SAAS,EAChB,UAAU,GAAG,SAAS,EACtB,KAAK,GAAG,2BAA2B,EACnC,IAAI,GAAG,0BAA0B,EACjC,SAAS,GAAG,KAAK,EACjB,QAAQ,GAAG,KAAK,EAChB,6CAA6C,GAAG,KAAK,EACnC;IAClB,MAAM,SAAS,GAAG,QAAQ,IAAI,IAAI,CAAC;IACnC,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;IAC9B,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAElD,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,OAAO,EAAE,CAAC,CAAC;IACtD,IAAI,CAAC,SAAS,EAAE;QACZ,IAAI,KAAK,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ,EAAE;YACxC,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;SACpD;QAED,IAAI,IAAI,EAAE;YACN,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;SACnD;KACJ;IACD,OAAO,CAAC,GAAG,EAAE,CAAC;IAKd,IAAI,gBAAgB,GAAkB,CAAC,SAAS,IAAI,CAAC,MAAM,0BAA0B,CAAC,WAAW,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,IAAI,IAAI,CAAC;QAC7H,CAAC,CAAC,OAAO;QACT,CAAC,CAAC,IAAI,CAAC;IAEX,IAAI,gBAAgB,IAAI,IAAI;QACxB,MAAM,OAAO,CAAC;YACV,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC;YAC9C,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC;YAC7C,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,gCAAgC,CAAC;SACrD,EAAE,KAAK,IAAI,EAAE;YACV,IAAI,aAAa,GAA6B,IAAI,CAAC;YAEnD,IAAI;gBACA,IAAI,OAAO,KAAK,QAAQ,EAAE;oBACtB,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC;wBACtD,KAAK,EAAE,WAAW;wBAClB,IAAI,EAAE,UAAU;qBACnB,CAAC,CAAC;iBACN;qBAAM;oBACH,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC;wBACrD,KAAK,EAAE,WAAW;wBAClB,IAAI,EAAE,UAAU;wBAChB,GAAG,EAAE,OAAO;qBACf,CAAC,CAAC;iBACN;aACJ;YAAC,OAAO,GAAG,EAAE;gBACV,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,GAAG,CAAC,CAAC;aAChE;YAED,IAAI,aAAa,IAAI,IAAI,EAAE;gBACvB,MAAM,IAAI,KAAK,CAAC,2BAA2B,OAAO,SAAS,IAAI,GAAG,CAAC,CAAC;aACvE;YAED,IAAI,aAAa,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,EAAE;gBACrC,MAAM,IAAI,KAAK,CAAC,kCAAkC,OAAO,SAAS,IAAI,GAAG,CAAC,CAAC;aAC9E;YAED,gBAAgB,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC;QACnD,CAAC,CAAC,CAAC;IAEP,MAAM,eAAe,EAAE,CAAC;IAExB,MAAM,OAAO,CAAC;QACV,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,uCAAuC,CAAC;QAC5D,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,sCAAsC,CAAC;QAC3D,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,+CAA+C,CAAC;KACpE,EAAE,KAAK,IAAI,EAAE;QACV,MAAM,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACnC,MAAM,EAAE,CAAC,MAAM,CAAC,4BAA4B,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC;IAC7C,MAAM,iBAAiB,CAAC,WAAW,EAAE,UAAU,EAAE,gBAAiB,EAAE,SAAS,CAAC,CAAC;IAE/E,IAAI,CAAC,SAAS,EAAE;QACZ,MAAM,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAElC,MAAM,cAAc,CAAC;YACjB,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,qBAAqB,CAAC;YAC1C,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC;YACzC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC;SAClD,EAAE,KAAK,IAAI,EAAE;YACV,MAAM,eAAe,CAAC;gBAClB,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;gBAC7B,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;gBAC/C,cAAc,EAAE,IAAI;gBACpB,KAAK;gBACL,IAAI;aACP,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;KACN;IAED,IAAI,IAAI,IAAI,6CAA6C,EAAE;QACvD,MAAM,wBAAwB,CAAC,gBAAiB,CAAC,CAAC;QAClD,MAAM,qDAAqD,EAAE,CAAC;KACjE;IAED,OAAO,CAAC,GAAG,EAAE,CAAC;IACd,OAAO,CAAC,GAAG,EAAE,CAAC;IACd,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,OAAO,EAAE,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,EAAE,CAAC;IACd,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;AACrC,CAAC"}
1
+ {"version":3,"file":"DownloadCommand.js","sourceRoot":"","sources":["../../../src/cli/commands/DownloadCommand.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,SAAS,CAAC;AAE9B,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EACH,0BAA0B,EAAE,yBAAyB,EAAE,2BAA2B,EAAE,sBAAsB,EAAE,IAAI,EAAE,iBAAiB,EACnI,6BAA6B,EAChC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAC,eAAe,EAAC,MAAM,yCAAyC,CAAC;AACxE,OAAO,OAAO,MAAM,wBAAwB,CAAC;AAC7C,OAAO,EAAC,eAAe,EAAC,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAC,wBAAwB,EAAC,MAAM,+CAA+C,CAAC;AACvF,OAAO,EAAC,qBAAqB,EAAC,MAAM,sBAAsB,CAAC;AAC3D,OAAO,cAAc,MAAM,+BAA+B,CAAC;AAC3D,OAAO,EAAC,wBAAwB,EAAC,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAC,0BAA0B,EAAE,qDAAqD,EAAC,MAAM,kCAAkC,CAAC;AACnI,OAAO,EAAC,iBAAiB,EAAC,MAAM,2CAA2C,CAAC;AAC5E,OAAO,EAAC,WAAW,EAAC,MAAM,qCAAqC,CAAC;AAChE,OAAO,EAAC,yBAAyB,EAAC,MAAM,mDAAmD,CAAC;AAC5F,OAAO,EAAC,8BAA8B,EAAC,MAAM,wDAAwD,CAAC;AACtG,OAAO,EAAC,oBAAoB,EAAC,MAAM,qCAAqC,CAAC;AAkBzE,MAAM,CAAC,MAAM,eAAe,GAA+C;IACvE,OAAO,EAAE,UAAU;IACnB,QAAQ,EAAE,gDAAgD;IAC1D,OAAO,CAAC,KAAK;QACT,MAAM,qBAAqB,GAAG,wBAAwB,EAAE,CAAC;QAEzD,OAAO,KAAK;aACP,MAAM,CAAC,MAAM,EAAE;YACZ,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,yBAAyB;YAClC,WAAW,EAAE,iIAAiI;SACjJ,CAAC;aACD,MAAM,CAAC,SAAS,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,sBAAsB;YAC3E,WAAW,EAAE,0KAA0K;SAC1L,CAAC;aACD,MAAM,CAAC,MAAM,EAAE;YACZ,KAAK,EAAE,GAAG;YACV,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,2CAA2C;SAC3D,CAAC;aACD,MAAM,CAAC,YAAY,EAAE;YAClB,KAAK,EAAE,GAAG;YACV,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,gEAAgE;SAChF,CAAC;aACD,MAAM,CAAC,OAAO,EAAE;YACb,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,2BAA2B,IAAI,qBAAqB;YAC7D,MAAM,EAAE,OAAO,CAAC,QAAQ,KAAK,QAAQ,IAAI,CAAC,qBAAqB;YAC/D,WAAW,EAAE,2KAA2K;SAC3L,CAAC;aACD,MAAM,CAAC,MAAM,EAAE;YACZ,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,0BAA0B;YACnC,WAAW,EAAE,uGAAuG;SACvH,CAAC;aACD,MAAM,CAAC,WAAW,EAAE;YACjB,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,8CAA8C;SAC9D,CAAC;aACD,MAAM,CAAC,UAAU,EAAE;YAChB,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,kGAAkG;SAClH,CAAC;aACD,MAAM,CAAC,gBAAgB,EAAE;YACtB,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,+CAA+C;SAC/D,CAAC;aACD,MAAM,CAAC,+CAA+C,EAAE;YACrD,IAAI,EAAE,SAAS;YACf,MAAM,EAAE,IAAI;YACZ,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,8FAA8F;SAC9G,CAAC,CAAC;IACX,CAAC;IACD,OAAO,EAAE,uBAAuB;CACnC,CAAC;AAEF,gCAAgC;AAChC,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,EAC1C,IAAI,GAAG,yBAAyB,EAChC,OAAO,GAAG,sBAAsB,EAChC,IAAI,GAAG,SAAS,EAChB,UAAU,GAAG,SAAS,EACtB,KAAK,GAAG,2BAA2B,EACnC,IAAI,GAAG,0BAA0B,EACjC,SAAS,GAAG,KAAK,EACjB,QAAQ,GAAG,KAAK,EAChB,cAAc,GAAG,KAAK,EACtB,6CAA6C,GAAG,KAAK,EACnC;IAClB,MAAM,SAAS,GAAG,QAAQ,IAAI,IAAI,CAAC;IACnC,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAC;IAC/B,MAAM,kBAAkB,GAAG,yBAAyB,EAAE,CAAC;IACvD,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAElD,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,OAAO,EAAE,CAAC,CAAC;IACtD,IAAI,CAAC,SAAS,EAAE;QACZ,IAAI,KAAK,IAAI,QAAQ,KAAK,KAAK,EAAE;YAC7B,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;SACpD;QAED,IAAI,IAAI,EAAE;YACN,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;SACnD;KACJ;IACD,OAAO,CAAC,GAAG,EAAE,CAAC;IAEd,IAAI,gBAAgB,GAAkB,CAAC,SAAS,IAAI,CAAC,MAAM,0BAA0B,CAAC,WAAW,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,IAAI,IAAI,CAAC;QAC7H,CAAC,CAAC,OAAO;QACT,CAAC,CAAC,IAAI,CAAC;IAEX,IAAI,gBAAgB,IAAI,IAAI;QACxB,MAAM,OAAO,CAAC;YACV,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC;YAC9C,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC;YAC7C,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,gCAAgC,CAAC;SACrD,EAAE,KAAK,IAAI,EAAE;YACV,gBAAgB,GAAG,MAAM,oBAAoB,CAAC,WAAW,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACpF,CAAC,CAAC,CAAC;IAEP,MAAM,eAAe,EAAE,CAAC;IAExB,MAAM,OAAO,CAAC;QACV,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,uCAAuC,CAAC;QAC5D,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,sCAAsC,CAAC;QAC3D,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,+CAA+C,CAAC;KACpE,EAAE,KAAK,IAAI,EAAE;QACV,MAAM,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACnC,MAAM,EAAE,CAAC,MAAM,CAAC,6BAA6B,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC;IAC7C,MAAM,iBAAiB,CAAC,WAAW,EAAE,UAAU,EAAE,gBAAiB,EAAE,SAAS,CAAC,CAAC;IAE/E,MAAM,YAAY,GAAiB;QAC/B,kBAAkB;QAClB,YAAY,EAAE,IAAI;QAClB,QAAQ;QACR,IAAI,EAAE,IAAI;YACN,CAAC,CAAC,IAA2B;YAC7B,CAAC,CAAC,OAAO,CAAC,IAAI;QAClB,aAAa,EAAE;YACX,KAAK;YACL,IAAI;SACP;QACD,QAAQ,EAAE;YACN,IAAI;YACJ,OAAO,EAAE,gBAAiB;SAC7B;KACJ,CAAC;IAEF,IAAI,CAAC,SAAS,EAAE;QACZ,MAAM,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAElC,MAAM,cAAc,CAAC;YACjB,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,qBAAqB,CAAC;YAC1C,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC;YACzC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC;SAClD,EAAE,KAAK,IAAI,EAAE;YACV,MAAM,eAAe,CAAC,YAAY,EAAE;gBAChC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;gBAC/C,mBAAmB,EAAE,IAAI;gBACzB,qBAAqB,EAAE,KAAK;gBAC5B,0BAA0B,EAAE,KAAK;gBACjC,qCAAqC,EAAE,IAAI;aAC9C,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;KACN;IAED,IAAI,IAAI,IAAI,6CAA6C,EAAE;QACvD,MAAM,wBAAwB,CAAC,gBAAiB,CAAC,CAAC;QAClD,MAAM,qDAAqD,EAAE,CAAC;KACjE;IAED,IAAI,CAAC,cAAc,EAAE;QACjB,OAAO,CAAC,GAAG,EAAE,CAAC;QACd,OAAO,CAAC,GAAG,EAAE,CAAC;QACd,8BAA8B,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;KACtD;IAED,OAAO,CAAC,GAAG,EAAE,CAAC;IACd,OAAO,CAAC,GAAG,EAAE,CAAC;IACd,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,OAAO,EAAE,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,EAAE,CAAC;IACd,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;AACrC,CAAC"}
@@ -1,20 +1,14 @@
1
- import { defaultLlamaCppCudaSupport, defaultLlamaCppGitHubRepo, defaultLlamaCppMetalSupport, defaultLlamaCppRelease, defaultSkipDownload } from "../../config.js";
2
- import { getPrebuildBinPath } from "../../utils/getBin.js";
3
- import { DownloadLlamaCppCommand } from "./DownloadCommand.js";
1
+ import { defaultSkipDownload } from "../../config.js";
2
+ import { getLlama } from "../../bindings/getLlama.js";
4
3
  export const OnPostInstallCommand = {
5
4
  command: "postinstall",
6
5
  describe: false,
7
6
  async handler() {
8
7
  if (defaultSkipDownload)
9
8
  return;
10
- if (await getPrebuildBinPath() != null)
11
- return;
12
9
  try {
13
- await DownloadLlamaCppCommand({
14
- repo: defaultLlamaCppGitHubRepo,
15
- release: defaultLlamaCppRelease,
16
- metal: defaultLlamaCppMetalSupport,
17
- cuda: defaultLlamaCppCudaSupport
10
+ await getLlama({
11
+ progressLogs: true
18
12
  });
19
13
  }
20
14
  catch (err) {
@@ -1 +1 @@
1
- {"version":3,"file":"OnPostInstallCommand.js","sourceRoot":"","sources":["../../../src/cli/commands/OnPostInstallCommand.ts"],"names":[],"mappings":"AACA,OAAO,EACH,0BAA0B,EAAE,yBAAyB,EAAE,2BAA2B,EAAE,sBAAsB,EAAE,mBAAmB,EAClI,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAC,kBAAkB,EAAC,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAC,uBAAuB,EAAC,MAAM,sBAAsB,CAAC;AAI7D,MAAM,CAAC,MAAM,oBAAoB,GAAgD;IAC7E,OAAO,EAAE,aAAa;IACtB,QAAQ,EAAE,KAAK;IACf,KAAK,CAAC,OAAO;QACT,IAAI,mBAAmB;YACnB,OAAO;QAEX,IAAI,MAAM,kBAAkB,EAAE,IAAI,IAAI;YAClC,OAAO;QAEX,IAAI;YACA,MAAM,uBAAuB,CAAC;gBAC1B,IAAI,EAAE,yBAAyB;gBAC/B,OAAO,EAAE,sBAAsB;gBAC/B,KAAK,EAAE,2BAA2B;gBAClC,IAAI,EAAE,0BAA0B;aACnC,CAAC,CAAC;SACN;QAAC,OAAO,GAAG,EAAE;YACV,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SACnB;IACL,CAAC;CACJ,CAAC"}
1
+ {"version":3,"file":"OnPostInstallCommand.js","sourceRoot":"","sources":["../../../src/cli/commands/OnPostInstallCommand.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,mBAAmB,EAAC,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAC,QAAQ,EAAC,MAAM,4BAA4B,CAAC;AAIpD,MAAM,CAAC,MAAM,oBAAoB,GAAgD;IAC7E,OAAO,EAAE,aAAa;IACtB,QAAQ,EAAE,KAAK;IACf,KAAK,CAAC,OAAO;QACT,IAAI,mBAAmB;YACnB,OAAO;QAEX,IAAI;YACA,MAAM,QAAQ,CAAC;gBACX,YAAY,EAAE,IAAI;aACrB,CAAC,CAAC;SACN;QAAC,OAAO,GAAG,EAAE;YACV,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SACnB;IACL,CAAC;CACJ,CAAC"}
package/dist/config.d.ts CHANGED
@@ -1,24 +1,30 @@
1
+ import { LlamaLogLevel } from "./bindings/types.js";
1
2
  export declare const llamaDirectory: string;
2
3
  export declare const llamaToolchainsDirectory: string;
3
- export declare const llamaBinsDirectory: string;
4
+ export declare const llamaPrebuiltBinsDirectory: string;
5
+ export declare const llamaLocalBuildBinsDirectory: string;
4
6
  export declare const llamaBinsGrammarsDirectory: string;
5
7
  export declare const llamaCppDirectory: string;
6
8
  export declare const llamaCppGrammarsDirectory: string;
7
9
  export declare const tempDownloadDirectory: string;
8
10
  export declare const chatCommandHistoryFilePath: string;
9
- export declare const usedBinFlagJsonPath: string;
11
+ export declare const lastBuildInfoJsonPath: string;
10
12
  export declare const binariesGithubReleasePath: string;
11
- export declare const llamaCppDirectoryTagFilePath: string;
13
+ export declare const llamaCppDirectoryInfoFilePath: string;
12
14
  export declare const currentReleaseGitBundlePath: string;
13
15
  export declare const xpackDirectory: string;
14
16
  export declare const localXpacksStoreDirectory: string;
15
17
  export declare const localXpacksCacheDirectory: string;
18
+ export declare const buildMetadataFileName = ".buildMetadata.json";
16
19
  export declare const xpmVersion = "^0.16.3";
20
+ export declare const builtinLlamaCppGitHubRepo = "ggerganov/llama.cpp";
21
+ export declare const builtinLlamaCppRelease: string;
17
22
  export declare const isCI: boolean;
18
23
  export declare const defaultLlamaCppGitHubRepo: string;
19
24
  export declare const defaultLlamaCppRelease: string;
20
25
  export declare const defaultLlamaCppMetalSupport: boolean;
21
26
  export declare const defaultLlamaCppCudaSupport: boolean;
27
+ export declare const defaultLlamaCppDebugLogs: LlamaLogLevel;
22
28
  export declare const defaultSkipDownload: boolean;
23
29
  export declare const defaultXpacksStoreDirectory: string;
24
30
  export declare const defaultXpacksCacheDirectory: string;
package/dist/config.js CHANGED
@@ -4,33 +4,38 @@ import * as os from "os";
4
4
  import process from "process";
5
5
  import envVar from "env-var";
6
6
  import * as uuid from "uuid";
7
- import { getBinariesGithubRelease } from "./utils/binariesGithubRelease.js";
7
+ import { getBinariesGithubRelease } from "./bindings/utils/binariesGithubRelease.js";
8
+ import { LlamaLogLevel, LlamaLogLevelValues } from "./bindings/types.js";
8
9
  const __dirname = path.dirname(fileURLToPath(import.meta.url));
9
10
  const env = envVar.from(process.env);
10
11
  export const llamaDirectory = path.join(__dirname, "..", "llama");
11
12
  export const llamaToolchainsDirectory = path.join(llamaDirectory, "toolchains");
12
- export const llamaBinsDirectory = path.join(__dirname, "..", "llamaBins");
13
+ export const llamaPrebuiltBinsDirectory = path.join(__dirname, "..", "llamaBins");
14
+ export const llamaLocalBuildBinsDirectory = path.join(llamaDirectory, "localBuilds");
13
15
  export const llamaBinsGrammarsDirectory = path.join(__dirname, "..", "llama", "grammars");
14
16
  export const llamaCppDirectory = path.join(llamaDirectory, "llama.cpp");
15
17
  export const llamaCppGrammarsDirectory = path.join(llamaDirectory, "llama.cpp", "grammars");
16
18
  export const tempDownloadDirectory = path.join(os.tmpdir(), "node-llama-cpp", uuid.v4());
17
19
  export const chatCommandHistoryFilePath = path.join(os.homedir(), ".node-llama-cpp.chat_repl_history");
18
- export const usedBinFlagJsonPath = path.join(llamaDirectory, "usedBin.json");
20
+ export const lastBuildInfoJsonPath = path.join(llamaDirectory, "lastBuild.json");
19
21
  export const binariesGithubReleasePath = path.join(llamaDirectory, "binariesGithubRelease.json");
20
- export const llamaCppDirectoryTagFilePath = path.join(llamaDirectory, "llama.cpp.tag.json");
22
+ export const llamaCppDirectoryInfoFilePath = path.join(llamaDirectory, "llama.cpp.info.json");
21
23
  export const currentReleaseGitBundlePath = path.join(llamaDirectory, "gitRelease.bundle");
22
24
  export const xpackDirectory = path.join(llamaDirectory, "xpack");
23
25
  export const localXpacksStoreDirectory = path.join(xpackDirectory, "store");
24
26
  export const localXpacksCacheDirectory = path.join(xpackDirectory, "cache");
27
+ export const buildMetadataFileName = ".buildMetadata.json";
25
28
  export const xpmVersion = "^0.16.3";
29
+ export const builtinLlamaCppGitHubRepo = "ggerganov/llama.cpp";
30
+ export const builtinLlamaCppRelease = await getBinariesGithubRelease();
26
31
  export const isCI = env.get("CI")
27
32
  .default("false")
28
33
  .asBool();
29
34
  export const defaultLlamaCppGitHubRepo = env.get("NODE_LLAMA_CPP_REPO")
30
- .default("ggerganov/llama.cpp")
35
+ .default(builtinLlamaCppGitHubRepo)
31
36
  .asString();
32
37
  export const defaultLlamaCppRelease = env.get("NODE_LLAMA_CPP_REPO_RELEASE")
33
- .default(await getBinariesGithubRelease())
38
+ .default(builtinLlamaCppRelease)
34
39
  .asString();
35
40
  export const defaultLlamaCppMetalSupport = env.get("NODE_LLAMA_CPP_METAL")
36
41
  .default((process.platform === "darwin" && process.arch !== "x64") ? "true" : "false")
@@ -38,6 +43,9 @@ export const defaultLlamaCppMetalSupport = env.get("NODE_LLAMA_CPP_METAL")
38
43
  export const defaultLlamaCppCudaSupport = env.get("NODE_LLAMA_CPP_CUDA")
39
44
  .default("false")
40
45
  .asBool();
46
+ export const defaultLlamaCppDebugLogs = env.get("NODE_LLAMA_CPP_LOG_LEVEL")
47
+ .default(LlamaLogLevel.debug)
48
+ .asEnum(LlamaLogLevelValues);
41
49
  export const defaultSkipDownload = env.get("NODE_LLAMA_CPP_SKIP_DOWNLOAD")
42
50
  .default("false")
43
51
  .asBool();
@@ -1 +1 @@
1
- {"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,aAAa,EAAC,MAAM,KAAK,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,MAAM,MAAM,SAAS,CAAC;AAC7B,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAC,wBAAwB,EAAC,MAAM,kCAAkC,CAAC;AAE1E,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAE/D,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;AAGrC,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AAClE,MAAM,CAAC,MAAM,wBAAwB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,YAAY,CAAC,CAAC;AAChF,MAAM,CAAC,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;AAC1E,MAAM,CAAC,MAAM,0BAA0B,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;AAC1F,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;AACxE,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC;AAC5F,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;AACzF,MAAM,CAAC,MAAM,0BAA0B,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,mCAAmC,CAAC,CAAC;AACvG,MAAM,CAAC,MAAM,mBAAmB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;AAC7E,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,4BAA4B,CAAC,CAAC;AACjG,MAAM,CAAC,MAAM,4BAA4B,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,oBAAoB,CAAC,CAAC;AAC5F,MAAM,CAAC,MAAM,2BAA2B,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;AAC1F,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;AACjE,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;AAC5E,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;AAC5E,MAAM,CAAC,MAAM,UAAU,GAAG,SAAS,CAAC;AAEpC,MAAM,CAAC,MAAM,IAAI,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC;KAC5B,OAAO,CAAC,OAAO,CAAC;KAChB,MAAM,EAAE,CAAC;AACd,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAG,CAAC,GAAG,CAAC,qBAAqB,CAAC;KAClE,OAAO,CAAC,qBAAqB,CAAC;KAC9B,QAAQ,EAAE,CAAC;AAChB,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAG,CAAC,GAAG,CAAC,6BAA6B,CAAC;KACvE,OAAO,CAAC,MAAM,wBAAwB,EAAE,CAAC;KACzC,QAAQ,EAAE,CAAC;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAG,CAAC,GAAG,CAAC,sBAAsB,CAAC;KACrE,OAAO,CAAC,CAAC,OAAO,CAAC,QAAQ,KAAK,QAAQ,IAAI,OAAO,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC;KACrF,MAAM,EAAE,CAAC;AACd,MAAM,CAAC,MAAM,0BAA0B,GAAG,GAAG,CAAC,GAAG,CAAC,qBAAqB,CAAC;KACnE,OAAO,CAAC,OAAO,CAAC;KAChB,MAAM,EAAE,CAAC;AACd,MAAM,CAAC,MAAM,mBAAmB,GAAG,GAAG,CAAC,GAAG,CAAC,8BAA8B,CAAC;KACrE,OAAO,CAAC,OAAO,CAAC;KAChB,MAAM,EAAE,CAAC;AACd,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAG,CAAC,GAAG,CAAC,oCAAoC,CAAC;KACnF,OAAO,CAAC,yBAAyB,CAAC;KAClC,QAAQ,EAAE,CAAC;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAG,CAAC,GAAG,CAAC,oCAAoC,CAAC;KACnF,OAAO,CAAC,yBAAyB,CAAC;KAClC,QAAQ,EAAE,CAAC;AAChB,MAAM,CAAC,MAAM,8BAA8B,GAAG,8BAA8B,CAAC;AAC7E,MAAM,CAAC,MAAM,uBAAuB,GAAG,+FAA+F;IAClI,+HAA+H;IAC/H,mFAAmF,CAAC;AACxF,MAAM,CAAC,MAAM,UAAU,GAAG,gBAAgB,CAAC;AAC3C,MAAM,CAAC,MAAM,YAAY,GAAG,WAAW,CAAC;AAExC,MAAM,gBAAgB,GAAG,4CAA4C,CAAC;AACtE,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACjC,IAAI,EAAE,gBAAgB,GAAG,aAAa;CAChC,CAAC"}
1
+ {"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,aAAa,EAAC,MAAM,KAAK,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,MAAM,MAAM,SAAS,CAAC;AAC7B,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAC,wBAAwB,EAAC,MAAM,2CAA2C,CAAC;AACnF,OAAO,EAAC,aAAa,EAAE,mBAAmB,EAAC,MAAM,qBAAqB,CAAC;AAEvE,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAE/D,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;AAGrC,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AAClE,MAAM,CAAC,MAAM,wBAAwB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,YAAY,CAAC,CAAC;AAChF,MAAM,CAAC,MAAM,0BAA0B,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;AAClF,MAAM,CAAC,MAAM,4BAA4B,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;AACrF,MAAM,CAAC,MAAM,0BAA0B,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;AAC1F,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;AACxE,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC;AAC5F,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;AACzF,MAAM,CAAC,MAAM,0BAA0B,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,mCAAmC,CAAC,CAAC;AACvG,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,gBAAgB,CAAC,CAAC;AACjF,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,4BAA4B,CAAC,CAAC;AACjG,MAAM,CAAC,MAAM,6BAA6B,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,qBAAqB,CAAC,CAAC;AAC9F,MAAM,CAAC,MAAM,2BAA2B,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;AAC1F,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;AACjE,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;AAC5E,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;AAC5E,MAAM,CAAC,MAAM,qBAAqB,GAAG,qBAAqB,CAAC;AAC3D,MAAM,CAAC,MAAM,UAAU,GAAG,SAAS,CAAC;AACpC,MAAM,CAAC,MAAM,yBAAyB,GAAG,qBAAqB,CAAC;AAC/D,MAAM,CAAC,MAAM,sBAAsB,GAAG,MAAM,wBAAwB,EAAE,CAAC;AAEvE,MAAM,CAAC,MAAM,IAAI,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC;KAC5B,OAAO,CAAC,OAAO,CAAC;KAChB,MAAM,EAAE,CAAC;AACd,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAG,CAAC,GAAG,CAAC,qBAAqB,CAAC;KAClE,OAAO,CAAC,yBAAyB,CAAC;KAClC,QAAQ,EAAE,CAAC;AAChB,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAG,CAAC,GAAG,CAAC,6BAA6B,CAAC;KACvE,OAAO,CAAC,sBAAsB,CAAC;KAC/B,QAAQ,EAAE,CAAC;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAG,CAAC,GAAG,CAAC,sBAAsB,CAAC;KACrE,OAAO,CAAC,CAAC,OAAO,CAAC,QAAQ,KAAK,QAAQ,IAAI,OAAO,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC;KACrF,MAAM,EAAE,CAAC;AACd,MAAM,CAAC,MAAM,0BAA0B,GAAG,GAAG,CAAC,GAAG,CAAC,qBAAqB,CAAC;KACnE,OAAO,CAAC,OAAO,CAAC;KAChB,MAAM,EAAE,CAAC;AACd,MAAM,CAAC,MAAM,wBAAwB,GAAG,GAAG,CAAC,GAAG,CAAC,0BAA0B,CAAC;KACtE,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC;KAC5B,MAAM,CAAC,mBAAmB,CAAC,CAAC;AACjC,MAAM,CAAC,MAAM,mBAAmB,GAAG,GAAG,CAAC,GAAG,CAAC,8BAA8B,CAAC;KACrE,OAAO,CAAC,OAAO,CAAC;KAChB,MAAM,EAAE,CAAC;AACd,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAG,CAAC,GAAG,CAAC,oCAAoC,CAAC;KACnF,OAAO,CAAC,yBAAyB,CAAC;KAClC,QAAQ,EAAE,CAAC;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAG,CAAC,GAAG,CAAC,oCAAoC,CAAC;KACnF,OAAO,CAAC,yBAAyB,CAAC;KAClC,QAAQ,EAAE,CAAC;AAChB,MAAM,CAAC,MAAM,8BAA8B,GAAG,8BAA8B,CAAC;AAC7E,MAAM,CAAC,MAAM,uBAAuB,GAAG,+FAA+F;IAClI,+HAA+H;IAC/H,mFAAmF,CAAC;AACxF,MAAM,CAAC,MAAM,UAAU,GAAG,gBAAgB,CAAC;AAC3C,MAAM,CAAC,MAAM,YAAY,GAAG,WAAW,CAAC;AAExC,MAAM,gBAAgB,GAAG,4CAA4C,CAAC;AACtE,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACjC,IAAI,EAAE,gBAAgB,GAAG,aAAa;CAChC,CAAC"}
@@ -99,7 +99,7 @@ export class LlamaChat {
99
99
  ? new LlamaGrammarEvaluationState({ grammar })
100
100
  : undefined;
101
101
  let functionsGrammar = functionsEnabled
102
- ? new FunctionCallGrammar(functions, this._chatWrapper, false)
102
+ ? new FunctionCallGrammar(model._llama, functions, this._chatWrapper, false)
103
103
  : undefined;
104
104
  let functionsEvaluationState = (functionsEnabled && functionsGrammar != null)
105
105
  ? new LlamaGrammarEvaluationState({
@@ -226,7 +226,7 @@ export class LlamaChat {
226
226
  .map((stopTrigger) => disengageInitiallyEngagedFunctionMode.addStopTrigger(stopTrigger));
227
227
  if (initiallyEngagedFunctionMode) {
228
228
  inFunctionEvaluationMode = true;
229
- functionsGrammar = new FunctionCallGrammar(functions, this._chatWrapper, true);
229
+ functionsGrammar = new FunctionCallGrammar(model._llama, functions, this._chatWrapper, true);
230
230
  functionsEvaluationState = new LlamaGrammarEvaluationState({
231
231
  grammar: functionsGrammar
232
232
  });
@@ -310,7 +310,7 @@ export class LlamaChat {
310
310
  }
311
311
  if (shouldStopFunctionEvaluationMode) {
312
312
  inFunctionEvaluationMode = false;
313
- functionsGrammar = new FunctionCallGrammar(functions, this._chatWrapper, false);
313
+ functionsGrammar = new FunctionCallGrammar(model._llama, functions, this._chatWrapper, false);
314
314
  functionsEvaluationState = new LlamaGrammarEvaluationState({
315
315
  grammar: functionsGrammar
316
316
  });
@@ -530,7 +530,8 @@ async function compressHistoryToFitContextSize({ history, contextShiftSize, cont
530
530
  lastShiftMetadata: contextShiftLastEvaluationMetadata
531
531
  });
532
532
  if (!checkIfHistoryFitsContext(chatHistory))
533
- throw new Error("The default context shift strategy did not return a history that fits the context size");
533
+ throw new Error("The default context shift strategy did not return a history that fits the context size. " +
534
+ "This may happen due to the system prompt being too long");
534
535
  return {
535
536
  compressedHistory: chatHistory,
536
537
  metadata
@@ -0,0 +1 @@
1
+ {"version":3,"file":"LlamaChat.js","sourceRoot":"","sources":["../../../src/evaluator/LlamaChat/LlamaChat.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,iBAAiB,EAAE,aAAa,EAAE,UAAU,EAAC,MAAM,iBAAiB,CAAC;AAE7E,OAAO,EAAC,kBAAkB,EAAC,MAAM,mCAAmC,CAAC;AAKrE,OAAO,EAAC,gBAAgB,EAAC,MAAM,iCAAiC,CAAC;AACjE,OAAO,EAAC,2BAA2B,EAAC,MAAM,mCAAmC,CAAC;AAC9E,OAAO,EAAC,UAAU,EAAC,MAAM,qBAAqB,CAAC;AAE/C,OAAO,EAAC,sBAAsB,EAAC,MAAM,uCAAuC,CAAC;AAC7E,OAAO,EAAyB,oBAAoB,EAAC,MAAM,qCAAqC,CAAC;AAEjG,OAAO,EACH,4DAA4D,EAC/D,MAAM,gGAAgG,CAAC;AACxG,OAAO,EAAC,mBAAmB,EAAE,gCAAgC,EAAC,MAAM,gCAAgC,CAAC;AAsHrG,MAAM,0BAA0B,GAA2C;IACvE,IAAI,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,GAAG,EAAE,CAAC,CAAC;IAC9E,QAAQ,EAAE,sCAAsC;IAChD,sBAAsB,EAAE,IAAI;CAC/B,CAAC;AAuCF,MAAM,oBAAoB,GAAG,QAAQ,CAAC;AAGtC,MAAM,OAAO,SAAS;IAClB,gBAAgB,CAAkB,YAAY,CAAc;IAC5D,gBAAgB,CAAkB,kBAAkB,GAAG,IAAI,iBAAiB,EAAE,CAAC;IAC/E,gBAAgB,CAAkB,oBAAoB,CAAU;IAChE,gBAAgB,CAAS,SAAS,CAA8B;IAChD,SAAS,GAAG,IAAI,UAAU,EAAQ,CAAC;IAEnD,YAAmB,EACf,eAAe,EACf,WAAW,GAAG,MAAM,EACpB,mBAAmB,GAAG,IAAI,EACX;QACf,IAAI,eAAe,IAAI,IAAI;YACvB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QAEtD,IAAI,eAAe,CAAC,QAAQ;YACxB,MAAM,IAAI,aAAa,EAAE,CAAC;QAE9B,IAAI,CAAC,SAAS,GAAG,eAAe,CAAC;QACjC,IAAI,CAAC,oBAAoB,GAAG,mBAAmB,CAAC;QAEhD,IAAI,CAAC,kBAAkB,CAAC,GAAG,CACvB,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,cAAc,CAAC,GAAG,EAAE;YACzC,IAAI,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC,CACL,CAAC;QACF,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QAE1D,IAAI,CAAC,YAAY,GAAG,kBAAkB,CAAC,WAAW,EAAE,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/E,CAAC;IAEM,OAAO,CAAC,EAAC,eAAe,GAAG,IAAI,CAAC,oBAAoB,KAAiC,EAAE;QAC1F,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI;YACtB,OAAO;QAEX,IAAI,eAAe;YACf,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QAE7B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtB,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,CAAC;IACtC,CAAC;IAED,cAAc;IACP,CAAC,MAAM,CAAC,OAAO,CAAC;QACnB,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;IAED,IAAW,QAAQ;QACf,OAAO,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC;IAClC,CAAC;IAED,IAAW,WAAW;QAClB,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI;YACtB,MAAM,IAAI,aAAa,EAAE,CAAC;QAE9B,OAAO,IAAI,CAAC,YAAY,CAAC;IAC7B,CAAC;IAED,IAAW,QAAQ;QACf,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI;YACtB,MAAM,IAAI,aAAa,EAAE,CAAC;QAE9B,OAAO,IAAI,CAAC,SAAS,CAAC;IAC1B,CAAC;IAED,IAAW,OAAO;QACd,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;IACjC,CAAC;IAED,IAAW,KAAK;QACZ,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;IAC/B,CAAC;IAEM,KAAK,CAAC,gBAAgB,CACzB,OAA0B,EAC1B,EACI,OAAO,EACP,MAAM,EACN,SAAS,EACT,WAAW,EACX,IAAI,EACJ,IAAI,EACJ,OAAO,EACP,oBAAoB,GAAG,KAAK,EAC5B,aAAa,GAAG,EAAE,EAClB,kBAAkB,GAAG,CAAC,EACtB,SAAS,EACT,sBAAsB,EACtB,YAAY,GAAG,0BAA0B,EACzC,2BAA2B,EAAE,EACzB,OAAO,EAAE,kCAAkC,EAC3C,6CAA6C,GAAG,GAAG,EACtD,GAAG,EAAE,KACuC,EAAE;QAEnD,MAAM,gBAAgB,GAAG,CAAC,SAAS,IAAI,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAElF,IAAI,OAAO,IAAI,IAAI,IAAI,gBAAgB;YACnC,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;QAE7E,IAAI,MAAM,EAAE,OAAO;YACf,MAAM,IAAI,UAAU,EAAE,CAAC;QAE3B,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI;YACtB,MAAM,IAAI,aAAa,EAAE,CAAC;QAE9B,IAAI,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB;YACjD,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE;YACjB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QAE5C,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,IAAI,eAAe,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO;YAC5F,eAAe,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,OAAO;gBACb,QAAQ,EAAE,EAAE;aACf,CAAC,CAAC;QAEP,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;QACvC,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC;QAClC,MAAM,oBAAoB,GAAG;YACzB,GAAG,0BAA0B;YAC7B,GAAG,gBAAgB,CAAC,YAAY,CAAC;SACpC,CAAC;QACF,MAAM,EACF,UAAU,EAAE,uBAAuB,GAAG,EAAE,EACxC,kBAAkB,EAClB,eAAe,EACf,OAAO,EACP,gBAAgB,EAChB,eAAe,EAClB,GAA2B,aAAa,KAAK,KAAK;YAC/C,CAAC,CAAC,EAAC,UAAU,EAAE,CAAC,EAAC;YACjB,CAAC,CAAC,aAAa,CAAC;QACpB,MAAM,iBAAiB,GAAG,uCAAuC,CAAC,eAAe,CAAC,CAAC;QAEnF,MAAM,GAAG,GAAY,EAAE,CAAC;QACxB,MAAM,aAAa,GAAY,EAAE,CAAC;QAClC,IAAI,sBAAsB,GAAY,EAAE,CAAC;QACzC,MAAM,kBAAkB,GAAY,EAAE,CAAC;QACvC,MAAM,oBAAoB,GAAG,uBAAuB,GAAG,CAAC,CAAC;QACzD,MAAM,sBAAsB,GAAG,OAAO,IAAI,IAAI;YAC1C,CAAC,CAAC,IAAI,2BAA2B,CAAC,EAAC,OAAO,EAAC,CAAC;YAC5C,CAAC,CAAC,SAAS,CAAC;QAChB,IAAI,gBAAgB,GAAG,gBAAgB;YACnC,CAAC,CAAC,IAAI,mBAAmB,CAAC,KAAK,CAAC,MAAM,EAAE,SAAmC,EAAE,IAAI,CAAC,YAAY,EAAE,KAAK,CAAC;YACtG,CAAC,CAAC,SAAS,CAAC;QAChB,IAAI,wBAAwB,GAAG,CAAC,gBAAgB,IAAI,gBAAgB,IAAI,IAAI,CAAC;YACzE,CAAC,CAAC,IAAI,2BAA2B,CAAC;gBAC9B,OAAO,EAAE,gBAAgB;aAC5B,CAAC;YACF,CAAC,CAAC,SAAS,CAAC;QAChB,MAAM,eAAe,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACnD,MAAM,sBAAsB,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAC5D,MAAM,2BAA2B,GAAG,IAAI,sBAAsB,EAAE,CAAC;QACjE,MAAM,yBAAyB,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAC/D,MAAM,qCAAqC,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAC3E,MAAM,uBAAuB,GAAG,IAAI,sBAAsB,EAAE,CAAC;QAC7D,MAAM,+BAA+B,GAA6B,EAAE,CAAC;QACrE,MAAM,4BAA4B,GAA6B,EAAE,CAAC;QAElE,IAAI,eAAe,GAAG,CAAC,CAAC;QACxB,IAAI,iBAAiB,GAAG,IAAI,CAAC;QAC7B,IAAI,wBAAwB,GAAG,KAAK,CAAC;QACrC,IAAI,4BAA4B,GAAG,KAAK,CAAC;QACzC,IAAI,wBAAwB,GAAsB,eAAe,CAAC;QAClE,IAAI,8BAA8B,GAA8B,oBAAoB,CAAC,sBAAsB,CAAC;QAE5G,MAAM,gBAAgB,GAAG,GAAG,EAAE;YAC1B,IAAI,MAAM,EAAE,OAAO;gBACf,MAAM,IAAI,UAAU,EAAE,CAAC;YAE3B,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI;gBACtB,MAAM,IAAI,aAAa,EAAE,CAAC;QAClC,CAAC,CAAC;QAEF,MAAM,gBAAgB,GAAG,GAAG,EAAE;YAC1B,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI;gBACtB,MAAM,IAAI,aAAa,EAAE,CAAC;YAE9B,IAAI,YAAY,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,uBAAuB,CAAC,CAAC;YAEvD,IAAI,kBAAkB,IAAI,IAAI;gBAC1B,YAAY,GAAG,kBAAkB,CAAC,YAAY,CAAC,CAAC;YAEpD,IAAI,CAAC,eAAe,EAAE;gBAClB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;gBAEhC,IAAI,OAAO,IAAI,IAAI;oBACf,YAAY,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,KAAK,OAAO,CAAC,CAAC;aACtE;YAED,OAAO,YAAY,CAAC;QACxB,CAAC,CAAC;QAEF,MAAM,0CAA0C,GAAG,GAAG,EAAE;YACpD,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;gBAChB,OAAO,eAAe,CAAC;YAE3B,IAAI,aAAa,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAE1C,IAAI,OAAO,EAAE,oBAAoB,IAAI,oBAAoB;gBACrD,aAAa,GAAG,aAAa,CAAC,OAAO,EAAE,CAAC;YAE5C,IAAI,aAAa,KAAK,EAAE;gBACpB,OAAO,eAAe,CAAC;YAE3B,OAAO,qCAAqC,CACxC,eAAe,EACf,iBAAiB,GAAG,aAAa,CACpC,CAAC;QACN,CAAC,CAAC;QAEF,MAAM,4CAA4C,GAAG,GAAG,EAAE;YACtD,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC9C,uBAAuB,CAAC,oBAAoB,EAAE,CAAC;gBAC/C,uBAAuB,CAAC,mBAAmB,EAAE,CAAC;gBAE9C,IAAI,4BAA4B,GAAwE,IAAI,CAAC;gBAE7G,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;oBAC3C,uBAAuB,CAAC,gBAAgB,CAAC;wBACrC,IAAI,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;wBAC1C,MAAM,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;wBAC1B,cAAc,EAAE,CAAC,KAAK,CAAC;qBAC1B,CAAC,CAAC;oBAEH,IAAI,uBAAuB,CAAC,iBAAiB,EAAE;wBAC3C,4BAA4B,GAAG,uBAAuB,CAAC,iBAAiB,EAAE,CAAC;wBAC3E,uBAAuB,CAAC,mBAAmB,EAAE,CAAC;qBACjD;yBAAM,IAAI,CAAC,uBAAuB,CAAC,kBAAkB;wBAClD,MAAM;iBACb;gBAED,IAAI,4BAA4B,IAAI,IAAI,EAAE;oBACtC,MAAM,CAAC,2BAA2B,CAAC,GAAG,4BAA4B,CAAC;oBAEnE,IAAI,2BAA2B,IAAI,IAAI,EAAE;wBACrC,sBAAsB,GAAG,2BAA2B,CAAC,WAAW;6BAC3D,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE;4BACjB,IAAI,OAAO,WAAW,KAAK,QAAQ;gCAC/B,OAAO,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;;gCAEnC,OAAO,CAAC,WAAW,CAAC,CAAC;wBAC7B,CAAC,CAAC;6BACD,IAAI,CAAC,CAAC,CAAC,CAAC;wBAEb,MAAM,gBAAgB,GAAG,2BAA2B,CAAC,oBAAoB;6BACpE,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;4BAChB,IAAI,OAAO,UAAU,KAAK,QAAQ;gCAC9B,OAAO,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;;gCAElC,OAAO,UAAU,CAAC;wBAC1B,CAAC,CAAC;6BACD,IAAI,CAAC,CAAC,CAAC,CAAC;wBACb,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;wBACzB,aAAa,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,CAAC;qBAC3C;iBACJ;aACJ;QACL,CAAC,CAAC;QAEF,IAAI,OAAO,IAAI,IAAI;YACf,sBAAsB,CAAC,mBAAmB,CAAC,OAAO,CAAC,sBAAsB,EAAE,KAAK,CAAC,QAAQ,CAAC;iBACrF,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,sBAAsB,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC;QAElF,IAAI,SAAS,IAAI,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC;YACtD,2BAA2B,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QAEnG,iDAAiD;QACjD,OAAO,IAAI,EAAE;YACT,gBAAgB,EAAE,CAAC;YAEnB,IAAI,kBAAkB,GAAG,KAAK,CAAC;YAC/B,MAAM,iBAAiB,GAAG,eAAe,CAAC,uBAAuB,EAAE,CAAC;YACpE,MAAM,EACF,OAAO,EAAE,oBAAoB,EAC7B,sBAAsB,EACtB,MAAM,EAAE,mBAAmB,EAC3B,kBAAkB,EAClB,6BAA6B,EAC7B,eAAe,EACf,4BAA4B,EAC5B,qCAAqC,EACxC,GAAG,MAAM,gBAAgB,CAAC;gBACvB,eAAe,EAAE,0CAA0C,EAAE;gBAC7D,oBAAoB;gBACpB,8BAA8B;gBAC9B,kBAAkB,EAAE,aAAa,CAAC,MAAM,GAAG,iBAAiB,CAAC,MAAM;gBACnE,iBAAiB;gBACjB,WAAW,EAAE,IAAI,CAAC,YAAY;gBAC9B,kCAAkC;gBAClC,6CAA6C;gBAC7C,QAAQ,EAAE,IAAI,CAAC,SAAS;gBACxB,oBAAoB,EAAE,CAAC;gBACvB,SAAS,EAAE,gBAAgB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACnD,sBAAsB;aACzB,CAAC,CAAC;YACH,gBAAgB,EAAE,CAAC;YAEnB,IAAI,eAAe,KAAK,CAAC,EAAE;gBACvB,sBAAsB,CAAC,mBAAmB,CAAC,eAAe,EAAE,KAAK,CAAC,QAAQ,CAAC;qBACtE,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,uBAAuB,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC;gBAE/E,IAAI,gBAAgB,EAAE;oBAClB,4BAA4B,GAAG,4BAA4B,CAAC;oBAC5D,sBAAsB,CAAC,mBAAmB,CAAC,qCAAqC,EAAE,KAAK,CAAC,QAAQ,CAAC;yBAC5F,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,qCAAqC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC;oBAE7F,IAAI,4BAA4B,EAAE;wBAC9B,wBAAwB,GAAG,IAAI,CAAC;wBAChC,gBAAgB,GAAG,IAAI,mBAAmB,CACtC,KAAK,CAAC,MAAM,EACZ,SAAmC,EACnC,IAAI,CAAC,YAAY,EACjB,IAAI,CACP,CAAC;wBACF,wBAAwB,GAAG,IAAI,2BAA2B,CAAC;4BACvD,OAAO,EAAE,gBAAgB;yBAC5B,CAAC,CAAC;qBACN;iBACJ;aACJ;YAED,MAAM,MAAM,GAAG,CAAC,GAAG,mBAAmB,EAAE,GAAG,sBAAsB,EAAE,GAAG,aAAa,EAAE,GAAG,iBAAiB,CAAC,CAAC;YAC3G,eAAe,GAAG,kBAAkB,CAAC;YACrC,8BAA8B,GAAG,6BAA6B,CAAC;YAC/D,wBAAwB,GAAG,oBAAoB,CAAC;YAChD,MAAM,8BAA8B,GAAG,uCAAuC,CAAC,oBAAoB,CAAC,CAAC;YACrG,MAAM,iBAAiB,GAAY,EAAE,CAAC;YAEtC,sBAAsB,CAAC,mBAAmB,CAAC,sBAAsB,EAAE,KAAK,CAAC,QAAQ,CAAC;iBAC7E,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,sBAAsB,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC;YAE9E,IAAI,gBAAgB,IAAI,IAAI;gBACxB,sBAAsB,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,sBAAsB,EAAE,KAAK,CAAC,QAAQ,CAAC;qBAC9F,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,yBAAyB,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC;YAErF,IAAI,EAAC,mBAAmB,EAAC,GAAG,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAExE,8DAA8D;YAC9D,IAAI,mBAAmB,KAAK,MAAM,CAAC,MAAM,IAAI,mBAAmB,GAAG,CAAC;gBAChE,mBAAmB,IAAI,CAAC,CAAC;YAE7B,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,mBAAmB,CAAC,CAAC;YAEtC,IAAI,mBAAmB,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE;gBACrD,MAAM,IAAI,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;wBAC1C,KAAK,EAAE,mBAAmB;wBAC1B,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc;qBACrC,CAAC,CAAC,CAAC;gBACJ,gBAAgB,EAAE,CAAC;aACtB;YAGD,MAAM,kBAAkB,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC;gBACxE,WAAW,EAAE,IAAI,EAAE,IAAI;gBACvB,sBAAsB,EAAE,GAAG,EAAE;oBACzB,IAAI,wBAAwB;wBACxB,OAAO,wBAAwB,CAAC;oBAEpC,OAAO,sBAAsB,CAAC;gBAClC,CAAC;gBACD,aAAa,EAAE,CAAC,oBAAoB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;oBAC/C,YAAY,EAAE,gBAAgB;oBAC9B,OAAO;oBACP,gBAAgB;oBAChB,eAAe;iBAClB;gBACD,kBAAkB;gBAClB,aAAa,EAAE,IAAI;aACtB,CAAC,CAAC,CAAC;YAEJ,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,kBAAkB,EAAE;gBAC1C,gBAAgB,EAAE,CAAC;gBACnB,eAAe,EAAE,CAAC;gBAElB,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC,CAAC;gBACvB,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;gBACvC,MAAM,kBAAkB,GAAG,eAAe,CAAC,QAAQ,CAAC,EAAC,MAAM,EAAE,IAAI,EAAC,CAAC,CAAC;gBAEpE,IAAI,4BAA4B;oBAC5B,qCAAqC,CAAC,gBAAgB,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,cAAc,EAAE,eAAe,KAAK,CAAC,EAAC,CAAC,CAAC;gBAElH,IAAI,IAAI,KAAK,oBAAoB,IAAI,CACjC,CAAC,OAAO,EAAE,oBAAoB,IAAI,oBAAoB,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAChF,EAAE;oBACC,+BAA+B,CAAC,IAAI,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;iBACnF;qBAAM;oBACH,OAAO,+BAA+B,CAAC,MAAM,GAAG,CAAC;wBAC7C,+BAA+B,CAAC,KAAK,EAAG,CAAC,OAAO,EAAE,CAAC;iBAC1D;gBAED,2BAA2B,CAAC,gBAAgB,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,kBAAkB,EAAC,CAAC,CAAC;gBAEjF,IAAI,4BAA4B,IAAI,qCAAqC,CAAC,iBAAiB,EAAE;oBACzF,4BAA4B,GAAG,KAAK,CAAC;oBAErC,IAAI,gCAAgC,GAAG,CAAC,2BAA2B,CAAC,iBAAiB,CAAC;oBAEtF,IAAI,CAAC,gCAAgC,IAAI,gBAAgB,IAAI,gBAAgB,IAAI,IAAI,EAAE;wBACnF,MAAM,gBAAgB,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC,GAAG,kBAAkB,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC;wBAE9E,IAAI;4BACA,MAAM,YAAY,GAAG,gBAAgB,CAAC,gCAAgC,CAAC,gBAAgB,EAAE;gCACrF,8BAA8B,EAAE,IAAI;gCACpC,0BAA0B,EAAE,IAAI;6BACnC,CAAC,CAAC;4BAEH,MAAM,wBAAwB,GAC1B,IAAI,CAAC,YAAY,CAAC,2BAA2B,CAAC,EAAC,0BAA0B,EAAE,IAAI,EAAC,CAAC,CAAC;4BACtF,IAAI,wBAAwB,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE;gCAChD,gCAAgC,GAAG,IAAI,CAAC;6BAC3C;yBACJ;wBAAC,OAAO,GAAG,EAAE;4BACV,IAAI,CAAC,CAAC,GAAG,YAAY,gCAAgC,CAAC;gCAClD,MAAM,GAAG,CAAC;yBACjB;qBACJ;oBAED,IAAI,gCAAgC,EAAE;wBAClC,wBAAwB,GAAG,KAAK,CAAC;wBACjC,gBAAgB,GAAG,IAAI,mBAAmB,CACtC,KAAK,CAAC,MAAM,EACZ,SAAmC,EACnC,IAAI,CAAC,YAAY,EAAE,KAAK,CAC3B,CAAC;wBACF,wBAAwB,GAAG,IAAI,2BAA2B,CAAC;4BACvD,OAAO,EAAE,gBAAgB;yBAC5B,CAAC,CAAC;wBAEH,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC;wBAE9B,OAAO,4BAA4B,CAAC,MAAM,GAAG,CAAC;4BAC1C,4BAA4B,CAAC,KAAK,EAAG,CAAC,OAAO,EAAE,CAAC;wBAEpD,2BAA2B,CAAC,oBAAoB,EAAE,CAAC;wBACnD,2BAA2B,CAAC,mBAAmB,EAAE,CAAC;wBAElD,yBAAyB,CAAC,oBAAoB,EAAE,CAAC;wBACjD,yBAAyB,CAAC,mBAAmB,EAAE,CAAC;qBACnD;iBACJ;gBAED,IAAI,CAAC,wBAAwB,IAAI,gBAAgB,IAAI,gBAAgB,IAAI,IAAI;oBACzE,2BAA2B,CAAC,iBAAiB,IAAI,wBAAwB,IAAI,IAAI,EACnF;oBACE,wBAAwB,GAAG,IAAI,CAAC;oBAChC,4BAA4B,CAAC,IAAI,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;oBAE7E,sBAAsB,CAAC,mBAAmB,EAAE,CAAC;oBAC7C,sBAAsB,CAAC,oBAAoB,EAAE,CAAC;oBAE9C,aAAa,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,kBAAkB,EAAE,CAAC,CAAC;oBAE5D,MAAM,cAAc,GAAI,2BAA2B,CAAC,iBAAiB,EAAE,CAAC;oBACxE,MAAM,mBAAmB,GAAG,eAAe,CAAC,qBAAqB,EAAE,CAAC;oBAEpE,MAAM,6BAA6B,GAAG,gCAAgC,CAClE,cAAc,EACd,mBAAmB,EACnB,KAAK,CAAC,QAAQ,CACjB,CAAC;oBACF,aAAa,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,CAAC;oBAErD,MAAM,CAAC,iCAAiC,CAAC,GAAG,cAAc;yBACrD,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,oBAAoB,CAAC;yBACtD,MAAM,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC;yBACjE,IAAI,CAAC,CAAC,CAAC,CAAC;oBAEb,MAAM,sBAAsB,GACxB,CAAC,iCAAiC,IAAI,IAAI,IAAI,iCAAiC,CAAC,MAAM,KAAK,CAAC,CAAC;wBACzF,CAAC,CAAC,EAAE;wBACJ,CAAC,CAAC,OAAO,iCAAiC,KAAK,QAAQ;4BACnD,CAAC,CAAC,iCAAiC;4BACnC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,iCAAiC,CAAC,CAAC;oBAElE,kBAAkB,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,GAAG,sBAAsB,CAAC,CAAC,CAAC;oBAEtH,KAAK,MAAM,iBAAiB,IAAI,kBAAkB;wBAC9C,OAAO,CAAC,oCAAoC,CAAC,wBAAwB,EAAE,iBAAiB,CAAC,CAAC;iBACjG;qBAAM,IAAI,wBAAwB,EAAE;oBACjC,kBAAkB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;oBACnC,4BAA4B,CAAC,IAAI,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7E,yBAAyB,CAAC,gBAAgB,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,kBAAkB,EAAC,CAAC,CAAC;iBAClF;gBAED,IAAI,wBAAwB,IAAI,yBAAyB,CAAC,iBAAiB,IAAI,gBAAgB,IAAI,IAAI,EAAE;oBACrG,MAAM,gBAAgB,GAAG,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;oBAC9D,MAAM,YAAY,GAAG,gBAAgB,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;oBAE1E,IAAI,aAAa,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;oBAC1C,IAAI,0BAA0B,GAAG,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;oBAErE,IAAI,OAAO,EAAE,oBAAoB,IAAI,oBAAoB,EAAE;wBACvD,aAAa,GAAG,aAAa,CAAC,OAAO,EAAE,CAAC;wBACxC,0BAA0B,GAAG,0BAA0B,CAAC,OAAO,EAAE,CAAC;qBACrE;oBAED,OAAO;wBACH,QAAQ,EAAE,aAAa;wBACvB,cAAc,EAAE;4BACZ,aAAa,EAAE,qCAAqC,CAChD,wBAAwB,EACxB,8BAA8B,GAAG,0BAA0B,CAC9D;4BACD,YAAY,EAAE,qCAAqC,CAC/C,eAAe,EACf,iBAAiB,GAAG,aAAa,CACpC;4BACD,oBAAoB,EAAE,8BAA8B;yBACvD;wBAED,yCAAyC;wBACzC,YAAY,EAAE,YAAmF;wBAEjG,QAAQ,EAAE;4BACN,UAAU,EAAE,cAAc;yBAC7B;qBACJ,CAAC;iBACL;gBAED,IAAI,CAAC,wBAAwB;oBACzB,sBAAsB,CAAC,gBAAgB,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,kBAAkB,EAAC,CAAC,CAAC;gBAEhF,aAAa,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,kBAAkB,EAAE,CAAC,CAAC;gBAE5D,4CAA4C,EAAE,CAAC;gBAE/C,IAAI,sBAAsB,CAAC,iBAAiB,IAAI,KAAK,KAAK,QAAQ,EAAE;oBAChE,MAAM,cAAc,GAAI,sBAAsB,CAAC,iBAAiB,EAAE,CAAC;oBACnE,MAAM,mBAAmB,GAAG,eAAe,CAAC,qBAAqB,EAAE,CAAC;oBAEpE,MAAM,6BAA6B,GAAG,gCAAgC,CAClE,cAAc,EACd,mBAAmB,EACnB,KAAK,CAAC,QAAQ,CACjB,CAAC;oBACF,aAAa,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,CAAC;oBAErD,MAAM,CAAC,iCAAiC,CAAC,GAAG,cAAc;yBACrD,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,oBAAoB,CAAC;yBACtD,MAAM,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC;yBACjE,IAAI,CAAC,CAAC,CAAC,CAAC;oBAEb,4CAA4C,EAAE,CAAC;oBAE/C,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC;wBACxB,OAAO,EAAE,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC,CAAC;oBAErC,GAAG,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;oBAC3B,iBAAiB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;oBACzC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;oBAEzB,IAAI,aAAa,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;oBAC1C,IAAI,0BAA0B,GAAG,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;oBAErE,IAAI,OAAO,EAAE,oBAAoB,IAAI,oBAAoB,EAAE;wBACvD,aAAa,GAAG,aAAa,CAAC,OAAO,EAAE,CAAC;wBACxC,0BAA0B,GAAG,0BAA0B,CAAC,OAAO,EAAE,CAAC;qBACrE;oBAED,OAAO;wBACH,QAAQ,EAAE,aAAa;wBACvB,cAAc,EAAE;4BACZ,aAAa,EAAE,qCAAqC,CAChD,wBAAwB,EACxB,8BAA8B,GAAG,0BAA0B,CAC9D;4BACD,YAAY,EAAE,qCAAqC,CAC/C,eAAe,EACf,iBAAiB,GAAG,aAAa,CACpC;4BACD,oBAAoB,EAAE,8BAA8B;yBACvD;wBACD,QAAQ,EAAE;4BACN,4BAA4B,EAAE,iCAAiC;4BAC/D,UAAU,EAAE,KAAK,KAAK,QAAQ;gCAC1B,CAAC,CAAC,UAAU;gCACZ,CAAC,CAAC,uBAAuB;yBAChC;qBACJ,CAAC;iBACL;gBAED,MAAM,kBAAkB,GAAG,SAAS,IAAI,IAAI,IAAI,SAAS,GAAG,CAAC,IAAI,eAAe,IAAI,SAAS,CAAC;gBAE9F,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE;oBAClB,uBAAuB,CAAC,oBAAoB,EAAE,CAAC;oBAC/C,uBAAuB,CAAC,mBAAmB,EAAE,CAAC;oBAE9C,uBAAuB,CAAC,gBAAgB,CAAC;wBACrC,IAAI,EAAE,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC;wBACrC,MAAM,EAAE,aAAa;qBACxB,CAAC,CAAC;iBACN;gBAED,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,uBAAuB,CAAC,kBAAkB,CAAC,EAAE;oBACjG,4CAA4C,EAAE,CAAC;oBAE/C,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;wBAC1B,OAAO,EAAE,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC,CAAC;wBACjC,GAAG,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;wBAC3B,iBAAiB,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;wBACzC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;qBAC5B;iBACJ;gBAED,IAAI,kBAAkB,EAAE;oBACpB,IAAI,aAAa,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;oBAC1C,IAAI,0BAA0B,GAAG,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;oBAErE,IAAI,OAAO,EAAE,oBAAoB,IAAI,oBAAoB,EAAE;wBACvD,aAAa,GAAG,aAAa,CAAC,OAAO,EAAE,CAAC;wBACxC,0BAA0B,GAAG,0BAA0B,CAAC,OAAO,EAAE,CAAC;qBACrE;oBAED,OAAO;wBACH,QAAQ,EAAE,aAAa;wBACvB,cAAc,EAAE;4BACZ,aAAa,EAAE,qCAAqC,CAChD,wBAAwB,EACxB,8BAA8B,GAAG,0BAA0B,CAC9D;4BACD,YAAY,EAAE,qCAAqC,CAC/C,eAAe,EACf,iBAAiB,GAAG,aAAa,CACpC;4BACD,oBAAoB,EAAE,8BAA8B;yBACvD;wBACD,QAAQ,EAAE;4BACN,UAAU,EAAE,WAAW;yBAC1B;qBACJ,CAAC;iBACL;gBAED,IAAI,IAAI,CAAC,SAAS,CAAC,cAAc,IAAI,OAAO,CAAC,WAAW,EAAE;oBACtD,kBAAkB,GAAG,IAAI,CAAC;oBAC1B,MAAM;iBACT;aACJ;YAED,iBAAiB,GAAG,KAAK,CAAC;YAE1B,IAAI,kBAAkB;gBAClB,SAAS;YAEb,MAAM;SACT;QAED,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;IAC5E,CAAC;CACJ;AA8BD,SAAS,wBAAwB,CAA+B,WAAiB;IAC7E,IAAI,WAAW,CAAC,IAAI,KAAK,OAAO,EAAE;QAC9B,MAAM,cAAc,GAAsB,EAAC,GAAG,WAAW,EAAC,CAAC;QAC3D,cAAc,CAAC,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAC3D,IAAI,OAAO,IAAI,KAAK,QAAQ;gBACxB,OAAO,IAAI,CAAC;;gBAEZ,OAAO;oBACH,GAAG,IAAI;oBACP,GAAG,EAAE,SAAS;iBACjB,CAAC;QACV,CAAC,CAAC,CAAC;QAEH,OAAO,cAAsB,CAAC;KACjC;IAED,OAAO,WAAW,CAAC;AACvB,CAAC;AAED,KAAK,UAAU,+BAA+B,CAAC,EAC3C,OAAO,EACP,gBAAgB,EAChB,oBAAoB,EACpB,kCAAkC,EAClC,WAAW,EACX,SAAS,EACT,WAAW,EACX,SAAS,EACT,sBAAsB,EAWzB;IAIG,SAAS,yBAAyB,CAAC,OAA0B;QACzD,MAAM,EAAC,WAAW,EAAC,GAAG,WAAW,CAAC,mBAAmB,CAAC,OAAO,EAAE;YAC3D,kBAAkB,EAAE,SAAS;YAC7B,sBAAsB;SACzB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAE/C,OAAO,MAAM,CAAC,MAAM,IAAI,WAAW,GAAG,gBAAgB,CAAC;IAC3D,CAAC;IAED,IAAI,WAAW,GAAG,gBAAgB,IAAI,CAAC;QACnC,MAAM,IAAI,KAAK,CACX,qBAAqB,WAAW,iDAAiD,gBAAgB,GAAG,CACvG,CAAC;IAEN,IAAI,yBAAyB,CAAC,OAAO,CAAC;QAClC,OAAO;YACH,iBAAiB,EAAE,OAAO;YAC1B,QAAQ,EAAE,IAAI;SACjB,CAAC;IAEN,IAAI,oBAAoB,YAAY,QAAQ,EAAE;QAC1C,IAAI;YACA,MAAM,EAAC,WAAW,EAAE,QAAQ,EAAC,GAAG,MAAM,oBAAoB,CAAC;gBACvD,WAAW,EAAE,OAAO;gBACpB,cAAc,EAAE,WAAW,GAAG,gBAAgB;gBAC9C,SAAS;gBACT,WAAW;gBACX,iBAAiB,EAAE,kCAAkC;aACxD,CAAC,CAAC;YAEH,IAAI,yBAAyB,CAAC,WAAW,CAAC;gBACtC,OAAO;oBACH,iBAAiB,EAAE,WAAW;oBAC9B,QAAQ;iBACX,CAAC;YAEN,OAAO,CAAC,IAAI,CACR,2FAA2F;gBAC3F,qCAAqC,CACxC,CAAC;SACL;QAAC,OAAO,GAAG,EAAE;YACV,OAAO,CAAC,KAAK,CACT,sDAAsD;gBACtD,qCAAqC,EACrC,GAAG,CACN,CAAC;SACL;KACJ;SAAM,IAAI,oBAAoB,KAAK,sCAAsC;QACtE,OAAO,CAAC,IAAI,CACR,mCAAmC,oBAAoB,KAAK;YAC5D,qCAAqC,CACxC,CAAC;IAEN,MAAM,EAAC,WAAW,EAAE,QAAQ,EAAC,GAAG,MAAM,4DAA4D,CAAC;QAC/F,WAAW,EAAE,OAAO;QACpB,cAAc,EAAE,WAAW,GAAG,gBAAgB;QAC9C,SAAS;QACT,WAAW;QACX,iBAAiB,EAAE,kCAAkC;KACxD,CAAC,CAAC;IAEH,IAAI,CAAC,yBAAyB,CAAC,WAAW,CAAC;QACvC,MAAM,IAAI,KAAK,CACX,0FAA0F;YAC1F,yDAAyD,CAC5D,CAAC;IAEN,OAAO;QACH,iBAAiB,EAAE,WAAW;QAC9B,QAAQ;KACX,CAAC;AACN,CAAC;AAED,SAAS,gCAAgC,CACrC,cAA2F,EAC3F,mBAGC,EACD,SAAoB;IAEpB,IAAI,mBAAmB,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;QAChF,OAAO,EAAE,CAAC;SACT,IAAI,mBAAmB,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;QACrF,OAAO,mBAAmB,CAAC,MAAM,CAAC;SACjC,IAAI,mBAAmB,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,IAAI,mBAAmB,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;QACrF,OAAO,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAE/C,MAAM,gCAAgC,GAAG,cAAc,CAAC,SAAS,CAC7D,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,QAAQ,CAC5F,CAAC;IACF,MAAM,+BAA+B,GAAG,cAAc,CAAC,SAAS,CAC5D,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,QAAQ,CAC5F,CAAC;IAEF,IAAI,+BAA+B,GAAG,CAAC,IAAI,gCAAgC,GAAG,CAAC;QAC3E,OAAO,mBAAmB,CAAC,MAAM,CAAC;SACjC,IAAI,gCAAgC,GAAG,CAAC,IAAI,+BAA+B,GAAG,CAAC;QAChF,OAAO,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAE/C,MAAM,YAAY,GAAG,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACzD,IAAI,YAAY,CAAC,MAAM,KAAK,mBAAmB,CAAC,MAAM,CAAC,MAAM;QACzD,YAAY,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,mBAAmB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAEjF,OAAO,YAAY,CAAC;SACnB,IAAI,gCAAgC,GAAG,+BAA+B;QACvE,OAAO,YAAY,CAAC;IAExB,OAAO,mBAAmB,CAAC,MAAM,CAAC;AACtC,CAAC;AAED,SAAS,uCAAuC,CAAC,WAA8B;IAC3E,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO;QAChF,OAAO,EAAE,CAAC;IAEd,MAAM,qBAAqB,GAAG,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAsB,CAAC;IACvF,MAAM,aAAa,GAAG,qBAAqB,CAAC,QAAQ,CAAC;IAErD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,QAAQ;QACvF,OAAO,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAW,CAAC;IAE7D,OAAO,EAAE,CAAC;AACd,CAAC;AAED,SAAS,qCAAqC,CAAC,WAA8B,EAAE,YAAoB;IAC/F,MAAM,cAAc,GAAG,WAAW,CAAC,KAAK,EAAE,CAAC;IAC3C,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,IAAI,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO;QACzF,cAAc,CAAC,IAAI,CAAC;YAChB,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,EAAE;SACf,CAAC,CAAC;IAEP,MAAM,qBAAqB,GAAG,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAsB,CAAC;IAC7F,MAAM,wBAAwB,GAAG,EAAC,GAAG,qBAAqB,EAAC,CAAC;IAC5D,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,wBAAwB,CAAC;IAErE,MAAM,aAAa,GAAG,wBAAwB,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IAChE,wBAAwB,CAAC,QAAQ,GAAG,aAAa,CAAC;IAElD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,QAAQ,EAAE;QACzF,IAAI,YAAY,KAAK,EAAE;YACnB,aAAa,CAAC,GAAG,EAAE,CAAC;;YAEpB,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,YAAY,CAAC;KAC9D;SAAM,IAAI,YAAY,KAAK,EAAE;QAC1B,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAErC,OAAO,cAAc,CAAC;AAC1B,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,EAC5B,eAAe,EAAE,oBAAoB,EACrC,8BAA8B,EAAE,kBAAkB,GAAG,CAAC,EAAE,iBAAiB,EACzE,WAAW,EAAE,kCAAkC,EAAE,6CAA6C,EAC9F,QAAQ,EAAE,oBAAoB,GAAG,CAAC,EAAE,SAAS,EAAE,sBAAsB,EAOxE;IAMG,IAAI,QAAQ,IAAI,IAAI;QAChB,MAAM,IAAI,aAAa,EAAE,CAAC;IAE9B,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;IAC7B,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;IAEjC,IAAI,iBAAiB,IAAI,kCAAkC,IAAI,IAAI,IAAI,QAAQ,CAAC,gBAAgB,EAAE;QAC9F,MAAM,gBAAgB,GAAG,kCAAkC,CAAC,KAAK,EAAE,CAAC;QAEpE,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,IAAI,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO;YAC/F,gBAAgB,CAAC,IAAI,CAAC;gBAClB,IAAI,EAAE,OAAO;gBACb,QAAQ,EAAE,EAAE;aACf,CAAC,CAAC;QAEP,MAAM,EAAC,WAAW,EAAE,sBAAsB,EAAE,eAAe,EAAE,YAAY,EAAC,GAAG,WAAW,CAAC,mBAAmB,CAAC,gBAAgB,EAAE;YAC3H,kBAAkB,EAAE,SAAS;YAC7B,sBAAsB;SACzB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,MAAM,CAAC,MAAM,GAAG,kBAAkB,GAAG,oBAAoB,GAAG,OAAO,CAAC,WAAW,EAAE;YACjF,MAAM,EAAC,mBAAmB,EAAC,GAAG,QAAQ,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEpE,MAAM,4BAA4B,GAAG,mBAAmB,GAAG,MAAM,CAAC,MAAM,CAAC;YAEzE,IAAI,4BAA4B,IAAI,6CAA6C;gBAC7E,OAAO;oBACH,OAAO,EAAE,gBAAgB;oBACzB,sBAAsB;oBACtB,MAAM;oBACN,kBAAkB,EAAE,eAAe;oBACnC,6BAA6B,EAAE,8BAA8B;oBAC7D,eAAe,EAAE,eAAe,IAAI,EAAE;oBACtC,4BAA4B,EAAE,YAAY,EAAE,gBAAgB,IAAI,KAAK;oBACrE,qCAAqC,EAAE,YAAY,EAAE,yBAAyB,IAAI,EAAE;iBACvF,CAAC;SACT;KACJ;IAED,eAAe,GAAG,QAAQ,CAAC,gBAAgB;QACvC,CAAC,CAAC,eAAe,CAAC,KAAK,EAAE;QACzB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;IAEpD,IAAI,oBAAoB,CAAC,sBAAsB,IAAI,IAAI,EAAE;QACrD,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,IAAI,YAAY,QAAQ;YAClE,CAAC,CAAC,MAAM,oBAAoB,CAAC,IAAI,CAAC,QAAQ,CAAC;YAC3C,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC;QAEhC,MAAM,EAAC,iBAAiB,EAAE,QAAQ,EAAC,GAAG,MAAM,+BAA+B,CAAC;YACxE,OAAO,EAAE,eAAe;YACxB,gBAAgB,EAAE,IAAI,CAAC,GAAG,CAAC,gBAAgB,EAAE,oBAAoB,CAAC,GAAG,kBAAkB;YACvF,oBAAoB,EAAE,oBAAoB,CAAC,QAAQ;YACnD,kCAAkC,EAAE,oBAAoB,CAAC,sBAAsB;YAC/E,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,SAAS,EAAE,KAAK,CAAC,QAAQ;YACzB,WAAW,EAAE,WAAW;YACxB,SAAS;YACT,sBAAsB;SACzB,CAAC,CAAC;QAEH,MAAM,EAAC,WAAW,EAAE,sBAAsB,EAAE,eAAe,EAAE,YAAY,EAAC,GAAG,WAAW,CAAC,mBAAmB,CAAC,iBAAiB,EAAE;YAC5H,kBAAkB,EAAE,SAAS;YAC7B,sBAAsB;SACzB,CAAC,CAAC;QAEH,OAAO;YACH,OAAO,EAAE,iBAAiB;YAC1B,sBAAsB;YACtB,MAAM,EAAE,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC;YAC5C,kBAAkB,EAAE,eAAe;YACnC,6BAA6B,EAAE,QAAQ;YACvC,eAAe,EAAE,eAAe,IAAI,EAAE;YACtC,4BAA4B,EAAE,YAAY,EAAE,gBAAgB,IAAI,KAAK;YACrE,qCAAqC,EAAE,YAAY,EAAE,yBAAyB,IAAI,EAAE;SACvF,CAAC;KACL;IAED;QACI,MAAM,EAAC,WAAW,EAAE,sBAAsB,EAAE,eAAe,EAAE,YAAY,EAAC,GAAG,WAAW,CAAC,mBAAmB,CAAC,eAAe,EAAE;YAC1H,kBAAkB,EAAE,SAAS;YAC7B,sBAAsB;SACzB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAEpD,IAAI,MAAM,CAAC,MAAM,GAAG,kBAAkB,GAAG,oBAAoB,GAAG,OAAO,CAAC,WAAW;YAC/E,OAAO;gBACH,OAAO,EAAE,eAAe;gBACxB,sBAAsB;gBACtB,MAAM;gBACN,kBAAkB,EAAE,eAAe;gBACnC,6BAA6B,EAAE,8BAA8B;gBAC7D,eAAe,EAAE,eAAe,IAAI,EAAE;gBACtC,4BAA4B,EAAE,YAAY,EAAE,gBAAgB,IAAI,KAAK;gBACrE,qCAAqC,EAAE,YAAY,EAAE,yBAAyB,IAAI,EAAE;aACvF,CAAC;KACT;IAED,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,IAAI,YAAY,QAAQ;QAClE,CAAC,CAAC,MAAM,oBAAoB,CAAC,IAAI,CAAC,QAAQ,CAAC;QAC3C,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC;IAEhC,MAAM,EAAC,iBAAiB,EAAE,QAAQ,EAAC,GAAG,MAAM,+BAA+B,CAAC;QACxE,OAAO,EAAE,eAAe;QACxB,gBAAgB,EAAE,IAAI,CAAC,GAAG,CAAC,gBAAgB,EAAE,oBAAoB,CAAC,GAAG,kBAAkB;QACvF,oBAAoB,EAAE,oBAAoB,CAAC,QAAQ;QACnD,kCAAkC,EAAE,oBAAoB,CAAC,sBAAsB;QAC/E,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,SAAS,EAAE,KAAK,CAAC,QAAQ;QACzB,WAAW,EAAE,WAAW;QACxB,SAAS;QACT,sBAAsB;KACzB,CAAC,CAAC;IAEH,MAAM,EAAC,WAAW,EAAE,sBAAsB,EAAE,eAAe,EAAE,YAAY,EAAC,GAAG,WAAW,CAAC,mBAAmB,CAAC,iBAAiB,EAAE;QAC5H,kBAAkB,EAAE,SAAS;QAC7B,sBAAsB;KACzB,CAAC,CAAC;IAEH,OAAO;QACH,OAAO,EAAE,iBAAiB;QAC1B,sBAAsB;QACtB,MAAM,EAAE,WAAW,CAAE,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC7C,kBAAkB,EAAE,eAAe;QACnC,6BAA6B,EAAE,QAAQ;QACvC,eAAe,EAAE,eAAe,IAAI,EAAE;QACtC,4BAA4B,EAAE,YAAY,EAAE,gBAAgB,IAAI,KAAK;QACrE,qCAAqC,EAAE,YAAY,EAAE,yBAAyB,IAAI,EAAE;KACvF,CAAC;AACN,CAAC"}
@@ -2,10 +2,11 @@ import { LlamaGrammar } from "../../LlamaGrammar.js";
2
2
  import { ChatModelFunctions } from "../../../types.js";
3
3
  import { ChatWrapper } from "../../../ChatWrapper.js";
4
4
  import { LlamaChatResponseFunctionCall } from "../LlamaChat.js";
5
+ import { Llama } from "../../../bindings/Llama.js";
5
6
  export declare class FunctionCallGrammar<const Functions extends ChatModelFunctions> extends LlamaGrammar {
6
7
  private readonly _functions;
7
8
  private readonly _chatWrapper;
8
- constructor(functions: Functions, chatWrapper: ChatWrapper, initialFunctionCallEngaged: boolean);
9
+ constructor(llama: Llama, functions: Functions, chatWrapper: ChatWrapper, initialFunctionCallEngaged: boolean);
9
10
  parseFunctionCall(callText: string): LlamaChatResponseFunctionCall<Functions>;
10
11
  parseFunctionNameFromPartialCall(callText: string, { enableInternalBuiltinFunctions, initialFunctionCallEngaged }?: {
11
12
  enableInternalBuiltinFunctions?: boolean;
@@ -9,9 +9,10 @@ import { GbnfVerbatimText } from "../../../utils/gbnfJson/terminals/GbnfVerbatim
9
9
  export class FunctionCallGrammar extends LlamaGrammar {
10
10
  _functions;
11
11
  _chatWrapper;
12
- constructor(functions, chatWrapper, initialFunctionCallEngaged) {
12
+ constructor(llama, functions, chatWrapper, initialFunctionCallEngaged) {
13
13
  const grammar = getGbnfGrammarForFunctionCalls(functions, chatWrapper, initialFunctionCallEngaged);
14
14
  super({
15
+ llama,
15
16
  grammar,
16
17
  stopGenerationTriggers: [LlamaText(chatWrapper.settings.functions.call.suffix, "\n".repeat(4))],
17
18
  trimWhitespaceSuffix: true
@@ -0,0 +1 @@
1
+ {"version":3,"file":"FunctionCallGrammar.js","sourceRoot":"","sources":["../../../../src/evaluator/LlamaChat/utils/FunctionCallGrammar.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,YAAY,EAAC,MAAM,uBAAuB,CAAC;AACnD,OAAO,EAAC,SAAS,EAAC,MAAM,6BAA6B,CAAC;AACtD,OAAO,EAAC,+BAA+B,EAAC,MAAM,kEAAkE,CAAC;AAEjH,OAAO,EAAC,oBAAoB,EAAC,MAAM,iDAAiD,CAAC;AACrF,OAAO,EAAC,oCAAoC,EAAC,MAAM,uEAAuE,CAAC;AAE3H,OAAO,EAAC,WAAW,EAAC,MAAM,kDAAkD,CAAC;AAE7E,OAAO,EAAC,MAAM,EAAC,MAAM,6CAA6C,CAAC;AAEnE,OAAO,EAAC,gBAAgB,EAAC,MAAM,uDAAuD,CAAC;AAIvF,MAAM,OAAO,mBAAgE,SAAQ,YAAY;IAC5E,UAAU,CAAY;IACtB,YAAY,CAAc;IAE3C,YAAmB,KAAY,EAAE,SAAoB,EAAE,WAAwB,EAAE,0BAAmC;QAChH,MAAM,OAAO,GAAG,8BAA8B,CAAC,SAAS,EAAE,WAAW,EAAE,0BAA0B,CAAC,CAAC;QAEnG,KAAK,CAAC;YACF,KAAK;YACL,OAAO;YACP,sBAAsB,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/F,oBAAoB,EAAE,IAAI;SAC7B,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;QAEhC,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC9B,CAAC;IAEM,iBAAiB,CAAC,QAAgB;QACrC,IAAI,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,mBAAmB;YAC7D,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,KAAK,GAAG;YAE7F,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAEjC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC;YACtE,MAAM,IAAI,gCAAgC,CACtC,mEAAmE,IAAI,CAAC,YAAY,CAAC,WAAW,gBAAgB,EAChH,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,YAAY,EACjB,QAAQ,CACX,CAAC;QAEN,MAAM,iBAAiB,GAAG,QAAQ,CAAC,OAAO,CACtC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,EACtD,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAC1D,CAAC;QAEF,IAAI,iBAAiB,GAAG,CAAC;YACrB,MAAM,IAAI,gCAAgC,CACtC,yDAAyD,IAAI,CAAC,YAAY,CAAC,WAAW,gBAAgB,EACtG,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,YAAY,EACjB,QAAQ,CACX,CAAC;QAEN,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAC/B,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EACvD,iBAAiB,CACQ,CAAC;QAE9B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,YAAY,CAAC;YAC7C,MAAM,IAAI,gCAAgC,CACtC,kBAAkB,YAAY,2CAA2C,EACzE,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,YAAY,EACjB,QAAQ,CACX,CAAC;QAEN,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAErD,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC;QACpE,MAAM,eAAe,GAAG,CAAC,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAE7F,IAAI,eAAe,GAAG,CAAC,IAAI,eAAe,GAAG,iBAAiB,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM;YAC1H,MAAM,IAAI,gCAAgC,CACtC,iEAAiE,IAAI,CAAC,YAAY,CAAC,WAAW,gBAAgB,EAC9G,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,YAAY,EACjB,QAAQ,CACX,CAAC;QAEN,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAC/B,iBAAiB,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EACjF,eAAe,CAClB,CAAC;QAEF,IAAI,cAAc,CAAC,MAAM,IAAI,IAAI,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC;YACjE,MAAM,IAAI,gCAAgC,CACtC,gEAAgE,YAAY,GAAG,EAC/E,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,YAAY,EACjB,QAAQ,CACX,CAAC;aACD,IAAI,cAAc,CAAC,MAAM,IAAI,IAAI;YAClC,OAAO;gBACH,YAAY;gBACZ,MAAM,EAAE,SAAgB;gBACxB,GAAG,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,GAAG,UAAU,CAAC,MAAM,CAAC;aAC9D,CAAC;QAEN,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAExC,+BAA+B,CAAC,MAAM,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC;QAE/D,OAAO;YACH,YAAY;YACZ,MAAM,EAAE,MAAa;YACrB,GAAG,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,GAAG,UAAU,CAAC,MAAM,CAAC;SAC9D,CAAC;IACN,CAAC;IAEM,gCAAgC,CAAC,QAAgB,EAAE,EACtD,8BAA8B,GAAG,KAAK,EACtC,0BAA0B,GAAG,KAAK,KAIlC,EAAE;QACF,IAAI,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,mBAAmB;YAC7D,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,KAAK,GAAG;YAE7F,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAEjC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC;YACtE,MAAM,IAAI,gCAAgC,CACtC,mEAAmE,IAAI,CAAC,YAAY,CAAC,WAAW,gBAAgB,EAChH,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,YAAY,EACjB,QAAQ,CACX,CAAC;QAEN,MAAM,iBAAiB,GAAG,QAAQ,CAAC,OAAO,CACtC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,EACtD,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAC1D,CAAC;QAEF,IAAI,iBAAiB,GAAG,CAAC;YACrB,MAAM,IAAI,gCAAgC,CACtC,yDAAyD,IAAI,CAAC,YAAY,CAAC,WAAW,gBAAgB,EACtG,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,YAAY,EACjB,QAAQ,CACX,CAAC;QAEN,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAC/B,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EACvD,iBAAiB,CACQ,CAAC;QAE9B,IAAI,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QAErE,IAAI,CAAC,iBAAiB,IAAI,8BAA8B;YACpD,iBAAiB,KAAK,MAAM,CAAC,MAAM,CAC/B,IAAI,CAAC,YAAY,CAAC,2BAA2B,CAAC,EAAC,0BAA0B,EAAC,CAAC,EAC3E,YAAY,CACf,CAAC;QAEN,IAAI,CAAC,iBAAiB;YAClB,MAAM,IAAI,gCAAgC,CACtC,kBAAkB,YAAY,2CAA2C,EACzE,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,YAAY,EACjB,QAAQ,CACX,CAAC;QAEN,OAAO,YAAY,CAAC;IACxB,CAAC;IAEO,kBAAkB;QACtB,KAAK,MAAM,aAAa,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE;YACtD,IAAI,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAC3F,MAAM,IAAI,KAAK,CAAC,kBAAkB,aAAa,sCAAsC,CAAC,CAAC;SAC9F;IACL,CAAC;CACJ;AAED,SAAS,8BAA8B,CACnC,SAAoB,EAAE,WAAwB,EAAE,0BAAmC;IAEnF,MAAM,gBAAgB,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAEpD,MAAM,YAAY,GAAmB,EAAE,CAAC;IAExC,SAAS,sBAAsB,CAAC,YAAoB,EAAE,cAA0C;QAC5F,IAAI,cAAc,CAAC,MAAM,IAAI,IAAI,EAAE;YAC/B,MAAM,cAAc,GAAG,oCAAoC,CAAC,cAAc,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;YAErG,YAAY,CAAC,IAAI,CACb,IAAI,WAAW,CAAC;gBACZ,IAAI,gBAAgB,CAAC,YAAY,GAAG,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE;gBAClG,cAAc,CAAC,OAAO,CAAC,gBAAgB,CAAC;aAC3C,CAAC,CACL,CAAC;SACL;;YACG,YAAY,CAAC,IAAI,CAAC,IAAI,gBAAgB,CAAC,YAAY,GAAG,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;IACjH,CAAC;IAED,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC;QAClE,sBAAsB,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;IAEzD,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,2BAA2B,CAAC,EAAC,0BAA0B,EAAC,CAAC,CAAC;QAC9H,sBAAsB,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;IAEzD,MAAM,WAAW,GAAG,IAAI,MAAM,CAAC,YAAY,CAAC,CAAC;IAE7C,MAAM,YAAY,GAAG,IAAI,WAAW,CAAC;QACjC,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5E,IAAI,gBAAgB,CAAC,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE;QAC7E,WAAW,CAAC,OAAO,CAAC,gBAAgB,CAAC;QACrC,IAAI,gBAAgB,CAAC,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE;KAChF,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,YAAY,CAAC,UAAU,EAAE,CAAC;IAE9C,OAAO,gBAAgB,CAAC,gBAAgB,CAAC,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC;AAC3F,CAAC;AAED,MAAM,OAAO,gCAA6E,SAAQ,KAAK;IACnF,SAAS,CAAY;IACrB,WAAW,CAAc;IACzB,QAAQ,CAAS;IAEjC,YAAmB,OAAe,EAAE,SAAoB,EAAE,WAAwB,EAAE,QAAgB;QAChG,KAAK,CAAC,OAAO,CAAC,CAAC;QAEf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC7B,CAAC;CACJ"}
@@ -96,10 +96,28 @@ export async function eraseFirstResponseAndKeepFirstSystemChatContextShiftStrate
96
96
  }
97
97
  }
98
98
  }
99
+ function compressLastModelResponse(minCharactersToKeep = 20) {
100
+ const lastHistoryItem = res[res.length - 1];
101
+ if (lastHistoryItem == null || lastHistoryItem.type !== "model")
102
+ return;
103
+ const lastResponseItem = lastHistoryItem.response[lastHistoryItem.response.length - 1];
104
+ if (lastResponseItem == null || typeof lastResponseItem !== "string")
105
+ return;
106
+ const nextTextLength = lastResponseItem.length - charactersLeftToRemove;
107
+ const charactersToRemoveFromText = charactersLeftToRemove + Math.max(0, nextTextLength - minCharactersToKeep);
108
+ const newText = truncateTextAndRoundToWords(lastResponseItem, charactersToRemoveFromText);
109
+ if (newText.length < lastResponseItem.length) {
110
+ lastHistoryItem.response[lastHistoryItem.response.length - 1] = newText;
111
+ charactersLeftToRemove -= lastResponseItem.length - newText.length;
112
+ }
113
+ }
99
114
  compressFunctionCalls();
100
115
  if (charactersLeftToRemove <= 0)
101
116
  return res;
102
117
  compressFirstModelResponse();
118
+ if (charactersLeftToRemove <= 0)
119
+ return res;
120
+ compressLastModelResponse();
103
121
  return res;
104
122
  }
105
123
  });
@@ -0,0 +1 @@
1
+ {"version":3,"file":"eraseFirstResponseAndKeepFirstSystemChatContextShiftStrategy.js","sourceRoot":"","sources":["../../../../../src/evaluator/LlamaChat/utils/contextShiftStrategies/eraseFirstResponseAndKeepFirstSystemChatContextShiftStrategy.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,kDAAkD,EAAC,MAAM,yEAAyE,CAAC;AAC3I,OAAO,EAAC,2BAA2B,EAAC,MAAM,kDAAkD,CAAC;AAG7F,MAAM,CAAC,KAAK,UAAU,4DAA4D,CAAC,EAC/E,WAAW,EACX,cAAc,EACd,SAAS,EACT,WAAW,EACX,iBAAiB,EAOpB;IAIG,IAAI,6BAA6B,GAAG,CAAC,CAAC;IACtC,IAAI,qBAAqB,CAAC,iBAAiB,CAAC;QACxC,6BAA6B,GAAG,iBAAiB,CAAC,uBAAuB,CAAC;IAE9E,MAAM,EAAC,sBAAsB,EAAE,qBAAqB,EAAC,GAAG,MAAM,kDAAkD,CAAC;QAC7G,WAAW;QACX,gBAAgB,EAAE,cAAc;QAChC,6BAA6B;QAC7B,SAAS;QACT,WAAW;QACX,mBAAmB,CAAC,EAAC,WAAW,EAAE,kBAAkB,EAAC;YACjD,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3D,IAAI,sBAAsB,GAAG,kBAAkB,CAAC;YAEhD,SAAS,qBAAqB;gBAC1B,KAAK,IAAI,CAAC,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE;oBACpE,MAAM,WAAW,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;oBAE3B,IAAI,WAAW,CAAC,IAAI,KAAK,OAAO;wBAC5B,SAAS;oBAEb,KAAK,IAAI,CAAC,GAAG,WAAW,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE;wBACrF,MAAM,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;wBAErC,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,KAAK,cAAc;4BACxD,SAAS;wBAEb,MAAM,iBAAiB,GAAG,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;wBAE3C,IAAI,iBAAiB,IAAI,IAAI;4BACzB,SAAS;wBAEb,MAAM,UAAU,GAAG,WAAW,CAAC,6BAA6B,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;wBAElG,IAAI,UAAU,CAAC,MAAM,GAAG,iBAAiB,EAAE;4BACvC,IAAI,CAAC,GAAG,GAAG,UAAU,CAAC;4BACtB,sBAAsB,IAAI,iBAAiB,GAAG,UAAU,CAAC,MAAM,CAAC;yBACnE;qBACJ;iBACJ;YACL,CAAC;YAED,SAAS,0CAA0C,CAAC,KAAa;gBAC7D,KAAK,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;oBACjC,MAAM,WAAW,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;oBAE3B,IAAI,WAAW,CAAC,IAAI,KAAK,OAAO;wBAC5B,MAAM,CAAC,iEAAiE;oBAE5E,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,CAAC,IAAI,KAAK,QAAQ;wBACxC,MAAM,CAAC,gCAAgC;oBAE3C,IAAI,WAAW,CAAC,IAAI,KAAK,MAAM,IAAI,WAAW,CAAC,IAAI,KAAK,QAAQ,EAAE;wBAC9D,MAAM,OAAO,GAAG,2BAA2B,CAAC,WAAW,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;wBAEtF,IAAI,OAAO,KAAK,EAAE,EAAE;4BAChB,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;4BACjB,CAAC,EAAE,CAAC;4BACJ,sBAAsB,IAAI,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC;yBACrD;6BAAM,IAAI,OAAO,CAAC,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE;4BACjD,sBAAsB,IAAI,WAAW,CAAC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;4BACnE,WAAW,CAAC,IAAI,GAAG,OAAO,CAAC;yBAC9B;qBACJ;yBAAM;wBACH,KAAK,CAAC,WAA2B,CAAC,CAAC;qBACtC;iBACJ;YACL,CAAC;YAED,SAAS,0BAA0B;gBAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,IAAI,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE;oBAC/D,MAAM,WAAW,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;oBAC3B,MAAM,iBAAiB,GAAG,CAAC,KAAK,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC;oBAE/C,IAAI,WAAW,CAAC,IAAI,KAAK,OAAO;wBAC5B,SAAS;oBAEb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,QAAQ,CAAC,MAAM,IAAI,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE;wBAChF,MAAM,IAAI,GAAkD,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;wBACpF,MAAM,UAAU,GAAG,CAAC,KAAK,WAAW,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;wBAEzD,IAAI,iBAAiB,IAAI,UAAU;4BAC/B,SAAS;wBAEb,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE;4BAC1B,MAAM,OAAO,GAAG,2BAA2B,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;4BAE1E,IAAI,OAAO,KAAK,EAAE,EAAE;gCAChB,WAAW,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gCAClC,CAAC,EAAE,CAAC;gCACJ,sBAAsB,IAAI,IAAI,CAAC,MAAM,CAAC;6BACzC;iCAAM,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;gCACrC,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC;gCAClC,sBAAsB,IAAI,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;6BAC1D;yBACJ;6BAAM,IAAI,IAAI,CAAC,IAAI,KAAK,cAAc,EAAE;4BACrC,WAAW,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;4BAClC,CAAC,EAAE,CAAC;4BACJ,sBAAsB,IAAI,IAAI,CAAC,GAAG,EAAE,MAAM;gCACtC,WAAW,CAAC,6BAA6B,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC;yBAC7F;qBACJ;oBAED,IAAI,WAAW,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;wBACnC,qDAAqD;wBACrD,sDAAsD;wBACtD,0CAA0C,CAAC,CAAC,CAAC,CAAC;wBAC9C,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;wBACjB,CAAC,EAAE,CAAC;qBACP;iBACJ;YACL,CAAC;YAED,SAAS,yBAAyB,CAAC,sBAA8B,EAAE;gBAC/D,MAAM,eAAe,GAAG,GAAG,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAE5C,IAAI,eAAe,IAAI,IAAI,IAAI,eAAe,CAAC,IAAI,KAAK,OAAO;oBAC3D,OAAO;gBAEX,MAAM,gBAAgB,GAAG,eAAe,CAAC,QAAQ,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAEvF,IAAI,gBAAgB,IAAI,IAAI,IAAI,OAAO,gBAAgB,KAAK,QAAQ;oBAChE,OAAO;gBAEX,MAAM,cAAc,GAAG,gBAAgB,CAAC,MAAM,GAAG,sBAAsB,CAAC;gBACxE,MAAM,0BAA0B,GAAG,sBAAsB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,cAAc,GAAG,mBAAmB,CAAC,CAAC;gBAC9G,MAAM,OAAO,GAAG,2BAA2B,CAAC,gBAAgB,EAAE,0BAA0B,CAAC,CAAC;gBAE1F,IAAI,OAAO,CAAC,MAAM,GAAG,gBAAgB,CAAC,MAAM,EAAE;oBAC1C,eAAe,CAAC,QAAQ,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC;oBACxE,sBAAsB,IAAI,gBAAgB,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;iBACtE;YACL,CAAC;YAED,qBAAqB,EAAE,CAAC;YAExB,IAAI,sBAAsB,IAAI,CAAC;gBAC3B,OAAO,GAAG,CAAC;YAEf,0BAA0B,EAAE,CAAC;YAE7B,IAAI,sBAAsB,IAAI,CAAC;gBAC3B,OAAO,GAAG,CAAC;YAEf,yBAAyB,EAAE,CAAC;YAE5B,OAAO,GAAG,CAAC;QACf,CAAC;KACJ,CAAC,CAAC;IAEH,MAAM,WAAW,GAAwB;QACrC,uBAAuB,EAAE,sBAAsB;KAClD,CAAC;IAEF,OAAO;QACH,WAAW,EAAE,qBAAqB;QAClC,QAAQ,EAAE,WAAW;KACxB,CAAC;AACN,CAAC;AAMD,SAAS,qBAAqB,CAAC,QAAa;IACxC,OAAO,QAAQ,IAAI,IAAI,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,OAAO,QAAQ,CAAC,uBAAuB,KAAK,QAAQ,CAAC;AACpH,CAAC"}
@@ -78,6 +78,8 @@ export class LlamaChatSession {
78
78
  */
79
79
  async promptWithMeta(prompt, { functions, documentFunctionParams, onToken, signal, maxTokens, temperature, topK, topP, grammar, trimWhitespaceSuffix = false, repeatPenalty, evaluationPriority } = {}) {
80
80
  this._ensureNotDisposed();
81
+ if (grammar != null && grammar._llama !== this.model._llama)
82
+ throw new Error("The LlamaGrammar used by passed to this function was created with a different Llama instance than the one used by this sequence's model. Make sure you use the same Llama instance for both the model and the grammar.");
81
83
  return await withLock(this, "prompt", async () => {
82
84
  this._ensureNotDisposed();
83
85
  if (this._chat == null)
@@ -0,0 +1 @@
1
+ {"version":3,"file":"LlamaChatSession.js","sourceRoot":"","sources":["../../../src/evaluator/LlamaChatSession/LlamaChatSession.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,iBAAiB,EAAE,aAAa,EAAE,UAAU,EAAE,QAAQ,EAAC,MAAM,iBAAiB,CAAC;AACvF,OAAO,EAAC,uBAAuB,EAAC,MAAM,iBAAiB,CAAC;AAGxD,OAAO,EAAC,8BAA8B,EAAC,MAAM,+CAA+C,CAAC;AAG7F,OAAO,EAAC,SAAS,EAAkD,MAAM,2BAA2B,CAAC;AAgIrG,MAAM,OAAO,gBAAgB;IACzB,gBAAgB,CAAkB,kBAAkB,GAAG,IAAI,iBAAiB,EAAE,CAAC;IAC/E,gBAAgB,CAAkB,oBAAoB,CAAU;IAChE,gBAAgB,CAAkB,aAAa,CAAuC;IACtF,gBAAgB,CAAS,YAAY,CAAoB;IACzD,gBAAgB,CAAS,eAAe,CAAuC;IAC/E,gBAAgB,CAAS,KAAK,CAAmB;IAEjC,SAAS,GAAG,IAAI,UAAU,EAAQ,CAAC;IAEnD;;OAEG;IACH,YAAmB,EACf,eAAe,EACf,WAAW,GAAG,MAAM,EACpB,YAAY,GAAG,uBAAuB,EACtC,mBAAmB,GAAG,IAAI,EAC1B,YAAY,EACU;QACtB,IAAI,eAAe,IAAI,IAAI;YACvB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QAEtD,IAAI,eAAe,CAAC,QAAQ;YACxB,MAAM,IAAI,aAAa,EAAE,CAAC;QAE9B,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,YAAY,GAAG,CAAC;gBACjB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,YAAY;aACrB,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,GAAG,IAAI,SAAS,CAAC;YACvB,mBAAmB;YACnB,WAAW;YACX,eAAe;SAClB,CAAC,CAAC;QAEH,IAAI,CAAC,oBAAoB,GAAG,mBAAmB,CAAC;QAEhD,IAAI,CAAC,kBAAkB,CAAC,GAAG,CACvB,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,cAAc,CAAC,GAAG,EAAE;YACrC,IAAI,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC,CACL,CAAC;QACF,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IAC9D,CAAC;IAEM,OAAO,CAAC,EAAC,eAAe,GAAG,IAAI,CAAC,oBAAoB,KAAiC,EAAE;QAC1F,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI;YAClB,OAAO;QAEX,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAC,eAAe,EAAC,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAElB,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,CAAC;IACtC,CAAC;IAED,cAAc;IACP,CAAC,MAAM,CAAC,OAAO,CAAC;QACnB,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;IAED,IAAW,QAAQ;QACf,OAAO,IAAI,CAAC,KAAK,IAAI,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;IACrD,CAAC;IAED,IAAW,QAAQ;QACf,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI;YAClB,MAAM,IAAI,aAAa,EAAE,CAAC;QAE9B,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;IAC/B,CAAC;IAED,IAAW,OAAO;QACd,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;IACjC,CAAC;IAED,IAAW,KAAK;QACZ,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;IAC/B,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,MAAM,CAA4E,MAAc,EAAE,EAC3G,SAAS,EACT,sBAAsB,EACtB,OAAO,EACP,MAAM,EACN,SAAS,EACT,WAAW,EACX,IAAI,EACJ,IAAI,EACJ,OAAO,EACP,oBAAoB,GAAG,KAAK,EAC5B,aAAa,KACsB,EAAE;QACrC,MAAM,EAAC,YAAY,EAAC,GAAG,MAAM,IAAI,CAAC,cAAc,CAAY,MAAM,EAAE;YAChE,uEAAuE;YACvE,SAAS,EAAE,SAAsB;YACjC,sBAAsB,EAAE,sBAAmC;YAE3D,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,oBAAoB,EAAE,aAAa;SACpG,CAAC,CAAC;QAEH,OAAO,YAAY,CAAC;IACxB,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,cAAc,CAA4E,MAAc,EAAE,EACnH,SAAS,EACT,sBAAsB,EACtB,OAAO,EACP,MAAM,EACN,SAAS,EACT,WAAW,EACX,IAAI,EACJ,IAAI,EACJ,OAAO,EACP,oBAAoB,GAAG,KAAK,EAC5B,aAAa,EACb,kBAAkB,KACiB,EAAE;QACrC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAE1B,IAAI,OAAO,IAAI,IAAI,IAAI,OAAO,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,CAAC,MAAM;YACvD,MAAM,IAAI,KAAK,CAAC,wNAAwN,CAAC,CAAC;QAE9O,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,IAAI,EAAE;YAC7C,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAE1B,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI;gBAClB,MAAM,IAAI,aAAa,EAAE,CAAC;YAE9B,IAAI,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC;YAC1C,IAAI,cAAc,GAAG,8BAA8B,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YAC/E,IAAI,2BAA2B,GAAG,cAAc,EAAE,aAAa,IAAI,IAAI;gBACnE,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,8BAA8B,CAAC,cAAc,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;YAE5E,cAAc,CAAC,IAAI,CAAC;gBAChB,IAAI,EAAE,OAAO;gBACb,QAAQ,EAAE,EAAE;aACf,CAAC,CAAC;YAEH,IAAI,2BAA2B,IAAI,IAAI;gBACnC,2BAA2B,CAAC,IAAI,CAAC;oBAC7B,IAAI,EAAE,OAAO;oBACb,QAAQ,EAAE,EAAE;iBACf,CAAC,CAAC;YAEP,iDAAiD;YACjD,OAAO,IAAI,EAAE;gBACT,MAAM,EACF,YAAY,EACZ,cAAc,EAAE,qBAAqB,EACrC,QAAQ,EACX,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAY,cAAc,EAAE;oBAC7D,SAAS;oBACT,sBAAsB;oBACtB,OAAO,EAAE,OAAoB;oBAC7B,OAAO;oBACP,MAAM;oBACN,aAAa;oBACb,IAAI;oBACJ,IAAI;oBACJ,SAAS;oBACT,WAAW;oBACX,oBAAoB;oBACpB,YAAY,EAAE;wBACV,GAAG,IAAI,CAAC,aAAa;wBACrB,sBAAsB,EAAE,cAAc,EAAE,oBAAoB;qBAC/D;oBACD,kBAAkB;oBAClB,2BAA2B,EAAE;wBACzB,OAAO,EAAE,2BAA2B;wBACpC,6CAA6C,EAAE,IAAI;qBACtD;iBACJ,CAAC,CAAC;gBACH,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAE1B,cAAc,GAAG,qBAAqB,CAAC;gBACvC,cAAc,GAAG,cAAc,CAAC,YAAY,CAAC;gBAE7C,IAAI,YAAY,IAAI,IAAI,EAAE;oBACtB,MAAM,kBAAkB,GAAG,SAAS,EAAE,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;oBAElE,IAAI,kBAAkB,IAAI,IAAI;wBAC1B,MAAM,IAAI,KAAK,CAAC,qCAAqC,YAAY,CAAC,YAAY,wBAAwB,CAAC,CAAC;oBAE5G,MAAM,kBAAkB,GAAG,MAAM,kBAAkB,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;oBACjF,IAAI,CAAC,kBAAkB,EAAE,CAAC;oBAE1B,cAAc,GAAG,4BAA4B,CAAC;wBAC1C,WAAW,EAAE,cAAc;wBAC3B,YAAY,EAAE,YAAY,CAAC,YAAY;wBACvC,mBAAmB,EAAE,kBAAkB,CAAC,WAAW;wBACnD,UAAU,EAAE,YAAY,CAAC,MAAM;wBAC/B,UAAU,EAAE,kBAAkB;wBAC9B,GAAG,EAAE,YAAY,CAAC,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,0BAA0B,CACrE,YAAY,CAAC,YAAY,EACzB,YAAY,CAAC,MAAM,EACnB,kBAAkB,CACrB;qBACJ,CAAC,CAAC;oBAEH,2BAA2B,GAAG,4BAA4B,CAAC;wBACvD,WAAW,EAAE,cAAc,CAAC,aAAa;wBACzC,YAAY,EAAE,YAAY,CAAC,YAAY;wBACvC,mBAAmB,EAAE,kBAAkB,CAAC,WAAW;wBACnD,UAAU,EAAE,YAAY,CAAC,MAAM;wBAC/B,UAAU,EAAE,kBAAkB;wBAC9B,GAAG,EAAE,YAAY,CAAC,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,0BAA0B,CACrE,YAAY,CAAC,YAAY,EACzB,YAAY,CAAC,MAAM,EACnB,kBAAkB,CACrB;qBACJ,CAAC,CAAC;oBACH,cAAc,CAAC,YAAY,GAAG,cAAc,CAAC;oBAC7C,cAAc,CAAC,aAAa,GAAG,2BAA2B,CAAC;oBAE3D,SAAS;iBACZ;gBAED,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;gBACtC,IAAI,CAAC,YAAY,GAAG,cAAc,CAAC;gBAEnC,MAAM,qBAAqB,GAAG,wBAAwB,CAAC,cAAc,CAAC,CAAC;gBAEvE,OAAO;oBACH,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;oBACxC,YAAY,EAAE,qBAAqB,CAAC,QAAQ;yBACvC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC;yBAC1D,IAAI,CAAC,EAAE,CAAC;oBACb,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,4BAA4B,EAAE,QAAQ,CAAC,4BAA4B;iBACtE,CAAC;aACL;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAEM,cAAc;QACjB,OAAO,eAAe,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC;IAEM,8BAA8B;QACjC,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI;YAC5B,OAAO,IAAI,CAAC;QAEhB,OAAO,eAAe,CAAC,IAAI,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC;IAChE,CAAC;IAEM,cAAc,CAAC,WAA8B;QAChD,IAAI,CAAC,YAAY,GAAG,eAAe,CAAC,WAAW,CAAC,CAAC;QACjD,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;IACrC,CAAC;IAED,gBAAgB;IACR,kBAAkB;QACtB,IAAI,IAAI,CAAC,QAAQ;YACb,MAAM,IAAI,aAAa,EAAE,CAAC;IAClC,CAAC;CACJ;AAED,SAAS,4BAA4B,CAAC,EAClC,WAAW,EACX,YAAY,EACZ,mBAAmB,EACnB,UAAU,EACV,UAAU,EACV,GAAG,EAQN;IACG,MAAM,cAAc,GAAG,WAAW,CAAC,KAAK,EAAE,CAAC;IAC3C,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,IAAI,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO;QACzF,cAAc,CAAC,IAAI,CAAC;YAChB,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,EAAE;SACf,CAAC,CAAC;IAEP,MAAM,qBAAqB,GAAG,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAsB,CAAC;IAC7F,MAAM,wBAAwB,GAAG,EAAC,GAAG,qBAAqB,EAAC,CAAC;IAC5D,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,wBAAwB,CAAC;IAErE,MAAM,aAAa,GAAG,wBAAwB,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IAChE,wBAAwB,CAAC,QAAQ,GAAG,aAAa,CAAC;IAElD,aAAa,CAAC,IAAI,CAAC;QACf,IAAI,EAAE,cAAc;QACpB,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,mBAAmB;QAChC,MAAM,EAAE,UAAU;QAClB,MAAM,EAAE,UAAU;QAClB,GAAG;KACN,CAAC,CAAC;IAEH,OAAO,cAAc,CAAC;AAC1B,CAAC;AAED,SAAS,wBAAwB,CAAC,WAA8B;IAC5D,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO;QAChF,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;IAE1E,OAAO,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAsB,CAAC;AACpE,CAAC"}
@@ -0,0 +1 @@
1
+ {"version":3,"file":"defineChatSessionFunction.js","sourceRoot":"","sources":["../../../../src/evaluator/LlamaChatSession/utils/defineChatSessionFunction.ts"],"names":[],"mappings":"AAGA,MAAM,UAAU,yBAAyB,CAAkD,EACvF,WAAW,EACX,MAAM,EACN,OAAO,EAKV;IACG,OAAO;QACH,WAAW;QACX,MAAM;QACN,OAAO;KACV,CAAC;AACN,CAAC"}
@@ -1,9 +1,9 @@
1
1
  import { DisposeAggregator, EventRelay, withLock, DisposedError } from "lifecycle-utils";
2
2
  import { removeNullFields } from "../../utils/removeNullFields.js";
3
- import { AddonContext } from "../LlamaBins.js";
4
3
  import { compareTokens } from "../../utils/compareTokens.js";
5
4
  import { resolveBatchItemsPrioritizingStrategy } from "./utils/resolveBatchItemsPrioritizingStrategy.js";
6
5
  export class LlamaContext {
6
+ /** @internal */ _llama;
7
7
  /** @internal */ _ctx;
8
8
  /** @internal */ _onReclaimUnusedSequenceId = new EventRelay();
9
9
  /** @internal */ _model;
@@ -28,11 +28,12 @@ export class LlamaContext {
28
28
  constructor({ model, sequences = 1, seed = null, contextSize = model.trainContextSize, batchSize = contextSize, threads = 6, batching: { dispatchSchedule: batchingDispatchSchedule = "nextTick", itemsPrioritizingStrategy: batchingItemsPrioritizingStrategy = "maximumParallelism" } = {}, _embedding, _noSeed }) {
29
29
  if (model.disposed)
30
30
  throw new DisposedError();
31
+ this._llama = model._llama;
31
32
  this._model = model;
32
33
  this._totalSequences = Math.max(1, Math.floor(sequences));
33
34
  this._contextSize = Math.max(2, contextSize);
34
35
  this._batchSize = Math.max(batchSize, this._totalSequences);
35
- this._ctx = new AddonContext(this._model._model, removeNullFields({
36
+ this._ctx = new this._llama._bindings.AddonContext(this._model._model, removeNullFields({
36
37
  seed: seed != null ? Math.max(-1, Math.floor(seed)) : undefined,
37
38
  contextSize: contextSize * this._totalSequences,
38
39
  batchSize: this._batchSize,
@@ -489,6 +490,8 @@ export class LlamaContextSequence {
489
490
  const resolvedGrammarEvaluationState = grammarEvaluationState instanceof Function
490
491
  ? grammarEvaluationState()
491
492
  : grammarEvaluationState;
493
+ if (resolvedGrammarEvaluationState != null && resolvedGrammarEvaluationState._llama !== this.model._llama)
494
+ throw new Error("The LlamaGrammar used by passed to this function was created with a different Llama instance than the one used by this sequence's model. Make sure you use the same Llama instance for both the model and the grammar.");
492
495
  return this._context._ctx.sampleToken(batchLogitIndex, removeNullFields({
493
496
  temperature,
494
497
  topK,