@promptbook/legacy-documents 0.104.0-3 → 0.104.0-5

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.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
@@ -4046,6 +4046,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;
 }
@@ -4155,6 +4164,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
@@ -4181,6 +4196,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}`);
@@ -6305,8 +6325,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