Add azure open ai base path for custom domains

vSpaceWallet · Nov 4, 2024 · 04879b2 · 04879b2
1 parent 4488ee0
commit 04879b2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.env.example b/.env.example
@@ -32,6 +32,5 @@ AZURE_OPENAI_API_KEY=your-azure-openai-api-key
 AZURE_OPENAI_API_INSTANCE_NAME=your-instance-name
 AZURE_OPENAI_API_DEPLOYMENT_NAME=your-deployment-name
 AZURE_OPENAI_API_VERSION=2024-08-01-preview
-
 # Optional: Azure OpenAI Base Path (if using a different domain)
-# AZURE_OPENAI_BASE_PATH=https://your-custom-domain.com/openai/deployments
+# AZURE_OPENAI_API_BASE_PATH=https://your-custom-domain.com/openai/deployments
diff --git a/src/agent/utils.ts b/src/agent/utils.ts
@@ -34,6 +34,7 @@ export const initChatModelWithConfig = async (
           azureOpenAIApiDeploymentName:
             config.azureConfig.azureOpenAIApiDeploymentName,
           azureOpenAIApiVersion: config.azureConfig.azureOpenAIApiVersion,
+          azureOpenAIBasePath: config.azureConfig.azureOpenAIBasePath,
         }
       : {}),
   });
@@ -159,14 +160,14 @@ export const getModelConfig = (
     azureOpenAIApiInstanceName: string;
     azureOpenAIApiDeploymentName: string;
     azureOpenAIApiVersion: string;
+    azureOpenAIBasePath?: string;
   };
 } => {
   const customModelName = config.configurable?.customModelName as string;
   if (!customModelName) {
     throw new Error("Model name is missing in config.");
   }
 
-  // Handle Azure OpenAI models
   if (customModelName.startsWith("azure/")) {
     const actualModelName = customModelName.replace("azure/", "");
     return {
@@ -180,11 +181,11 @@ export const getModelConfig = (
           process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME || "",
         azureOpenAIApiVersion:
           process.env.AZURE_OPENAI_API_VERSION || "2024-08-01-preview",
+        azureOpenAIBasePath: process.env.AZURE_OPENAI_API_BASE_PATH,
       },
     };
   }
 
-  // Handle existing model providers
   if (customModelName.includes("gpt-")) {
     return {
       modelName: customModelName,