@promptbook/openai 0.103.0-47 → 0.103.0-48

package/esm/index.es.js

@@ -1,5 +1,4 @@
 import colors from 'colors';
-import { forEver } from 'waitasecond';
 import spaceTrim$1, { spaceTrim } from 'spacetrim';
 import { randomBytes } from 'crypto';
 import Bottleneck from 'bottleneck';
@@ -20,7 +19,7 @@ const BOOK_LANGUAGE_VERSION = '2.0.0';
  * @generated
  * @see https://github.com/webgptorg/promptbook
  */
-const PROMPTBOOK_ENGINE_VERSION = '0.103.0-47';
+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
@@ -2361,17 +2360,17 @@ const OPENAI_MODELS = exportJson({
         },
         /**/
         /*/
-      {
-          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',
@@ -3732,15 +3731,19 @@ class OpenAiAssistantExecutionTools extends OpenAiExecutionTools {
             },
         });
     }
-    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)',
@@ -3750,8 +3753,19 @@ class OpenAiAssistantExecutionTools extends OpenAiExecutionTools {
             },
         });
         console.log('!!! Updated Assistant:', updatedAssistant);
+
         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.`);
@@ -3846,6 +3860,95 @@ class OpenAiAssistantExecutionTools extends OpenAiExecutionTools {
             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
      */