@promptbook/cli 0.114.0-2 → 0.114.0-20

package/src/llm-providers/azure-openai/AzureOpenAiExecutionTools.ts

@@ -1,4 +1,4 @@
-import { AzureKeyCredential, OpenAIClient } from '@azure/openai';
+import type { OpenAIClient } from '@azure/openai';
 import Bottleneck from 'bottleneck';
 import colors from 'colors'; // <- TODO: [🔶] Make system to put color and style to both node and browser
 import type { ChatParticipant } from '../../book-components/Chat/types/ChatParticipant';
@@ -18,6 +18,7 @@ import type { string_completion_prompt } from '../../types/string_prompt';
 import type { string_title } from '../../types/string_title';
 import type { string_date_iso8601 } from '../../types/string_token';
 import { $getCurrentDate } from '../../utils/misc/$getCurrentDate';
+import { createLazyModuleLoader } from '../../utils/misc/createLazyModuleLoader';
 import { keepTypeImported } from '../../utils/organization/keepTypeImported';
 import { templateParameters } from '../../utils/parameters/templateParameters';
 import { exportJson } from '../../utils/serialization/exportJson';
@@ -35,6 +36,15 @@ const AZURE_OPENAI_PROVIDER_PROFILE: ChatParticipant = {
 
 keepTypeImported<Usage>();
 
+/**
+ * Loads the Azure OpenAI SDK (`@azure/openai`) on demand
+ *
+ * Note: [🐌] Loaded lazily to keep the startup of the `ptbk` CLI utility fast
+ *
+ * @private internal utility of `AzureOpenAiExecutionTools`
+ */
+const loadAzureOpenAiModule = createLazyModuleLoader(() => import('@azure/openai'));
+
 /**
  * Execution Tools for calling Azure OpenAI API.
  *
@@ -77,6 +87,8 @@ export class AzureOpenAiExecutionTools implements LlmExecutionTools /* <- TODO:
 
     public async getClient(): Promise<OpenAIClient> {
         if (this.client === null) {
+            const { AzureKeyCredential, OpenAIClient } = await loadAzureOpenAiModule();
+
             this.client = new OpenAIClient(
                 `https://${this.options.resourceName}.openai.azure.com/`,
                 new AzureKeyCredential(this.options.apiKey),

package/src/llm-providers/openai/OpenAiAgentKitExecutionTools.ts

@@ -1,4 +1,4 @@
-import { Agent as AgentFromKit, run, setDefaultOpenAIClient, setDefaultOpenAIKey } from '@openai/agents';
+import type { Agent as AgentFromKit } from '@openai/agents';
 import { NotYetImplementedError } from '../../errors/NotYetImplementedError';
 import { PipelineExecutionError } from '../../errors/PipelineExecutionError';
 import type { CallChatModelStreamOptions, LlmExecutionTools } from '../../execution/LlmExecutionTools';
@@ -23,6 +23,7 @@ import { OpenAiAgentKitExecutionToolsOutputTypeMapper } from './OpenAiAgentKitEx
 import { OpenAiAgentKitExecutionToolsToolBuilder } from './OpenAiAgentKitExecutionToolsToolBuilder';
 import type { OpenAiCompatibleExecutionToolsNonProxiedOptions } from './OpenAiCompatibleExecutionToolsOptions';
 import { OpenAiVectorStoreHandler } from './OpenAiVectorStoreHandler';
+import { loadOpenAiAgentsModule } from './utils/loadOpenAiAgentsModule';
 
 /**
  * Constant for default agent kit model name.
@@ -211,7 +212,8 @@ export class OpenAiAgentKitExecutionTools extends OpenAiVectorStoreHandler imple
             });
         }
 
-        const agentKitTools = this.buildAgentKitTools({ tools, vectorStoreId });
+        const { Agent: AgentFromKit } = await loadOpenAiAgentsModule();
+        const agentKitTools = await this.buildAgentKitTools({ tools, vectorStoreId });
         const openAiAgentKitAgent = new AgentFromKit({
             name,
             model: this.agentKitModelName,
@@ -275,6 +277,7 @@ export class OpenAiAgentKitExecutionTools extends OpenAiVectorStoreHandler imple
             agentName: agentForRun.name,
             input: inputItems,
         };
+        const { run } = await loadOpenAiAgentsModule();
         const streamResult = await run(agentForRun, inputItems, {
             stream: true,
             maxTurns: 200,
@@ -433,6 +436,7 @@ export class OpenAiAgentKitExecutionTools extends OpenAiVectorStoreHandler imple
      * Ensures the AgentKit SDK is wired to the OpenAI client and API key.
      */
     private async ensureAgentKitDefaults(): Promise<void> {
+        const { setDefaultOpenAIClient, setDefaultOpenAIKey } = await loadOpenAiAgentsModule();
         const client = await this.getClient();
         setDefaultOpenAIClient(client as TODO_any);
 

package/src/llm-providers/openai/OpenAiAgentKitExecutionToolsToolBuilder.ts

@@ -1,5 +1,4 @@
 import type { Tool as AgentKitTool } from '@openai/agents';
-import { Agent as AgentFromKit, fileSearchTool, tool as agentKitTool, webSearchTool } from '@openai/agents';
 import { spaceTrim } from 'spacetrim';
 import { parseToolExecutionEnvelope } from '../../commitments/_common/toolExecutionEnvelope';
 import type { ToolCallProgressUpdate } from '../../commitments/_common/toolRuntimeContext';
@@ -26,6 +25,7 @@ import type { TODO_any } from '../../utils/organization/TODO_any';
 import type { OpenAiAgentKitExecutionToolsOptions } from './OpenAiAgentKitExecutionToolsOptions';
 import type { OpenAiCompatibleExecutionToolsNonProxiedOptions } from './OpenAiCompatibleExecutionToolsOptions';
 import { buildToolInvocationScript } from './utils/buildToolInvocationScript';
+import { loadOpenAiAgentsModule } from './utils/loadOpenAiAgentsModule';
 
 /**
  * Constant for default model used for nested DeepSearch tool invocations.
@@ -33,7 +33,7 @@ import { buildToolInvocationScript } from './utils/buildToolInvocationScript';
 const DEFAULT_DEEP_SEARCH_MODEL_NAME = 'o4-mini-deep-research';
 
 /**
- * Tool name used by the Book commitment-backed DeepSearch capability.
+ * Tool name used by the DeepSearch capability.
  */
 const DEEP_SEARCH_TOOL_NAME = 'deep_search' as string_javascript_name;
 
@@ -98,11 +98,12 @@ export class OpenAiAgentKitExecutionToolsToolBuilder {
     /**
      * Builds the tool list for AgentKit, including hosted file search when applicable.
      */
-    public buildAgentKitTools(options: {
+    public async buildAgentKitTools(options: {
         readonly tools?: ModelRequirements['tools'];
         readonly vectorStoreId?: string;
-    }): Array<AgentKitTool> {
+    }): Promise<Array<AgentKitTool>> {
         const { tools, vectorStoreId } = options;
+        const { fileSearchTool, tool: agentKitTool } = await loadOpenAiAgentsModule();
         const agentKitTools: Array<AgentKitTool> = [];
 
         if (vectorStoreId) {
@@ -117,7 +118,7 @@ export class OpenAiAgentKitExecutionToolsToolBuilder {
 
         for (const toolDefinition of tools) {
             if (this.isDeepSearchToolDefinition(toolDefinition)) {
-                agentKitTools.push(this.createDeepSearchAgentKitTool(toolDefinition));
+                agentKitTools.push(await this.createDeepSearchAgentKitTool(toolDefinition));
                 continue;
             }
 
@@ -374,9 +375,10 @@ export class OpenAiAgentKitExecutionToolsToolBuilder {
     }
 
     /**
-     * Creates the native Agent SDK tool used for `USE DEEPSEARCH`.
+     * Creates the native Agent SDK tool used for the DeepSearch capability.
      */
-    private createDeepSearchAgentKitTool(toolDefinition: AgentKitToolDefinition): AgentKitTool {
+    private async createDeepSearchAgentKitTool(toolDefinition: AgentKitToolDefinition): Promise<AgentKitTool> {
+        const { Agent: AgentFromKit, webSearchTool } = await loadOpenAiAgentsModule();
         const deepSearchAgent = new AgentFromKit({
             name: 'DeepSearch',
             model: DEFAULT_DEEP_SEARCH_MODEL_NAME,

package/src/llm-providers/openai/OpenAiAssistantExecutionTools.ts

@@ -1,4 +1,4 @@
-import OpenAI from 'openai';
+import type OpenAI from 'openai';
 import { NotAllowed } from '../../errors/NotAllowed';
 import { NotYetImplementedError } from '../../errors/NotYetImplementedError';
 import { PipelineExecutionError } from '../../errors/PipelineExecutionError';

package/src/llm-providers/openai/OpenAiAssistantExecutionToolsStreamRunner.ts

@@ -1,5 +1,5 @@
 import colors from 'colors'; // <- TODO: [🔶] Make system to put color and style to both node and browser
-import OpenAI from 'openai';
+import type OpenAI from 'openai';
 import { PipelineExecutionError } from '../../errors/PipelineExecutionError';
 import type { ChatPromptResult } from '../../execution/PromptResult';
 import type { Prompt } from '../../types/Prompt';

package/src/llm-providers/openai/OpenAiAssistantExecutionToolsToolRunner.ts

@@ -1,5 +1,5 @@
 import colors from 'colors'; // <- TODO: [🔶] Make system to put color and style to both node and browser
-import OpenAI from 'openai';
+import type OpenAI from 'openai';
 import { spaceTrim } from 'spacetrim';
 import { parseToolExecutionEnvelope } from '../../commitments/_common/toolExecutionEnvelope';
 import type { ToolCallProgressUpdate } from '../../commitments/_common/toolRuntimeContext';

package/src/llm-providers/openai/OpenAiCompatibleNonChatPromptCaller.ts

@@ -1,5 +1,5 @@
 import colors from 'colors';
-import OpenAI from 'openai';
+import type OpenAI from 'openai';
 import { assertsError } from '../../errors/assertsError';
 import { PipelineExecutionError } from '../../errors/PipelineExecutionError';
 import type { AvailableModel } from '../../execution/AvailableModel';

package/src/llm-providers/openai/OpenAiCompatibleRequestManager.ts

@@ -1,11 +1,21 @@
 import Bottleneck from 'bottleneck';
 import colors from 'colors';
 import type { ClientOptions } from 'openai';
-import OpenAI from 'openai';
+import type OpenAI from 'openai';
 import { API_REQUEST_TIMEOUT, CONNECTION_RETRIES_LIMIT, DEFAULT_MAX_REQUESTS_PER_MINUTE } from '../../config';
 import { assertsError } from '../../errors/assertsError';
+import { createLazyModuleLoader } from '../../utils/misc/createLazyModuleLoader';
 import type { OpenAiCompatibleExecutionToolsNonProxiedOptions } from './OpenAiCompatibleExecutionToolsOptions';
 
+/**
+ * Loads the OpenAI SDK (`openai`) on demand
+ *
+ * Note: [🐌] Loaded lazily to keep the startup of the `ptbk` CLI utility fast
+ *
+ * @private internal utility of `OpenAiCompatibleRequestManager`
+ */
+const loadOpenAiModule = createLazyModuleLoader(() => import('openai'));
+
 /**
  * Manages OpenAI-compatible client creation plus shared retry and rate-limit behavior.
  *
@@ -36,6 +46,7 @@ export class OpenAiCompatibleRequestManager {
                 maxRetries: CONNECTION_RETRIES_LIMIT,
             } as ClientOptions;
 
+            const { default: OpenAI } = await loadOpenAiModule();
             this.client = new OpenAI(enhancedOptions);
         }
 

package/src/llm-providers/openai/utils/buildToolInvocationScript.ts

@@ -1,3 +1,4 @@
+import { spaceTrim } from 'spacetrim';
 import {
     TOOL_PROGRESS_TOKEN_ARGUMENT,
     TOOL_PROGRESS_TOKEN_PARAMETER,
@@ -16,26 +17,28 @@ export function buildToolInvocationScript(options: {
 }): string {
     const { functionName, functionArgsExpression } = options;
 
-    return `
-        const args = ${functionArgsExpression};
-        const runtimeContextRaw =
-            typeof ${TOOL_RUNTIME_CONTEXT_PARAMETER} === 'undefined'
-                ? undefined
-                : ${TOOL_RUNTIME_CONTEXT_PARAMETER};
+    return spaceTrim(
+        (block) => `
+            const args = ${block(functionArgsExpression)};
+            const runtimeContextRaw =
+                typeof ${TOOL_RUNTIME_CONTEXT_PARAMETER} === 'undefined'
+                    ? undefined
+                    : ${TOOL_RUNTIME_CONTEXT_PARAMETER};
 
-        if (runtimeContextRaw !== undefined && args && typeof args === 'object' && !Array.isArray(args)) {
-            args.${TOOL_RUNTIME_CONTEXT_ARGUMENT} = runtimeContextRaw;
-        }
+            if (runtimeContextRaw !== undefined && args && typeof args === 'object' && !Array.isArray(args)) {
+                args.${TOOL_RUNTIME_CONTEXT_ARGUMENT} = runtimeContextRaw;
+            }
 
-        const toolProgressTokenRaw =
-            typeof ${TOOL_PROGRESS_TOKEN_PARAMETER} === 'undefined'
-                ? undefined
-                : ${TOOL_PROGRESS_TOKEN_PARAMETER};
+            const toolProgressTokenRaw =
+                typeof ${TOOL_PROGRESS_TOKEN_PARAMETER} === 'undefined'
+                    ? undefined
+                    : ${TOOL_PROGRESS_TOKEN_PARAMETER};
 
-        if (toolProgressTokenRaw !== undefined && args && typeof args === 'object' && !Array.isArray(args)) {
-            args.${TOOL_PROGRESS_TOKEN_ARGUMENT} = toolProgressTokenRaw;
-        }
+            if (toolProgressTokenRaw !== undefined && args && typeof args === 'object' && !Array.isArray(args)) {
+                args.${TOOL_PROGRESS_TOKEN_ARGUMENT} = toolProgressTokenRaw;
+            }
 
-        return await ${functionName}(args);
-    `;
+            return await ${functionName}(args);
+        `,
+    );
 }

package/src/llm-providers/openai/utils/callOpenAiCompatibleChatModel.ts

@@ -1,5 +1,5 @@
 import colors from 'colors';
-import OpenAI from 'openai';
+import type OpenAI from 'openai';
 import { assertsError } from '../../../errors/assertsError';
 import { PipelineExecutionError } from '../../../errors/PipelineExecutionError';
 import type { AvailableModel } from '../../../execution/AvailableModel';

package/src/llm-providers/openai/utils/loadOpenAiAgentsModule.ts

@@ -0,0 +1,11 @@
+import { createLazyModuleLoader } from '../../../utils/misc/createLazyModuleLoader';
+
+/**
+ * Loads the OpenAI AgentKit SDK (`@openai/agents`) on demand
+ *
+ * Note: [🐌] The AgentKit SDK is one of the heaviest dependencies of Promptbook, loading it eagerly would slow down
+ *       every single run of the `ptbk` CLI utility even when no AgentKit agent is used
+ *
+ * @private internal utility of `@promptbook/openai`
+ */
+export const loadOpenAiAgentsModule = createLazyModuleLoader(() => import('@openai/agents'));

package/src/llm-providers/openai/utils/uploadFilesToOpenAi.ts

@@ -1,4 +1,4 @@
-import OpenAI from 'openai';
+import type OpenAI from 'openai';
 
 /**
  * Uploads files to OpenAI and returns their IDs