From 0b7e0cadc0ea3836445815bc1a2cedc8e3ec4bc6 Mon Sep 17 00:00:00 2001
From: Bannings <zhangao0086@gmail.com>
Date: Fri, 22 Dec 2023 14:17:15 +0800
Subject: [PATCH] Fix Azure OpenAI Provider BASE_MODELS (#1813)

Co-authored-by: xifan <xifan@gaoding.com>
---
 api/core/model_providers/providers/azure_openai_provider.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/api/core/model_providers/providers/azure_openai_provider.py b/api/core/model_providers/providers/azure_openai_provider.py
index 0e3d20881c..25621e7251 100644
--- a/api/core/model_providers/providers/azure_openai_provider.py
+++ b/api/core/model_providers/providers/azure_openai_provider.py
@@ -24,6 +24,8 @@ from models.provider import ProviderType, ProviderModel, ProviderQuotaType
 BASE_MODELS = [
     'gpt-4',
     'gpt-4-32k',
+    'gpt-4-1106-preview',
+    'gpt-4-vision-preview',
     'gpt-35-turbo',
     'gpt-35-turbo-16k',
     'text-davinci-003',