aimyapi 0.1.5 → 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/example/example1.ts

@@ -1,7 +1,7 @@
+import path from "path";
 import AIMyAPI from "../lib";
-import { API } from "./sales_api_impl";
 import * as APIExports from "./sales_api";
-import path from "path";
+import { API } from "./sales_api_impl";
 
 // Simpler example using no support for chat history; 
 
@@ -20,7 +20,7 @@ import path from "path";
     console.log("Print 'hello' three times.")
     await aimyapi.processRequest("Print 'hello' three times.");
 
-    console.log("What was our best month in 2020")
+    console.log("What month had the highest sales in 2020 for any business unit?")
     await aimyapi.processRequest("What was our best month in 2020");
 
     console.log("Which business unit had the highest sales in 2020?")

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.5",
+  "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",

package/prompts/create-task-prompt.md

@@ -1,7 +1,5 @@
-You are an expert coder, writing typescript code to accomplish tasks for the user.
-We fulfill user requests by writing code that does what the user asked by performing operations on an API.
-Respond only with code - do not include any extra explanations or questions outside of the code.
-if you need to talk to the user, do so using functions provided by the API.
+You are a helpful assistant who speaks only in Typescript.
+You fulfill user requests by writing code. 
 You comment your code and are sure to properly escape your strings and follow good programming practices.
 
 Here is the API that you will use: