@promptbook/openai 0.103.0-46 → 0.103.0-48

package/umd/index.umd.js

@@ -1,8 +1,8 @@
 (function (global, factory) {
-    typeof exports === 'object' && typeof module !== 'undefined' ? factory(exports, require('colors'), require('waitasecond'), require('spacetrim'), require('crypto'), require('bottleneck'), require('openai'), require('socket.io-client')) :
-    typeof define === 'function' && define.amd ? define(['exports', 'colors', 'waitasecond', 'spacetrim', 'crypto', 'bottleneck', 'openai', 'socket.io-client'], factory) :
-    (global = typeof globalThis !== 'undefined' ? globalThis : global || self, factory(global["promptbook-openai"] = {}, global.colors, global.waitasecond, global.spaceTrim, global.crypto, global.Bottleneck, global.OpenAI, global.socket_ioClient));
-})(this, (function (exports, colors, waitasecond, spaceTrim, crypto, Bottleneck, OpenAI, socket_ioClient) { 'use strict';
+    typeof exports === 'object' && typeof module !== 'undefined' ? factory(exports, require('colors'), require('spacetrim'), require('crypto'), require('bottleneck'), require('openai'), require('socket.io-client')) :
+    typeof define === 'function' && define.amd ? define(['exports', 'colors', 'spacetrim', 'crypto', 'bottleneck', 'openai', 'socket.io-client'], factory) :
+    (global = typeof globalThis !== 'undefined' ? globalThis : global || self, factory(global["promptbook-openai"] = {}, global.colors, global.spaceTrim, global.crypto, global.Bottleneck, global.OpenAI, global.socket_ioClient));
+})(this, (function (exports, colors, spaceTrim, crypto, Bottleneck, OpenAI, socket_ioClient) { 'use strict';
 
     function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'default' in e ? e : { 'default': e }; }
 
@@ -25,7 +25,7 @@
      * @generated
      * @see https://github.com/webgptorg/promptbook
      */
-    const PROMPTBOOK_ENGINE_VERSION = '0.103.0-46';
+    const PROMPTBOOK_ENGINE_VERSION = '0.103.0-48';
     /**
      * TODO: string_promptbook_version should be constrained to the all versions of Promptbook engine
      * Note: [💞] Ignore a discrepancy between file name and entity name
@@ -479,9 +479,6 @@
             throw new Error(`${channelName} channel is greater than 255, it is ${value}`);
         }
     }
-    /**
-     * TODO: [🧠][🚓] Is/which combination it better to use asserts/check, validate or is utility function?
-     */
 
     /**
      * Color object represents an RGB color with alpha channel
@@ -2369,17 +2366,17 @@
             },
             /**/
             /*/
-          {
-              modelTitle: 'tts-1-hd-1106',
-              modelName: 'tts-1-hd-1106',
-          },
-          /**/
+            {
+                modelTitle: 'tts-1-hd-1106',
+                modelName: 'tts-1-hd-1106',
+            },
+            /**/
             /*/
-          {
-              modelTitle: 'tts-1-hd',
-              modelName: 'tts-1-hd',
-          },
-          /**/
+            {
+                modelTitle: 'tts-1-hd',
+                modelName: 'tts-1-hd',
+            },
+            /**/
             /**/
             {
                 modelVariant: 'CHAT',
@@ -3565,7 +3562,7 @@
      *
      * This is useful for calling OpenAI API with a single assistant, for more wide usage use `OpenAiExecutionTools`.
      *
-     * Note: [🦖] There are several different things in Promptbook:
+     * !!! Note: [🦖] There are several different things in Promptbook:
      * - `Agent` - which represents an AI Agent with its source, memories, actions, etc. Agent is a higher-level abstraction which is internally using:
      * - `LlmExecutionTools` - which wraps one or more LLM models and provides an interface to execute them
      * - `AgentLlmExecutionTools` - which is a specific implementation of `LlmExecutionTools` that wraps another LlmExecutionTools and applies agent-specific system prompts and requirements
@@ -3671,17 +3668,21 @@
                     console.info('connect', stream.currentEvent);
                 }
             });
+            /*
             stream.on('messageDelta', (messageDelta) => {
-                var _a;
-                if (this.options.isVerbose &&
+                if (
+                    this.options.isVerbose &&
                     messageDelta &&
                     messageDelta.content &&
                     messageDelta.content[0] &&
-                    messageDelta.content[0].type === 'text') {
-                    console.info('messageDelta', (_a = messageDelta.content[0].text) === null || _a === void 0 ? void 0 : _a.value);
+                    messageDelta.content[0].type === 'text'
+                ) {
+                    console.info('messageDelta', messageDelta.content[0].text?.value);
                 }
+
                 // <- TODO: [🐚] Make streaming and running tasks working
             });
+            */
             stream.on('messageCreated', (message) => {
                 if (this.options.isVerbose) {
                     console.info('messageCreated', message);
@@ -3736,15 +3737,19 @@
                 },
             });
         }
-        async playground() {
+        /*
+        public async playground() {
             const client = await this.getClient();
+
             // List all assistants
             const assistants = await client.beta.assistants.list();
             console.log('!!! Assistants:', assistants);
+
             // Get details of a specific assistant
             const assistantId = 'asst_MO8fhZf4dGloCfXSHeLcIik0';
             const assistant = await client.beta.assistants.retrieve(assistantId);
             console.log('!!! Assistant Details:', assistant);
+
             // Update an assistant
             const updatedAssistant = await client.beta.assistants.update(assistantId, {
                 name: assistant.name + '(M)',
@@ -3754,71 +3759,103 @@
                 },
             });
             console.log('!!! Updated Assistant:', updatedAssistant);
-            await waitasecond.forEver();
+
+            await forEver();
+        }
+        */
+        /**
+         * Get an existing assistant tool wrapper
+         */
+        getAssistant(assistantId) {
+            return new OpenAiAssistantExecutionTools({
+                ...this.options,
+                assistantId,
+            });
         }
         async createNewAssistant(options) {
             if (!this.isCreatingNewAssistantsAllowed) {
                 throw new NotAllowed(`Creating new assistants is not allowed. Set \`isCreatingNewAssistantsAllowed: true\` in options to enable this feature.`);
             }
             // await this.playground();
-            const { name, instructions } = options;
+            const { name, instructions, knowledgeSources } = options;
             const client = await this.getClient();
-            /*/
-            //TODO: !!!
-            async function downloadFile(url: string, folder = './tmp'): Promise<string> {
-                const filename = path.basename(url.split('?')[0]);
-                const filepath = path.join(folder, filename);
-
-                if (!fs.existsSync(folder)) fs.mkdirSync(folder);
-
-                const res = await fetch(url);
-                if (!res.ok) throw new Error(`Download error: ${url}`);
-                const buffer = await res.arrayBuffer();
-                fs.writeFileSync(filepath, Buffer.from(buffer));
-                console.log(`📥 File downloaded: ${filename}`);
-
-                return filepath;
-            }
-
-            async function uploadFileToOpenAI(filepath: string) {
-                const file = await client.files.create({
-                    file: fs.createReadStream(filepath),
-                    purpose: 'assistants',
+            let vectorStoreId;
+            // If knowledge sources are provided, create a vector store with them
+            if (knowledgeSources && knowledgeSources.length > 0) {
+                if (this.options.isVerbose) {
+                    console.info(`📚 Creating vector store with ${knowledgeSources.length} knowledge sources...`);
+                }
+                // Create a vector store
+                const vectorStore = await client.beta.vectorStores.create({
+                    name: `${name} Knowledge Base`,
                 });
-                console.log(`⬆️  File uploaded to OpenAI: ${file.filename} (${file.id})`);
-                return file;
-            }
-
-            // 🌐 URL addresses of files to upload
-            const fileUrls = [
-                'https://raw.githubusercontent.com/vercel/next.js/canary/packages/next/README.md',
-                'https://raw.githubusercontent.com/openai/openai-cookbook/main/examples/How_to_call_the_Assistants_API_with_Node.js.ipynb',
-            ];
-
-            // 1️⃣ Download files from URL
-            const localFiles = [];
-            for (const url of fileUrls) {
-                const filepath = await downloadFile(url);
-                localFiles.push(filepath);
-            }
-
-            // 2️⃣ Upload files to OpenAI
-            const uploadedFiles = [];
-            for (const filepath of localFiles) {
-                const file = await uploadFileToOpenAI(filepath);
-                uploadedFiles.push(file.id);
+                vectorStoreId = vectorStore.id;
+                if (this.options.isVerbose) {
+                    console.info(`✅ Vector store created: ${vectorStoreId}`);
+                }
+                // Upload files from knowledge sources to the vector store
+                const fileStreams = [];
+                for (const source of knowledgeSources) {
+                    try {
+                        // Check if it's a URL
+                        if (source.startsWith('http://') || source.startsWith('https://')) {
+                            // Download the file
+                            const response = await fetch(source);
+                            if (!response.ok) {
+                                console.error(`Failed to download ${source}: ${response.statusText}`);
+                                continue;
+                            }
+                            const buffer = await response.arrayBuffer();
+                            const filename = source.split('/').pop() || 'downloaded-file';
+                            const blob = new Blob([buffer]);
+                            const file = new File([blob], filename);
+                            fileStreams.push(file);
+                        }
+                        else {
+                            // Assume it's a local file path
+                            // Note: This will work in Node.js environment
+                            // For browser environments, this would need different handling
+                            const fs = await import('fs');
+                            const fileStream = fs.createReadStream(source);
+                            fileStreams.push(fileStream);
+                        }
+                    }
+                    catch (error) {
+                        console.error(`Error processing knowledge source ${source}:`, error);
+                    }
+                }
+                // Batch upload files to the vector store
+                if (fileStreams.length > 0) {
+                    try {
+                        await client.beta.vectorStores.fileBatches.uploadAndPoll(vectorStoreId, {
+                            files: fileStreams,
+                        });
+                        if (this.options.isVerbose) {
+                            console.info(`✅ Uploaded ${fileStreams.length} files to vector store`);
+                        }
+                    }
+                    catch (error) {
+                        console.error('Error uploading files to vector store:', error);
+                    }
+                }
             }
-            /**/
-            // alert('!!!! Creating new OpenAI assistant');
-            // 3️⃣ Create assistant with uploaded files
-            const assistant = await client.beta.assistants.create({
+            // Create assistant with vector store attached
+            const assistantConfig = {
                 name,
                 description: 'Assistant created via Promptbook',
                 model: 'gpt-4o',
                 instructions,
                 tools: [/* TODO: [🧠] Maybe add { type: 'code_interpreter' }, */ { type: 'file_search' }],
-                // !!!! file_ids: uploadedFiles,
-            });
+            };
+            // Attach vector store if created
+            if (vectorStoreId) {
+                assistantConfig.tool_resources = {
+                    file_search: {
+                        vector_store_ids: [vectorStoreId],
+                    },
+                };
+            }
+            const assistant = await client.beta.assistants.create(assistantConfig);
             console.log(`✅ Assistant created: ${assistant.id}`);
             // TODO: !!!! Try listing existing assistants
             // TODO: !!!! Try marking existing assistants by DISCRIMINANT
@@ -3829,6 +3866,95 @@
                 assistantId: assistant.id,
             });
         }
+        async updateAssistant(options) {
+            if (!this.isCreatingNewAssistantsAllowed) {
+                throw new NotAllowed(`Updating assistants is not allowed. Set \`isCreatingNewAssistantsAllowed: true\` in options to enable this feature.`);
+            }
+            const { assistantId, name, instructions, knowledgeSources } = options;
+            const client = await this.getClient();
+            let vectorStoreId;
+            // If knowledge sources are provided, create a vector store with them
+            // TODO: [🧠] Reuse vector store creation logic from createNewAssistant
+            if (knowledgeSources && knowledgeSources.length > 0) {
+                if (this.options.isVerbose) {
+                    console.info(`📚 Creating vector store for update with ${knowledgeSources.length} knowledge sources...`);
+                }
+                // Create a vector store
+                const vectorStore = await client.beta.vectorStores.create({
+                    name: `${name} Knowledge Base`,
+                });
+                vectorStoreId = vectorStore.id;
+                if (this.options.isVerbose) {
+                    console.info(`✅ Vector store created: ${vectorStoreId}`);
+                }
+                // Upload files from knowledge sources to the vector store
+                const fileStreams = [];
+                for (const source of knowledgeSources) {
+                    try {
+                        // Check if it's a URL
+                        if (source.startsWith('http://') || source.startsWith('https://')) {
+                            // Download the file
+                            const response = await fetch(source);
+                            if (!response.ok) {
+                                console.error(`Failed to download ${source}: ${response.statusText}`);
+                                continue;
+                            }
+                            const buffer = await response.arrayBuffer();
+                            const filename = source.split('/').pop() || 'downloaded-file';
+                            const blob = new Blob([buffer]);
+                            const file = new File([blob], filename);
+                            fileStreams.push(file);
+                        }
+                        else {
+                            // Assume it's a local file path
+                            // Note: This will work in Node.js environment
+                            // For browser environments, this would need different handling
+                            const fs = await import('fs');
+                            const fileStream = fs.createReadStream(source);
+                            fileStreams.push(fileStream);
+                        }
+                    }
+                    catch (error) {
+                        console.error(`Error processing knowledge source ${source}:`, error);
+                    }
+                }
+                // Batch upload files to the vector store
+                if (fileStreams.length > 0) {
+                    try {
+                        await client.beta.vectorStores.fileBatches.uploadAndPoll(vectorStoreId, {
+                            files: fileStreams,
+                        });
+                        if (this.options.isVerbose) {
+                            console.info(`✅ Uploaded ${fileStreams.length} files to vector store`);
+                        }
+                    }
+                    catch (error) {
+                        console.error('Error uploading files to vector store:', error);
+                    }
+                }
+            }
+            const assistantUpdate = {
+                name,
+                instructions,
+                tools: [/* TODO: [🧠] Maybe add { type: 'code_interpreter' }, */ { type: 'file_search' }],
+            };
+            if (vectorStoreId) {
+                assistantUpdate.tool_resources = {
+                    file_search: {
+                        vector_store_ids: [vectorStoreId],
+                    },
+                };
+            }
+            const assistant = await client.beta.assistants.update(assistantId, assistantUpdate);
+            if (this.options.isVerbose) {
+                console.log(`✅ Assistant updated: ${assistant.id}`);
+            }
+            return new OpenAiAssistantExecutionTools({
+                ...this.options,
+                isCreatingNewAssistantsAllowed: false,
+                assistantId: assistant.id,
+            });
+        }
         /**
          * Discriminant for type guards
          */