@promptbook/openai 0.103.0-47 → 0.103.0-49

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-49';
 /**
  * TODO: string_promptbook_version should be constrained to the all versions of Promptbook engine
  * Note: [💞] Ignore a discrepancy between file name and entity name
@@ -575,6 +574,9 @@ class Color {
         if (hex.length === 3) {
             return Color.fromHex3(hex);
         }
+        if (hex.length === 4) {
+            return Color.fromHex4(hex);
+        }
         if (hex.length === 6) {
             return Color.fromHex6(hex);
         }
@@ -595,6 +597,19 @@ class Color {
         const b = parseInt(hex.substr(2, 1), 16) * 16;
         return take(new Color(r, g, b));
     }
+    /**
+     * Creates a new Color instance from color in hex format with 4 digits (with alpha channel)
+     *
+     * @param color in hex for example `09df`
+     * @returns Color object
+     */
+    static fromHex4(hex) {
+        const r = parseInt(hex.substr(0, 1), 16) * 16;
+        const g = parseInt(hex.substr(1, 1), 16) * 16;
+        const b = parseInt(hex.substr(2, 1), 16) * 16;
+        const a = parseInt(hex.substr(3, 1), 16) * 16;
+        return take(new Color(r, g, b, a));
+    }
     /**
      * Creates a new Color instance from color in hex format with 6 color digits (without alpha channel)
      *
@@ -785,7 +800,8 @@ class Color {
      * @returns true if the value is a valid hex color string (e.g., `#009edd`, `#fff`, etc.)
      */
     static isHexColorString(value) {
-        return typeof value === 'string' && /^#(?:[0-9a-fA-F]{3}){1,2}$/.test(value);
+        return (typeof value === 'string' &&
+            /^#(?:[0-9a-fA-F]{3}|[0-9a-fA-F]{4}|[0-9a-fA-F]{6}|[0-9a-fA-F]{8})$/.test(value));
     }
     /**
      * Creates new Color object
@@ -1092,6 +1108,7 @@ const PROMPTBOOK_COLOR = Color.fromHex('#79EAFD');
 ({
     TITLE: Color.fromHex('#244EA8'),
     LINE: Color.fromHex('#eeeeee'),
+    SEPARATOR: Color.fromHex('#cccccc'),
     COMMITMENT: Color.fromHex('#DA0F78'),
     PARAMETER: Color.fromHex('#8e44ad'),
 });
@@ -1471,7 +1488,7 @@ function deepClone(objectValue) {
     TODO: [🧠] Is there a better implementation?
     > const propertyNames = Object.getOwnPropertyNames(objectValue);
     > for (const propertyName of propertyNames) {
-    >     const value = (objectValue as really_any)[propertyName];
+    >     const value = (objectValue as chococake)[propertyName];
     >     if (value && typeof value === 'object') {
     >         deepClone(value);
     >     }
@@ -2361,17 +2378,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',
@@ -3557,11 +3574,12 @@ class OpenAiExecutionTools extends OpenAiCompatibleExecutionTools {
  *
  * 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
  * - `OpenAiAssistantExecutionTools` - which is a specific implementation of `LlmExecutionTools` for OpenAI models with assistant capabilities, recommended for usage in `Agent` or `AgentLlmExecutionTools`
+ * - `RemoteAgent` - which is an `Agent` that connects to a Promptbook Agents Server
  *
  * @public exported from `@promptbook/openai`
  */
@@ -3596,6 +3614,12 @@ class OpenAiAssistantExecutionTools extends OpenAiExecutionTools {
      * Calls OpenAI API to use a chat model.
      */
     async callChatModel(prompt) {
+        return this.callChatModelStream(prompt, () => { });
+    }
+    /**
+     * Calls OpenAI API to use a chat model with streaming.
+     */
+    async callChatModelStream(prompt, onProgress) {
         var _a, _b, _c;
         if (this.options.isVerbose) {
             console.info('💬 OpenAI callChatModel call', { prompt });
@@ -3663,21 +3687,24 @@ class OpenAiAssistantExecutionTools extends OpenAiExecutionTools {
                 console.info('connect', stream.currentEvent);
             }
         });
-        /*
-        stream.on('messageDelta', (messageDelta) => {
-            if (
-                this.options.isVerbose &&
-                messageDelta &&
-                messageDelta.content &&
-                messageDelta.content[0] &&
-                messageDelta.content[0].type === 'text'
-            ) {
-                console.info('messageDelta', messageDelta.content[0].text?.value);
+        stream.on('textDelta', (textDelta, snapshot) => {
+            if (this.options.isVerbose && textDelta.value) {
+                console.info('textDelta', textDelta.value);
             }
-
-            // <- TODO: [🐚] Make streaming and running tasks working
+            const chunk = {
+                content: textDelta.value || '',
+                modelName: 'assistant',
+                timing: {
+                    start,
+                    complete: $getCurrentDate(),
+                },
+                usage: UNCERTAIN_USAGE,
+                rawPromptContent,
+                rawRequest,
+                rawResponse: snapshot,
+            };
+            onProgress(chunk);
         });
-        */
         stream.on('messageCreated', (message) => {
             if (this.options.isVerbose) {
                 console.info('messageCreated', message);
@@ -3713,7 +3740,7 @@ class OpenAiAssistantExecutionTools extends OpenAiExecutionTools {
         }
         return exportJson({
             name: 'promptResult',
-            message: `Result of \`OpenAiAssistantExecutionTools.callChatModel\``,
+            message: `Result of \`OpenAiAssistantExecutionTools.callChatModelStream\``,
             order: [],
             value: {
                 content: resultContent,
@@ -3732,15 +3759,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 +3781,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.`);
@@ -3837,9 +3879,98 @@ class OpenAiAssistantExecutionTools extends OpenAiExecutionTools {
         }
         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
-        // TODO: !!!! Allow to update and reconnect to existing assistants
+        // TODO: [🐱‍🚀] Try listing existing assistants
+        // TODO: [🐱‍🚀] Try marking existing assistants by DISCRIMINANT
+        // TODO: [🐱‍🚀] Allow to update and reconnect to existing assistants
+        return new OpenAiAssistantExecutionTools({
+            ...this.options,
+            isCreatingNewAssistantsAllowed: false,
+            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,
@@ -3963,7 +4094,7 @@ class DatabaseError extends Error {
     }
 }
 /**
- * TODO: !!!! Explain that NotFoundError (!!! and other specific errors) has priority over DatabaseError in some contexts
+ * TODO: [🐱‍🚀] Explain that NotFoundError ([🐱‍🚀] and other specific errors) has priority over DatabaseError in some contexts
  */
 
 /**