braintrust 0.0.26 → 0.0.28

package/dist/oai.js

@@ -1,104 +0,0 @@
-"use strict";
-// TODO REPLACE WITH autoevals version
-// https://github.com/braintrustdata/braintrust/issues/218
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.cachedChatCompletion = exports.openAI = void 0;
-const path = __importStar(require("path"));
-const sqlite3 = __importStar(require("sqlite3"));
-const fs = __importStar(require("fs"));
-const cache_1 = require("./cache");
-const openai_1 = require("openai");
-const OAI_CACHE = path.join(cache_1.CACHE_PATH, "oai.sqlite");
-let _db = null;
-function openCache() {
-    return __awaiter(this, void 0, void 0, function* () {
-        if (_db === null) {
-            fs.mkdirSync(cache_1.CACHE_PATH, { recursive: true });
-            _db = new sqlite3.Database(OAI_CACHE);
-            yield new Promise((resolve, reject) => {
-                _db.run("CREATE TABLE IF NOT EXISTS cache (params text, response text)", (err) => {
-                    if (err) {
-                        reject(err);
-                    }
-                    else {
-                        resolve(undefined);
-                    }
-                });
-            });
-        }
-        return _db;
-    });
-}
-let _openai = null;
-function openAI() {
-    if (_openai === null && process.env.OPENAI_API_KEY) {
-        const config = new openai_1.Configuration({ apiKey: process.env.OPENAI_API_KEY });
-        _openai = new openai_1.OpenAIApi(config);
-    }
-    return _openai;
-}
-exports.openAI = openAI;
-function cachedChatCompletion(args) {
-    return __awaiter(this, void 0, void 0, function* () {
-        const db = yield openCache();
-        const param_key = JSON.stringify(args);
-        const query = `SELECT response FROM "cache" WHERE params=?`;
-        const resp = yield new Promise((resolve, reject) => {
-            db.get(query, [param_key], (err, row) => {
-                if (err) {
-                    reject(err);
-                }
-                else {
-                    resolve(row);
-                }
-            });
-        });
-        if (resp) {
-            return JSON.parse(resp.response);
-        }
-        const openai = openAI();
-        if (openai === null) {
-            return new Error("OPENAI_API_KEY not set");
-        }
-        const completion = yield openai.createChatCompletion(args);
-        const data = completion.data;
-        db.run(`INSERT INTO "cache" VALUES (?, ?)`, [
-            param_key,
-            JSON.stringify(data),
-        ]);
-        return data;
-    });
-}
-exports.cachedChatCompletion = cachedChatCompletion;