@promptbook/markdown-utils 0.105.0-1 → 0.105.0-4

package/umd/index.umd.js

@@ -24,7 +24,7 @@
      * @generated
      * @see https://github.com/webgptorg/promptbook
      */
-    const PROMPTBOOK_ENGINE_VERSION = '0.105.0-1';
+    const PROMPTBOOK_ENGINE_VERSION = '0.105.0-4';
     /**
      * TODO: string_promptbook_version should be constrained to the all versions of Promptbook engine
      * Note: [💞] Ignore a discrepancy between file name and entity name
@@ -1118,6 +1118,7 @@
         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
     /**
@@ -3494,74 +3495,90 @@
      * 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 rxjs.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;
     }
     /**