diff --git a/backend/danswer/direct_qa/open_ai.py b/backend/danswer/direct_qa/open_ai.py index 528d6818b9..54c938b034 100644 --- a/backend/danswer/direct_qa/open_ai.py +++ b/backend/danswer/direct_qa/open_ai.py @@ -15,12 +15,12 @@ from openai.error import Timeout from danswer.chunking.models import InferenceChunk from danswer.configs.app_configs import INCLUDE_METADATA -from danswer.configs.model_configs import GEN_AI_MAX_OUTPUT_TOKENS -from danswer.configs.model_configs import GEN_AI_MODEL_VERSION from danswer.configs.model_configs import API_BASE_OPENAI from danswer.configs.model_configs import API_TYPE_OPENAI from danswer.configs.model_configs import API_VERSION_OPENAI from danswer.configs.model_configs import AZURE_DEPLOYMENT_ID +from danswer.configs.model_configs import GEN_AI_MAX_OUTPUT_TOKENS +from danswer.configs.model_configs import GEN_AI_MODEL_VERSION from danswer.direct_qa.exceptions import OpenAIKeyMissing from danswer.direct_qa.interfaces import DanswerAnswer from danswer.direct_qa.interfaces import DanswerQuote @@ -42,9 +42,10 @@ logger = setup_logger() F = TypeVar("F", bound=Callable) + if API_BASE_OPENAI: openai.api_base = API_BASE_OPENAI -if API_TYPE_OPENAI in ["azure"]: #TODO: Azure AD support ["azure_ad", "azuread"] +if API_TYPE_OPENAI in ["azure"]: # TODO: Azure AD support ["azure_ad", "azuread"] openai.api_type = API_TYPE_OPENAI openai.api_version = API_VERSION_OPENAI