feat: add load method

Add a method to load a set of tool definitions into a program.

Signed-off-by: Nick Hale <[email protected]>

Author: njhale

src/gptscript.ts

@@ -252,6 +252,100 @@ export class GPTScript {
         }
     }
 
+    /**
+     * Loads a tool or tools with the specified options.
+     *
+     * @param {ToolDef | ToolDef[]} toolDefs - The tool or tools to load.
+     * @param {string} [content] - The content of the tool.
+     * @param {boolean} [disableCache] - Whether to disable the cache.
+     * @param {string} [subTool] - The sub-tool to use.
+     * @param {string} [file] - The file to load.
+     * @return {Promise<LoadResponse>} The loaded program.
+     */
+    // async load(
+    //     toolDefs: ToolDef | ToolDef[],
+    //     disableCache?: boolean,
+    //     content?: string,
+    //     subTool?: string,
+    //     file?: string
+    // ): Promise<LoadResponse> {
+    //     if (!this.ready) {
+    //         this.ready = await this.testGPTScriptURL(20);
+    //     }
+    //     const r: Run = new RunSubcommand("load", toolDefs, {}, GPTScript.serverURL);
+    //     const requestPayload: any = { toolDefs: Array.isArray(toolDefs) ? toolDefs : [toolDefs] };
+    //     if (content) requestPayload.content = content;
+    //     if (disableCache !== undefined) requestPayload.disableCache = disableCache;
+    //     if (subTool) requestPayload.subTool = subTool;
+    //     if (file) requestPayload.file = file;
+    //
+    //     r.request(requestPayload);
+    //     return r.json() as Promise<LoadResponse>;
+    // }
+    /**
+     * Loads a file into a Program.
+     *
+     * @param {string} fileName - The name of the file to load.
+     * @param {boolean} [disableCache] - Whether to disable the cache.
+     * @param {string} [subTool] - The sub-tool to use.
+     * @return {Promise<LoadResponse>} The loaded program.
+     */
+    async loadFile(
+        fileName: string,
+        disableCache?: boolean,
+        subTool?: string
+    ): Promise<LoadResponse> {
+        return this.load({ file: fileName, disableCache, subTool });
+    }
+
+    /**
+     * Loads content into a Program.
+     *
+     * @param {string} content - The content to load.
+     * @param {boolean} [disableCache] - Whether to disable the cache.
+     * @param {string} [subTool] - The sub-tool to use.
+     * @return {Promise<LoadResponse>} The loaded program.
+     */
+    async loadContent(
+        content: string,
+        disableCache?: boolean,
+        subTool?: string
+    ): Promise<LoadResponse> {
+        return this.load({ content, disableCache, subTool });
+    }
+
+    /**
+     * Loads tools into a Program.
+     *
+     * @param {ToolDef[]} toolDefs - The tools to load.
+     * @param {boolean} [disableCache] - Whether to disable the cache.
+     * @param {string} [subTool] - The sub-tool to use.
+     * @return {Promise<LoadResponse>} The loaded program.
+     */
+    async loadTools(
+        toolDefs: ToolDef[],
+        disableCache?: boolean,
+        subTool?: string
+    ): Promise<LoadResponse> {
+        return this.load({ toolDefs, disableCache, subTool });
+    }
+
+    /**
+     * Helper method to handle the common logic for loading.
+     *
+     * @param {any} payload - The payload to send in the request.
+     * @return {Promise<LoadResponse>} The loaded program.
+     */
+    private async load(payload: any): Promise<LoadResponse> {
+        if (!this.ready) {
+            this.ready = await this.testGPTScriptURL(20);
+        }
+        const r: Run = new RunSubcommand("load", payload.toolDefs || [], {}, GPTScript.serverURL);
+
+        r.request(payload);
+        return (await r.json()) as LoadResponse;
+    }
+
     private async testGPTScriptURL(count: number): Promise<boolean> {
         while (count > 0) {
             try {
@@ -812,6 +906,10 @@ export interface PromptResponse {
     responses: Record<string, string>
 }
 
+export interface LoadResponse {
+    program: Program;
+}
+
 export function getEnv(key: string, def: string = ""): string {
     let v = process.env[key] || ""
     if (v == "") {