@promptbook/cli 0.114.0-1 → 0.114.0-3

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.
@@ -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;
             }
 
@@ -376,7 +377,8 @@ export class OpenAiAgentKitExecutionToolsToolBuilder {
     /**
      * Creates the native Agent SDK tool used for `USE DEEPSEARCH`.
      */
-    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/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