Merge branch 'main' into yunkim/llmobs-safely-create-span

ddtrace/internal/packages.py

@@ -135,6 +135,14 @@ def _package_file_mapping():
         return None
 
 
+@callonce
+def _third_party_packages() -> set:
+    from gzip import decompress
+    from importlib.resources import read_binary
+
+    return set(decompress(read_binary("ddtrace.internal", "third-party.tar.gz")).decode("utf-8").splitlines())
+
+
 def filename_to_package(filename):
     # type: (str) -> t.Optional[Distribution]
 
@@ -169,6 +177,15 @@ def is_stdlib(path: Path) -> bool:
     )
 
 
+@cached()
+def is_third_party(path: Path) -> bool:
+    package = filename_to_package(str(path))
+    if package is None:
+        return False
+
+    return package.name in _third_party_packages()
+
+
 @cached()
 def is_distribution_available(name: str) -> bool:
     """Determine if a distribution is available in the current environment."""

tests/.suitespec.json

@@ -56,6 +56,7 @@
             "ddtrace/internal/metrics.py",
             "ddtrace/internal/module.py",
             "ddtrace/internal/packages.py",
+            "ddtrace/internal/third-party.tar.gz",
             "ddtrace/internal/periodic.py",
             "ddtrace/internal/rate_limiter.py",
             "ddtrace/internal/safety.py",

tests/internal/test_packages.py

@@ -5,6 +5,7 @@
 import pytest
 
 from ddtrace.internal import packages
+from ddtrace.internal.packages import _third_party_packages
 from ddtrace.internal.packages import get_distributions
 
 
@@ -86,3 +87,10 @@ def test_filename_to_package():
     assert package is None or package.name == "protobuf"
 
     del packages._package_file_mapping.__closure__[0].cell_contents.__callonce_result__
+
+
+def test_third_party_packages():
+    assert 4000 < len(_third_party_packages()) < 5000
+
+    assert "requests" in _third_party_packages()
+    assert "nota3rdparty" not in _third_party_packages()