openai 1.0.0 → 2.0.0

package/dist/index.js

@@ -1,193 +1,27 @@
-(function (factory) {
-    if (typeof module === "object" && typeof module.exports === "object") {
-        var v = factory(require, exports);
-        if (v !== undefined) module.exports = v;
-    }
-    else if (typeof define === "function" && define.amd) {
-        define(["require", "exports", "tslib", "stream", "form-data", "node-fetch", "./types"], factory);
-    }
-})(function (require, exports) {
-    "use strict";
-    Object.defineProperty(exports, "__esModule", { value: true });
-    exports.OpenAI = void 0;
-    const tslib_1 = require("tslib");
-    const stream_1 = require("stream");
-    const form_data_1 = (0, tslib_1.__importDefault)(require("form-data"));
-    const node_fetch_1 = (0, tslib_1.__importDefault)(require("node-fetch"));
-    const baseUrl = 'https://api.openai.com';
-    const defaultVersion = 'v1';
-    (0, tslib_1.__exportStar)(require("./types"), exports);
-    class OpenAI {
-        url;
-        headers;
-        constructor(apiKey, organizationId, version = defaultVersion) {
-            this.headers = {
-                'authorization': `Bearer ${apiKey}`,
-                'content-type': 'application/json',
-            };
-            if (organizationId) {
-                this.headers['openai-organization'] = organizationId;
-            }
-            this.url = `${baseUrl}/${version}`;
-        }
-        getEngines() {
-            return this.request('/engines', 'GET').then((r) => r.data);
-        }
-        getEngine(engine) {
-            return this.request(`/engines/${engine}`, 'GET');
-        }
-        complete(engine, options) {
-            return this.request(`/engines/${engine}/completions`, 'POST', options);
-        }
-        completeFromModel(fineTunedModel, options) {
-            return this.request(`/completions`, 'POST', { ...options, model: fineTunedModel });
-        }
-        async completeAndStream(engine, options) {
-            const request = await this.requestRaw(`/engines/${engine}/completions`, 'POST', { ...options, stream: true });
-            return request.body.pipe(this.eventStreamTransformer());
-        }
-        async completeFromModelAndStream(fineTunedModel, options) {
-            const request = await this.requestRaw(`/completions`, 'POST', {
-                ...options,
-                model: fineTunedModel,
-                stream: true,
-            });
-            return request.body.pipe(this.eventStreamTransformer());
-        }
-        async contentFilter(content, user) {
-            const completion = await this.complete('content-filter-alpha-c4', {
-                prompt: `<|endoftext|>${content}\n--\nLabel:`,
-                temperature: 0,
-                max_tokens: 1,
-                top_p: 1,
-                frequency_penalty: 0,
-                presence_penalty: 0,
-                logprobs: 10,
-                user,
-            });
-            let label = Number(completion.choices[0].text);
-            if (label === 2) {
-                const logprobs = completion.choices[0].logprobs?.top_logprobs[0];
-                if (logprobs && logprobs['2'] < -0.355) {
-                    if (logprobs['0'] && logprobs['1']) {
-                        label = logprobs['0'] >= logprobs['1'] ? 0 : 1;
-                    }
-                    else if (logprobs['0']) {
-                        label = 0;
-                    }
-                    else if (logprobs['1']) {
-                        label = 1;
-                    }
-                }
-            }
-            if (![0, 1, 2].includes(label)) {
-                label = 2;
-            }
-            return label;
-        }
-        search(engine, options) {
-            return this.request(`/engines/${engine}/search`, 'POST', options).then((r) => r.data);
-        }
-        classify(options) {
-            return this.request('/classifications', 'POST', options);
-        }
-        answer(options) {
-            return this.request('/answers', 'POST', options);
-        }
-        getFiles() {
-            return this.request('/files', 'GET').then((r) => r.data);
-        }
-        uploadFile(file, jsonlines, purpose) {
-            const data = new form_data_1.default();
-            let fileJsonlines;
-            if (Array.isArray(jsonlines)) {
-                if (typeof jsonlines[0] === 'object') {
-                    jsonlines = jsonlines.map((j) => JSON.stringify(j));
-                }
-                fileJsonlines = jsonlines.join('\n');
-            }
-            else {
-                fileJsonlines = jsonlines;
-            }
-            data.append('file', fileJsonlines, file);
-            data.append('purpose', purpose);
-            return this.request('/files', 'POST', data);
-        }
-        getFile(fileId) {
-            return this.request(`/files/${fileId}`, 'GET');
-        }
-        deleteFile(fileId) {
-            return this.request(`/files/${fileId}`, 'DELETE');
-        }
-        finetune(options) {
-            return this.request(`/fine-tunes`, 'POST', options);
-        }
-        getFinetunes() {
-            return this.request('/fine-tunes', 'GET').then((r) => r.data);
-        }
-        getFinetune(finetuneId) {
-            return this.request(`/fine-tunes/${finetuneId}`, 'GET');
-        }
-        cancelFinetune(finetuneId) {
-            return this.request(`/fine-tunes/${finetuneId}/cancel`, 'POST');
-        }
-        getFinetuneEvents(finetuneId) {
-            return this.request(`/fine-tunes/${finetuneId}/events`, 'GET').then((r) => r.data);
-        }
-        async requestRaw(path, method, body) {
-            let headers = { ...this.headers };
-            if (body instanceof form_data_1.default) {
-                delete headers['content-type'];
-                headers = body.getHeaders(headers);
-            }
-            else if (!['string', 'undefined'].includes(typeof body)) {
-                body = JSON.stringify(body);
-            }
-            const response = await (0, node_fetch_1.default)(this.url + path, {
-                headers,
-                method,
-                body: body,
-            });
-            if (!response.ok) {
-                let errorBody;
-                try {
-                    const { error: { message }, } = await response.json();
-                    errorBody = message;
-                }
-                catch {
-                    try {
-                        errorBody = await response.text();
-                    }
-                    catch {
-                        errorBody = 'Failed to get body as text';
-                    }
-                }
-                throw new Error(`OpenAI did not return ok: ${response.status} ~ Error body: ${errorBody}`);
-            }
-            return response;
-        }
-        async request(path, method, body) {
-            const response = await this.requestRaw(path, method, body);
-            return response.json();
-        }
-        eventStreamTransformer() {
-            const dataHeader = Buffer.from('data: ');
-            return new stream_1.Transform({
-                transform: function (chunk, _, callback) {
-                    if (chunk.length >= dataHeader.length &&
-                        dataHeader.compare(chunk, undefined, dataHeader.length) === 0) {
-                        if (this.prevChunk) {
-                            const completion = JSON.parse(this.prevChunk.toString());
-                            this.push(completion.choices[0].text);
-                            this.prevChunk = undefined;
-                        }
-                        chunk = chunk.slice(dataHeader.length);
-                    }
-                    this.prevChunk = this.prevChunk ? Buffer.concat([this.prevChunk, chunk]) : chunk;
-                    callback();
-                },
-            });
-        }
-    }
-    exports.OpenAI = OpenAI;
-});
+"use strict";
+/* tslint:disable */
+/* eslint-disable */
+/**
+ * OpenAI API
+ * APIs for sampling from and fine-tuning language models
+ *
+ * The version of the OpenAPI document: 1.0.0
+ *
+ *
+ * NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
+ * https://openapi-generator.tech
+ * Do not edit the class manually.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !exports.hasOwnProperty(p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./api"), exports);
+__exportStar(require("./configuration"), exports);

package/index.ts

@@ -0,0 +1,18 @@
+/* tslint:disable */
+/* eslint-disable */
+/**
+ * OpenAI API
+ * APIs for sampling from and fine-tuning language models
+ *
+ * The version of the OpenAPI document: 1.0.0
+ * 
+ *
+ * NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
+ * https://openapi-generator.tech
+ * Do not edit the class manually.
+ */
+
+
+export * from "./api";
+export * from "./configuration";
+

package/package.json

@@ -1,45 +1,27 @@
 {
-    "name": "openai",
-    "version": "1.0.0",
-    "description": "Tiny OpenAI API wrapper",
-    "main": "dist/index.js",
-    "scripts": {
-        "start": "tsc -d --watch",
-        "build": "tsc -d",
-        "clean": "rm -rf dist",
-        "lint": "eslint . --fix --cache && prettier --write src",
-        "lint:nofix": "eslint . && prettier --check src"
-    },
-    "repository": {
-        "type": "git",
-        "url": "git+https://github.com/ceifa/openai.git"
-    },
-    "author": "ceifa",
-    "license": "MIT",
-    "keywords": [
-        "gpt-3",
-        "gpt",
-        "ai"
-    ],
-    "bugs": {
-        "url": "https://github.com/ceifa/openai/issues"
-    },
-    "homepage": "https://github.com/ceifa/openai#readme",
-    "devDependencies": {
-        "@types/node": "16.10.5",
-        "@types/node-fetch": "2.5.12",
-        "@typescript-eslint/eslint-plugin": "5.0.0",
-        "@typescript-eslint/parser": "5.0.0",
-        "eslint": "8.0.0",
-        "eslint-config-prettier": "8.3.0",
-        "eslint-plugin-prettier": "4.0.0",
-        "eslint-plugin-sort-imports-es6-autofix": "0.6.0",
-        "prettier": "2.4.1",
-        "typescript": "4.4.4"
-    },
-    "dependencies": {
-        "form-data": "4.0.0",
-        "node-fetch": "2.6.5",
-        "tslib": "2.3.1"
-    }
+  "name": "openai",
+  "version": "2.0.0",
+  "description": "Node.js library for the OpenAI API",
+  "keywords": [
+    "openai",
+    "open",
+    "ai",
+    "gpt-3",
+    "gpt3"
+  ],
+  "author": "OpenAI",
+  "license": "MIT",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "scripts": {
+    "build": "tsc --outDir dist/"
+  },
+  "dependencies": {
+    "axios": "^0.25.0",
+    "form-data": "^4.0.0"
+  },
+  "devDependencies": {
+    "@types/node": "^12.11.5",
+    "typescript": "^3.6.4"
+  }
 }

package/tsconfig.json

@@ -0,0 +1,17 @@
+{
+  "compilerOptions": {
+    "declaration": true,
+    "target": "es6",
+    "module": "commonjs",
+    "noImplicitAny": true,
+    "outDir": "dist",
+    "rootDir": ".",
+    "typeRoots": [
+      "node_modules/@types"
+    ]
+  },
+  "exclude": [
+    "dist",
+    "node_modules"
+  ]
+}

package/dist/types.d.ts

@@ -1,175 +0,0 @@
-export declare type EngineId = 'davinci' | 'curie' | 'babbage' | 'ada' | string;
-export interface Engine {
-    id: EngineId;
-    object: 'engine';
-    created?: Date;
-    max_replicas?: number;
-    owner: string;
-    permissions: unknown;
-    ready: boolean;
-    ready_replicas: unknown;
-    replicas: unknown;
-}
-export interface List<T> {
-    object: 'list';
-    data: T[];
-}
-export interface CompletionRequest {
-    prompt?: string | string[];
-    max_tokens?: number;
-    temperature?: number;
-    top_p?: number;
-    n?: number;
-    logprobs?: number;
-    echo?: boolean;
-    stop?: string | string[];
-    presence_penalty?: number;
-    frequency_penalty?: number;
-    best_of?: number;
-    logit_bias?: Record<string, unknown>;
-    user?: string;
-}
-export interface LogProbs {
-    tokens: string[];
-    token_logprobs: number[];
-    top_logprobs: Array<Record<string, number>>;
-    text_offset: number[];
-}
-export interface Choice {
-    text: string;
-    index: number;
-    logprobs: LogProbs | null;
-    finish_reason: string | null;
-}
-export interface Completion {
-    id: string;
-    object: 'text_completion';
-    created: number;
-    model: string;
-    choices: Choice[];
-}
-export interface SearchRequest {
-    query: string;
-    documents?: string[];
-    file?: string;
-    max_rerank?: number;
-    return_metadata?: boolean;
-}
-export interface SearchDocument {
-    document: number;
-    object: 'search_result';
-    score: number;
-}
-export interface ClassificationRequest {
-    model: string;
-    query: string;
-    examples?: string[];
-    file?: string;
-    labels?: string[] | null;
-    search_model?: string;
-    temperature?: number;
-    logprobs?: number;
-    max_examples?: number;
-    logit_bias?: Record<string, unknown>;
-    return_prompt?: boolean;
-    return_metadata?: boolean;
-    expand?: string[];
-}
-export interface ClassificationExample {
-    document: number;
-    label: string;
-    text: string;
-}
-export interface Classification {
-    completion: string;
-    label: string;
-    model: string;
-    object: 'classification';
-    search_model: string;
-    selected_examples: ClassificationExample[];
-}
-export interface AnswerRequest {
-    model: string;
-    question: string;
-    examples: Array<[string, string]>;
-    examples_context: string;
-    documents?: string[];
-    file?: string;
-    search_model?: string;
-    max_rerank?: number;
-    temperature?: number;
-    logprobs?: number;
-    max_tokens?: number;
-    stop?: string | string[];
-    n?: number;
-    logit_bias?: Record<string, unknown>;
-    return_metadata?: boolean;
-    return_prompt?: boolean;
-    expand?: string[];
-}
-export interface AnswerDocument {
-    document: number;
-    text: string;
-}
-export interface Answer {
-    answers: string[];
-    completion: string | Completion;
-    model: string;
-    object: 'answer';
-    search_model: string;
-    prompt: string;
-    selected_documents: AnswerDocument[];
-}
-export declare type FilePurpose = 'search' | 'answers' | 'classifications' | 'fine-tune';
-export declare type JsonLines = string | string[] | unknown[];
-export interface File {
-    id: string;
-    object: string;
-    bytes: number;
-    created_at: number;
-    filename: string;
-    purpose: FilePurpose;
-}
-export interface Hyperparams {
-    n_epochs?: number;
-    batch_size?: number;
-    learning_rate_multiplier?: number;
-    use_packing?: boolean;
-    prompt_loss_weight?: number;
-}
-export interface FineTuneRequest extends Hyperparams {
-    training_file: string;
-    validation_file?: string;
-    model?: string;
-    compute_classification_metrics?: boolean;
-    classification_n_classes?: number;
-    classification_positive_class?: string;
-    classification_betas?: number[];
-}
-export interface FineTuneEvent {
-    object: 'fine-tune-event';
-    created_at: number;
-    level: string;
-    message: string;
-}
-export interface FineTune {
-    id: string;
-    object: 'fine-tune';
-    model: string;
-    created_at: number;
-    events: FineTuneEvent[];
-    fine_tuned_model: string;
-    hyperparams: Hyperparams;
-    organization_id: string;
-    result_files: File[];
-    status: string;
-    validation_files: File[];
-    training_files: File[];
-    updated_at: number;
-    user_id: string;
-}
-export declare const enum ContentLabel {
-    Safe = 0,
-    Sensitive = 1,
-    Unsafe = 2
-}

package/dist/types.js

@@ -1,12 +0,0 @@
-(function (factory) {
-    if (typeof module === "object" && typeof module.exports === "object") {
-        var v = factory(require, exports);
-        if (v !== undefined) module.exports = v;
-    }
-    else if (typeof define === "function" && define.amd) {
-        define(["require", "exports"], factory);
-    }
-})(function (require, exports) {
-    "use strict";
-    Object.defineProperty(exports, "__esModule", { value: true });
-});