@promptbook/documents 0.105.0-0 → 0.105.0-3

package/esm/index.es.js

@@ -26,7 +26,7 @@ const BOOK_LANGUAGE_VERSION = '2.0.0';
  * @generated
  * @see https://github.com/webgptorg/promptbook
  */
-const PROMPTBOOK_ENGINE_VERSION = '0.105.0-0';
+const PROMPTBOOK_ENGINE_VERSION = '0.105.0-3';
 /**
  * TODO: string_promptbook_version should be constrained to the all versions of Promptbook engine
  * Note: [💞] Ignore a discrepancy between file name and entity name
@@ -988,6 +988,7 @@ const PROMPTBOOK_COLOR = Color.fromString('promptbook');
     SEPARATOR: Color.fromHex('#cccccc'),
     COMMITMENT: Color.fromHex('#DA0F78'),
     PARAMETER: Color.fromHex('#8e44ad'),
+    CODE_BLOCK: Color.fromHex('#7700ffff'),
 });
 // <- TODO: [🧠][🈵] Using `Color` here increases the package size approx 3kb, maybe remove it
 /**
@@ -1309,6 +1310,7 @@ function $execCommandNormalizeOptions(options) {
     let args = [];
     let timeout;
     let isVerbose;
+    let env;
     if (typeof options === 'string') {
         // TODO: [1] DRY default values
         command = options;
@@ -1316,6 +1318,7 @@ function $execCommandNormalizeOptions(options) {
         crashOnError = true;
         timeout = Infinity; // <- TODO: [⏳]
         isVerbose = DEFAULT_IS_VERBOSE;
+        env = undefined;
     }
     else {
         /*
@@ -1332,6 +1335,7 @@ function $execCommandNormalizeOptions(options) {
         crashOnError = (_b = options.crashOnError) !== null && _b !== void 0 ? _b : true;
         timeout = (_c = options.timeout) !== null && _c !== void 0 ? _c : Infinity;
         isVerbose = (_d = options.isVerbose) !== null && _d !== void 0 ? _d : DEFAULT_IS_VERBOSE;
+        env = options.env;
     }
     // TODO: /(-[a-zA-Z0-9-]+\s+[^\s]*)|[^\s]*/g
     const _ = Array.from(command.matchAll(/(".*")|([^\s]*)/g))
@@ -1350,7 +1354,7 @@ function $execCommandNormalizeOptions(options) {
     if (/^win/.test(process.platform) && ['npm', 'npx'].includes(command)) {
         command = `${command}.cmd`;
     }
-    return { command, humanReadableCommand, args, cwd, crashOnError, timeout, isVerbose };
+    return { command, humanReadableCommand, args, cwd, crashOnError, timeout, isVerbose, env };
 }
 // TODO: This should show type error> execCommandNormalizeOptions({ command: '', commands: [''] });
 
@@ -1371,7 +1375,7 @@ function $execCommand(options) {
     }
     return new Promise((resolve, reject) => {
         // eslint-disable-next-line prefer-const
-        const { command, humanReadableCommand, args, cwd, crashOnError, timeout, isVerbose = DEFAULT_IS_VERBOSE, } = $execCommandNormalizeOptions(options);
+        const { command, humanReadableCommand, args, cwd, crashOnError, timeout, isVerbose = DEFAULT_IS_VERBOSE, env, } = $execCommandNormalizeOptions(options);
         if (timeout !== Infinity) {
             // TODO: In waitasecond forTime(Infinity) should be equivalent to forEver()
             forTime(timeout).then(() => {
@@ -1389,7 +1393,11 @@ function $execCommand(options) {
             console.info(colors.yellow(cwd) + ' ' + colors.green(command) + ' ' + colors.blue(args.join(' ')));
         }
         try {
-            const commandProcess = spawn(command, args, { cwd, shell: true });
+            const commandProcess = spawn(command, args, {
+                cwd,
+                shell: true,
+                env: env ? { ...process.env, ...env } : process.env,
+            });
             if (isVerbose) {
                 commandProcess.on('message', (message) => {
                     console.info({ message });
@@ -3994,74 +4002,90 @@ function addUsage(...usageItems) {
  * in real-time through an observable.
  *
  * @param llmTools - The LLM tools to be intercepted and tracked
- * @returns An augmented version of the tools that includes usage tracking capabilities
+ * @returns Full proxy of the tools with added usage tracking capabilities
  * @public exported from `@promptbook/core`
  */
 function countUsage(llmTools) {
     let totalUsage = ZERO_USAGE;
     const spending = new Subject();
-    const proxyTools = {
-        get title() {
-            return `${llmTools.title} (+usage)`;
-            // <- TODO: [🧈] Maybe standartize the suffix when wrapping `LlmExecutionTools` up
-            // <- TODO: [🧈][🧠] Does it make sense to suffix "(+usage)"?
-        },
-        get description() {
-            return `${llmTools.description} (+usage)`;
-            // <- TODO: [🧈] Maybe standartize the suffix when wrapping `LlmExecutionTools` up
-            // <- TODO: [🧈][🧠] Does it make sense to suffix "(+usage)"?
-        },
-        checkConfiguration() {
-            return /* not await */ llmTools.checkConfiguration();
-        },
-        listModels() {
-            return /* not await */ llmTools.listModels();
-        },
-        spending() {
-            return spending.asObservable();
-        },
-        getTotalUsage() {
-            // <- Note: [🥫] Not using getter `get totalUsage` but `getTotalUsage` to allow this object to be proxied
-            return totalUsage;
+    // Create a Proxy to intercept all property access and ensure full proxying of all properties
+    const proxyTools = new Proxy(llmTools, {
+        get(target, prop, receiver) {
+            // Handle title property
+            if (prop === 'title') {
+                return `${target.title} (+usage)`;
+                // <- TODO: [🧈] Maybe standartize the suffix when wrapping `LlmExecutionTools` up
+                // <- TODO: [🧈][🧠] Does it make sense to suffix "(+usage)"?
+            }
+            // Handle description property
+            if (prop === 'description') {
+                return `${target.description} (+usage)`;
+                // <- TODO: [🧈] Maybe standartize the suffix when wrapping `LlmExecutionTools` up
+                // <- TODO: [🧈][🧠] Does it make sense to suffix "(+usage)"?
+            }
+            // Handle spending method (new method added by this wrapper)
+            if (prop === 'spending') {
+                return () => {
+                    return spending.asObservable();
+                };
+            }
+            // Handle getTotalUsage method (new method added by this wrapper)
+            if (prop === 'getTotalUsage') {
+                // <- Note: [🥫] Not using getter `get totalUsage` but `getTotalUsage` to allow this object to be proxied
+                return () => {
+                    return totalUsage;
+                };
+            }
+            // Handle callChatModel method with usage counting
+            if (prop === 'callChatModel' && target.callChatModel !== undefined) {
+                return async (prompt) => {
+                    // console.info('[🚕] callChatModel through countTotalUsage');
+                    const promptResult = await target.callChatModel(prompt);
+                    totalUsage = addUsage(totalUsage, promptResult.usage);
+                    spending.next(promptResult.usage);
+                    return promptResult;
+                };
+            }
+            // Handle callCompletionModel method with usage counting
+            if (prop === 'callCompletionModel' && target.callCompletionModel !== undefined) {
+                return async (prompt) => {
+                    // console.info('[🚕] callCompletionModel through countTotalUsage');
+                    const promptResult = await target.callCompletionModel(prompt);
+                    totalUsage = addUsage(totalUsage, promptResult.usage);
+                    spending.next(promptResult.usage);
+                    return promptResult;
+                };
+            }
+            // Handle callEmbeddingModel method with usage counting
+            if (prop === 'callEmbeddingModel' && target.callEmbeddingModel !== undefined) {
+                return async (prompt) => {
+                    // console.info('[🚕] callEmbeddingModel through countTotalUsage');
+                    const promptResult = await target.callEmbeddingModel(prompt);
+                    totalUsage = addUsage(totalUsage, promptResult.usage);
+                    spending.next(promptResult.usage);
+                    return promptResult;
+                };
+            }
+            // Handle callImageGenerationModel method with usage counting
+            if (prop === 'callImageGenerationModel' && target.callImageGenerationModel !== undefined) {
+                return async (prompt) => {
+                    // console.info('[🚕] callImageGenerationModel through countTotalUsage');
+                    const promptResult = await target.callImageGenerationModel(prompt);
+                    totalUsage = addUsage(totalUsage, promptResult.usage);
+                    spending.next(promptResult.usage);
+                    return promptResult;
+                };
+            }
+            // <- Note: [🤖]
+            // For all other properties and methods, delegate to the original target
+            const value = Reflect.get(target, prop, receiver);
+            // If it's a function, bind it to the target to preserve context
+            if (typeof value === 'function') {
+                return value.bind(target);
+            }
+            return value;
         },
-    };
-    if (llmTools.callChatModel !== undefined) {
-        proxyTools.callChatModel = async (prompt) => {
-            // console.info('[🚕] callChatModel through countTotalUsage');
-            const promptResult = await llmTools.callChatModel(prompt);
-            totalUsage = addUsage(totalUsage, promptResult.usage);
-            spending.next(promptResult.usage);
-            return promptResult;
-        };
-    }
-    if (llmTools.callCompletionModel !== undefined) {
-        proxyTools.callCompletionModel = async (prompt) => {
-            // console.info('[🚕] callCompletionModel through countTotalUsage');
-            const promptResult = await llmTools.callCompletionModel(prompt);
-            totalUsage = addUsage(totalUsage, promptResult.usage);
-            spending.next(promptResult.usage);
-            return promptResult;
-        };
-    }
-    if (llmTools.callEmbeddingModel !== undefined) {
-        proxyTools.callEmbeddingModel = async (prompt) => {
-            // console.info('[🚕] callEmbeddingModel through countTotalUsage');
-            const promptResult = await llmTools.callEmbeddingModel(prompt);
-            totalUsage = addUsage(totalUsage, promptResult.usage);
-            spending.next(promptResult.usage);
-            return promptResult;
-        };
-    }
-    if (llmTools.callImageGenerationModel !== undefined) {
-        proxyTools.callImageGenerationModel = async (prompt) => {
-            // console.info('[🚕] callImageGenerationModel through countTotalUsage');
-            const promptResult = await llmTools.callImageGenerationModel(prompt);
-            totalUsage = addUsage(totalUsage, promptResult.usage);
-            spending.next(promptResult.usage);
-            return promptResult;
-        };
-    }
-    // <- Note: [🤖]
+    });
     return proxyTools;
 }
 /**