feat(openai): baseUrl for DallEAPIWrapper (#6685)

langchain-ai · Sep 4, 2024 · 8ddb89d · 8ddb89d
1 parent 890d24c
commit 8ddb89d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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;