@promptbook/core 0.104.0-3 → 0.104.0-4

package/esm/index.es.js

@@ -27,7 +27,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-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
@@ -3744,6 +3744,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;
 }
@@ -3853,6 +3862,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
@@ -3879,6 +3894,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}`);
@@ -6304,8 +6324,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
 
@@ -16918,6 +16939,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}"`);
@@ -17005,6 +17029,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;
 }
@@ -17043,6 +17072,11 @@ function limitTotalUsage(llmTools, options = {}) {
             throw new LimitReachedError('Cannot call `callEmbeddingModel` because you are not allowed to spend any cost');
         };
     }
+    if (proxyTools.callImageGenerationModel !== undefined) {
+        proxyTools.callImageGenerationModel = async (prompt) => {
+            throw new LimitReachedError('Cannot call `callImageGenerationModel` because you are not allowed to spend any cost');
+        };
+    }
     // <- Note: [🤖]
     return proxyTools;
 }