Merge branch 'master' into syonfox-dev

syonfox · Dec 25, 2022 · ea8540d · ea8540d
2 parents de619c0 + 04cf547
commit ea8540d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/index.d.ts b/index.d.ts
@@ -2,4 +2,6 @@ declare module "gpt-3-encoder" {
   export function encode(text: string): number[];
 
   export function decode(tokens: number[]): string;
+
+  export function countTokens(text: string): number;
 }
diff --git a/index.js b/index.js
@@ -1,6 +1,7 @@
-const { encode, decode } = require("./Encoder");
+const { encode, decode, countTokens } = require("./Encoder");
 
 module.exports = {
   encode,
   decode,
+  countTokens,
 };