aimyapi 0.1.6 → 0.1.7

package/dist/index.d.ts

@@ -5,6 +5,7 @@ export declare const generateCode: (queryText: string, userChatHistory: ChatComp
 export interface AIMyAPIOptions {
     apiObject: object;
     apiExports: object;
+    apiWhitelist?: string[];
     apiGlobals?: object;
     apiDefFilePath: string;
     apiGlobalName?: string;

package/dist/index.js

@@ -5,8 +5,8 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generateCode = exports.createBasePrompt = void 0;
 require('dotenv').config();
-const openai_1 = require("openai");
 const fs_1 = __importDefault(require("fs"));
+const openai_1 = require("openai");
 const path_1 = __importDefault(require("path"));
 const sandbox_1 = __importDefault(require("./sandbox"));
 const sandbox_wrappers_1 = require("./sandbox-wrappers");
@@ -91,14 +91,14 @@ async function createWithAPI(options) {
     options = {
         apiGlobalName: "api",
         apiGlobals: {},
+        apiWhitelist: Object.getOwnPropertyNames(Object.getPrototypeOf(options.apiObject)).filter((f) => f !== "constructor" && !f.startsWith("_")),
         debug: false,
         ...options
     };
     if (!options.apiObject || !options.apiDefFilePath) {
         throw new Error("apiObject and apiFilePath are required");
     }
-    const { apiObject, apiGlobalName, apiExports, apiDefFilePath, apiDocsPath, debug } = options;
-    const whitelist = Object.getOwnPropertyNames(Object.getPrototypeOf(apiObject)).filter((f) => f !== "constructor" && !f.startsWith("_"));
+    const { apiObject, apiWhitelist, apiGlobalName, apiExports, apiDefFilePath, apiDocsPath, debug } = options;
     const createTaskPrompt = createBasePrompt(apiDefFilePath, apiDocsPath);
     return {
         generateCode: async function (queryText, userChatHistory, currentContext = undefined) {
@@ -127,7 +127,7 @@ async function createWithAPI(options) {
                 ...apiGlobals,
                 [apiGlobalName]: {
                     value: apiObject,
-                    whitelist
+                    apiWhitelist
                 }
             }, options.debug);
             try {

package/lib/index.ts

@@ -1,13 +1,11 @@
 require('dotenv').config()
-import { ChatCompletionRequestMessage, Configuration, OpenAIApi } from "openai";
 import fs from 'fs';
+import { ChatCompletionRequestMessage, Configuration, OpenAIApi } from "openai";
 import path from 'path';
 
 import createSandbox from "./sandbox";
 import { rejectOpenPromises } from './sandbox-wrappers';
 
-
-
 import { QuickJSWASMModule, getQuickJS } from "quickjs-emscripten";
 
 const configuration = new Configuration({
@@ -102,6 +100,7 @@ let QuickJS:QuickJSWASMModule = null;
 export interface AIMyAPIOptions {
     apiObject: object;
     apiExports: object;
+    apiWhitelist?: string[];      // List of functions to expose to the user. Necessary to pass this yourself if you are extending your api from another class.
     apiGlobals?: object;
     apiDefFilePath: string;     // Full path to the api definition file so that it can be loaded into the prompt
     apiGlobalName?: string;    
@@ -123,6 +122,7 @@ async function createWithAPI(options:AIMyAPIOptions): Promise<CreateWithAPIExpor
     options = {
         apiGlobalName: "api",
         apiGlobals: {},
+        apiWhitelist: Object.getOwnPropertyNames(Object.getPrototypeOf(options.apiObject)).filter((f) => f !== "constructor" && !f.startsWith("_")),
         debug: false,
         ...options
     };
@@ -131,9 +131,8 @@ async function createWithAPI(options:AIMyAPIOptions): Promise<CreateWithAPIExpor
         throw new Error("apiObject and apiFilePath are required");
     }
 
-    const {apiObject, apiGlobalName, apiExports, apiDefFilePath, apiDocsPath, debug} = options;
+    const {apiObject, apiWhitelist, apiGlobalName, apiExports, apiDefFilePath, apiDocsPath, debug} = options;
 
-    const whitelist =  Object.getOwnPropertyNames(Object.getPrototypeOf(apiObject)).filter((f) => f !== "constructor" && !f.startsWith("_"));
     const createTaskPrompt = createBasePrompt( apiDefFilePath, apiDocsPath);
 
     return {
@@ -167,7 +166,7 @@ async function createWithAPI(options:AIMyAPIOptions): Promise<CreateWithAPIExpor
                 ...apiGlobals,
                 [apiGlobalName]: {
                     value: apiObject,
-                    whitelist
+                    apiWhitelist
                 }
             }, options.debug);
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aimyapi",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "AI My API! Use LLMs to create small programs that safely* run and make calls against your API.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",