picoLLM Inference Engine
Node.js API

API Reference for the picoLLM Node.js SDK (npm)

PicoLLM

class PicoLLM {}

Class for the picoLLM Inference Engine.

picoLLM can be initialized using the class constructor(). Resources should be cleaned when you are done using the release() method.

PicoLLM.constructor()

PicoLLM.constructor(
  accessKey: string,
  modelPath: string,
  options: PicoLLMInitOptions = {}
)

picoLLM constructor.

Parameters

• accessKey string : AccessKey obtained from Picovoice Console.
• modelPath string : Absolute path to the file containing LLM parameters (.pllm).
• options PicoLLMOptions : Optional init configuration arguments, see PicoLLMInitOptions for details.

Returns

• PicoLLM : An instance of the picoLLM engine.

Throws

• PicoLLMError: If an error is encountered.

PicoLLM.generate()

async PicoLLM.generate(
  prompt: string,
  options: PicoLLMGenerateOptions = {}
): Promise<PicoLLMCompletion>

Given a text prompt and a set of generation parameters, creates a completion text and relevant metadata.

Parameters

• prompt string : Prompt text.
• options PicoLLMGenerateOptions : Generate configuration arguments, see PicoLLMGenerateOptions for details.

Returns

• PicoLLMCompletion : Completion result.

Throws

• PicoLLMError: If an error is encountered.

PicoLLM.interrupt()

PicoLLM.interrupt(): void

Interrupts .generate() if generation is in progress. Otherwise, it has no effect.

Throws

• PicoLLMError: If interrupt fails.

PicoLLM.tokenize()

PicoLLM.tokenize(
  text: string,
  bos: boolean,
  eos: boolean
): number[]

Tokenizes a given text using the model's tokenizer. This is a low-level function meant for benchmarking and advanced usage. .generate() should be used when possible.

Parameters

• text string : Text string.
• bos boolean : If set to true, the tokenizer prepends the beginning of the sentence token to the result.
• eso boolean : If set to true, the tokenizer appends the end of the sentence token to the result.

Returns

• number[] : Tokens representing the input text.

Throws

• PicoLLMError: If an error is encountered.

PicoLLM.forward()

PicoLLM.forward(token: number): number[]

Performs a single forward pass given a token and returns the logits. This is a low-level function for benchmarking and advanced usage. .generate() should be used when possible.

Parameters

• token number : Input token.

Returns

• number[] : Logits.

Throws

• PicoLLMError: If an error is encountered.

PicoLLM.reset()

PicoLLM.reset(): void

Resets the internal state of LLM. It should be called in conjunction with .forward() when processing a new sequence of tokens. This is a low-level function for benchmarking and advanced usage. .generate() should be used when possible.

Throws

• PicoLLMError: If an error is encountered.

PicoLLM.getDialog()

PicoLLM.getDialog(
  mode?: string,
  history: number = 0,
  system?: string
): Dialog

Return the Dialog object corresponding to the loaded model. The model needs to be instruction-tuned and have a specific chat template.

Parameters

• mode string? : Some models (e.g., phi-2) define multiple chat template models. For example, phi-2 allows both qa and chat templates.
• history number? : refers to the number of latest back-and-forths to include in the prompt. Setting history to undefined will embed the entire dialog in the prompt.
• system system : System instruction to embed in the prompt for configuring the model's responses.

Returns

• Dialog : Constructed dialog object.

Throws

• PicoLLMError: If an error is encountered.

PicoLLM.release()

PicoLLM.release(): void

Releases resources acquired by the picoLLM Node.js SDK.

PicoLLM.listAvailableDevices()

static PicoLLM.listAvailableDevices(): string[]

Lists all available devices that picoLLM can use for inference. Each entry in the list can be the device argument of the constructor.

Returns

• string[] : all available devices that picoLLM can use for inference.

Throws

• PicoLLMError: If an error is encountered.

PicoLLM.contextLength

get contextLength(): number

Model's context length.

PicoLLM.maxTopChoices

get maxTopChoices(): number

Maximum number of top choices for generate.

PicoLLM.model

get model(): string

Model's name.

PicoLLM.version

get version(): string

picoLLM version.

PicoLLMEndpoint

enum PicoLLMEndpoint {
  END_OF_SENTENCE,
  COMPLETION_TOKEN_LIMIT_REACHED,
  STOP_PHRASE_ENCOUNTERED,
}

Enum of picoLLM endpoints.

PicoLLMInitOptions

type PicoLLMInitOptions = {
  device?: string;
  libraryPath?: string;
};

picoLLM init options type.

• device string? : String representation of the device (e.g., CPU or GPU) to use for inference. If set to best, picoLLM picks the most suitable device. If set to gpu, the engine uses the first available GPU device. To select a specific GPU device, set this argument to gpu:${GPU_INDEX}, where ${GPU_INDEX} is the index of the target GPU. If set to cpu, the engine will run on the CPU with the default number of threads. To specify the number of threads, set this argument to cpu:${NUM_THREADS}, where ${NUM_THREADS} is the desired number of threads.
• libraryPath string? : Absolute path to picoLLM's dynamic library.

PicoLLMGenerateOptions

type PicoLLMGenerateOptions = {
  completionTokenLimit?: number;
  stopPhrases?: string[];
  seed?: number;
  presencePenalty?: number;
  frequencyPenalty?: number;
  temperature?: number;
  topP?: number;
  numTopChoices?: number;
  streamCallback?: (token: string) => void;
}

picoLLM generate options type.

• completionTokenLimit number : Maximum number of tokens in the completion. If the generation process stops due to reaching this limit, the .endpoint parameter in PicoLLMCompletion output will be PicoLLMEndpoint.COMPLETION_TOKEN_LIMIT_REACHED. Set to undefined to impose no limit.
• stopPhrases string[] : The generation process stops when it encounters any of these phrases in the completion. The already generated completion, including the encountered stop phrase, will be returned. The endpoint parameter in PicoLLMCompletion output will be PicoLLMEndpoint.STOP_PHRASE_ENCOUNTERED. Set to undefined to turn off this feature.
• seed number : The internal random number generator uses it as its seed if set to a positive integer value. Seeding enforces deterministic outputs. Set to undefined for randomized outputs for a given prompt.
• presencePenalty number : It penalizes logits already appearing in the partial completion if set to a positive value. If set to 0 or undefined, it has no effect.
• frequencyPenalty number : If set to a positive floating-point value, it penalizes logits proportional to the frequency of their appearance in the partial completion. If set to 0 or undefined, it has no effect.
• temperature number : Sampling temperature. Temperature is a non-negative floating-point value that controls the randomness of the sampler. A higher temperature smoothens the samplers' output, increasing the randomness. In contrast, a lower temperature creates a narrower distribution and reduces variability. Setting it to 0 or undefined selects the maximum logit during sampling.
• topP number : A positive floating-point number within 0, and 1. It restricts the sampler's choices to high-probability logits that form the topP portion of the probability mass. Hence, it avoids randomly selecting unlikely logits. A value of 1 or undefined enables the sampler to pick any token with non-zero probability turning off the feature.
• numTopChoices number : If set to a positive value, picoLLM returns the list of the highest probability tokens for any generated token. Set to 0 to turn off the feature. The maximum number of top choices is .maxTopChoices.
• streamCallback (token: string) => void : If not set to undefined, picoLLM executes this callback every time a new piece of completion string becomes available.

PicoLLMUsage

type PicoLLMUsage = {
  promptTokens: number;
  completionTokens: number;
}

picoLLM usage type.

• promptTokens number : Number of tokens in the prompt.
• completionTokens number : Number of tokens in the completion.

PicoLLMToken

type PicoLLMToken = {
  token: string;
  logProb: number;
}

picoLLM token type.

• token string : Token string.
• logProb number : Log probability.

PicoLLMCompletionToken

type PicoLLMCompletionToken = {
  token: PicoLLMToken;
  topChoices: PicoLLMToken[];
}

picoLLM completion token type.

• token PicoLLMToken : PicoLLMToken.
• topChoices PicoLLMToken[] : Top PicoLLMToken choices.

PicoLLMCompletion

type PicoLLMCompletion = {
  usage: PicoLLMUsage;
  endpoint: PicoLLMEndpoint;
  completionTokens: PicoLLMCompletionToken[];
  completion: string;
}

picoLLM completion type.

• usage PicoLLMUsage : Usage information.
• endpoint PicoLLMEndpoint[] : Reason for ending the generation process.
• completionTokens PicoLLMCompletionToken[] : Generated tokens within completion and top alternative tokens.
• completion string : Completion string.

PicoLLMError

class PicoLLMError extends Error { }

Error thrown if an error occurs within picoLLM Inference Engine.

Errors:

class PicoLLMActivationError           extends PicoLLMError { }
class PicoLLMActivationLimitError      extends PicoLLMError { }
class PicoLLMActivationRefusedError    extends PicoLLMError { }
class PicoLLMActivationThrottledError  extends PicoLLMError { }
class PicoLLMIOError                   extends PicoLLMError { }
class PicoLLMInvalidArgumentError      extends PicoLLMError { }
class PicoLLMInvalidStateError         extends PicoLLMError { }
class PicoLLMKeyError                  extends PicoLLMError { }
class PicoLLMMemoryError               extends PicoLLMError { }
class PicoLLMRuntimeError              extends PicoLLMError { }
class PicoLLMStopIterationError        extends PicoLLMError { }

Dialog

class Dialog {}

Dialog is a helper class that stores a chat dialog and formats it according to an instruction-tuned LLM's chat template. Dialog is the base class. Each supported instruction-tuned LLM has an accompanying concrete subclass.

Dialog.constructor

Dialog.constructor(
  history?: number,
  system?: string
)

Dialog constructor.

Parameters

• history number? : The number of latest back-and-forths to include in the prompt. Setting history to undefined will embed the entire dialog in the prompt.
• system string? : Instruction to embed in the prompt for configuring the model's responses.

Returns

• Dialog: An instance of Dialog class.

Dialog.addHumanRequest

function addHumanRequest(content: string): void

Adds human's request to the dialog.

Parameters

• content string : Human's request.

Dialog.addLLMResponse

function addLLMResponse(content: string): void

Adds LLM's response to the dialog.

Parameters

• content string : LLM's response.

Dialog.prompt

function prompt(): string

Creates a prompt string given parameters passed the constructor and dialog's content.

Returns

• string : Formatted prompt.

GemmaChatDialog

class GemmaChatDialog extends Dialog {}

Dialog helper for gemma-2b-it.

Llama2ChatDialog

class Llama2ChatDialog extends Dialog {}

Dialog helper for llama-2-7b-chat.

Llama3ChatDialog

class Llama3ChatDialog extends Dialog {}

Dialog helper for llama-3-8b-instruct.

MistralChatDialog

class MistralChatDialog extends Dialog {}

Dialog helper for mistral-7b-instruct-v0.1 and mistral-7b-instruct-v0.2.

MixtralChatDialog

class MixtralChatDialog extends Dialog {}

Dialog helper for mixtral-8x7b-instruct-v0.1.

Phi2Dialog

class Phi2Dialog extends Dialog {}

Dialog helper for phi-2 base class.

Phi2Dialog.constructor

Phi2Dialog.constructor(
  humanTag: string,
  llmTag: string,
  history?: number,
  system?: string
)

typescript constructor.

Parameters

• humanTag string : Tag to classify human requests.
• llmTag string :Tag to classify llm responses.
• history number? : The number of latest back-and-forths to include in the prompt. Setting history to undefined will embed the entire dialog in the prompt.
• system string? : Instruction to embed in the prompt for configuring the model's responses.

Returns

• Phi2Dialog: An instance of Phi2Dialog class.

Phi2QADialog

class Phi2QADialog extends Phi2Dialog {}

Dialog helper for phi-2 qa mode.

Phi2ChatDialog

class Phi2ChatDialog extends Phi2Dialog {}

Dialog helper for phi-2 chat mode.