First upload version 0.0.1
This commit is contained in:
22
node_modules/node-llama-cpp/dist/utils/getQueuedTokensBeforeStopTrigger.js
generated
vendored
Normal file
22
node_modules/node-llama-cpp/dist/utils/getQueuedTokensBeforeStopTrigger.js
generated
vendored
Normal file
@@ -0,0 +1,22 @@
|
||||
export function getQueuedTokensBeforeStopTrigger(triggeredStops, partiallyFreeTokens, tokenizer) {
|
||||
if (partiallyFreeTokens.tokens.length === 0 && partiallyFreeTokens.text.length === 0)
|
||||
return [];
|
||||
else if (partiallyFreeTokens.tokens.length !== 0 && partiallyFreeTokens.text.length === 0)
|
||||
return partiallyFreeTokens.tokens;
|
||||
else if (partiallyFreeTokens.tokens.length === 0 && partiallyFreeTokens.text.length !== 0)
|
||||
return tokenizer(partiallyFreeTokens.text, false, "trimLeadingSpace");
|
||||
const triggerThatStartsWithStringIndex = triggeredStops.findIndex((trigger) => trigger.stopTrigger.length > 0 && typeof trigger.stopTrigger[0] === "string");
|
||||
const triggerThatStartsWithTokenIndex = triggeredStops.findIndex((trigger) => trigger.stopTrigger.length > 0 && typeof trigger.stopTrigger[0] !== "string");
|
||||
if (triggerThatStartsWithTokenIndex > 0 && triggerThatStartsWithStringIndex < 0)
|
||||
return partiallyFreeTokens.tokens;
|
||||
else if (triggerThatStartsWithStringIndex > 0 && triggerThatStartsWithTokenIndex < 0)
|
||||
return tokenizer(partiallyFreeTokens.text, false, "trimLeadingSpace");
|
||||
const stringTokens = tokenizer(partiallyFreeTokens.text, false, "trimLeadingSpace");
|
||||
if (stringTokens.length === partiallyFreeTokens.tokens.length &&
|
||||
stringTokens.every((value, index) => value === partiallyFreeTokens.tokens[index]))
|
||||
return stringTokens;
|
||||
else if (triggerThatStartsWithStringIndex < triggerThatStartsWithTokenIndex)
|
||||
return stringTokens;
|
||||
return partiallyFreeTokens.tokens;
|
||||
}
|
||||
//# sourceMappingURL=getQueuedTokensBeforeStopTrigger.js.map
|
||||
Reference in New Issue
Block a user