Fix startup exception + mypy

backend/danswer/background/connector_deletion.py

@@ -90,7 +90,7 @@ def _update_multi_indexed_docs() -> None:
         def _get_user(
             credential: Credential,
         ) -> str:
-            if credential.public_doc:
+            if credential.public_doc or not credential.user:
                 return PUBLIC_DOC_PAT
 
             return str(credential.user.id)

backend/danswer/llm/azure.py

@@ -1,3 +1,4 @@
+import os
 from typing import Any
 
 from langchain.chat_models.azure_openai import AzureChatOpenAI
@@ -22,6 +23,11 @@ def __init__(
         *args: list[Any],
         **kwargs: dict[str, Any]
     ):
+        # set a dummy API key if not specified so that LangChain doesn't throw an
+        # exception when trying to initialize the LLM which would prevent the API
+        # server from starting up
+        if not api_key:
+            api_key = os.environ.get("OPENAI_API_KEY") or "dummy_api_key"
         self._llm = AzureChatOpenAI(
             model=model_version,
             openai_api_type="azure",

backend/danswer/llm/openai.py

@@ -1,3 +1,4 @@
+import os
 from typing import Any
 
 from langchain.chat_models.openai import ChatOpenAI
@@ -16,6 +17,11 @@ def __init__(
         *args: list[Any],
         **kwargs: dict[str, Any]
     ):
+        # set a dummy API key if not specified so that LangChain doesn't throw an
+        # exception when trying to initialize the LLM which would prevent the API
+        # server from starting up
+        if not api_key:
+            api_key = os.environ.get("OPENAI_API_KEY") or "dummy_api_key"
         self._llm = ChatOpenAI(
             model=model_version,
             openai_api_key=api_key,