From 620280db926e3c22d2a7cffdbb5cc6aa52e5f645 Mon Sep 17 00:00:00 2001 From: Yuhong Sun Date: Tue, 15 Aug 2023 01:32:09 -0700 Subject: [PATCH] Fix formatting according to precommit hooks (#303) --- backend/danswer/direct_qa/open_ai.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/backend/danswer/direct_qa/open_ai.py b/backend/danswer/direct_qa/open_ai.py index 528d6818b90..54c938b0346 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