new: replace LANGFUSE env var by wdoc's LANGFUSE var twice instead of just at runtime, just in case

Signed-off-by: thiswillbeyourgithub
<[email protected]>

Author: thiswillbeyourgithub

wdoc/utils/env.py

@@ -198,3 +198,13 @@ def __getattribute__(self, name):
     # if not k.startswith("WDOC_"):
     #     continue
     assert is_bearable(v, env.__dataclass_fields__[k].type), v
+
+# If langfuse env variables are set AND WDOC_LANGFUSE_PUBLIC_KEY etc are set: we replace langfuse's env variable to make sure any underlyng lib use wdoc's instead
+for k in [
+    "LANGFUSE_PUBLIC_KEY",
+    "LANGFUSE_SECRET_KEY",
+    "LANGFUSE_HOST",
+]:
+    newk = "WDOC_" + k
+    if newk in os.environ and os.environ[newk]:
+        os.environ[k] = os.environ[newk]

wdoc/utils/misc.py

@@ -971,7 +971,7 @@ def thinking_answer_parser(output: str, strict: bool = False) -> dict:
 @optional_typecheck
 def create_langfuse_callback(version: str) -> None:
     assert not is_private
-    # replace langfuse's env variable if set for wdoc
+    # replace langfuse's env variable if set for wdoc, this is already done in env.py but doing it here also at runtime
     for k in [
         "LANGFUSE_PUBLIC_KEY",
         "LANGFUSE_SECRET_KEY",