Merge branch 'main' into zachg/llmobs_annotate_context_fix

Author: ZStriker19

.github/workflows/pytorch_gpu_tests.yml

@@ -32,10 +32,8 @@ jobs:
         run: rustup update stable && rustup default stable && rustup component add rustfmt clippy
 
       - name: Install hatch
-        uses: pypa/hatch@257e27e51a6a5616ed08a39a408a21c35c9931bc # install
-        with:
-          version: "1.12.0"
-      
+        run: pip install hatch==1.12.0
+
       - name: Install PyTorch
         run: pip install torch
 

.github/workflows/unit_tests.yml

@@ -29,11 +29,6 @@ jobs:
       - name: Install latest stable toolchain and rustfmt
         run: rustup update stable && rustup default stable && rustup component add rustfmt clippy
 
-      - name: Install hatch
-        uses: pypa/hatch@257e27e51a6a5616ed08a39a408a21c35c9931bc # install
-        with:
-          version: "1.12.0"
-
       - name: Install coreutils for MacOS to get sha256sum
         if: matrix.os == 'macos-latest'
         run: brew install coreutils
@@ -45,6 +40,9 @@ jobs:
           sudo apt-get update
           sudo apt-get install -y python3.14 python3.14-dev
 
+      - name: Install hatch
+        run: pip install hatch==1.12.0
+
       - name: Run tests
         run: hatch run +py=${{ matrix.python-version }} multiple_os_tests:test
 

setup.py

@@ -664,8 +664,15 @@ def build_rust(self):
                 if src_file.exists():
                     newest_source_time = max(newest_source_time, src_file.stat().st_mtime)
 
-            # Only rebuild if source files are newer than the destination
-            should_build = newest_source_time > library_mtime
+            required_headers = ["common.h"]
+            if "profiling" in rust_features:
+                required_headers.append("profiling.h")
+
+            include_dir = CARGO_TARGET_DIR / "include" / "datadog"
+            headers_exist = include_dir.exists() and all((include_dir / header).exists() for header in required_headers)
+
+            # Only rebuild if source files are newer than the destination OR if any required header is missing
+            should_build = newest_source_time > library_mtime or not headers_exist
 
         if should_build:
             # Create and run the CustomBuildRust command