feat: adjust impure shell deps and autodocs workflow

.github/workflows/autodocs.yaml

@@ -20,7 +20,7 @@ jobs:
     - name: Install Protocol Buffers compiler
       run: |
         sudo apt-get update
-        sudo apt-get install -y protobuf-compiler libprotobuf-dev
+        sudo apt-get install -y protobuf-compiler libprotobuf-dev libavutil-dev pkg-config
 
     - name: Install Launcher
       id: install-launcher

nix/impure-shell.nix

@@ -8,6 +8,11 @@
   which,
   cudaPackages,
   openssl,
+  ffmpeg,
+  llvmPackages,
+  gcc,
+  gcc-unwrapped,
+  stdenv,
   pkg-config,
   poetry,
   protobuf,
@@ -53,6 +58,11 @@ mkShell {
   buildInputs =
     [
       openssl.dev
+      ffmpeg.dev
+      llvmPackages.libclang
+      gcc.cc
+      gcc-unwrapped
+      stdenv
     ]
     ++ (with python3.pkgs; [
       venvShellHook
@@ -81,7 +91,15 @@ mkShell {
 
   inputsFrom = [ server ];
 
-  env = lib.optionalAttrs withCuda {
+  env = {
+    LIBCLANG_PATH = "${llvmPackages.libclang.lib}/lib";
+    CPATH = "${gcc-unwrapped}/lib/gcc/${stdenv.hostPlatform.config}/${gcc-unwrapped.version}/include";
+    BINDGEN_EXTRA_CLANG_ARGS = builtins.concatStringsSep " " [
+      "-I${gcc.libc.dev}/include"
+      "-I${gcc}/lib/gcc/x86_64-unknown-linux-gnu/${gcc.version}/include"
+      "-I${llvmPackages.libclang.lib}/lib/clang/${llvmPackages.libclang.version}/include"
+    ];
+  } // lib.optionalAttrs withCuda {
     CUDA_HOME = "${lib.getDev cudaPackages.cuda_nvcc}";
     TORCH_CUDA_ARCH_LIST = lib.concatStringsSep ";" python3.pkgs.torch.cudaCapabilities;
   };