@promptbook/wizard 0.104.0-3 → 0.104.0-5

package/esm/index.es.js

@@ -36,7 +36,7 @@ const BOOK_LANGUAGE_VERSION = '2.0.0';
  * @generated
  * @see https://github.com/webgptorg/promptbook
  */
-const PROMPTBOOK_ENGINE_VERSION = '0.104.0-3';
+const PROMPTBOOK_ENGINE_VERSION = '0.104.0-5';
 /**
  * TODO: string_promptbook_version should be constrained to the all versions of Promptbook engine
  * Note: [💞] Ignore a discrepancy between file name and entity name
@@ -2526,6 +2526,7 @@ class RemoteLlmExecutionTools {
     }
 }
 /**
+ * TODO: !!!! Deprecate pipeline server and all of its components
  * TODO: Maybe use `$exportJson`
  * TODO: [🧠][🛍] Maybe not `isAnonymous: boolean` BUT `mode: 'ANONYMOUS'|'COLLECTION'`
  * TODO: [🍓] Allow to list compatible models with each variant
@@ -9417,6 +9418,15 @@ function countUsage(llmTools) {
             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;
 }
@@ -9526,6 +9536,12 @@ class MultipleLlmExecutionTools {
     callEmbeddingModel(prompt) {
         return this.callCommonModel(prompt);
     }
+    /**
+     * Calls the best available embedding model
+     */
+    callImageGenerationModel(prompt) {
+        return this.callCommonModel(prompt);
+    }
     // <- Note: [🤖]
     /**
      * Calls the best available model
@@ -9552,6 +9568,11 @@ class MultipleLlmExecutionTools {
                             continue llm;
                         }
                         return await llmExecutionTools.callEmbeddingModel(prompt);
+                    case 'IMAGE_GENERATION':
+                        if (llmExecutionTools.callImageGenerationModel === undefined) {
+                            continue llm;
+                        }
+                        return await llmExecutionTools.callImageGenerationModel(prompt);
                     // <- case [🤖]:
                     default:
                         throw new UnexpectedError(`Unknown model variant "${prompt.modelRequirements.modelVariant}" in ${llmExecutionTools.title}`);
@@ -11255,8 +11276,9 @@ async function executeAttempts(options) {
                                     $ongoingTaskResult.$resultString = $ongoingTaskResult.$completionResult.content;
                                     break variant;
                                 case 'EMBEDDING':
+                                case 'IMAGE_GENERATION':
                                     throw new PipelineExecutionError(spaceTrim$1((block) => `
-                                                Embedding model can not be used in pipeline
+                                                ${modelRequirements.modelVariant} model can not be used in pipeline
 
                                                 This should be catched during parsing
 
@@ -19109,6 +19131,9 @@ function cacheLlmTools(llmTools, options = {}) {
             case 'EMBEDDING':
                 promptResult = await llmTools.callEmbeddingModel(prompt);
                 break variant;
+            case 'IMAGE_GENERATION':
+                promptResult = await llmTools.callImageGenerationModel(prompt);
+                break variant;
             // <- case [🤖]:
             default:
                 throw new PipelineExecutionError(`Unknown model variant "${prompt.modelRequirements.modelVariant}"`);
@@ -19145,12 +19170,13 @@ function cacheLlmTools(llmTools, options = {}) {
                 }
             }
             catch (error) {
+                assertsError(error);
                 // If validation throws an unexpected error, don't cache
                 shouldCache = false;
                 if (isVerbose) {
                     console.info('Not caching result due to validation error for key:', key, {
                         content: promptResult.content,
-                        validationError: error instanceof Error ? error.message : String(error),
+                        validationError: serializeError(error),
                     });
                 }
             }
@@ -19196,6 +19222,11 @@ function cacheLlmTools(llmTools, options = {}) {
             return /* not await */ callCommonModel(prompt);
         };
     }
+    if (llmTools.callImageGenerationModel !== undefined) {
+        proxyTools.callImageGenerationModel = async (prompt) => {
+            return /* not await */ callCommonModel(prompt);
+        };
+    }
     // <- Note: [🤖]
     return proxyTools;
 }