Limit blockbuster to calls originated from langchain_core

libs/core/pyproject.toml

@@ -118,7 +118,7 @@ grandalf = "^0.8"
 responses = "^0.25.0"
 pytest-socket = "^0.7.0"
 pytest-xdist = "^3.6.1"
-blockbuster = "~1.5.9"
+blockbuster = "~1.5.11"
 [[tool.poetry.group.test.dependencies.numpy]]
 version = "^1.24.0"
 python = "<3.12"

libs/core/tests/unit_tests/conftest.py

@@ -12,7 +12,7 @@
 
 @pytest.fixture(autouse=True)
 def blockbuster() -> Iterator[BlockBuster]:
-    with blockbuster_ctx() as bb:
+    with blockbuster_ctx("langchain_core") as bb:
         for func in ["os.stat", "os.path.abspath"]:
             (
                 bb.functions[func]

libs/core/tests/unit_tests/runnables/test_tracing_interops.py

@@ -312,7 +312,9 @@ def _check_posts(self) -> None:
             "other_thing": "RunnableParallel<chain_result,other_thing>",
             "after": "RunnableSequence",
         }
-        assert len(posts) == sum(1 if isinstance(n, str) else len(n) for n in name_order)
+        assert len(posts) == sum(
+            1 if isinstance(n, str) else len(n) for n in name_order
+        )
         prev_dotted_order = None
         dotted_order_map = {}
         id_map = {}

libs/core/tests/unit_tests/test_setup.py

@@ -0,0 +1,15 @@
+import time
+
+import pytest
+from blockbuster import BlockingError
+
+from langchain_core import sys_info
+
+
+async def test_blockbuster_setup():
+    """Check if blockbuster is correctly setup."""
+    # Blocking call outside of langchain_core is allowed.
+    time.sleep(0.01)  # noqa: ASYNC251
+    with pytest.raises(BlockingError):
+        # Blocking call from langchain_core raises BlockingError.
+        sys_info.print_sys_info()