needed to update the wdoc private mode env variable

Author: thiswillbeyourgithub

WDoc/utils/llm.py

@@ -20,6 +20,7 @@
 from .logger import whi, red, yel
 from .typechecker import optional_typecheck
 from .flags import is_verbose
+from .env import WDOC_PRIVATE_MODE
 
 TESTING_LLM = "testing/testing"
 
@@ -70,6 +71,9 @@ def load_llm(
 
     if private:
         assert api_base, "If private is set, api_base must be set too"
+        assert WDOC_PRIVATE_MODE
+        assert "WDOC_PRIVATE_MODE" in os.environ, "Missing env variable WDOC_PRIVATE_MODE"
+        assert os.environ["WDOC_PRIVATE_MODE"] == "true", "Wrong value for env variable WDOC_PRIVATE_MODE"
     if api_base:
         red(f"Will use custom api_base {api_base}")
     if not (f"{backend.upper()}_API_KEY" in os.environ and os.environ[f"{backend.upper()}_API_KEY"]):
@@ -86,6 +90,7 @@ def load_llm(
             f"private is on so overwriting {backend.upper()}_API_KEY from environment variables")
         assert os.environ[f"{backend.upper()}_API_KEY"] == "REDACTED_BECAUSE_WDOC_IN_PRIVATE_MODE"
     else:
+        assert not WDOC_PRIVATE_MODE
         assert "WDOC_PRIVATE_MODE" not in os.environ or os.environ["WDOC_PRIVATE_MODE"] == "false"
 
     if not private and backend == "openai" and api_base is None:

WDoc/utils/loaders.py

@@ -71,7 +71,7 @@
 from .logger import whi, yel, red, logger
 from .flags import is_verbose, is_linux, is_debug
 from .errors import TimeoutPdfLoaderError
-from .env import WDOC_MAX_PDF_LOADER_TIMEOUT, WDOC_EMPTY_LOADER
+from .env import WDOC_MAX_PDF_LOADER_TIMEOUT, WDOC_EMPTY_LOADER, WDOC_PRIVATE_MODE
 
 try:
     import pdftotext
@@ -1837,10 +1837,7 @@ def transcribe_audio_deepgram(
 ) -> dict:
     "Use whisper to transcribe an audio file"
     whi(f"Calling deepgram to transcribe {audio_path}")
-    if "WDOC_PRIVATE_MODE" in os.environ:
-        assert os.environ["WDOC_PRIVATE_MODE"] == "false", (
-            "Private mode detected, aborting before trying to use deepgram's API"
-        )
+    assert not WDOC_PRIVATE_MODE, "Private mode detected, aborting before trying to use deepgram's API"
     assert "DEEPGRAM_API_KEY" in os.environ and not os.environ[
         "DEEPGRAM_API_KEY"] == "REDACTED_BECAUSE_WDOC_IN_PRIVATE_MODE", "No environment variable DEEPGRAM_API_KEY found"
 
@@ -1909,11 +1906,7 @@ def transcribe_audio_whisper(
         prompt: Optional[str]) -> dict:
     "Use whisper to transcribe an audio file"
     whi(f"Calling openai's whisper to transcribe {audio_path}")
-    if "WDOC_PRIVATE_MODE" in os.environ:
-        assert os.environ["WDOC_PRIVATE_MODE"] == "false", (
-            "Private mode detected, aborting before trying to use openai's whisper"
-        )
-
+    assert not WDOC_PRIVATE_MODE, "Private mode detected, aborting before trying to use openai's whisper"
     assert "OPENAI_API_KEY" in os.environ and not os.environ[
         "OPENAI_API_KEY"] == "REDACTED_BECAUSE_WDOC_IN_PRIVATE_MODE", "No environment variable OPENAI_API_KEY found"