fix openai base env bug

api/core/embedding/openai_embedding.py

@@ -113,8 +113,8 @@ class OpenAIEmbedding(BaseEmbedding):
         self.deployment_name = deployment_name
         self.openai_api_key = openai_api_key
         # Use proxy openai base
-        if current_app.config['API_URL'] is not None:
-            openai.api_base = current_app.config['API_URL']
+        if current_app.config['OPENAI_API_BASE'] is not None:
+            openai.api_base = current_app.config['OPENAI_API_BASE']
 
     @handle_llm_exceptions
     def _get_query_embedding(self, query: str) -> List[float]:

api/core/llm/moderation.py

@@ -1,4 +1,5 @@
 import openai
+from flask import current_app
 from models.provider import ProviderName
 
 
@@ -8,6 +9,10 @@ class Moderation:
         self.provider = provider
         self.api_key = api_key
 
+        # Use proxy openai base
+        if current_app.config['OPENAI_API_BASE'] is not None:
+            openai.api_base = current_app.config['OPENAI_API_BASE']
+
         if self.provider == ProviderName.OPENAI.value:
             self.client = openai.Moderation
 

api/core/llm/provider/openai_provider.py

@@ -3,7 +3,7 @@ from typing import Optional, Union
 
 import openai
 from openai.error import AuthenticationError, OpenAIError
-
+from flask import current_app
 from core.llm.moderation import Moderation
 from core.llm.provider.base import BaseProvider
 from core.llm.provider.errors import ValidateFailedError
@@ -12,6 +12,9 @@ from models.provider import ProviderName
 
 class OpenAIProvider(BaseProvider):
     def get_models(self, model_id: Optional[str] = None) -> list[dict]:
+        # Use proxy openai base
+        if current_app.config['OPENAI_API_BASE'] is not None:
+            openai.api_base = current_app.config['OPENAI_API_BASE']
         credentials = self.get_credentials(model_id)
         response = openai.Model.list(**credentials)