reset env after no_env_var, important if wrapped block sets var

langchain-ai · Jul 30, 2024 · 454f8ac · 454f8ac
1 parent 571570f
commit 454f8ac
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/libs/ai-endpoints/tests/unit_tests/test_api_key.py b/libs/ai-endpoints/tests/unit_tests/test_api_key.py
@@ -16,6 +16,9 @@ def no_env_var(var: str) -> Generator[None, None, None]:
     finally:
         if val:
             os.environ[var] = val
+        else:
+            if var in os.environ:
+                del os.environ[var]
 
 
 @pytest.fixture(autouse=True)