diff --git a/backend/danswer/llm/openai.py b/backend/danswer/llm/openai.py index 6cea989869ba..06e252a852f1 100644 --- a/backend/danswer/llm/openai.py +++ b/backend/danswer/llm/openai.py @@ -1,5 +1,6 @@ import os from typing import Any +from typing import cast from langchain.chat_models.openai import ChatOpenAI @@ -28,7 +29,8 @@ class OpenAIGPT(LangChainChatLLM): self._llm = ChatOpenAI( model=model_version, openai_api_key=api_key, - openai_api_base=kwargs.get('endpoint', ''), + # Prefer using None which is the default value, endpoint could be empty string + openai_api_base=cast(str, kwargs.get("endpoint")) or None, max_tokens=max_output_tokens, temperature=temperature, request_timeout=timeout,