diff --git a/libs/langchain-openai/src/tools/dalle.ts b/libs/langchain-openai/src/tools/dalle.ts index 35d405807cc2..d3d45c6dba90 100644 --- a/libs/langchain-openai/src/tools/dalle.ts +++ b/libs/langchain-openai/src/tools/dalle.ts @@ -80,6 +80,10 @@ export interface DallEAPIWrapperParams extends ToolParams { * The organization to use */ organization?: string; + /** + * The base URL of the OpenAI API. + */ + baseUrl?: string; } /** @@ -141,6 +145,7 @@ export class DallEAPIWrapper extends Tool { apiKey: openAIApiKey, organization, dangerouslyAllowBrowser: true, + baseUrl: fields?.baseUrl, }; this.client = new OpenAIClient(clientConfig); this.model = fields?.model ?? fields?.modelName ?? this.model;