gptoken
TypeScript icon, indicating that this package has built-in type declarations

0.1.0 • Public • Published

GPToken is an expanded project based on the original

npm version

New npm version

JSDocs

Also check out the browser demo browser demo

GitHub last commit example workflow github

TLDR

npm install gptoken
const gptoken = require('gptoken')

let tokens = gptoken.encode("hello world, we all share a goal of life")
console.log("Tokens: ", tokens);
console.log("TokenStats: ", JSON.stringify(gptoken.tokenStats(tokens)));

//or browser demo
firefox ../node_modules/gptoken/browser.html

Or check out the full express demo

cd demo_app
npm install 
npm start

Fork Purpose

I have created this to add general gpt helper functionality as well as creat a compleat pakage.

The plan is to clean up and stabilize this original implementation.

I would like to then make this usefully for other models and features.

Roadmap

Improved performance More utilities function.

More research on how to interact with GPT models.

Add a simple elagent openai api integration so this can be a minimal frontend and backend base.

Intro

Javascript library for encoding and decoding text using Byte Pair Encoding (BPE), as used in GPT-2 and GPT-3 models by OpenAI. This is a fork of the original python implementation by OpenAI, which can be found here.

This fork includes additional features such as the countTokens and tokenStats functions, as well as updated documentation.

Installation

To install with npm:

npm install gptoken
or old
npm install @syonfox/GPT3-encoder

Overview

The main interface is defined in index.js or index.d.ts

The code is in Encoder.js

The Encoding data/ maps are in the bpe_data directory this is loaded by Encoder to perfrom the conversion.

There are useful scripts defined in pakage.json

The tests are using jest and are defined in Encoder.test.js

docs are built using jsdoc npm run doc and we need to cp browser.* docs/ after build so demo works on github pages

There are 2 demos one using nodejs npm run demo and one using the browserify version in a html page npm run browser

Usage

Compatible with Node >= 12

To use the library in your project, import it as follows:

const gptoken = require('gptoken');

TODO add nextjs browser node and react examples for browser ejs node and other syntax

Additional Features (check out the docs :)

In addition to the original encoding and decoding functions, this fork includes the following additional features:

countTokens(text: string): number This function returns the number of tokens in the provided text, after encoding it using BPE.

tokenStats(text: string): object This function returns an object containing statistics about the tokens in the provided text, after encoding it using BPE. The returned object includes the following properties:

  • count: the total number of tokens in the text.
  • unique: the number of unique tokens in the text.
  • frequencies: an object containing the frequency of each token in the text.
  • postions: an object mapping tokens to positions in the encoded string
  • tokens: same as the output to tokens

Compatibility

Node.js 12+

This library is compatible with both Node.js const gptoken = require('gptoken'); and browser environments, we have used webpack to build /dist/bundle.js 1.5 MB including the data. A compiled version for both environments is included in the package.

Browser

<script src="/js/gptoken/browser.js"></script> and

cp -r node_modules/gptoken ./public/js/gptoken or

app.use('/js/gptoken', express.static(path.join(__dirname, 'node_modules/gptoken')));'

Credits

This library was created as a fork of the original GPT-3-Encoder library by latitudegames.

Example

See browser.html and demo.js Note you may need to include it from the appropriate place in node modules / npm package name

import {encode, decode, countTokens, tokenStats} from "gptoken"
//or note you might need @syonfox/gpt-3-encoder if thats what you npm install
const {encode, decode, countTokens, tokenStats} = require('gptoken')

const str = 'This is an example sentence to try encoding out on!'
const encoded = encode(str)
console.log('Encoded this string looks like: ', encoded)

console.log('We can look at each token and what it represents')
for (let token of encoded) {
    console.log({token, string: decode([token])})
}

//example count tokens usage
if (countTokens(str) > 5) {
    console.log("String is over five tokens, inconcevable");
}

const decoded = decode(encoded)
console.log('We can decode it back into:\n', decoded)

Developers

I have added som other examples to the examples folder. Please take a look at package.json for how to do stuff

//the original repo
git clone https://github.com/syonfox/GPT-3-Encoder.git

cd GPT-3-Encoder

npm install # install dev deps (docs tests build)

npm run test # run tests
npm run docs # build docs

npm run build # builds it for the browser
npm run browser # launches demo in firefox
npm run demo # runs node.js demo


less Encoder.js # the main code is here

firefox ./docs/index.html # view docs locally

npm publish --access public # dev publish to npm


Performance

Built bpe_ranks in 100 ms

// using js loading (probably before cache) Loaded encoder in 121 ms Loaded bpe_ranks in 91 ms

// using fs loading Loaded encoder in 32 ms Loaded bpe_ranks in 44 ms

//back to js loading Loaded encoder in 35 ms Loaded bpe_ranks in 40 ms

todo

More stats that work well with this token representation.

Clean up and keep it simple.

Here are some additional suggestions for improving the GPT-3 Encoder:

  • Add more unit tests to ensure the correctness and reliability of the code. This can be particularly important for the encode and decode functions, which are the main functions of the encoder.
  • Add more documentation and examples to help users understand how to use the encoder and integrate it into their own projects. This could include additional JSDoc comments, as well as additional documentation in the README file and/or GitHub Pages.
  • Consider adding support for other languages and character sets. Currently, the encoder only supports ASCII characters, but there may be a demand for support for other languages and character sets.
  • Explore potential optimizations and performance improvements for the encode and decode functions. Some ideas might include using faster data structures (such as a hash map or a trie), implementing more efficient algorithms, or using multi-threading or web workers to take advantage of multiple cores or processors.
  • Consider adding support for other models or use cases. For example, you could add support for other OpenAI models ( such as GPT-2 or GPT-3) or for other applications of BPE encoding (such as machine translation or natural language processing).

Package Sidebar

Install

npm i gptoken

Weekly Downloads

563

Version

0.1.0

License

MIT

Unpacked Size

5.42 MB

Total Files

15

Last publish

Collaborators

  • syonfox