Export from index.js

syonfox · Oct 31, 2020 · 392263a · 392263a
1 parent de921e5
commit 392263a
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 5 deletions.
diff --git a/Encoder.d.ts b/Encoder.d.ts
diff --git a/index.d.ts b/index.d.ts
@@ -0,0 +1,5 @@
+declare module "gpt-3-encoder" {
+  export function encode(text: string): string[];
+
+  export function decode(tokens: string[]): string;
+}
diff --git a/index.js b/index.js
@@ -0,0 +1,6 @@
+const { encode, decode } = require("./Encoder");
+
+module.exports = {
+  encode,
+  decode,
+};
diff --git a/package.json b/package.json
@@ -2,8 +2,8 @@
   "name": "gpt-3-encoder",
   "version": "1.1.3",
   "description": "Javascript BPE Encoder Decoder for GPT-2 / GPT-3",
-  "main": "Encoder.js",
-  "types": "Encoder.d.ts",
+  "main": "index.js",
+  "types": "index.d.ts",
   "files": [
     "encoder.json",
     "vocab.bpe"