feat: add nodejs commonjs module support

Author: magicds

cjs/index.js

@@ -0,0 +1,152 @@
+const fetchEventSource = require('@microsoft/fetch-event-source');
+const fetch = require('node-fetch');
+
+module.exports = class Api2d {
+    // 设置key和apiBaseUrl
+    constructor(key = null, apiBaseUrl = null, timeout = 60000) {
+        this.key = key;
+        this.apiBaseUrl = apiBaseUrl || (key && key.startsWith('fk') ? 'https://stream.api2d.net' : 'https://api.openai.com');
+        this.timeout = timeout;
+        this.controller = new AbortController();
+    }
+
+    // set key
+    setKey(key) {
+        this.key = key;
+    }
+
+    // set apiBaseUrl
+    setApiBaseUrl(apiBaseUrl) {
+        this.apiBaseUrl = apiBaseUrl;
+    }
+
+    setTimeout(timeout) {
+        this.timeout = parseInt(timeout) || 60 * 1000;
+    }
+
+    abort() {
+        this.controller.abort();
+    }
+
+    // Completion
+    async completion(options) {
+        // 拼接目标URL
+        const url = this.apiBaseUrl + '/v1/chat/completions';
+        // 拼接headers
+        const headers = {
+            'Content-Type': 'application/json',
+            Authorization: 'Bearer ' + this.key
+        };
+
+        const { onMessage, onEnd, model, ...restOptions } = options;
+
+        // 如果是流式返回，且有回调函数
+        if (restOptions.stream && onMessage) {
+            // 返回一个 Promise
+            return new Promise(async (resolve, reject) => {
+                try {
+                    let chars = '';
+                    console.log('in stream');
+                    // 使用 fetchEventSource 发送请求
+                    const timeout_handle = setTimeout(() => {
+                        this.controller.abort();
+                        // throw new Error( "Timeout "+ this.timeout );
+                        reject(new Error(`[408]:Timeout by ${this.timeout} ms`));
+                    }, this.timeout);
+                    const response = await fetchEventSource(url, {
+                        signal: this.controller.signal,
+                        method: 'POST',
+                        headers: { ...headers, Accept: 'text/event-stream' },
+                        body: JSON.stringify({ ...restOptions, model: model || 'gpt-3.5-turbo' }),
+                        async onopen(response) {
+                            if (response.status != 200) {
+                                throw new Error(`[${response.status}]:${response.statusText}`);
+                            }
+                        },
+                        onmessage: (e) => {
+                            if (timeout_handle) {
+                                clearTimeout(timeout_handle);
+                            }
+                            if (e.data == '[DONE]') {
+                                // console.log( 'DONE' );
+                                if (onEnd) onEnd(chars);
+                                resolve(chars);
+                            } else {
+                                // console.log( e.data );
+                                const event = JSON.parse(e.data);
+                                if (event.choices[0].delta.content) chars += event.choices[0].delta.content;
+                                if (onMessage) onMessage(chars);
+                            }
+                        },
+                        onerror: (error) => {
+                            console.log(error);
+                            throw new Error(String(error)?.match(/\[(\d+)\]/)?.[1] ? error : `[500]:${error}`);
+                        }
+                    });
+
+                    // const ret = await response.json();
+                } catch (error) {
+                    console.log(error);
+                    reject(error);
+                }
+            });
+        } else {
+            // 使用 fetch 发送请求
+            const response = await fetch(url, {
+                signal: this.controller.signal,
+                method: 'POST',
+                headers: headers,
+                body: JSON.stringify({ ...restOptions, model: model || 'gpt-3.5-turbo' })
+            });
+            const timeout_handle = setTimeout(() => {
+                this.controller.abort();
+            }, this.timeout);
+            const ret = await response.json();
+            clearTimeout(timeout_handle);
+            return ret;
+        }
+    }
+
+    async embeddings(options) {
+        // 拼接目标URL
+        const url = this.apiBaseUrl + '/v1/embeddings';
+        // 拼接headers
+        const headers = {
+            'Content-Type': 'application/json',
+            Authorization: 'Bearer ' + this.key
+        };
+        const { model, ...restOptions } = options;
+        // 使用 fetch 发送请求
+        const response = await fetch(url, {
+            signal: this.controller.signal,
+            method: 'POST',
+            headers: headers,
+            body: JSON.stringify({ ...restOptions, model: model || 'text-embedding-ada-002' })
+        });
+        const timeout_handle = setTimeout(() => {
+            this.controller.abort();
+        }, this.timeout);
+        const ret = await response.json();
+        clearTimeout(timeout_handle);
+        return ret;
+    }
+
+    async billing() {
+        const url = this.apiBaseUrl + '/dashboard/billing/credit_grants';
+        const headers = {
+            'Content-Type': 'application/json',
+            Authorization: 'Bearer ' + this.key
+        };
+        const response = await fetch(url, {
+            signal: this.controller.signal,
+            method: 'GET',
+            headers: headers
+        });
+        const timeout_handle = setTimeout(() => {
+            this.controller.abort();
+        }, this.timeout);
+        const ret = await response.json();
+        clearTimeout(timeout_handle);
+        return ret;
+    }
+};

package.json

@@ -2,12 +2,15 @@
   "name": "api2d",
   "version": "0.1.11",
   "description": "pure browser sdk for api2d and openai",
-  "main": "index.js",
+  "main": "cjs/index.js",
+  "module": "index.js",
+  "types": "index.d.ts",
   "repository": "https://github.com/easychen/api2d-js",
   "author": "EasyChen",
   "license": "MIT",
   "private": false,
   "dependencies": {
-    "@microsoft/fetch-event-source": "^2.0.1"
+    "@microsoft/fetch-event-source": "^2.0.1",
+    "node-fetch": "^2.6.9"
   }
 }